lbpool.go

package cbytebuf

import "github.com/koykov/lbpool"

// LBPool is a pool implementation based on lbpool.Pool.
type LBPool struct {
	Size          uint
	ReleaseFactor float32

	p lbpool.Pool
}

var (
	// LBP is a default instance of LB pool for simple cases.
	// Just call cbytebuf.LBAcquire() and cbytebuf.LBRelease().
	LBP = LBPool{Size: 1000}

	_, _ = LBAcquire, LBRelease
)

// Get old byte buffer from the LB pool or create a new byte buffer.
func (p *LBPool) Get() *CByteBuf {
	p.p.Size = p.Size
	p.p.ReleaseFactor = p.ReleaseFactor
	v := p.p.Get()
	if v != nil {
		if b, ok := v.(*CByteBuf); ok {
			metricsHandler.PoolAcquire(uint64(b.h.Cap))
			return b
		}
	}
	return &CByteBuf{}
}

// Put byte buffer back to the LB pool.
//
// Using data returned from the buffer after putting is unsafe.
func (p *LBPool) Put(b *CByteBuf) {
	if b.h.Data == 0 {
		return
	}
	b.ResetLen()
	add := p.p.Put(b)
	if add {
		metricsHandler.PoolRelease(uint64(b.h.Cap))
	}
}

// LBAcquire gets byte buffer from default LB pool instance.
func LBAcquire() *CByteBuf {
	return LBP.Get()
}

// LBRelease puts byte buffer back to default LB pool instance.
func LBRelease(b *CByteBuf) {
	LBP.Put(b)
}