package redis_test

import (
	"os"
	"testing"

	"apigo.cc/go/config"
	"apigo.cc/go/redis"
)

func BenchmarkGetSet(b *testing.B) {
	os.Setenv("REDIS_TEST", "redis://:@localhost:6379/2?timeout=10ms&logSlow=10us")
	_ = config.Load("redis", nil)
	rd := redis.GetRedis("test", nil)
	rd.DEL("bench_key")

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		rd.SET("bench_key", "bench_value")
		_ = rd.GET("bench_key").String()
	}
	b.StopTimer()
	rd.DEL("bench_key")
}

func BenchmarkIDMaker(b *testing.B) {
	os.Setenv("REDIS_TEST", "redis://:@localhost:6379/2?timeout=10ms&logSlow=10us")
	_ = config.Load("redis", nil)
	rd := redis.GetRedis("test", nil)
	im := redis.NewIDMaker(rd)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		im.Get(16)
	}
}