fix(bytesBuffer): limit all bytes buffer to 256k

paragor · Sep 8, 2024 · eeff934 · eeff934
1 parent 2aab161
commit eeff934
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/pkg/cache/cache.go b/pkg/cache/cache.go
@@ -43,6 +43,9 @@ var bufferPool = sync.Pool{
 func getBytesBuffer() ([]byte, func()) {
 	responseBytesBuffer := bufferPool.Get().([]byte)[:0]
 	return responseBytesBuffer, func() {
+		if cap(responseBytesBuffer) > 256*1024 {
+			return
+		}
 		bufferPool.Put(responseBytesBuffer[:0])
 	}
 }
diff --git a/pkg/cachebehavior/cache.go b/pkg/cachebehavior/cache.go
@@ -219,6 +219,9 @@ var bufferPool = sync.Pool{
 func getBytesBuffer() ([]byte, func()) {
 	responseBytesBuffer := bufferPool.Get().([]byte)[:0]
 	return responseBytesBuffer, func() {
+		if cap(responseBytesBuffer) > 256*1024 {
+			return
+		}
 		bufferPool.Put(responseBytesBuffer[:0])
 	}
 }

diff --git a/pkg/upstream/upstream.go b/pkg/upstream/upstream.go
@@ -118,6 +118,9 @@ var bufferPool = sync.Pool{
 func getBytesBuffer() ([]byte, func()) {
 	responseBytesBuffer := bufferPool.Get().([]byte)[:0]
 	return responseBytesBuffer, func() {
+		if cap(responseBytesBuffer) > 256*1024 {
+			return
+		}
 		bufferPool.Put(responseBytesBuffer[:0])
 	}
 }