pool: Add NewFunc (#59)

sunshineplan · web-flow · commit dcca62f9b11c · 2024-06-13T16:28:29.000+08:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,11 @@ type Pool[T any] struct {`
`7`	`7`	`}`
`8`	`8`
`9`	`9`	`func New[T any]() *Pool[T] {`
`10`		`- return &Pool[T]{&sync.Pool{New: func() any { return new(T) }}}`
	`10`	`+ return NewFunc(func() *T { return new(T) })`
	`11`	`+}`
	`12`	`+`
	`13`	`+func NewFunc[T any](fn func() T) Pool[T] {`
	`14`	`+ return &Pool[T]{&sync.Pool{New: func() any { return fn() }}}`
`11`	`15`	`}`
`12`	`16`
`13`	`17`	`func (p Pool[T]) Get() T {`