|
|
package model
|
|
|
|
|
|
import (
|
|
|
"net/url"
|
|
|
"strconv"
|
|
|
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
|
|
|
"github.com/jinzhu/gorm"
|
|
|
)
|
|
|
|
|
|
// Setting 系统设置模型
|
|
|
type Setting struct {
|
|
|
gorm.Model
|
|
|
Type string `gorm:"not null"`
|
|
|
Name string `gorm:"unique;not null;index:setting_key"`
|
|
|
Value string `gorm:"size:65535"`
|
|
|
}
|
|
|
|
|
|
// IsTrueVal 返回设置的值是否为真
|
|
|
func IsTrueVal(val string) bool {
|
|
|
return val == "1" || val == "true"
|
|
|
}
|
|
|
|
|
|
// GetSettingByName 用 Name 获取设置值
|
|
|
func GetSettingByName(name string) string {
|
|
|
return GetSettingByNameFromTx(DB, name)
|
|
|
}
|
|
|
|
|
|
// GetSettingByNameFromTx 用 Name 获取设置值，使用事务
|
|
|
func GetSettingByNameFromTx(tx *gorm.DB, name string) string {
|
|
|
var setting Setting
|
|
|
|
|
|
// 优先从缓存中查找
|
|
|
cacheKey := "setting_" + name
|
|
|
if optionValue, ok := cache.Get(cacheKey); ok {
|
|
|
return optionValue.(string)
|
|
|
}
|
|
|
|
|
|
// 尝试数据库中查找
|
|
|
if tx == nil {
|
|
|
tx = DB
|
|
|
if tx == nil {
|
|
|
return ""
|
|
|
}
|
|
|
}
|
|
|
|
|
|
result := tx.Where("name = ?", name).First(&setting)
|
|
|
if result.Error == nil {
|
|
|
_ = cache.Set(cacheKey, setting.Value, -1)
|
|
|
return setting.Value
|
|
|
}
|
|
|
|
|
|
return ""
|
|
|
}
|
|
|
|
|
|
// GetSettingByNameWithDefault 用 Name 获取设置值, 取不到时使用缺省值
|
|
|
func GetSettingByNameWithDefault(name, fallback string) string {
|
|
|
res := GetSettingByName(name)
|
|
|
if res == "" {
|
|
|
return fallback
|
|
|
}
|
|
|
return res
|
|
|
}
|
|
|
|
|
|
// GetSettingByNames 用多个 Name 获取设置值
|
|
|
func GetSettingByNames(names ...string) map[string]string {
|
|
|
var queryRes []Setting
|
|
|
res, miss := cache.GetSettings(names, "setting_")
|
|
|
|
|
|
if len(miss) > 0 {
|
|
|
DB.Where("name IN (?)", miss).Find(&queryRes)
|
|
|
for _, setting := range queryRes {
|
|
|
res[setting.Name] = setting.Value
|
|
|
}
|
|
|
}
|
|
|
|
|
|
_ = cache.SetSettings(res, "setting_")
|
|
|
return res
|
|
|
}
|
|
|
|
|
|
// GetSettingByType 获取一个或多个分组的所有设置值
|
|
|
func GetSettingByType(types []string) map[string]string {
|
|
|
var queryRes []Setting
|
|
|
res := make(map[string]string)
|
|
|
|
|
|
DB.Where("type IN (?)", types).Find(&queryRes)
|
|
|
for _, setting := range queryRes {
|
|
|
res[setting.Name] = setting.Value
|
|
|
}
|
|
|
|
|
|
return res
|
|
|
}
|
|
|
|
|
|
// GetSiteURL 获取站点地址
|
|
|
func GetSiteURL() *url.URL {
|
|
|
base, err := url.Parse(GetSettingByName("siteURL"))
|
|
|
if err != nil {
|
|
|
base, _ = url.Parse("https://cloudreve.org")
|
|
|
}
|
|
|
return base
|
|
|
}
|
|
|
|
|
|
// GetIntSetting 获取整形设置值，如果转换失败则返回默认值defaultVal
|
|
|
func GetIntSetting(key string, defaultVal int) int {
|
|
|
res, err := strconv.Atoi(GetSettingByName(key))
|
|
|
if err != nil {
|
|
|
return defaultVal
|
|
|
}
|
|
|
return res
|
|
|
}
|
