package cache

import (
	"fmt"
	"time"

	"gitee.com/kyugao/ujson"
	"github.com/gomodule/redigo/redis"
	"github.com/jinzhu/configor"
	"github.com/rs/zerolog/log"
)

var Config = struct {
	Ip      string
	Port    string
	Db      string
	Session struct {
		Db                      string
		TempSessionAliveSeconds int
		ActiveSeconds           int
		Refresh                 bool
	}
}{}

var pool redis.Pool

func init() {
	err := configor.Load(&Config, "conf/cache.yml")
	if err != nil {
		panic(err)
	}
	log.Info().Msgf("config=%s", ujson.ToJsonRawString(&Config))
	address := fmt.Sprintf("%s:%s", Config.Ip, Config.Port)
	pool = redis.Pool{
		Dial: func() (c redis.Conn, err error) {
			c, err = redis.Dial("tcp", address)
			if err != nil {
				panic(err)
			}
			return
		},
		TestOnBorrow:    nil,
		MaxIdle:         3,
		MaxActive:       20,
		IdleTimeout:     240 * time.Second,
		Wait:            false,
		MaxConnLifetime: 0,
	}
}

func GetConn() (conn redis.Conn) {
	conn = GetConnWithNum(Config.Db)
	return
}

func GetConnWithNum(dbNum string) (conn redis.Conn) {
	conn = pool.Get()
	conn.Do("SELECT", dbNum)
	return
}