refactoring

This commit is contained in:
Chris Lu
2018-12-06 00:44:41 -08:00
parent ffa2827ab1
commit c28e8a2397
3 changed files with 42 additions and 30 deletions

View File

@@ -27,16 +27,7 @@ func LoadConfiguration(config *viper.Viper) {
return
}
enabledQueue := ""
for _, queue := range MessageQueues {
if config.GetBool(queue.GetName() + ".enabled") {
if enabledQueue == "" {
enabledQueue = queue.GetName()
} else {
glog.Fatalf("Notification message queue is enabled for both %s and %s", enabledQueue, queue.GetName())
}
}
}
validateOneEnabledQueue(config)
for _, queue := range MessageQueues {
if config.GetBool(queue.GetName() + ".enabled") {
@@ -52,3 +43,16 @@ func LoadConfiguration(config *viper.Viper) {
}
}
func validateOneEnabledQueue(config *viper.Viper) {
enabledQueue := ""
for _, queue := range MessageQueues {
if config.GetBool(queue.GetName() + ".enabled") {
if enabledQueue == "" {
enabledQueue = queue.GetName()
} else {
glog.Fatalf("Notification message queue is enabled for both %s and %s", enabledQueue, queue.GetName())
}
}
}
}