Збільшення швидкодії Redis-у через розділення на домени

import (
	"sync/atomic"
	"testing"

	"github.com/go-redis/redis"
	"github.com/stretchr/testify/require"
)

var (
	keys = []string{
		"1",
		"2",
		"3",
		"4",
		"5",
	}
	keyCount = uint32(len(keys))
)

// https://github.com/go-redis/redis#quickstart
func Client(addr string) (*redis.Client, error) {
	client := redis.NewClient(&redis.Options{
		Addr:     addr, // "localhost:6379",
		Password: "",   // no password set
		DB:       0,    // use default DB
	})

	err := client.Ping().Err()
	if err != nil {
		return nil, err
	}

	return client, err
}

func BenchmarkRedis1Increment(b *testing.B) {
	const (
		addr1 = "redis1:6379"
	)

	benchmarkIncrement(b, addr1, addr1, addr1)
}

func BenchmarkRedisAllIncrement(b *testing.B) {
	const (
		addr1 = "redis1:6379"
		addr2 = "redis2:6379"
		addr3 = "redis3:6379"
	)

	benchmarkIncrement(b, addr1, addr2, addr3)
}

func benchmarkIncrement(b *testing.B, addrs ...string) {
	b.Helper()

	var (
		clientCount = uint32(len(addrs))
		clients     = make([]*redis.Client, clientCount)
	)

	for i, addr := range addrs {
		var client, err = Client(addr)
		require.NoError(b, err)
		defer client.Close()

		var flushAllErr = client.FlushAll().Err()
		require.NoError(b, flushAllErr)

		clients[i] = client
	}

	var counter = uint32(0)

	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			var index = atomic.AddUint32(&counter, 1)

			var client = clients[index%clientCount]

			var err = client.Incr(keys[index%keyCount]).Err()

			require.NoError(b, err)
		}
	})
}