135 lines
2.5 KiB
Go
135 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
conf *Config
|
|
)
|
|
|
|
func InitConfig() error {
|
|
// 加载配置
|
|
err := readConfigFile()
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("read config file failed: %w", err)
|
|
}
|
|
|
|
initLogger()
|
|
initUniqueId()
|
|
|
|
return nil
|
|
}
|
|
|
|
func defaultConfig() *Config {
|
|
return &Config{
|
|
Server: ServerConfig{
|
|
Port: "8080",
|
|
BaseUrl: "/",
|
|
WsUrl: "/ws",
|
|
},
|
|
JWT: JwtConfig{
|
|
Secret: "qwemwddlalmskadklsdklamlkdsakld",
|
|
AccessExpiry: 24 * time.Hour,
|
|
},
|
|
UniqueID: UniqueIDConfig{
|
|
DataCenterID: 1,
|
|
WorkerID: 1,
|
|
},
|
|
Postgres: PostgresConfig{
|
|
Port: 5432,
|
|
SslMode: false,
|
|
TimeZone: "Asia/Shanghai",
|
|
MaxOpenConns: 20,
|
|
MaxIdleConns: 10,
|
|
ConnMaxLifetime: 30 * time.Minute,
|
|
},
|
|
Log: LogConfig{
|
|
Level: "info",
|
|
Dir: "logs",
|
|
MaxSize: 100,
|
|
MaxBackups: 10,
|
|
MaxAge: 30,
|
|
Compress: true,
|
|
},
|
|
}
|
|
}
|
|
|
|
func readConfigFile() error {
|
|
// 配置文件的目录
|
|
prefix := "resources"
|
|
env := GetEnv() // 获取当前环境
|
|
|
|
// 设置默认值
|
|
conf = defaultConfig() // 先给默认结构体
|
|
|
|
// 默认配置文件
|
|
confFileRelPath := filepath.Join(prefix, "application.yaml")
|
|
|
|
// 创建 viper 实例
|
|
v := viper.New()
|
|
|
|
// 首先加载默认的 application.yaml
|
|
v.SetConfigFile(confFileRelPath)
|
|
v.SetConfigType("yaml")
|
|
|
|
// 读取默认配置文件
|
|
if err := v.ReadInConfig(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// 根据环境加载不同的配置文件
|
|
switch env {
|
|
case "dev":
|
|
v.SetConfigFile(filepath.Join(prefix, "application-dev.yaml"))
|
|
case "build":
|
|
v.SetConfigFile(filepath.Join(prefix, "application-build.yaml"))
|
|
}
|
|
|
|
// 如果需要，合并环境特定的配置文件
|
|
if err := v.MergeInConfig(); err != nil {
|
|
return fmt.Errorf("merge env config failed: %w", err)
|
|
}
|
|
|
|
if err := v.Unmarshal(conf); err != nil {
|
|
return fmt.Errorf("unmarshal config failed: %w", err)
|
|
}
|
|
|
|
fmt.Println("✅️ 加载配置成功")
|
|
// 设置环境变量
|
|
conf.Env = env
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetEnv 获取 GO_ENV 环境变量，默认返回 "dev"
|
|
func GetEnv() string {
|
|
e := os.Getenv("GO_ENV")
|
|
if len(e) == 0 {
|
|
// 如果没有设置 GO_ENV，使用默认的 "dev" 配置
|
|
return "dev"
|
|
}
|
|
return e
|
|
}
|
|
|
|
// GetConf 获取配置信息
|
|
func GetConf() *Config {
|
|
return conf
|
|
}
|
|
|
|
// GetServerPort 项目启动端口
|
|
func GetServerPort() string {
|
|
port := conf.Server.Port
|
|
if strings.HasPrefix(port, ":") {
|
|
return port
|
|
}
|
|
return ":" + port
|
|
}
|