fix

zhangqian

2023-08-16 a9053682a59a20d5994bae402de549b46d4ac969

 conf/config.go

@@ -8,6 +8,7 @@
   "flag"
   "github.com/spf13/viper"
   "log"
   "os"
)

var (
@@ -111,7 +112,22 @@
   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()
}

// WriteConfig 回写配置
@@ -133,3 +149,9 @@
   log.Printf("   GrpcServiceAddr:       %+v", Conf.GrpcServiceAddr)
   log.Println("......................................................")
}

func CheckConfig() {
   if Conf.GrpcServiceAddr.Aps == "" {
      log.Fatalf("%v 尚未配置", "GrpcServiceAddr.Aps")
   }
}