X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=dmaap-mediator-producer%2Finternal%2Fconfig%2Fconfig.go;h=dfd2505b6c38d0fd886cab4c36e7cdc150749279;hb=0af0d6013c831bfea032c34ce85e1dcf28b10855;hp=6969f9fa7c3a9859e45c898e0ef4f3f4d214c6dc;hpb=cce95ff706e7b7d703b0bf1cfa6ce855cc2d9b68;p=nonrtric.git diff --git a/dmaap-mediator-producer/internal/config/config.go b/dmaap-mediator-producer/internal/config/config.go index 6969f9fa..dfd2505b 100644 --- a/dmaap-mediator-producer/internal/config/config.go +++ b/dmaap-mediator-producer/internal/config/config.go @@ -22,19 +22,28 @@ package config import ( "os" + "strconv" + + log "github.com/sirupsen/logrus" ) type Config struct { - LogLevel string - JobResultUri string + LogLevel log.Level + InfoProducerHost string + InfoProducerPort int InfoCoordinatorAddress string + MRHost string + MRPort int } func New() *Config { return &Config{ - LogLevel: getEnv("LOG_LEVEL", "Info"), - JobResultUri: getEnv("JOB_RESULT_URI", ""), + LogLevel: getLogLevel(), + InfoProducerHost: getEnv("INFO_PRODUCER_HOST", ""), + InfoProducerPort: getEnvAsInt("INFO_PRODUCER_PORT", 8085), InfoCoordinatorAddress: getEnv("INFO_COORD_ADDR", "http://enrichmentservice:8083"), + MRHost: getEnv("MR_HOST", "http://message-router.onap"), + MRPort: getEnvAsInt("MR_PORT", 3904), } } @@ -45,3 +54,24 @@ func getEnv(key string, defaultVal string) string { return defaultVal } + +func getEnvAsInt(name string, defaultVal int) int { + valueStr := getEnv(name, "") + if value, err := strconv.Atoi(valueStr); err == nil { + return value + } else if valueStr != "" { + log.Warnf("Invalid int value: %v for variable: %v. Default value: %v will be used", valueStr, name, defaultVal) + } + + return defaultVal +} + +func getLogLevel() log.Level { + logLevelStr := getEnv("LOG_LEVEL", "Info") + if loglevel, err := log.ParseLevel(logLevelStr); err == nil { + return loglevel + } else { + log.Warnf("Invalid log level: %v. Log level will be Info!", logLevelStr) + return log.InfoLevel + } +}