buffer/pool.go

package buffer

import (
	"errors"
	"sync"
)

type noPool struct {
	allocSize int
}

type pool struct {
	sp *sync.Pool
}

func newPool(allocSize int) *pool {
	sp := &sync.Pool{
		New: func() any {
			return make([]byte, allocSize)
		},
	}

	return &pool{sp: sp}
}

func (p noPool) Get() ([]byte, error) {
	return make([]byte, p.allocSize), nil
}

func (noPool) Put([]byte) {
}

func (p *pool) Get() ([]byte, error) {
	b, ok := p.sp.Get().([]byte)
	if !ok {
		return nil, errors.New("invalid resource received from pool")
	}

	return b, nil
}

func (p *pool) Put(b []byte) {
	p.sp.Put(b)
}