Moves buffer pool to separate file

ksysoev · Apr 20, 2024 · e8f2b65 · e8f2b65
1 parent 99faff9
commit e8f2b65
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/channel/buffers.go b/channel/buffers.go
@@ -0,0 +1,29 @@
+package channel
+
+import (
+	"bytes"
+	"sync"
+)
+
+type bufferPool struct {
+	pool *sync.Pool
+}
+
+func newBufferPool() *bufferPool {
+	return &bufferPool{
+		pool: &sync.Pool{
+			New: func() interface{} {
+				return &bytes.Buffer{}
+			},
+		},
+	}
+}
+
+func (p *bufferPool) get() *bytes.Buffer {
+	return p.pool.Get().(*bytes.Buffer)
+}
+
+func (p *bufferPool) put(b *bytes.Buffer) {
+	b.Reset()
+	p.pool.Put(b)
+}