rpc_util: return unused buffers to memory pool

grpc · May 23, 2023 · 2678efb · 2678efb
1 parent 5535416
commit 2678efb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/shared_buffer_pool.go b/shared_buffer_pool.go
@@ -122,7 +122,9 @@ func (p *bufferPool) Get(size int) []byte {
 	bs := p.Pool.Get().(*[]byte)
 
 	if cap(*bs) < size {
-		*bs = make([]byte, size)
+		p.Pool.Put(bs)
+
+		return make([]byte, size)
 	}
 
 	return (*bs)[:size]