crate/synchelper/pool.go

package synchelper

import (
	"sync"
)

// NewPool 初始化 Pool,
// newFn 是资源的构造函数,
// putCond 返回 true 时表示资源可以被放回 Pool 中.
func NewPool[T any](newFn func() any, putCond func(v T) bool) Pool[T] {
	return Pool[T]{
		pool:    &sync.Pool{New: newFn},
		putCond: putCond,
	}
}

// Pool 是通用的资源池
type Pool[T any] struct {
	pool    *sync.Pool
	putCond func(v T) bool
}

// Get 获取资源
func (pool Pool[T]) Get() T {
	return pool.pool.Get().(T)
}

// Put 放回资源
func (pool Pool[T]) Put(v T) {
	if !pool.putCond(v) {
		return
	}
	pool.pool.Put(v)
}