orbyte/pbuf/pbuf.go

// Package pbuf is a lightweight pooled buffer.
package pbuf

import (
	"bytes"

	"github.com/fumiama/orbyte"
)

var bufferPool = NewBufferPool[struct{}]()

type (
	OBuffer = orbyte.Item[Buffer]
	Buffer  = UserBuffer[struct{}]
	Bytes   = UserBytes[struct{}]
)

type BufferPool[USRDAT any] struct {
	p *orbyte.Pool[UserBuffer[USRDAT]]
}

func NewBufferPool[USRDAT any]() BufferPool[USRDAT] {
	return BufferPool[USRDAT]{
		p: orbyte.NewPool[UserBuffer[USRDAT]](bufpooler[USRDAT]{}),
	}
}

// NewBuffer wraps bytes.NewBuffer into Item.
func NewBuffer(buf []byte) *OBuffer {
	return bufferPool.NewBuffer(buf)
}

// InvolveBuffer involve external *bytes.Buffer into Item.
func InvolveBuffer(buf *bytes.Buffer) *OBuffer {
	return bufferPool.InvolveBuffer(buf)
}

// ParseBuffer convert external *bytes.Buffer into Item.
func ParseBuffer(buf *bytes.Buffer) *OBuffer {
	return bufferPool.ParseBuffer(buf)
}

// NewBytes alloc sz bytes.
func NewBytes(sz int) Bytes {
	return bufferPool.NewBytes(sz)
}

// InvolveBytes involve outside buf into pool.
func InvolveBytes(b ...byte) Bytes {
	return bufferPool.InvolveBytes(b...)
}

// ParseBytes convert outside bytes to Bytes safely
// without adding it into pool.
func ParseBytes(b ...byte) Bytes {
	return bufferPool.ParseBytes(b...)
}