056de22edb
* Feat: retrieve nodes from data table * Feat: master node ping slave node in REST API * Feat: master send scheduled ping request * Feat: inactive nodes recover loop * Modify: remove database operations from aria2 RPC caller implementation * Feat: init aria2 client in master node * Feat: Round Robin load balancer * Feat: create and monitor aria2 task in master node * Feat: salve receive and handle heartbeat * Fix: Node ID will be 0 in download record generated in older version * Feat: sign request headers with all `X-` prefix * Feat: API call to slave node will carry meta data in headers * Feat: call slave aria2 rpc method from master * Feat: get slave aria2 task status Feat: encode slave response data using gob * Feat: aria2 callback to master node / cancel or select task to slave node * Fix: use dummy aria2 client when caller initialize failed in master node * Feat: slave aria2 status event callback / salve RPC auth * Feat: prototype for slave driven filesystem * Feat: retry for init aria2 client in master node * Feat: init request client with global options * Feat: slave receive async task from master * Fix: competition write in request header * Refactor: dependency initialize order * Feat: generic message queue implementation * Feat: message queue implementation * Feat: master waiting slave transfer result * Feat: slave transfer file in stateless policy * Feat: slave transfer file in slave policy * Feat: slave transfer file in local policy * Feat: slave transfer file in OneDrive policy * Fix: failed to initialize update checker http client * Feat: list slave nodes for dashboard * Feat: test aria2 rpc connection in slave * Feat: add and save node * Feat: add and delete node in node pool * Fix: temp file cannot be removed when aria2 task fails * Fix: delete node in admin panel * Feat: edit node and get node info * Modify: delete unused settings
91 lines
2 KiB
Go
91 lines
2 KiB
Go
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 {
|
||
var setting Setting
|
||
|
||
// 优先从缓存中查找
|
||
cacheKey := "setting_" + name
|
||
if optionValue, ok := cache.Get(cacheKey); ok {
|
||
return optionValue.(string)
|
||
}
|
||
|
||
// 尝试数据库中查找
|
||
if DB != nil {
|
||
result := DB.Where("name = ?", name).First(&setting)
|
||
if result.Error == nil {
|
||
_ = cache.Set(cacheKey, setting.Value, -1)
|
||
return setting.Value
|
||
}
|
||
}
|
||
|
||
return ""
|
||
}
|
||
|
||
// 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
|
||
}
|