mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-04 16:00:24 +08:00
124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package models
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
|
||
_ "github.com/go-sql-driver/mysql"
|
||
"github.com/iwind/TeaGo/Tea"
|
||
"github.com/iwind/TeaGo/dbs"
|
||
"github.com/iwind/TeaGo/types"
|
||
)
|
||
|
||
type SysSettingDAO dbs.DAO
|
||
|
||
func NewSysSettingDAO() *SysSettingDAO {
|
||
return dbs.NewDAO(&SysSettingDAO{
|
||
DAOObject: dbs.DAOObject{
|
||
DB: Tea.Env,
|
||
Table: "edgeSysSettings",
|
||
Model: new(SysSetting),
|
||
PkName: "id",
|
||
},
|
||
}).(*SysSettingDAO)
|
||
}
|
||
|
||
var SharedSysSettingDAO *SysSettingDAO
|
||
|
||
func init() {
|
||
dbs.OnReady(func() {
|
||
SharedSysSettingDAO = NewSysSettingDAO()
|
||
})
|
||
}
|
||
|
||
// 设置配置
|
||
func (this *SysSettingDAO) UpdateSetting(tx *dbs.Tx, codeFormat string, valueJSON []byte, codeFormatArgs ...interface{}) error {
|
||
if len(codeFormatArgs) > 0 {
|
||
codeFormat = fmt.Sprintf(codeFormat, codeFormatArgs...)
|
||
}
|
||
|
||
countRetries := 3
|
||
var lastErr error
|
||
for i := 0; i < countRetries; i++ {
|
||
settingId, err := this.Query(tx).
|
||
Attr("code", codeFormat).
|
||
ResultPk().
|
||
FindInt64Col(0)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if settingId == 0 {
|
||
// 新建
|
||
op := NewSysSettingOperator()
|
||
op.Code = codeFormat
|
||
op.Value = valueJSON
|
||
err = this.Save(tx, op)
|
||
if err != nil {
|
||
lastErr = err
|
||
|
||
// 因为错误的原因可能是因为code冲突，所以这里我们继续执行
|
||
continue
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 修改
|
||
op := NewSysSettingOperator()
|
||
op.Id = settingId
|
||
op.Value = valueJSON
|
||
err = this.Save(tx, op)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
return lastErr
|
||
}
|
||
|
||
// 读取配置
|
||
func (this *SysSettingDAO) ReadSetting(tx *dbs.Tx, code string, codeFormatArgs ...interface{}) (valueJSON []byte, err error) {
|
||
if len(codeFormatArgs) > 0 {
|
||
code = fmt.Sprintf(code, codeFormatArgs...)
|
||
}
|
||
col, err := this.Query(tx).
|
||
Attr("code", code).
|
||
Result("value").
|
||
FindStringCol("")
|
||
return []byte(col), err
|
||
}
|
||
|
||
// 对比配置中的数字大小
|
||
func (this *SysSettingDAO) CompareInt64Setting(tx *dbs.Tx, code string, anotherValue int64) (int8, error) {
|
||
valueJSON, err := this.ReadSetting(tx, code)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
value := types.Int64(string(valueJSON))
|
||
if value > anotherValue {
|
||
return 1, nil
|
||
}
|
||
if value < anotherValue {
|
||
return -1, nil
|
||
}
|
||
return 0, nil
|
||
}
|
||
|
||
// 读取全局配置
|
||
func (this *SysSettingDAO) ReadGlobalConfig(tx *dbs.Tx) (*serverconfigs.GlobalConfig, error) {
|
||
globalConfigData, err := this.ReadSetting(tx, systemconfigs.SettingCodeServerGlobalConfig)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if len(globalConfigData) == 0 {
|
||
return &serverconfigs.GlobalConfig{}, nil
|
||
}
|
||
config := &serverconfigs.GlobalConfig{}
|
||
err = json.Unmarshal(globalConfigData, config)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return config, nil
|
||
}
|