pool: Add NewFunc (#59)

sunshineplan · Jun 13, 2024 · dcca62f · dcca62f
1 parent 1a3d91b
commit dcca62f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pool/pool.go b/pool/pool.go
@@ -7,7 +7,11 @@ type Pool[T any] struct {
 }
 
 func New[T any]() *Pool[T] {
-	return &Pool[T]{&sync.Pool{New: func() any { return new(T) }}}
+	return NewFunc(func() *T { return new(T) })
+}
+
+func NewFunc[T any](fn func() *T) *Pool[T] {
+	return &Pool[T]{&sync.Pool{New: func() any { return fn() }}}
 }
 
 func (p *Pool[T]) Get() *T {