mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-02-03 02:54:41 +08:00
312 lines
8.0 KiB
Go
312 lines
8.0 KiB
Go
// Package sys
|
||
// @Link https://github.com/bufanyun/hotgo
|
||
// @Copyright Copyright (c) 2023 HotGo CLI
|
||
// @Author Ms <133814250@qq.com>
|
||
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
||
package sys
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/gogf/gf/v2/database/gdb"
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gtime"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
"hotgo/internal/consts"
|
||
"hotgo/internal/dao"
|
||
"hotgo/internal/library/payment"
|
||
"hotgo/internal/library/storager"
|
||
"hotgo/internal/library/token"
|
||
"hotgo/internal/library/wechat"
|
||
"hotgo/internal/model"
|
||
"hotgo/internal/model/entity"
|
||
"hotgo/internal/model/input/sysin"
|
||
"hotgo/internal/service"
|
||
"hotgo/utility/simple"
|
||
)
|
||
|
||
type sSysConfig struct{}
|
||
|
||
func NewSysConfig() *sSysConfig {
|
||
return &sSysConfig{}
|
||
}
|
||
|
||
func init() {
|
||
service.RegisterSysConfig(NewSysConfig())
|
||
}
|
||
|
||
// InitConfig 初始化一些系统启动就需要用到的配置
|
||
func (s *sSysConfig) InitConfig(ctx context.Context) {
|
||
wx, err := s.GetWechat(ctx)
|
||
if err != nil {
|
||
g.Log().Fatalf(ctx, "init wechat conifg fail：%+v", err)
|
||
}
|
||
wechat.SetConfig(wx)
|
||
|
||
pay, err := s.GetPay(ctx)
|
||
if err != nil {
|
||
g.Log().Fatalf(ctx, "init pay conifg fail：%+v", err)
|
||
}
|
||
payment.SetConfig(pay)
|
||
|
||
upload, err := s.GetUpload(ctx)
|
||
if err != nil {
|
||
g.Log().Fatalf(ctx, "init upload conifg fail：%+v", err)
|
||
}
|
||
storager.SetConfig(upload)
|
||
|
||
tk, err := s.GetLoadToken(ctx)
|
||
if err != nil {
|
||
g.Log().Fatalf(ctx, "init token conifg fail：%+v", err)
|
||
}
|
||
token.SetConfig(tk)
|
||
}
|
||
|
||
// GetLogin 获取登录配置
|
||
func (s *sSysConfig) GetLogin(ctx context.Context) (conf *model.LoginConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "login"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetWechat 获取微信配置
|
||
func (s *sSysConfig) GetWechat(ctx context.Context) (conf *model.WechatConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "wechat"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetPay 获取支付配置
|
||
func (s *sSysConfig) GetPay(ctx context.Context) (conf *model.PayConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "pay"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetSms 获取短信配置
|
||
func (s *sSysConfig) GetSms(ctx context.Context) (conf *model.SmsConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "sms"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetGeo 获取地理配置
|
||
func (s *sSysConfig) GetGeo(ctx context.Context) (conf *model.GeoConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "geo"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetUpload 获取上传配置
|
||
func (s *sSysConfig) GetUpload(ctx context.Context) (conf *model.UploadConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "upload"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetSmtp 获取邮件配置
|
||
func (s *sSysConfig) GetSmtp(ctx context.Context) (conf *model.EmailConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "smtp"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
if err = gconv.Scan(models.List, &conf); err != nil {
|
||
return
|
||
}
|
||
|
||
conf.Addr = fmt.Sprintf("%s:%d", conf.Host, conf.Port)
|
||
|
||
return
|
||
}
|
||
|
||
// GetBasic 获取基础配置
|
||
func (s *sSysConfig) GetBasic(ctx context.Context) (conf *model.BasicConfig, err error) {
|
||
models, err := s.GetConfigByGroup(ctx, sysin.GetConfigInp{Group: "basic"})
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = gconv.Scan(models.List, &conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadTCP 获取本地tcp配置
|
||
func (s *sSysConfig) GetLoadTCP(ctx context.Context) (conf *model.TCPConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "tcp").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadCache 获取本地缓存配置
|
||
func (s *sSysConfig) GetLoadCache(ctx context.Context) (conf *model.CacheConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "cache").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadGenerate 获取本地生成配置
|
||
func (s *sSysConfig) GetLoadGenerate(ctx context.Context) (conf *model.GenerateConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "hggen").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadToken 获取本地token配置
|
||
func (s *sSysConfig) GetLoadToken(ctx context.Context) (conf *model.TokenConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "token").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadLog 获取本地日志配置
|
||
func (s *sSysConfig) GetLoadLog(ctx context.Context) (conf *model.LogConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "hotgo.log").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetLoadServeLog 获取本地服务日志配置
|
||
func (s *sSysConfig) GetLoadServeLog(ctx context.Context) (conf *model.ServeLogConfig, err error) {
|
||
err = g.Cfg().MustGet(ctx, "hotgo.serveLog").Scan(&conf)
|
||
return
|
||
}
|
||
|
||
// GetConfigByGroup 获取指定分组的配置
|
||
func (s *sSysConfig) GetConfigByGroup(ctx context.Context, in sysin.GetConfigInp) (res *sysin.GetConfigModel, err error) {
|
||
if in.Group == "" {
|
||
err = gerror.New("分组不能为空")
|
||
return
|
||
}
|
||
|
||
var models []*entity.SysConfig
|
||
if err = dao.SysConfig.Ctx(ctx).Fields("key", "value", "type").Where("group", in.Group).Scan(&models); err != nil {
|
||
err = gerror.Wrapf(err, "获取配置分组[ %v ]失败，请稍后重试！", in.Group)
|
||
return
|
||
}
|
||
|
||
res = new(sysin.GetConfigModel)
|
||
if len(models) > 0 {
|
||
res.List = make(g.Map, len(models))
|
||
for _, v := range models {
|
||
val, err := s.ConversionType(ctx, v)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
res.List[v.Key] = val
|
||
}
|
||
}
|
||
|
||
res.List = simple.FilterMaskDemo(ctx, res.List)
|
||
return
|
||
}
|
||
|
||
// ConversionType 转换类型
|
||
func (s *sSysConfig) ConversionType(ctx context.Context, models *entity.SysConfig) (value interface{}, err error) {
|
||
if models == nil {
|
||
err = gerror.New("数据不存在")
|
||
return
|
||
}
|
||
return consts.ConvType(models.Value, models.Type), nil
|
||
}
|
||
|
||
// UpdateConfigByGroup 更新指定分组的配置
|
||
func (s *sSysConfig) UpdateConfigByGroup(ctx context.Context, in sysin.UpdateConfigInp) (err error) {
|
||
if in.Group == "" {
|
||
err = gerror.New("分组不能为空")
|
||
return
|
||
}
|
||
var (
|
||
mod = dao.SysConfig.Ctx(ctx)
|
||
models []*entity.SysConfig
|
||
)
|
||
|
||
if err = mod.Where("group", in.Group).Scan(&models); err != nil {
|
||
return
|
||
}
|
||
|
||
err = dao.SysConfig.Transaction(ctx, func(ctx context.Context, tx gdb.TX) (err error) {
|
||
for k, v := range in.List {
|
||
row := s.getConfigByKey(k, models)
|
||
// 新增
|
||
if row == nil {
|
||
//row.Id = 0
|
||
//row.Key = k
|
||
//row.Value = gconv.String(v)
|
||
//row.Group = in.Group
|
||
//row.Status = consts.StatusEnabled
|
||
//row.CreatedAt = gtime.Now()
|
||
//row.UpdatedAt = gtime.Now()
|
||
//_, err := dao.SysConfig.Ctx(ctx).Data(row).Insert()
|
||
//if err != nil {
|
||
// err = gerror.Wrap(err, consts.ErrorORM)
|
||
// return err
|
||
//}
|
||
//continue
|
||
err = gerror.Newf("暂不支持从前台添加变量，请先在数据库表[%v]中配置变量：%v", dao.SysConfig.Table(), k)
|
||
return
|
||
}
|
||
|
||
// 更新
|
||
_, err = dao.SysConfig.Ctx(ctx).Where("id", row.Id).Data(g.Map{"value": v, "updated_at": gtime.Now()}).Update()
|
||
if err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
return s.syncUpdate(ctx, in)
|
||
})
|
||
return
|
||
}
|
||
|
||
func (s *sSysConfig) getConfigByKey(key string, models []*entity.SysConfig) *entity.SysConfig {
|
||
if len(models) == 0 {
|
||
return nil
|
||
}
|
||
|
||
for _, v := range models {
|
||
if key == v.Key {
|
||
return v
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// syncUpdate 同步更新一些加载配置
|
||
func (s *sSysConfig) syncUpdate(ctx context.Context, in sysin.UpdateConfigInp) (err error) {
|
||
switch in.Group {
|
||
case "wechat":
|
||
wx, err := s.GetWechat(ctx)
|
||
if err == nil {
|
||
wechat.SetConfig(wx)
|
||
}
|
||
case "pay":
|
||
pay, err := s.GetPay(ctx)
|
||
if err == nil {
|
||
payment.SetConfig(pay)
|
||
}
|
||
case "upload":
|
||
upload, err := s.GetUpload(ctx)
|
||
if err == nil {
|
||
storager.SetConfig(upload)
|
||
}
|
||
}
|
||
|
||
if err != nil {
|
||
err = gerror.Newf("syncUpdate %v conifg fail：%+v", in.Group, err.Error())
|
||
}
|
||
return
|
||
}
|