152 lines
3.8 KiB
Go
152 lines
3.8 KiB
Go
package cacheservice
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"go-admin/app/admin/models"
|
||
"go-admin/common/const/rediskey"
|
||
"go-admin/common/global"
|
||
"go-admin/common/helper"
|
||
models2 "go-admin/models"
|
||
|
||
"github.com/bytedance/sonic"
|
||
"github.com/go-admin-team/go-admin-core/logger"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// 加载config缓存
|
||
func InitConfigCache(db *gorm.DB) {
|
||
datas := []models.SysConfig{}
|
||
db.Model(&models.SysConfig{}).Find(&datas)
|
||
for _, item := range datas {
|
||
cacheKey := fmt.Sprintf(rediskey.SysConfigKey, item.ConfigKey)
|
||
|
||
content, err := sonic.Marshal(item)
|
||
|
||
if err != nil {
|
||
logger.Error(err)
|
||
continue
|
||
}
|
||
|
||
if len(content) > 0 {
|
||
helper.DefaultRedis.SetString(cacheKey, item.ConfigValue)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 获取 sys config缓存
|
||
func GetConfigCacheByKey(db *gorm.DB, key string) models.SysConfig {
|
||
cacheKey := fmt.Sprintf(rediskey.SysConfigKey, key)
|
||
val, _ := helper.DefaultRedis.GetString(key)
|
||
result := models.SysConfig{}
|
||
|
||
if val == "" {
|
||
sonic.Unmarshal([]byte(val), &result)
|
||
}
|
||
|
||
if result.Id == 0 {
|
||
if err := db.Model(&result).Where("config_key = ?", key).First(&result).Error; err != nil {
|
||
logger.Errorf("GetConfigCacheByKey %s", err.Error())
|
||
}
|
||
}
|
||
|
||
if result.Id > 0 {
|
||
content, err := sonic.Marshal(result)
|
||
|
||
if err != nil {
|
||
logger.Errorf("GetConfigCacheByKey %s", err.Error())
|
||
} else {
|
||
helper.DefaultRedis.SetString(cacheKey, string(content))
|
||
}
|
||
}
|
||
|
||
return result
|
||
}
|
||
|
||
// 获取缓存交易对
|
||
// symbolType 0-现货 1-合约
|
||
func GetTradeSet(exchangeType string, symbol string, symbolType int) (models2.TradeSet, error) {
|
||
// 定义返回结果和val变量
|
||
result := models2.TradeSet{}
|
||
val := ""
|
||
|
||
// 根据交易对类型选择不同的key
|
||
switch symbolType {
|
||
case 0:
|
||
key := fmt.Sprintf(global.TICKER_SPOT, exchangeType, symbol)
|
||
val, _ = helper.DefaultRedis.GetString(key)
|
||
case 1:
|
||
key := fmt.Sprintf(global.TICKER_FUTURES, exchangeType, symbol)
|
||
val, _ = helper.DefaultRedis.GetString(key)
|
||
}
|
||
|
||
// 如果val不为空，则解析val为TradeSet结构体
|
||
if val != "" {
|
||
if err := sonic.Unmarshal([]byte(val), &result); err != nil {
|
||
return result, err
|
||
}
|
||
} else {
|
||
// 如果val为空，则返回错误信息
|
||
return result, errors.New("未找到交易对信息")
|
||
}
|
||
|
||
// 返回结果
|
||
return result, nil
|
||
}
|
||
|
||
func GetSystemSetting(db *gorm.DB) (models.LineSystemSetting, error) {
|
||
key := fmt.Sprintf(rediskey.SystemSetting)
|
||
val, _ := helper.DefaultRedis.GetString(key)
|
||
setting := models.LineSystemSetting{}
|
||
|
||
if val != "" {
|
||
sonic.UnmarshalString(val, &setting)
|
||
}
|
||
|
||
if setting.Id > 0 {
|
||
return setting, nil
|
||
}
|
||
|
||
var err error
|
||
setting, err = ResetSystemSetting(db)
|
||
if err != nil {
|
||
return setting, err
|
||
}
|
||
|
||
return setting, nil
|
||
}
|
||
func ResetSystemSetting(db *gorm.DB) (models.LineSystemSetting, error) {
|
||
setting := models.LineSystemSetting{}
|
||
if err := db.Model(&setting).First(&setting).Error; err != nil {
|
||
return setting, err
|
||
}
|
||
|
||
settVal, _ := sonic.MarshalString(&setting)
|
||
|
||
if settVal != "" {
|
||
if err := helper.DefaultRedis.SetString(rediskey.SystemSetting, settVal); err != nil {
|
||
logger.Error("redis添加系统设置失败", err)
|
||
}
|
||
}
|
||
return models.LineSystemSetting{}, nil
|
||
}
|
||
|
||
// 获取反单配置
|
||
func GetReverseSetting(db *gorm.DB) (models.LineReverseSetting, error) {
|
||
setting := models.LineReverseSetting{}
|
||
|
||
helper.DefaultRedis.HGetAsObject(rediskey.ReverseSetting, &setting)
|
||
|
||
if setting.ReverseOrderType == "" {
|
||
if err := db.Model(&setting).First(&setting).Error; err != nil {
|
||
logger.Error("获取反单配置失败", err)
|
||
}
|
||
|
||
if err := helper.DefaultRedis.SetHashWithTags(rediskey.ReverseSetting, &setting); err != nil {
|
||
logger.Error("redis添加反单配置失败", err)
|
||
}
|
||
}
|
||
|
||
return setting, nil
|
||
}
|