结构调整添加cmd

tools/config/config.go

@@ -0,0 +1,76 @@
+package config
+
+import (
+	"fmt"
+	"github.com/spf13/viper"
+	"io/ioutil"
+	"log"
+	"os"
+	"strings"
+)
+
+var cfgDatabase *viper.Viper
+var cfgApplication *viper.Viper
+var cfgJwt *viper.Viper
+var cfgLog *viper.Viper
+
+//func init() {
+//	InitConfig("settings.dev")
+//}
+
+//载入配置文件
+func ConfigSetup(path string) {
+	viper.SetConfigFile(path)
+	content, err := ioutil.ReadFile(path)
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Read config file fail: %s", err.Error()))
+	}
+	//Replace environment variables
+	err = viper.ReadConfig(strings.NewReader(os.ExpandEnv(string(content))))
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Parse config file fail: %s", err.Error()))
+	}
+	//}
+	//
+	//func InitConfig(fileName string) {
+	//	viper.SetConfigName(fileName)
+	//	viper.AddConfigPath("/config")
+	//	err := viper.ReadInConfig()
+	//	if err != nil {
+	//		log.Println(err)
+	//	}
+
+	cfgDatabase = viper.Sub("settings.database")
+	if cfgDatabase == nil {
+		panic("config not found settings.database")
+	}
+	DatabaseConfig = InitDatabase(cfgDatabase)
+
+	cfgApplication = viper.Sub("settings.application")
+	if cfgApplication == nil {
+		panic("config not found settings.application")
+	}
+	ApplicationConfig = InitApplication(cfgApplication)
+
+	cfgJwt = viper.Sub("settings.jwt")
+	if cfgJwt == nil {
+		panic("config not found settings.jwt")
+	}
+	JwtConfig = InitJwt(cfgJwt)
+
+	cfgLog = viper.Sub("settings.log")
+	if cfgLog == nil {
+		panic("config not found settings.log")
+	}
+	LogConfig = InitLog(cfgLog)
+}
+
+func SetApplicationIsInit() {
+	SetConfig("./config", "settings.application.isInit", false)
+}
+
+func SetConfig(configPath string, key string, value interface{}) {
+	viper.AddConfigPath(configPath)
+	viper.Set(key, value)
+	viper.WriteConfig()
+}