mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-25 19:15:34 +00:00
e92f4c6af8
Pool is a type-safe wrapper over sync.Pool. Updates tailscale/corp#11038 Updates #cleanup Signed-off-by: Joe Tsai <joetsai@digital-static.net>
32 lines
707 B
Go
32 lines
707 B
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package syncs
|
|
|
|
import "sync"
|
|
|
|
// Pool is the generic version of [sync.Pool].
|
|
type Pool[T any] struct {
|
|
pool sync.Pool
|
|
|
|
// New optionally specifies a function to generate
|
|
// a value when Get would otherwise return the zero value of T.
|
|
// It may not be changed concurrently with calls to Get.
|
|
New func() T
|
|
}
|
|
|
|
// Get selects an arbitrary item from the Pool, removes it from the Pool,
|
|
// and returns it to the caller. See [sync.Pool.Get].
|
|
func (p *Pool[T]) Get() T {
|
|
x, ok := p.pool.Get().(T)
|
|
if !ok && p.New != nil {
|
|
x = p.New()
|
|
}
|
|
return x
|
|
}
|
|
|
|
// Put adds x to the pool.
|
|
func (p *Pool[T]) Put(x T) {
|
|
p.pool.Put(x)
|
|
}
|