From b1eefb173bd55121718962a4ea856c4873793811 Mon Sep 17 00:00:00 2001
From: bestgopher <84328409@qq.com>
Date: Mon, 12 Aug 2024 17:22:16 +0800
Subject: [PATCH] refactor: add New function for bpool
Signed-off-by: bestgopher <84328409@qq.com>
---
internal/bpool/bpool.go | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/internal/bpool/bpool.go b/internal/bpool/bpool.go
index aa826fba..12cf577a 100644
--- 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)
}