1 files changed, 50 insertions, 0 deletions

diff --git a/workerpool/workerpool_bench_test.go b/workerpool/workerpool_bench_test.go
new file mode 100644
index 0000000..1fbc672
--- /dev/null
+++ b/workerpool/workerpool_bench_test.go
@@ -0,0 +1,50 @@
+package workerpool
+
+import (
+        "context"
+        "fmt"
+        "testing"
+)
+
+type benchTask struct {
+        id string
+}
+
+func (t *benchTask) Execute(_ context.Context) error { return nil }
+func (t *benchTask) GetID() string                   { return t.id }
+
+func BenchmarkWorkerPool(b *testing.B) {
+        b.ReportAllocs()
+
+        pool, _ := NewWorkerPool[*benchTask](b.Context(),
+                WithNumWorkers(4),
+                WithTaskQueueSize(256),
+        )
+        pool.Start()
+
+        task := &benchTask{id: "bench"}
+
+        for b.Loop() {
+                pool.Push(task)
+        }
+
+        pool.StopAndWait()
+}
+
+func BenchmarkWorkerPool_Push(b *testing.B) {
+        b.ReportAllocs()
+
+        pool, _ := NewWorkerPool[*benchTask](b.Context(),
+                WithNumWorkers(8),
+                WithTaskQueueSize(1024),
+        )
+        pool.Start()
+        defer pool.StopAndWait()
+
+        b.RunParallel(func(pb *testing.PB) {
+                task := &benchTask{id: fmt.Sprintf("bench-%d", b.N)}
+                for pb.Next() {
+                        pool.Push(task)
+                }
+        })
+}