Function BenchmarkPubSubConcurrentCreation

pool_pubsub_bench_test.go:240–271

BenchmarkPubSubConcurrentCreation benchmarks concurrent PubSub creation

(b *testing.B)

// BenchmarkPubSubConcurrentCreation benchmarks concurrent PubSub creation
func BenchmarkPubSubConcurrentCreation(b *testing.B) {
	ctx := context.Background()
	client := benchmarkClient(32)
	defer client.Close()

	concurrencyLevels := []int{1, 2, 4, 8, 16}

	for _, concurrency := range concurrencyLevels {
		b.Run(fmt.Sprintf("Concurrency_%d", concurrency), func(b *testing.B) {
			b.ResetTimer()
			b.ReportAllocs()

			var wg sync.WaitGroup
			semaphore := make(chan struct{}, concurrency)

			for i := 0; i < b.N; i++ {
				wg.Add(1)
				semaphore <- struct{}{}

				go func() {
					defer wg.Done()
					defer func() { <-semaphore }()

					pubsub := client.Subscribe(ctx, "test-channel")
					pubsub.Close()
				}()
			}

			wg.Wait()
		})
	}
}

// BenchmarkPubSubMultipleChannels benchmarks subscribing to multiple channels
func BenchmarkPubSubMultipleChannels(b *testing.B) {