fix errors in usage of zstd

[mt.git] / internal / mkserialize / mkserialize.go

diff --git a/internal/mkserialize/mkserialize.go b/internal/mkserialize/mkserialize.go
index a3bd65747e5f69ad8d505e570ca9679fe30eae88..23a9d39ca105e7c223a0a156d47f6ee40c3ca2b0 100644 (file)
--- a/internal/mkserialize/mkserialize.go
+++ b/internal/mkserialize/mkserialize.go
@@ -82,10 +82,10 @@ func structPragma(c *ast.Comment, sp *[]func(), expr string, de bool) {
                if de {
                        fmt.Println("{ r, err := zstd.NewReader(byteReader{r}); chk(err)")
                        *sp = append(*sp, func() {
-                               fmt.Println("chk(r.Close()) }")
+                               fmt.Println("r.Close() }")
                        })
                } else {
-                       fmt.Println("{ w := zstd.NewWriter(w)")
+                       fmt.Println("{ w, err := zstd.NewWriter(w); chk(err)")
                        *sp = append(*sp, func() {
                                fmt.Println("chk(w.Close()) }")
                        })