package fgtm

import (
	"errors"

	"gitee.com/wu-jin-feng/fgin/fgin_plugs/fgtm/fgtm_logger"
	"gitee.com/wu-jin-feng/fgin/fgin_plugs/fgtm/fgtm_mysql"
	"gitee.com/wu-jin-feng/fgin/fgin_plugs/fgtm/fgtm_redis"
	"gitee.com/wu-jin-feng/fgin/fgin_plugs/fgtm/fgtm_service"
	"github.com/redis/go-redis/v9"
	"gorm.io/gorm"
)

var tranType string

// 初始化日志器
func InitFgtmLogger(logger fgtm_logger.FgTmLogger) {
	fgtm_logger.SetLogger(logger)
}

// 以redis为存储
func InitRedisTm(redisCli redis.UniversalClient, tranName string) {
	tranType = "redis"
	fgtm_redis.SetFgtmGlobalRedisObj(redisCli, tranName)
}

// 以mysql为存储
func InitMysqlTm(mysqlCli *gorm.DB, tranName, transTb, tranActionsTb string) {
	tranType = "mysql"
	fgtm_mysql.SetFgtmGlobalMysqlObj(mysqlCli, tranName, transTb, tranActionsTb)
}

// 获取事务对象
func GetFgtm() (fgtm_service.FgtmInterface, error) {
	switch tranType {
	case "redis":
		return fgtm_redis.NewFgtmRedis(), nil
	case "mysql":
		return fgtm_mysql.NewFgtmMysql(), nil
	default:
		return nil, errors.New("请检查fgtm配置，事务存储类型只支持redis、mysql")
	}
}