Function BenchmarkPoolGetPutWithMinIdle

pool_pubsub_bench_test.go:100–140

BenchmarkPoolGetPutWithMinIdle benchmarks pool operations with MinIdleConns

(b *testing.B)

// BenchmarkPoolGetPutWithMinIdle benchmarks pool operations with MinIdleConns
func BenchmarkPoolGetPutWithMinIdle(b *testing.B) {
	ctx := context.Background()

	configs := []struct {
		poolSize     int
		minIdleConns int
	}{
		{8, 2},
		{16, 4},
		{32, 8},
		{64, 16},
	}

	for _, config := range configs {
		b.Run(fmt.Sprintf("Pool_%d_MinIdle_%d", config.poolSize, config.minIdleConns), func(b *testing.B) {
			connPool := pool.NewConnPool(&pool.Options{
				Dialer:             dummyDialer,
				PoolSize:           int32(config.poolSize),
				MaxConcurrentDials: config.poolSize,
				MinIdleConns:       int32(config.minIdleConns),
				PoolTimeout:        time.Second,
				DialTimeout:        time.Second,
				ConnMaxIdleTime:    time.Hour,
			})
			defer connPool.Close()

			b.ResetTimer()
			b.ReportAllocs()

			b.RunParallel(func(pb *testing.PB) {
				for pb.Next() {
					cn, err := connPool.Get(ctx)
					if err != nil {
						b.Fatal(err)
					}
					connPool.Put(ctx, cn)
				}
			})
		})
	}
}

// BenchmarkPoolConcurrentGetPut benchmarks pool under high concurrency
func BenchmarkPoolConcurrentGetPut(b *testing.B) {