| | |
| | | "flag" |
| | | "github.com/spf13/viper" |
| | | "log" |
| | | "os" |
| | | ) |
| | | |
| | | var ( |
| | |
| | | } |
| | | |
| | | GrpcServiceAddr struct { |
| | | Aps string // jwt签名 |
| | | Aps string // aps服务地址 |
| | | Admin string // admin服务地址 |
| | | } |
| | | |
| | | config struct { |
| | |
| | | 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 回写配置 |
| | |
| | | log.Printf(" GrpcServiceAddr: %+v", Conf.GrpcServiceAddr) |
| | | log.Println("......................................................") |
| | | } |
| | | |
| | | func CheckConfig() { |
| | | if Conf.GrpcServiceAddr.Aps == "" { |
| | | log.Fatalf("%v 尚未配置", "GrpcServiceAddr.Aps") |
| | | } |
| | | } |