internal/bpool: add New function (#465)

Signed-off-by: bestgopher <84328409@qq.com>
coder · Aug 21, 2024 · 641f4f5 · 641f4f5
1 parent 4ce1d90
commit 641f4f5
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/internal/bpool/bpool.go b/internal/bpool/bpool.go
@@ -5,15 +5,16 @@ import (
 	"sync"
 )
 
-var bpool sync.Pool
+var bpool = sync.Pool{
+	New: func() any {
+		return &bytes.Buffer{}
+	},
+}
 
 // Get returns a buffer from the pool or creates a new one if
 // the pool is empty.
 func Get() *bytes.Buffer {
 	b := bpool.Get()
-	if b == nil {
-		return &bytes.Buffer{}
-	}
 	return b.(*bytes.Buffer)
 }