fix: didn't fully reuse buffer memory

bytedance · Oct 30, 2023 · 1895d9a · 1895d9a
1 parent 53c974e
commit 1895d9a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/internal/encoder/stream.go b/internal/encoder/stream.go
@@ -36,7 +36,8 @@ func NewStreamEncoder(w io.Writer) *StreamEncoder {
 
 // Encode encodes interface{} as JSON to io.Writer
 func (enc *StreamEncoder) Encode(val interface{}) (err error) {
-    out := newBytes()
+    buf := newBytes()
+    out := buf
 
     /* encode into the buffer */
     err = EncodeInto(&out, val, enc.Opts)
@@ -79,6 +80,6 @@ func (enc *StreamEncoder) Encode(val interface{}) (err error) {
     }
 
 free_bytes:
-    freeBytes(out)
+    freeBytes(buf)
     return err
 }