import (
"flag"
+ "fmt"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
"os"
type ConfigChangeCB func(filename string)
+type SDLNotificationCB func(string, ...string)
+
var ConfigChangeListeners []ConfigChangeCB
func parseCmd() string {
}
l.Info("Using config file: %s", viper.ConfigFileUsed())
- updatemtypes := func() {
+ updateMTypes := func() {
var mtypes []mtype
- viper.UnmarshalKey("rmr.mtypes", &mtypes)
+ viper.UnmarshalKey("messaging.mtypes", &mtypes)
if len(mtypes) > 0 {
l.Info("Config mtypes before RICMessageTypes:%d RicMessageTypeToName:%d", len(RICMessageTypes), len(RicMessageTypeToName))
}
}
- updatemtypes()
+ updateMTypes()
viper.WatchConfig()
viper.OnConfigChange(func(e fsnotify.Event) {
l.Info("config file %s changed ", e.Name)
- updatemtypes()
- Logger.SetLevel(viper.GetInt("logger.level"))
+ updateMTypes()
+ if viper.IsSet("controls.logger.level") {
+ Logger.SetLevel(viper.GetInt("controls.logger.level"))
+ } else {
+ Logger.SetLevel(viper.GetInt("logger.level"))
+ }
+
if len(ConfigChangeListeners) > 0 {
for _, f := range ConfigChangeListeners {
go f(e.Name)
ConfigChangeListeners = append(ConfigChangeListeners, f)
}
+func PublishConfigChange(appName, eventJson string) error {
+ channel := fmt.Sprintf("CM_UPDATE:%s", appName)
+ if err := Sdl.StoreAndPublish(channel, eventJson, appName, eventJson); err != nil {
+ Logger.Error("Sdl.Store failed: %v", err)
+ return err
+ }
+ return nil
+}
+
+func GetPortData(pname string) (d PortData) {
+ var getPolicies = func(policies []interface{}) (plist []int) {
+ for _, p := range policies {
+ plist = append(plist, int(p.(float64)))
+ }
+ return plist
+ }
+
+ if viper.IsSet("messaging") == false {
+ if pname == "http" {
+ d.Port = 8080
+ }
+ if pname == "rmrdata" {
+ d.Port = 4560
+ }
+ return
+ }
+
+ for _, v := range viper.GetStringMap("messaging")["ports"].([]interface{}) {
+ if n, ok := v.(map[string]interface{})["name"].(string); ok && n == pname {
+ d.Name = n
+ if p, _ := v.(map[string]interface{})["port"].(float64); ok {
+ d.Port = int(p)
+ }
+ if m, _ := v.(map[string]interface{})["maxSize"].(float64); ok {
+ d.MaxSize = int(m)
+ }
+ if m, _ := v.(map[string]interface{})["threadType"].(float64); ok {
+ d.ThreadType = int(m)
+ }
+ if m, _ := v.(map[string]interface{})["lowLatency"].(bool); ok {
+ d.LowLatency = bool(m)
+ }
+ if m, _ := v.(map[string]interface{})["fastAck"].(bool); ok {
+ d.FastAck = bool(m)
+ }
+ if m, _ := v.(map[string]interface{})["maxRetryOnFailure"].(float64); ok {
+ d.MaxRetryOnFailure = int(m)
+ }
+ if policies, ok := v.(map[string]interface{})["policies"]; ok {
+ d.Policies = getPolicies(policies.([]interface{}))
+ }
+ }
+ }
+ return
+}
+
+func (*Configurator) SetSDLNotificationCB(appName string, sdlNotificationCb SDLNotificationCB) error {
+ return Sdl.Subscribe(sdlNotificationCb, fmt.Sprintf("CM_UPDATE:%s", appName))
+}
+
func (*Configurator) GetString(key string) string {
return viper.GetString(key)
}
func (*Configurator) GetStringMap(key string) map[string]interface{} {
return viper.GetStringMap(key)
}
+
+func (*Configurator) IsSet(key string) bool {
+ return viper.IsSet(key)
+}