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
  18 package configuration
  19
  20 import (
  21         "fmt"
  22         "github.com/spf13/viper"
  23 )
  24
  25 type Configuration struct {
  26         Logging struct {
  27                 LogLevel string
  28         }
  29         Http struct {
  30                 Port int
  31         }
  32         Rmr struct {
  33                 Port       int
  34                 MaxMsgSize int
  35         }
  36         NotificationResponseBuffer int
  37         BigRedButtonTimeoutSec  int
  38 }
  39
  40 func ParseConfiguration() *Configuration{
  41         viper.SetConfigType("yaml")
  42         viper.SetConfigName("configuration")
  43         viper.AddConfigPath("E2Manager/resources/")
  44         viper.AddConfigPath("./resources/")  //For production
  45         viper.AddConfigPath("../resources/") //For test under Docker
  46         err := viper.ReadInConfig()
  47         if err != nil {
  48                 panic(fmt.Sprintf("#http_server.parseConfiguration - failed to read configuration file: %s\n", err))
  49         }
  50
  51         config := Configuration{}
  52         config.fillRmrConfig(viper.Sub("rmr"))
  53         config.fillHttpConfig(viper.Sub("http"))
  54         config.fillLoggingConfig(viper.Sub("logging"))
  55
  56         config.NotificationResponseBuffer = viper.GetInt("notificationResponseBuffer")
  57         config.BigRedButtonTimeoutSec = viper.GetInt("bigRedButtonTimeoutSec")
  58
  59         return &config
  60 }
  61
  62 func (c *Configuration)fillLoggingConfig(logConfig *viper.Viper) {
  63         if logConfig == nil {
  64                 panic(fmt.Sprintf("#http_server.fillLoggingConfig - failed to fill logging configuration: The entry 'logging' not found\n"))
  65         }
  66         c.Logging.LogLevel = logConfig.GetString("logLevel")
  67 }
  68
  69 func (c *Configuration)fillHttpConfig(httpConfig *viper.Viper) {
  70         if httpConfig == nil {
  71                 panic(fmt.Sprintf("#http_server.fillHttpConfig - failed to fill HTTP configuration: The entry 'http' not found\n"))
  72         }
  73         c.Http.Port = httpConfig.GetInt("port")
  74 }
  75
  76 func (c *Configuration)fillRmrConfig(rmrConfig *viper.Viper) {
  77         if rmrConfig == nil {
  78                 panic(fmt.Sprintf("#http_server.fillRmrConfig - failed to fill RMR configuration: The entry 'rmr' not found\n"))
  79         }
  80         c.Rmr.Port = rmrConfig.GetInt("port")
  81         c.Rmr.MaxMsgSize = rmrConfig.GetInt("maxMsgSize")
  82 }
  83