bench_test.go

package sharding_test

import (
	"math/rand"
	"testing"
	"time"

	"github.com/go-pg/sharding/v8"

	"github.com/go-pg/pg/v10"
)

func benchmarkDB() *pg.DB {
	return pg.Connect(&pg.Options{
		User:         "postgres",
		Database:     "postgres",
		DialTimeout:  30 * time.Second,
		ReadTimeout:  10 * time.Second,
		WriteTimeout: 10 * time.Second,
		PoolSize:     10,
		PoolTimeout:  30 * time.Second,
	})
}

func BenchmarkGopg(b *testing.B) {
	db := benchmarkDB()
	defer db.Close()

	b.ResetTimer()

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			_, err := db.Exec("SELECT 1")
			if err != nil {
				b.Fatal(err)
			}
		}
	})
}

func BenchmarkCluster(b *testing.B) {
	db := benchmarkDB()
	defer db.Close()

	cluster := sharding.NewCluster([]*pg.DB{db}, 1)
	defer cluster.Close()

	b.ResetTimer()

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			_, err := cluster.Shard(0).Exec("SELECT 1")
			if err != nil {
				b.Fatal(err)
			}
		}
	})
}

var sink *sharding.SubCluster

func BenchmarkSubCluster(b *testing.B) {
	db := benchmarkDB()
	defer db.Close()

	cluster := sharding.NewCluster([]*pg.DB{db}, 1)
	defer cluster.Close()

	b.RunParallel(func(pb *testing.PB) {
		n := rand.Int63()
		for pb.Next() {
			sink = cluster.SubCluster(n, 32)
		}
	})
}

func BenchmarkNewUUID(b *testing.B) {
	tm := time.Now()

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			_ = sharding.NewUUID(0, tm)
		}
	})
}