E2Manager/configuration/configuration.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   4 //
   5 // Licensed under the Apache License, Version 2.0 (the "License");
   6 // you may not use this file except in compliance with the License.
   7 // You may obtain a copy of the License at
   8 //
   9 //      http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16
  17 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  18 //  platform project (RICP).
  19
  20
  21 package configuration
  22
  23 import (
  24         "fmt"
  25         "github.com/spf13/viper"
  26 )
  27
  28 type Configuration struct {
  29         Logging struct {
  30                 LogLevel string
  31         }
  32         Http struct {
  33                 Port int
  34         }
  35         Rmr struct {
  36                 Port       int
  37                 MaxMsgSize int
  38         }
  39         RoutingManager struct {
  40                 BaseUrl    string
  41         }
  42         RicId struct {
  43                 PlmnId string
  44                 RicNearRtId string
  45         }
  46         NotificationResponseBuffer   int
  47         BigRedButtonTimeoutSec       int
  48         MaxConnectionAttempts        int
  49         MaxRnibConnectionAttempts    int
  50         RnibRetryIntervalMs          int
  51         KeepAliveResponseTimeoutMs       int
  52         KeepAliveDelayMs             int
  53         E2TInstanceDeletionTimeoutMs int
  54 }
  55
  56 func ParseConfiguration() *Configuration {
  57         viper.SetConfigType("yaml")
  58         viper.SetConfigName("configuration")
  59         viper.AddConfigPath("E2Manager/resources/")
  60         viper.AddConfigPath("./resources/")     //For production
  61         viper.AddConfigPath("../resources/")    //For test under Docker
  62         viper.AddConfigPath("../../resources/") //For test under Docker
  63         err := viper.ReadInConfig()
  64         if err != nil {
  65                 panic(fmt.Sprintf("#configuration.ParseConfiguration - failed to read configuration file: %s\n", err))
  66         }
  67
  68         config := Configuration{}
  69         config.populateRmrConfig(viper.Sub("rmr"))
  70         config.populateHttpConfig(viper.Sub("http"))
  71         config.populateLoggingConfig(viper.Sub("logging"))
  72         config.populateRoutingManagerConfig(viper.Sub("routingManager"))
  73         config.NotificationResponseBuffer = viper.GetInt("notificationResponseBuffer")
  74         config.BigRedButtonTimeoutSec = viper.GetInt("bigRedButtonTimeoutSec")
  75         config.MaxConnectionAttempts = viper.GetInt("maxConnectionAttempts")
  76         config.MaxRnibConnectionAttempts = viper.GetInt("maxRnibConnectionAttempts")
  77         config.RnibRetryIntervalMs = viper.GetInt("rnibRetryIntervalMs")
  78         config.KeepAliveResponseTimeoutMs = viper.GetInt("keepAliveResponseTimeoutMs")
  79         config.KeepAliveDelayMs = viper.GetInt("KeepAliveDelayMs")
  80         config.E2TInstanceDeletionTimeoutMs = viper.GetInt("e2tInstanceDeletionTimeoutMs")
  81         return &config
  82 }
  83
  84 func (c *Configuration) populateLoggingConfig(logConfig *viper.Viper) {
  85         if logConfig == nil {
  86                 panic(fmt.Sprintf("#configuration.populateLoggingConfig - failed to populate logging configuration: The entry 'logging' not found\n"))
  87         }
  88         c.Logging.LogLevel = logConfig.GetString("logLevel")
  89 }
  90
  91 func (c *Configuration) populateHttpConfig(httpConfig *viper.Viper) {
  92         if httpConfig == nil {
  93                 panic(fmt.Sprintf("#configuration.populateHttpConfig - failed to populate HTTP configuration: The entry 'http' not found\n"))
  94         }
  95         c.Http.Port = httpConfig.GetInt("port")
  96 }
  97
  98 func (c *Configuration) populateRmrConfig(rmrConfig *viper.Viper) {
  99         if rmrConfig == nil {
 100                 panic(fmt.Sprintf("#configuration.populateRmrConfig - failed to populate RMR configuration: The entry 'rmr' not found\n"))
 101         }
 102         c.Rmr.Port = rmrConfig.GetInt("port")
 103         c.Rmr.MaxMsgSize = rmrConfig.GetInt("maxMsgSize")
 104 }
 105
 106 func (c *Configuration) populateRoutingManagerConfig(rmConfig *viper.Viper) {
 107         if rmConfig == nil {
 108                 panic(fmt.Sprintf("#configuration.populateRoutingManagerConfig - failed to populate Routing Manager configuration: The entry 'routingManager' not found\n"))
 109         }
 110         c.RoutingManager.BaseUrl = rmConfig.GetString("baseUrl")
 111 }