mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 16:00:25 +08:00
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
package entity
|
||
|
||
import (
|
||
"encoding/json"
|
||
"mayfly-go/pkg/model"
|
||
"strconv"
|
||
)
|
||
|
||
const (
|
||
ConfigKeyAccountLoginSecurity string = "AccountLoginSecurity" // 账号登录安全配置
|
||
ConfigKeyUseLoginCaptcha string = "UseLoginCaptcha" // 是否使用登录验证码
|
||
ConfigKeyUseLoginOtp string = "UseLoginOtp" // 是否开启otp双因素校验
|
||
ConfigKeyDbQueryMaxCount string = "DbQueryMaxCount" // 数据库查询的最大数量
|
||
ConfigKeyDbSaveQuerySQL string = "DbSaveQuerySQL" // 数据库是否记录查询相关sql
|
||
ConfigUseWartermark string = "UseWartermark" // 是否使用水印
|
||
)
|
||
|
||
type Config struct {
|
||
model.Model
|
||
Name string `json:"name"` // 配置名
|
||
Key string `json:"key"` // 配置key
|
||
Params string `json:"params" gorm:"column:params;type:varchar(1000)"`
|
||
Value string `json:"value" gorm:"column:value;type:varchar(1000)"`
|
||
Remark string `json:"remark"`
|
||
}
|
||
|
||
func (a *Config) TableName() string {
|
||
return "t_sys_config"
|
||
}
|
||
|
||
// 若配置信息不存在, 则返回传递的默认值.
|
||
func (c *Config) BoolValue(defaultValue bool) bool {
|
||
// 如果值不存在，则返回默认值
|
||
if c.Id == 0 {
|
||
return defaultValue
|
||
}
|
||
return convertBool(c.Value, defaultValue)
|
||
}
|
||
|
||
// 值返回json map
|
||
func (c *Config) GetJsonMap() map[string]string {
|
||
var res map[string]string
|
||
if c.Id == 0 || c.Value == "" {
|
||
return res
|
||
}
|
||
_ = json.Unmarshal([]byte(c.Value), &res)
|
||
return res
|
||
}
|
||
|
||
// 获取配置的int值，如果配置值非int或不存在，则返回默认值
|
||
func (c *Config) IntValue(defaultValue int) int {
|
||
// 如果值不存在，则返回默认值
|
||
if c.Id == 0 {
|
||
return defaultValue
|
||
}
|
||
return convertInt(c.Value, defaultValue)
|
||
}
|
||
|
||
type AccountLoginSecurity struct {
|
||
UseCaptcha bool // 是否使用登录验证码
|
||
UseOtp bool // 是否双因素校验
|
||
OtpIssuer string // otp发行人
|
||
LoginFailCount int // 允许失败次数
|
||
LoginFailMin int // 登录失败指定次数后禁止的分钟数
|
||
}
|
||
|
||
// 转换为AccountLoginSecurity结构体
|
||
func (c *Config) ToAccountLoginSecurity() *AccountLoginSecurity {
|
||
jm := c.GetJsonMap()
|
||
als := new(AccountLoginSecurity)
|
||
als.UseCaptcha = convertBool(jm["useCaptcha"], true)
|
||
als.UseOtp = convertBool(jm["useOtp"], false)
|
||
als.LoginFailCount = convertInt(jm["loginFailCount"], 5)
|
||
als.LoginFailMin = convertInt(jm["loginFailMin"], 10)
|
||
otpIssuer := jm["otpIssuer"]
|
||
if otpIssuer == "" {
|
||
otpIssuer = "mayfly-go"
|
||
}
|
||
als.OtpIssuer = otpIssuer
|
||
return als
|
||
}
|
||
|
||
// 转换配置中的值为bool类型（默认"1"或"true"为true，其他为false）
|
||
func convertBool(value string, defaultValue bool) bool {
|
||
if value == "" {
|
||
return defaultValue
|
||
}
|
||
return value == "1" || value == "true"
|
||
}
|
||
|
||
// 转换配置值中的值为int
|
||
func convertInt(value string, defaultValue int) int {
|
||
if value == "" {
|
||
return defaultValue
|
||
}
|
||
if intV, err := strconv.Atoi(value); err != nil {
|
||
return defaultValue
|
||
} else {
|
||
return intV
|
||
}
|
||
}
|