zhangqian
2023-08-18 6556134a5790821f93fa65ce15186f51111e768c
conf/config.go
@@ -8,6 +8,7 @@
   "flag"
   "github.com/spf13/viper"
   "log"
   "os"
)
var (
@@ -58,7 +59,8 @@
   }
   GrpcServiceAddr struct {
      Aps string // jwt签名
      Aps   string // aps服务地址
      Admin string // admin服务地址
   }
   config struct {
@@ -111,6 +113,20 @@
   if err := Viper.Unmarshal(&Conf); err != nil {
      log.Fatalf("Unmarshal err:%v", err)
   }
   DBHost := os.Getenv("DB_HOST")
   DBName := os.Getenv("DB_NAME")
   DBPort := os.Getenv("DB_PORT")
   DBUser := os.Getenv("DB_USER")
   DBPasswd := os.Getenv("DB_PASSWD")
   if len(DBHost) > 0 &&
      len(DBName) > 0 &&
      len(DBPort) > 0 &&
      len(DBUser) > 0 &&
      len(DBPasswd) > 0 {
      Conf.Mysql.Dsn = DBUser + ":" + DBPasswd + "@tcp(" + DBHost + ":" + DBPort + ")/" + DBName + "?charset=utf8&parseTime=True&loc=Local"
   }
   ShowConfig()
   CheckConfig()
}