Function BenchmarkPoolConcurrentGetPut

pool_pubsub_bench_test.go:143–177

BenchmarkPoolConcurrentGetPut benchmarks pool under high concurrency

(b *testing.B)

// BenchmarkPoolConcurrentGetPut benchmarks pool under high concurrency
func BenchmarkPoolConcurrentGetPut(b *testing.B) {
	ctx := context.Background()

	connPool := pool.NewConnPool(&pool.Options{
		Dialer:             dummyDialer,
		PoolSize:           int32(32),
		MaxConcurrentDials: 32,
		PoolTimeout:        time.Second,
		DialTimeout:        time.Second,
		ConnMaxIdleTime:    time.Hour,
		MinIdleConns:       int32(0),
	})
	defer connPool.Close()

	b.ResetTimer()
	b.ReportAllocs()

	// Test with different levels of concurrency
	concurrencyLevels := []int{1, 2, 4, 8, 16, 32, 64}

	for _, concurrency := range concurrencyLevels {
		b.Run(fmt.Sprintf("Concurrency_%d", concurrency), func(b *testing.B) {
			b.SetParallelism(concurrency)
			b.RunParallel(func(pb *testing.PB) {
				for pb.Next() {
					cn, err := connPool.Get(ctx)
					if err != nil {
						b.Fatal(err)
					}
					connPool.Put(ctx, cn)
				}
			})
		})
	}
}

// =============================================================================
// PUBSUB BENCHMARKS