62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package config
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"log/slog"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
type Config struct {
|
||
Mqtt mqtt
|
||
}
|
||
|
||
// MQTT客户端配置
|
||
type mqtt struct {
|
||
Topic topic
|
||
ClientId string
|
||
Address string
|
||
Username string
|
||
Password string
|
||
KeepAlive uint16 // 保活时间间隔,单位s
|
||
ConnectRetryDelay uint16 // 连接重试延时,单位s
|
||
ConnectTimeout uint16 // 连接操作超时,单位s
|
||
}
|
||
|
||
type topic struct {
|
||
App string
|
||
}
|
||
|
||
var Cfg Config
|
||
|
||
// 获取配置文件名称,从运行flag参数config中获取，若未提供，使用默认'dev'
|
||
func getConfigName() string {
|
||
configName := ""
|
||
flag.StringVar(&configName, "config", "dev", "config name, eg: -config test")
|
||
flag.Parse()
|
||
if configName == "" {
|
||
configName = "dev"
|
||
}
|
||
slog.Info("读取配置文件", "配置文件名称", configName)
|
||
return configName
|
||
}
|
||
|
||
// 加载配置
|
||
func LoadConfig() {
|
||
cnf := viper.New()
|
||
cnf.SetConfigName(getConfigName())
|
||
cnf.SetConfigType("yml")
|
||
cnf.AddConfigPath("./config/")
|
||
cnf.AddConfigPath(".")
|
||
err := cnf.ReadInConfig()
|
||
if err != nil {
|
||
panic(fmt.Errorf("读取配置文件错误: %w", err))
|
||
}
|
||
err = cnf.Unmarshal(&Cfg)
|
||
if err != nil {
|
||
panic(fmt.Errorf("解析配置文件错误: %w", err))
|
||
}
|
||
slog.Info("成功加载配置", "config", Cfg)
|
||
}
|