20 Star 167 Fork 28

qiqi/orange

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
init.go 1.23 KB
一键复制 编辑 原始数据 按行查看 历史
qiqi 提交于 2020-05-08 01:16 . 更新配置包版本
package cfg
import (
"fmt"
"gitee.com/zhucheer/cfg"
"gitee.com/zhucheer/orange/utils"
)
var Config = cfg.New("")
var ConfigDef = cfg.New("")
func init() {
NewFlagParam()
// 注册配置文件参数
SetStringFlag("config", "./config/config.toml", "config file path")
}
func StartConfig() {
configPath := GetStringFlag("config")
if isFile, _ := utils.FileExists(configPath); isFile == false {
fmt.Println(fmt.Sprintf("\033[0;33m%v \033[0m", "[WARNING] "+configPath+" not found, please check the config path"))
Config = cfg.NewTomlData(defaultConfig)
} else {
Config = cfg.New(configPath)
}
if Config == nil {
panic("config init failed")
}
// 加载默认配置对象
ConfigDef = cfg.NewTomlData(defaultConfig)
}
// 门面方法快速调用
// GetString 获取字符型配置
func GetString(key string, defaultVal string) string {
if Config.Exists(key) {
return Config.GetString(key)
}
return defaultVal
}
// GetInt 获取int类型配置
func GetInt(key string, defaultVal int) int {
if Config.Exists(key) {
return Config.GetInt(key)
}
return defaultVal
}
// GetBool 获取bool类型配置
func GetBool(key string, defaultVal bool) bool {
if Config.Exists(key) {
return Config.GetBool(key)
}
return defaultVal
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/zhucheer/orange.git
git@gitee.com:zhucheer/orange.git
zhucheer
orange
orange
v0.2.4

搜索帮助