fix: check CanSizeReuse in FreeBytes

bytedance · Jul 9, 2024 · cc6330c · cc6330c
1 parent 3a695bd
commit cc6330c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/internal/encoder/vars/stack.go b/internal/encoder/vars/stack.go
@@ -119,8 +119,10 @@ func NewBuffer() *bytes.Buffer {
 }
 
 func FreeBytes(p *[]byte) {
-	(*p) = (*p)[:0]
-	bytesPool.Put(p)
+	if rt.CanSizeResue(cap(*p)) {
+		(*p) = (*p)[:0]
+		bytesPool.Put(p)
+	}
 }
 
 func FreeStack(p *Stack) {
@@ -129,8 +131,10 @@ func FreeStack(p *Stack) {
 }
 
 func FreeBuffer(p *bytes.Buffer) {
-	p.Reset()
-	bufferPool.Put(p)
+	if rt.CanSizeResue(cap(p.Bytes())) {
+		p.Reset()
+		bufferPool.Put(p)
+	}
 }
 
 var (