summaryrefslogtreecommitdiff
path: root/pool.go
diff options
context:
space:
mode:
Diffstat (limited to 'pool.go')
-rw-r--r--pool.go33
1 files changed, 33 insertions, 0 deletions
diff --git a/pool.go b/pool.go
new file mode 100644
index 0000000..c196085
--- /dev/null
+++ b/pool.go
@@ -0,0 +1,33 @@
+// Copyright (C) 2022-2023 Luke Shumaker <lukeshu@lukeshu.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+package typedsync
+
+import (
+ "sync"
+)
+
+type Pool[T any] struct {
+ New func() T
+
+ inner sync.Pool
+}
+
+func (p *Pool[T]) Get() (val T, ok bool) {
+ _val := p.inner.Get()
+ switch {
+ case _val != nil:
+ //nolint:forcetypeassert // Typed wrapper around untyped lib.
+ return _val.(T), true
+ case p.New != nil:
+ return p.New(), true
+ default:
+ var zero T
+ return zero, false
+ }
+}
+
+func (p *Pool[T]) Put(val T) {
+ p.inner.Put(val)
+}