1 files changed, 38 insertions, 0 deletions

diff --git a/core/pool.go b/core/pool.go
new file mode 100644
index 0000000..484b342
--- /dev/null
+++ b/core/pool.go
@@ -0,0 +1,38 @@
+package core
+
+import "sync"
+
+
+const (
+        DefaultBufferSize = 1024
+        MaxBufferSize = 64 * 1024
+)
+
+var bytesPool = sync.Pool{
+        New: func () any {
+                b := make([]byte, DefaultBufferSize)
+                return &b
+        },
+}
+
+func GetBuffer(size int) []byte {
+        bufPtr := bytesPool.Get().(*[]byte)
+
+        buf := *bufPtr
+
+
+        if size > cap(buf) {
+                buf = make([]byte, size)
+        }
+
+        return buf[:size]
+}
+
+
+func PutBuffer(buf []byte) {
+        if cap(buf) < MaxBufferSize{
+                // reset
+                buf = buf[:0]
+                bytesPool.Put(&buf)
+        }
+}