102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/goravel/framework/facades"
|
|
|
|
"goravel/app/models"
|
|
)
|
|
|
|
// GetConfigValue 从数据库获取配置值
|
|
// group: 配置分组
|
|
// key: 配置键
|
|
// defaultValue: 默认值（如果配置不存在）
|
|
func GetConfigValue(group, key string, defaultValue string) string {
|
|
// 使用 recover 来捕获可能的 panic（例如在构建时数据库不可用）
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
// 静默处理，返回默认值
|
|
}
|
|
}()
|
|
|
|
// 尝试检查数据库连接是否可用，如果不可用则直接返回默认值
|
|
// 这样可以避免在构建时执行数据库查询
|
|
orm := facades.Orm()
|
|
if orm == nil {
|
|
return defaultValue
|
|
}
|
|
|
|
var config models.Config
|
|
err := orm.Query().Where("group", group).Where("key", key).First(&config)
|
|
if err != nil {
|
|
return defaultValue
|
|
}
|
|
if config.Value == "" {
|
|
return defaultValue
|
|
}
|
|
return config.Value
|
|
}
|
|
|
|
// GetConfigValueInt 从数据库获取配置值（整数类型）
|
|
func GetConfigValueInt(group, key string, defaultValue int) int {
|
|
// 使用 recover 来捕获可能的 panic（例如在构建时数据库不可用）
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
// 静默处理，返回默认值
|
|
}
|
|
}()
|
|
|
|
// 尝试检查数据库连接是否可用，如果不可用则直接返回默认值
|
|
// 这样可以避免在构建时执行数据库查询
|
|
orm := facades.Orm()
|
|
if orm == nil {
|
|
return defaultValue
|
|
}
|
|
|
|
var config models.Config
|
|
err := orm.Query().Where("group", group).Where("key", key).First(&config)
|
|
if err != nil {
|
|
return defaultValue
|
|
}
|
|
if config.Value == "" {
|
|
return defaultValue
|
|
}
|
|
// 简单的字符串转整数，实际可以使用更完善的转换
|
|
value := 0
|
|
_, err = fmt.Sscanf(config.Value, "%d", &value)
|
|
if err != nil {
|
|
return defaultValue
|
|
}
|
|
return value
|
|
}
|
|
|
|
// GetConfigValueBool 从数据库获取配置值（布尔类型）
|
|
func GetConfigValueBool(group, key string, defaultValue bool) bool {
|
|
// 使用 recover 来捕获可能的 panic（例如在构建时数据库不可用）
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
// 静默处理，返回默认值
|
|
}
|
|
}()
|
|
|
|
// 尝试检查数据库连接是否可用，如果不可用则直接返回默认值
|
|
// 这样可以避免在构建时执行数据库查询
|
|
orm := facades.Orm()
|
|
if orm == nil {
|
|
return defaultValue
|
|
}
|
|
|
|
var config models.Config
|
|
err := orm.Query().Where("group", group).Where("key", key).First(&config)
|
|
if err != nil {
|
|
return defaultValue
|
|
}
|
|
if config.Value == "" {
|
|
return defaultValue
|
|
}
|
|
// 支持 "1", "true", "True" 等格式
|
|
value := config.Value
|
|
return value == "1" || value == "true" || value == "True" || value == "TRUE"
|
|
}
|