mirror of
https://github.com/hanzokv/go.git
synced 2026-09-24 00:31:51 +00:00
zeekay
24f201b8e5
Update module path from github.com/redis/go-redis/v9 to github.com/hanzoai/kv-go/v9 for use as the Hanzo KV Go SDK. - Update go.mod module declaration - Update all Go imports across 260 files - Update all submodule go.mod files (extra/*, example/*, internal/customvet)
101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package pool_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoai/kv-go/v9/internal/pool"
|
|
)
|
|
|
|
type poolGetPutBenchmark struct {
|
|
poolSize int
|
|
}
|
|
|
|
func (bm poolGetPutBenchmark) String() string {
|
|
return fmt.Sprintf("pool=%d", bm.poolSize)
|
|
}
|
|
|
|
func BenchmarkPoolGetPut(b *testing.B) {
|
|
ctx := context.Background()
|
|
benchmarks := []poolGetPutBenchmark{
|
|
{1},
|
|
{2},
|
|
{8},
|
|
{32},
|
|
{64},
|
|
{128},
|
|
}
|
|
for _, bm := range benchmarks {
|
|
b.Run(bm.String(), func(b *testing.B) {
|
|
connPool := pool.NewConnPool(&pool.Options{
|
|
Dialer: dummyDialer,
|
|
PoolSize: int32(bm.poolSize),
|
|
MaxConcurrentDials: bm.poolSize,
|
|
PoolTimeout: time.Second,
|
|
DialTimeout: 1 * time.Second,
|
|
ConnMaxIdleTime: time.Hour,
|
|
})
|
|
|
|
b.ResetTimer()
|
|
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
cn, err := connPool.Get(ctx)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
connPool.Put(ctx, cn)
|
|
}
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
type poolGetRemoveBenchmark struct {
|
|
poolSize int
|
|
}
|
|
|
|
func (bm poolGetRemoveBenchmark) String() string {
|
|
return fmt.Sprintf("pool=%d", bm.poolSize)
|
|
}
|
|
|
|
func BenchmarkPoolGetRemove(b *testing.B) {
|
|
ctx := context.Background()
|
|
benchmarks := []poolGetRemoveBenchmark{
|
|
{1},
|
|
{2},
|
|
{8},
|
|
{32},
|
|
{64},
|
|
{128},
|
|
}
|
|
|
|
for _, bm := range benchmarks {
|
|
b.Run(bm.String(), func(b *testing.B) {
|
|
connPool := pool.NewConnPool(&pool.Options{
|
|
Dialer: dummyDialer,
|
|
PoolSize: int32(bm.poolSize),
|
|
MaxConcurrentDials: bm.poolSize,
|
|
PoolTimeout: time.Second,
|
|
DialTimeout: 1 * time.Second,
|
|
ConnMaxIdleTime: time.Hour,
|
|
})
|
|
|
|
b.ResetTimer()
|
|
rmvErr := errors.New("Bench test remove")
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
cn, err := connPool.Get(ctx)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
connPool.Remove(ctx, cn, rmvErr)
|
|
}
|
|
})
|
|
})
|
|
}
|
|
}
|