2018-09-16 08:18:30 +00:00
|
|
|
package notification
|
2018-08-13 08:20:49 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
2018-09-16 08:37:35 +00:00
|
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
|
|
"github.com/golang/protobuf/proto"
|
2018-09-21 08:56:43 +00:00
|
|
|
"github.com/spf13/viper"
|
2018-08-13 08:20:49 +00:00
|
|
|
)
|
|
|
|
|
2018-09-16 08:37:35 +00:00
|
|
|
type MessageQueue interface {
|
|
|
|
// GetName gets the name to locate the configuration in filer.toml file
|
|
|
|
GetName() string
|
|
|
|
// Initialize initializes the file store
|
|
|
|
Initialize(configuration util.Configuration) error
|
|
|
|
SendMessage(key string, message proto.Message) error
|
|
|
|
}
|
|
|
|
|
2018-08-13 08:20:49 +00:00
|
|
|
var (
|
|
|
|
MessageQueues []MessageQueue
|
|
|
|
|
|
|
|
Queue MessageQueue
|
|
|
|
)
|
|
|
|
|
2018-08-19 22:17:55 +00:00
|
|
|
func LoadConfiguration(config *viper.Viper) {
|
2018-08-13 08:20:49 +00:00
|
|
|
|
2018-08-23 06:54:00 +00:00
|
|
|
if config == nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-12-06 08:37:59 +00:00
|
|
|
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())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, queue := range MessageQueues {
|
|
|
|
if config.GetBool(queue.GetName() + ".enabled") {
|
|
|
|
viperSub := config.Sub(queue.GetName())
|
|
|
|
if err := queue.Initialize(viperSub); err != nil {
|
2018-12-06 07:24:25 +00:00
|
|
|
glog.Fatalf("Failed to initialize notification for %s: %+v",
|
2018-12-06 08:37:59 +00:00
|
|
|
queue.GetName(), err)
|
2018-08-13 08:20:49 +00:00
|
|
|
}
|
2018-12-06 08:37:59 +00:00
|
|
|
Queue = queue
|
|
|
|
glog.V(0).Infof("Configure notification message queue for %s", queue.GetName())
|
2018-08-13 08:20:49 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|