x_admin/server/config/init.go

package config

import (
	"flag"
	"fmt"
	"log"

	"github.com/spf13/viper"
)

type config struct {
	APP   *appConfig
	DB    *dbConfig
	REDIS *redisConfig
	FILE  *fileConfig
	GeTui *geTuiConfig
	Email *emailConfig
}

var Config = loadConfig(config{
	APP:   &AppConfig,
	DB:    &DBConfig,
	REDIS: &RedisConfig,
	FILE:  &FileConfig,
	GeTui: &GeTuiConfig,
	Email: &EmailConfig,
})

func loadConfig(config config) config {
	var envFilePath string
	flag.StringVar(&envFilePath, "env", "", "-env 配置文件路径，默认运行目录下的.env文件")
	flag.Parse()
	if envFilePath == "" {
		envFilePath = ".env"
	}
	// viper.SetConfigType("yaml")
	viper.AddConfigPath(".")
	viper.SetConfigFile(envFilePath)
	viper.AutomaticEnv()
	err := viper.ReadInConfig()
	if err != nil {
		log.Fatal("loadConfig ReadInConfig err:", err)
	}

	err = viper.Unmarshal(&config)
	if err != nil {
		log.Fatal("loadConfig Unmarshal err:", err)
	}
	return config
}

func init() {
	fmt.Printf("AppConfig: %+v\n", AppConfig)
	fmt.Printf("DBConfig: %+v\n", DBConfig)
	fmt.Printf("RedisConfig: %+v\n", RedisConfig)
	fmt.Printf("FileConfig: %+v\n", FileConfig)
	fmt.Printf("GeTuiConfig: %+v\n", GeTuiConfig)
	fmt.Printf("EmailConfig: %+v\n", EmailConfig)
}