1 files changed, 65 insertions, 0 deletions

diff --git a/fanout/fanout_bench_test.go b/fanout/fanout_bench_test.go
new file mode 100644
index 0000000..0b88d87
--- /dev/null
+++ b/fanout/fanout_bench_test.go
@@ -0,0 +1,65 @@
+package fanout
+
+import (
+        "context"
+        "testing"
+)
+
+func BenchmarkFanOut_RoundRobin(b *testing.B) {
+        b.ReportAllocs()
+
+        fo, _ := NewFanOut[int](
+                WithWorkerCount(4),
+                WithBufferSize(64),
+                WithStrategy(RoundRobin),
+        )
+
+        ctx, cancel := context.WithCancel(context.Background())
+        defer cancel()
+
+        input := make(chan int, 64)
+        fo.Run(ctx, input)
+
+        // Drain outputs
+        for _, ch := range fo.Outputs() {
+                go func() {
+                        for range ch {
+                        }
+                }()
+        }
+
+        for b.Loop() {
+                input <- 1
+        }
+
+        close(input)
+}
+
+func BenchmarkFanOut_Broadcast(b *testing.B) {
+        b.ReportAllocs()
+
+        fo, _ := NewFanOut[int](
+                WithWorkerCount(4),
+                WithBufferSize(64),
+                WithStrategy(Broadcast),
+        )
+
+        ctx, cancel := context.WithCancel(context.Background())
+        defer cancel()
+
+        input := make(chan int, 64)
+        fo.Run(ctx, input)
+
+        for _, ch := range fo.Outputs() {
+                go func() {
+                        for range ch {
+                        }
+                }()
+        }
+
+        for b.Loop() {
+                input <- 1
+        }
+
+        close(input)
+}