Function BenchmarkPubSubMultipleChannels

pool_pubsub_bench_test.go:274–298

BenchmarkPubSubMultipleChannels benchmarks subscribing to multiple channels

(b *testing.B)

// BenchmarkPubSubMultipleChannels benchmarks subscribing to multiple channels
func BenchmarkPubSubMultipleChannels(b *testing.B) {
	ctx := context.Background()
	client := benchmarkClient(16)
	defer client.Close()

	channelCounts := []int{1, 5, 10, 25, 50, 100}

	for _, channelCount := range channelCounts {
		b.Run(fmt.Sprintf("Channels_%d", channelCount), func(b *testing.B) {
			// Prepare channel names
			channels := make([]string, channelCount)
			for i := 0; i < channelCount; i++ {
				channels[i] = fmt.Sprintf("channel-%d", i)
			}

			b.ResetTimer()
			b.ReportAllocs()

			for i := 0; i < b.N; i++ {
				pubsub := client.Subscribe(ctx, channels...)
				pubsub.Close()
			}
		})
	}
}

// BenchmarkPubSubReuse benchmarks reusing PubSub connections
func BenchmarkPubSubReuse(b *testing.B) {