pkg/xapp/config.go

   1 /*
   2 ==================================================================================
   3   Copyright (c) 2019 AT&T Intellectual Property.
   4   Copyright (c) 2019 Nokia
   5
   6    Licensed under the Apache License, Version 2.0 (the "License");
   7    you may not use this file except in compliance with the License.
   8    You may obtain a copy of the License at
   9
  10        http://www.apache.org/licenses/LICENSE-2.0
  11
  12    Unless required by applicable law or agreed to in writing, software
  13    distributed under the License is distributed on an "AS IS" BASIS,
  14    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15    See the License for the specific language governing permissions and
  16    limitations under the License.
  17 ==================================================================================
  18 */
  19
  20 package xapp
  21
  22 import (
  23         "flag"
  24         "fmt"
  25         "github.com/fsnotify/fsnotify"
  26         "github.com/spf13/viper"
  27         "os"
  28         "path/filepath"
  29 )
  30
  31 //-----------------------------------------------------------------------------
  32 //
  33 //-----------------------------------------------------------------------------
  34
  35 type mtype struct {
  36         Name string
  37         Id   int
  38 }
  39
  40 type Configurator struct {
  41 }
  42
  43 type ConfigChangeCB func(filename string)
  44
  45 type SDLNotificationCB func(string, ...string)
  46
  47 var ConfigChangeListeners []ConfigChangeCB
  48
  49 func parseCmd() string {
  50         var fileName *string
  51         fileName = flag.String("f", os.Getenv("CFG_FILE"), "Specify the configuration file.")
  52         flag.Parse()
  53
  54         return *fileName
  55 }
  56
  57 func LoadConfig() (l *Log) {
  58         l = NewLogger(filepath.Base(os.Args[0]))
  59         viper.SetConfigFile(parseCmd())
  60
  61         if err := viper.ReadInConfig(); err != nil {
  62                 l.Error("Reading config file failed: %v", err.Error())
  63         }
  64         l.Info("Using config file: %s", viper.ConfigFileUsed())
  65
  66         updatemtypes := func() {
  67                 var mtypes []mtype
  68                 viper.UnmarshalKey("rmr.mtypes", &mtypes)
  69
  70                 if len(mtypes) > 0 {
  71                         l.Info("Config mtypes before RICMessageTypes:%d RicMessageTypeToName:%d", len(RICMessageTypes), len(RicMessageTypeToName))
  72                         for _, v := range mtypes {
  73                                 nadd := false
  74                                 iadd := false
  75                                 if _, ok := RICMessageTypes[v.Name]; ok == false {
  76                                         nadd = true
  77                                 }
  78                                 if _, ok := RicMessageTypeToName[int(v.Id)]; ok == false {
  79                                         iadd = true
  80                                 }
  81                                 if iadd != nadd {
  82                                         l.Error("Config mtypes rmr.mtypes entry skipped due conflict with existing values %s(%t) %d(%t) ", v.Name, nadd, v.Id, iadd)
  83                                 } else if iadd {
  84                                         l.Info("Config mtypes rmr.mtypes entry added %s(%t) %d(%t) ", v.Name, nadd, v.Id, iadd)
  85                                         RICMessageTypes[v.Name] = int(v.Id)
  86                                         RicMessageTypeToName[int(v.Id)] = v.Name
  87                                 } else {
  88                                         l.Info("Config mtypes rmr.mtypes entry skipped %s(%t) %d(%t) ", v.Name, nadd, v.Id, iadd)
  89                                 }
  90                         }
  91                         l.Info("Config mtypes after RICMessageTypes:%d RicMessageTypeToName:%d", len(RICMessageTypes), len(RicMessageTypeToName))
  92                 }
  93         }
  94
  95         updatemtypes()
  96
  97         viper.WatchConfig()
  98         viper.OnConfigChange(func(e fsnotify.Event) {
  99                 l.Info("config file %s changed ", e.Name)
 100
 101                 updatemtypes()
 102                 Logger.SetLevel(viper.GetInt("controls.logger.level"))
 103                 if len(ConfigChangeListeners) > 0 {
 104                         for _, f := range ConfigChangeListeners {
 105                                 go f(e.Name)
 106                         }
 107                 }
 108         })
 109
 110         return
 111 }
 112
 113 func AddConfigChangeListener(f ConfigChangeCB) {
 114         if ConfigChangeListeners == nil {
 115                 ConfigChangeListeners = make([]ConfigChangeCB, 0)
 116         }
 117         ConfigChangeListeners = append(ConfigChangeListeners, f)
 118 }
 119
 120 func PublishConfigChange(appName, eventJson string) error {
 121         channel := fmt.Sprintf("CM_UPDATE:%s", appName)
 122         if err := Sdl.StoreAndPublish(channel, eventJson, appName, eventJson); err != nil {
 123                 Logger.Error("Sdl.Store failed: %v", err)
 124                 return err
 125         }
 126         return nil
 127 }
 128
 129 func GetPortData(pname string) (d PortData) {
 130         var getPolicies = func(policies []interface{}) (plist []int) {
 131                 for _, p := range policies {
 132                         plist = append(plist, int(p.(float64)))
 133                 }
 134                 return plist
 135         }
 136
 137         for _, v := range viper.GetStringMap("messaging")["ports"].([]interface{}) {
 138                 if n, ok := v.(map[string]interface{})["name"].(string); ok && n == pname {
 139                         d.Name = n
 140                         if p, _ := v.(map[string]interface{})["port"].(float64); ok {
 141                                 d.Port = int(p)
 142                         }
 143                         if m, _ := v.(map[string]interface{})["maxSize"].(float64); ok {
 144                                 d.MaxSize = int(m)
 145                         }
 146                         if m, _ := v.(map[string]interface{})["threadType"].(float64); ok {
 147                                 d.ThreadType = int(m)
 148                         }
 149                         if m, _ := v.(map[string]interface{})["lowLatency"].(bool); ok {
 150                                 d.LowLatency = bool(m)
 151                         }
 152                         if m, _ := v.(map[string]interface{})["fastAck"].(bool); ok {
 153                                 d.FastAck = bool(m)
 154                         }
 155                         if m, _ := v.(map[string]interface{})["maxRetryOnFailure"].(float64); ok {
 156                                 d.MaxRetryOnFailure = int(m)
 157                         }
 158                         if policies, ok := v.(map[string]interface{})["policies"]; ok {
 159                                 d.Policies = getPolicies(policies.([]interface{}))
 160                         }
 161                 }
 162         }
 163         return
 164 }
 165
 166 func (*Configurator) SetSDLNotificationCB(appName string, sdlNotificationCb SDLNotificationCB) error {
 167         return Sdl.Subscribe(sdlNotificationCb, fmt.Sprintf("CM_UPDATE:%s", appName))
 168 }
 169
 170 func (*Configurator) GetString(key string) string {
 171         return viper.GetString(key)
 172 }
 173
 174 func (*Configurator) GetInt(key string) int {
 175         return viper.GetInt(key)
 176 }
 177
 178 func (*Configurator) GetUint32(key string) uint32 {
 179         return viper.GetUint32(key)
 180 }
 181
 182 func (*Configurator) GetBool(key string) bool {
 183         return viper.GetBool(key)
 184 }
 185
 186 func (*Configurator) Get(key string) interface{} {
 187         return viper.Get(key)
 188 }
 189
 190 func (*Configurator) GetStringSlice(key string) []string {
 191         return viper.GetStringSlice(key)
 192 }
 193
 194 func (*Configurator) GetStringMap(key string) map[string]interface{} {
 195         return viper.GetStringMap(key)
 196 }