fixup! Cleanup.

internal/optimization/memory_pool.go

@@ -0,0 +1,95 @@
+// Package optimization provides memory and performance optimizations
+package optimization
+
+import (
+	"sync"
+)
+
+// StringBuilderPool provides a pool of reusable string builders
+type StringBuilderPool struct {
+	pool sync.Pool
+}
+
+// NewStringBuilderPool creates a new string builder pool
+func NewStringBuilderPool() *StringBuilderPool {
+	return &StringBuilderPool{
+		pool: sync.Pool{
+			New: func() any {
+				buf := make([]byte, 0, 256) // Pre-allocate 256 bytes
+				return &buf
+			},
+		},
+	}
+}
+
+// Get retrieves a string builder from the pool
+func (p *StringBuilderPool) Get() []byte {
+	bufPtr := p.pool.Get().(*[]byte)
+	return *bufPtr
+}
+
+// Put returns a string builder to the pool
+func (p *StringBuilderPool) Put(buf []byte) {
+	if cap(buf) < 4096 { // Don't pool overly large buffers
+		buf = buf[:0] // Reset length but keep capacity
+		p.pool.Put(&buf)
+	}
+}
+
+// ByteSlicePool provides a pool of reusable byte slices
+type ByteSlicePool struct {
+	pool sync.Pool
+	size int
+}
+
+// NewByteSlicePool creates a new byte slice pool with specified size
+func NewByteSlicePool(size int) *ByteSlicePool {
+	return &ByteSlicePool{
+		size: size,
+		pool: sync.Pool{
+			New: func() any {
+				buf := make([]byte, size)
+				return &buf
+			},
+		},
+	}
+}
+
+// Get retrieves a byte slice from the pool
+func (p *ByteSlicePool) Get() []byte {
+	bufPtr := p.pool.Get().(*[]byte)
+	return *bufPtr
+}
+
+// Put returns a byte slice to the pool
+func (p *ByteSlicePool) Put(buf []byte) {
+	if len(buf) == p.size {
+		p.pool.Put(&buf)
+	}
+}
+
+// Global pools for common use cases
+var (
+	globalStringBuilderPool = NewStringBuilderPool()
+	globalByteSlicePool     = NewByteSlicePool(2048)
+)
+
+// GetStringBuilder gets a string builder from the global pool
+func GetStringBuilder() []byte {
+	return globalStringBuilderPool.Get()
+}
+
+// PutStringBuilder returns a string builder to the global pool
+func PutStringBuilder(buf []byte) {
+	globalStringBuilderPool.Put(buf)
+}
+
+// GetByteSlice gets a byte slice from the global pool
+func GetByteSlice() []byte {
+	return globalByteSlicePool.Get()
+}
+
+// PutByteSlice returns a byte slice to the global pool
+func PutByteSlice(buf []byte) {
+	globalByteSlicePool.Put(buf)
+}