Check that zlib writer and readers close correctly because we re-use …

…them from the pool
fredli74 · Jul 27, 2024 · 7b82c97 · 7b82c97
1 parent 0d43dfd
commit 7b82c97
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/core/block.go b/core/block.go
@@ -176,7 +176,9 @@ func ZlibCompress(src bytearray.ByteArray) (dst bytearray.ByteArray) {
 	src.ReadSeek(0, bytearray.SEEK_SET)
 	zw := zpool.GetWriter(&dst)
 	CopyOrPanic(zw, &src)
-	zw.Close()
+	if err := zw.Close(); err != nil {
+		panic(err)
+	}
 	zpool.PutWriter(zw)
 	return dst
 }
@@ -186,7 +188,11 @@ func ZlibUncompress(src bytearray.ByteArray) (dst bytearray.ByteArray) {
 	if err != nil {
 		panic(err)
 	}
-	defer zr.Close()
+	defer func() {
+		if err := zr.Close(); err != nil {
+			panic(err)
+		}
+	}()
 	CopyOrPanic(&dst, zr)
 	return dst
 }