Added err handler in compression

ikkerens · Mar 1, 2019 · 563197b · 563197b
1 parent 512f5e0
commit 563197b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/compression.go b/compression.go
@@ -47,14 +47,16 @@ func (c *compressionReadWriter) writeVariable(w io.Writer, v reflect.Value) erro
 		return err
 	}
 
-	if err := handleVariableWriter(z, c.handler, v); err != nil {
+	if err = handleVariableWriter(z, c.handler, v); err != nil {
+		return err
+	}
+	if err = z.Close(); err != nil {
 		return err
 	}
-	z.Close()
 
 	lb := make([]byte, 4)
 	binary.BigEndian.PutUint32(lb, uint32(b.Len()))
-	if _, err := w.Write(lb); err != nil {
+	if _, err = w.Write(lb); err != nil {
 		return err
 	}
 	if _, err = w.Write(b.Bytes()); err != nil {