zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
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,7 +113,35 @@
   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"
   }
   AdminGrpc := os.Getenv("ADMIN_GRPC")
   if len(AdminGrpc) > 0 {
      Conf.GrpcServiceAddr.Admin = AdminGrpc
   }
   ApsGrpc := os.Getenv("GRPC_PORT")
   Host := os.Getenv("HOST")
   if len(Host) > 0 && len(ApsGrpc) > 0 {
      Conf.GrpcServiceAddr.Aps = Host + ":" + ApsGrpc
      //Conf.GrpcServiceAddr.Aps = DBUser + DBUser +
   }
   ShowConfig()
   CheckConfig()
}
// WriteConfig 回写配置
@@ -133,3 +163,9 @@
   log.Printf("   GrpcServiceAddr:       %+v", Conf.GrpcServiceAddr)
   log.Println("......................................................")
}
func CheckConfig() {
   if Conf.GrpcServiceAddr.Aps == "" {
      log.Fatalf("%v 尚未配置", "GrpcServiceAddr.Aps")
   }
}