syncy: add Pool

syncy/pool.go

@@ -0,0 +1,28 @@
+package syncy
+
+import "sync"
+
+// Pool is a simple wrapper around sync.Pool, providing type safety.
+type Pool[T any] struct {
+	pool *sync.Pool
+}
+
+func NewPool[T any](new func() T) *Pool[T] {
+	pool := &sync.Pool{
+		New: func() any {
+			return new()
+		},
+	}
+
+	return &Pool[T]{
+		pool: pool,
+	}
+}
+
+func (p *Pool[T]) Put(v T) {
+	p.pool.Put(v)
+}
+
+func (p *Pool[T]) Get() T {
+	return p.pool.Get().(T)
+}

syncy/pool_test.go

@@ -0,0 +1,27 @@
+package syncy_test
+
+import (
+	"testing"
+
+	"github.com/micvbang/go-helpy/syncy"
+)
+
+func TestPoolSingleResource(t *testing.T) {
+	p := syncy.NewPool(func() *int {
+		var n = 0
+		return &n
+	})
+
+	const expectedN = 10
+
+	for i := 0; i < expectedN; i++ {
+		n := p.Get()
+		*n += 1
+		p.Put(n)
+	}
+
+	n := p.Get()
+	if *n != expectedN {
+		t.Fatalf("expected n == %d, got n == %d", expectedN, *n)
+	}
+}