package zsocket

import "sync"

// BufferPool is a generic reusable buffer pool.

type BufferPool interface {

Get() any

Put(any)

}

type defaultBufferPool struct {

size int

pool sync.Pool

}

// NewBufferPool creates a sync.Pool-backed byte buffer pool.

func NewBufferPool(size int) BufferPool {

if size <= 0 {

size = 4096

}

p := &defaultBufferPool{size: size}

p.pool.New = func() any {

b := make([]byte, 0, p.size)

return &b

}

return p

}

func (p *defaultBufferPool) Get() any {

return p.pool.Get()

}

func (p *defaultBufferPool) Put(v any) {

b, ok := v.(*[]byte)

if !ok || b == nil {

return

}

buf := (*b)[:0]

if cap(buf) > p.size*4 {

buf = make([]byte, 0, p.size)

}

*b = buf

p.pool.Put(b)

}