Initial release of queue package

bench_test.go

@@ -0,0 +1,67 @@
+package queue_test
+
+import (
+	"context"
+	"sync"
+	"testing"
+	"time"
+
+	"git.belvedersky.ru/belvedersky/queue"
+)
+
+type benchTask struct {
+	ID int
+}
+
+// BenchmarkProduceConsume измеряет производительность очереди
+// при различном количестве воркеров.
+func BenchmarkProduceConsume(b *testing.B) {
+	benchmarks := []struct {
+		name    string
+		workers int
+	}{
+		{"workers_1", 1},
+		{"workers_4", 4},
+		{"workers_16", 16},
+	}
+
+	for _, bm := range benchmarks {
+		b.Run(bm.name, func(b *testing.B) {
+			ctx, cancel := context.WithCancel(context.Background())
+			defer cancel()
+
+			q := queue.NewQueue[benchTask](10)
+			var wg sync.WaitGroup
+			wg.Add(b.N)
+
+			q.Register(func(t benchTask) {
+				time.Sleep(100 * time.Microsecond) // имитация нагрузки
+				wg.Done()
+			})
+
+			q.HandleParallel(ctx, bm.workers)
+
+			b.ResetTimer()
+			for i := 0; i < b.N; i++ {
+				q.Produce(benchTask{ID: i})
+			}
+
+			wg.Wait()
+			q.Shutdown(context.Background())
+		})
+	}
+}
+
+// BenchmarkTryProduce тестирует неблокирующую постановку задач.
+func BenchmarkTryProduce(b *testing.B) {
+	ctx := context.Background()
+	q := queue.NewQueue[benchTask](10)
+	q.HandleParallel(ctx, 4)
+	defer q.Close()
+
+	task := benchTask{}
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		q.TryProduce(task, 5*time.Millisecond)
+	}
+}