Uplift of PMS source from ONAP
[nonrtric.git] / dmaap-mediator-producer / internal / config / config.go
1 // -
2 //   ========================LICENSE_START=================================
3 //   O-RAN-SC
4 //   %%
5 //   Copyright (C) 2021: Nordix Foundation
6 //   %%
7 //   Licensed under the Apache License, Version 2.0 (the "License");
8 //   you may not use this file except in compliance with the License.
9 //   You may obtain a copy of the License at
10 //
11 //        http://www.apache.org/licenses/LICENSE-2.0
12 //
13 //   Unless required by applicable law or agreed to in writing, software
14 //   distributed under the License is distributed on an "AS IS" BASIS,
15 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 //   See the License for the specific language governing permissions and
17 //   limitations under the License.
18 //   ========================LICENSE_END===================================
19 //
20
21 package config
22
23 import (
24         "os"
25         "strconv"
26
27         log "github.com/sirupsen/logrus"
28 )
29
30 type Config struct {
31         LogLevel                            string
32         InfoProducerSupervisionCallbackHost string
33         InfoProducerSupervisionCallbackPort int
34         InfoJobCallbackHost                 string
35         InfoJobCallbackPort                 int
36         InfoCoordinatorAddress              string
37 }
38
39 type ProducerRegistrationInfo struct {
40         InfoProducerSupervisionCallbackUrl string   `json:"info_producer_supervision_callback_url"`
41         SupportedInfoTypes                 []string `json:"supported_info_types"`
42         InfoJobCallbackUrl                 string   `json:"info_job_callback_url"`
43 }
44
45 func New() *Config {
46         return &Config{
47                 LogLevel:                            getEnv("LOG_LEVEL", "Info"),
48                 InfoProducerSupervisionCallbackHost: getEnv("INFO_PRODUCER_SUPERVISION_CALLBACK_HOST", ""),
49                 InfoProducerSupervisionCallbackPort: getEnvAsInt("INFO_PRODUCER_SUPERVISION_CALLBACK_PORT", 8085),
50                 InfoJobCallbackHost:                 getEnv("INFO_JOB_CALLBACK_HOST", ""),
51                 InfoJobCallbackPort:                 getEnvAsInt("INFO_JOB_CALLBACK_PORT", 8086),
52                 InfoCoordinatorAddress:              getEnv("INFO_COORD_ADDR", "http://enrichmentservice:8083"),
53         }
54 }
55
56 func getEnv(key string, defaultVal string) string {
57         if value, exists := os.LookupEnv(key); exists {
58                 return value
59         }
60
61         return defaultVal
62 }
63
64 func getEnvAsInt(name string, defaultVal int) int {
65         valueStr := getEnv(name, "")
66         if value, err := strconv.Atoi(valueStr); err == nil {
67                 return value
68         } else if valueStr != "" {
69                 log.Warnf("Invalid int value: %v for variable: %v. Default value: %v will be used", valueStr, name, defaultVal)
70         }
71
72         return defaultVal
73 }