1 Star 0 Fork 0

江苏岚江智能科技有限公司/ljmicro

加入 Gitee
与超过 1400万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
conf_loader_file.go 2.60 KB
一键复制 编辑 原始数据 按行查看 历史
吴文凯 提交于 2023-03-13 16:50 +08:00 . update
package conf
import (
"errors"
"fmt"
"gitee.com/jiangsu-lanjiang-intelligent/ljmicro/pkg/utils/xcodec"
"os"
"path"
"path/filepath"
)
// todo 监听文件变化
type ConfLoaderFile struct {
filePath string
codec string
sourceData []byte
config *Configuration
}
func (f *ConfLoaderFile) LoadConf(c *Configuration) error {
fmt.Println("-----------------------开始初始化File加载器-----------------------")
f.config = c
f.filePath = c.GetString("config.file.path")
if f.filePath == "" {
fmt.Println("未找到配置项[config.file.path]")
fmt.Println("开始尝试搜索配置文件")
if f.filePath = tryFindConfigFilePath(c); f.filePath == "" {
fmt.Println("未找到配置文件,停止初始化File配置加载器")
return ErrUnNeedLoad
}
fmt.Println(fmt.Sprintf("自动搜索到配置文件[%s]", f.filePath))
}
return f.loadConfigFile()
}
// tryFindConfigFilePath 尝试寻找配置文件
func tryFindConfigFilePath(c *Configuration) string {
filePath := c.GetStringd("config.file.name", "config.yaml")
for i := 0; i < 10; i++ {
// 检查配置文件是否存在
if _, err := os.ReadFile(filePath); err == nil {
return filePath
}
filePath = "../" + filePath
}
return ""
}
func (f *ConfLoaderFile) loadConfigFile() error {
absolutePath, err := filepath.Abs(f.filePath)
if err != nil {
return errors.New(fmt.Sprintf("get file config absolute path error. path:%s, err: %v", f.filePath, err))
}
fmt.Println(fmt.Sprintf("开始加载配置文件[%s]", absolutePath))
contentData, err := os.ReadFile(absolutePath)
if err != nil {
return errors.New(fmt.Sprintf("read file by absolute path error. absolute_path:%s, err: %v", absolutePath, err))
}
f.sourceData = contentData
// todo 判断是否需要输出配置文件内容
switch path.Ext(f.filePath) {
case ".yaml":
f.codec = xcodec.XcodecYaml
case ".json":
f.codec = xcodec.XcodecJson
default:
f.codec = xcodec.XcodecJson
}
conf := map[string]interface{}{}
if err := xcodec.Decode(f.codec, contentData, &conf); err != nil {
return errors.New(fmt.Sprintf("decode file config data failed: %v,codec: %s", err, f.codec))
}
f.setConfigByMap(conf)
fmt.Println("File配置加载器加载完成")
return nil
}
func (f *ConfLoaderFile) setConfigByMap(data map[string]interface{}) {
var recursionSetConfig func(key string, d map[string]interface{})
recursionSetConfig = func(key string, d map[string]interface{}) {
for k, v := range d {
if vv, ok := v.(map[string]interface{}); ok {
recursionSetConfig(key+k+".", vv)
} else {
f.config.SetConfig(key+k, v)
}
}
}
recursionSetConfig("", data)
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/jiangsu-lanjiang-intelligent/ljmicro.git
git@gitee.com:jiangsu-lanjiang-intelligent/ljmicro.git
jiangsu-lanjiang-intelligent
ljmicro
ljmicro
9f39f3984435

搜索帮助