package xredis

import (
	"context"
	"gitee.com/igolang/pkg/xredis/hooks"
	"github.com/redis/go-redis/v9"
	"time"
)

type RedisConfig struct {
	Addr         string `json:"addr"`
	Password     string `json:"password"`
	DB           int    `json:"DB"`
	DialTimeout  int64  `json:"dialTimeout"`
	ReadTimeout  int64  `json:"readTimeout"`
	WriteTimeout int64  `json:"writeTimeout"`
	IdleTimeout  int64  `json:"idleTimeout"` // 连接生效时间，默认5分钟
	MaxRetries   int    `json:"maxRetries"`
	PoolSize     int    `json:"poolSize"`
	MinIdleConn  int    `json:"minIdleConn"`
	KeyPrefix    string `json:"keyPrefix"`
}

type Client struct {
	*redis.Client
	keyPrefix *hooks.KeyPrefix
}

func NewClient(config *RedisConfig) (client *Client, err error) {
	options := &redis.Options{
		Addr:         config.Addr,
		Password:     config.Password,
		DB:           config.DB,
		DialTimeout:  time.Duration(config.DialTimeout) * time.Millisecond,
		ReadTimeout:  time.Duration(config.ReadTimeout) * time.Millisecond,
		WriteTimeout: time.Duration(config.WriteTimeout) * time.Millisecond,
		MaxRetries:   config.MaxRetries,
		PoolSize:     config.PoolSize,
		MinIdleConns: config.MinIdleConn,
	}
	rdb := redis.NewClient(options)
	hk := hooks.NewKeyPrefix(config.KeyPrefix)
	client = &Client{
		Client:    rdb,
		keyPrefix: hk,
	}

	client.AddHook(hk)

	return
}

func (c *Client) KeyPrefix() string {
	return c.keyPrefix.Prefix()
}

func (c *Client) EnableKeyPrefix() {
	c.keyPrefix.Enable()
}

func (c *Client) DisableKeyPrefix() {
	c.keyPrefix.Disable()
}

type Logging interface {
	Printf(ctx context.Context, format string, v ...interface{})
}

func SetLogger(logger Logging) {
	redis.SetLogger(logger)
}