conf_helper.go

/*
*	Description : 配置文件 yaml  TODO 测试
*	Author 		: ManGe
*	Mail 		: 2912882908@qq.com
**/

package gathertool

import (
	"fmt"
	"os"

	yaml "gopkg.in/yaml.v3"
)

// TODO 扩展到所有能支持的配置文件

var Config *conf

type conf struct {
	Path string
	Data map[string]any
}

func NewConf(appConfigPath string) error {
	Config = &conf{
		Path: appConfigPath,
		Data: make(map[string]any),
	}
	err := Config.Init()
	return err
}

func (c *conf) Init() error {
	if !FileExists(c.Path) {
		return fmt.Errorf("未找到配置文件: %v", c.Path)
	}
	Info("读取配置文件:", c.Path)
	//读取yaml文件到缓存中
	config, err := os.ReadFile(c.Path)
	if err != nil {
		ErrorF("读取配置文件 %v 失败", c.Path)
		return err
	}
	return yaml.Unmarshal(config, c.Data)
}

func (c *conf) GetInt(key string) int {
	if c.Data == nil {
		_ = c.Init()
	}
	return Any2Int(c.Data[key])
}

func (c *conf) Get(key string) any {
	if c.Data == nil {
		_ = c.Init()
	}
	return c.Data[key]
}

func (c *conf) GetStr(key string) string {
	if c.Data == nil {
		_ = c.Init()
	}
	return Any2String(c.Data[key])
}