Update MapBlk serialization to zstd

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

diff --git a/internal/mkserialize/mkserialize.go b/internal/mkserialize/mkserialize.go
index be1ba223258f47ecf3a311d7f07e87709d80b67e..a3bd65747e5f69ad8d505e570ca9679fe30eae88 100644 (file)
--- a/internal/mkserialize/mkserialize.go
+++ b/internal/mkserialize/mkserialize.go
@@ -78,6 +78,18 @@ func structPragma(c *ast.Comment, sp *[]func(), expr string, de bool) {
                                fmt.Println("chk(w.Close()) }")
                        })
                }
+       case "zstd":
+               if de {
+                       fmt.Println("{ r, err := zstd.NewReader(byteReader{r}); chk(err)")
+                       *sp = append(*sp, func() {
+                               fmt.Println("chk(r.Close()) }")
+                       })
+               } else {
+                       fmt.Println("{ w := zstd.NewWriter(w)")
+                       *sp = append(*sp, func() {
+                               fmt.Println("chk(w.Close()) }")
+                       })
+               }
        case "lenhdr":
                if arg != "8" && arg != "16" && arg != "32" {
                        error(c.Pos(), "usage: //mt:lenhdr (8|16|32)")