@@ -19,7 +19,6 @@ import (
|
||||
"github.com/grafana/grafana/pkg/plugins"
|
||||
"github.com/grafana/grafana/pkg/services/alerting"
|
||||
"github.com/grafana/grafana/pkg/services/cleanup"
|
||||
"github.com/grafana/grafana/pkg/services/eventpublisher"
|
||||
"github.com/grafana/grafana/pkg/services/notifications"
|
||||
"github.com/grafana/grafana/pkg/services/search"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
@@ -59,7 +58,6 @@ func (g *GrafanaServerImpl) Start() {
|
||||
search.Init()
|
||||
login.Init()
|
||||
social.NewOAuthService()
|
||||
eventpublisher.Init()
|
||||
plugins.Init()
|
||||
|
||||
closer, err := tracing.Init(setting.Cfg)
|
||||
|
||||
@@ -1,150 +0,0 @@
|
||||
package eventpublisher
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana/pkg/bus"
|
||||
"github.com/grafana/grafana/pkg/events"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
"github.com/streadway/amqp"
|
||||
)
|
||||
|
||||
var (
|
||||
url string
|
||||
exchange string
|
||||
conn *amqp.Connection
|
||||
channel *amqp.Channel
|
||||
)
|
||||
|
||||
func getConnection() (*amqp.Connection, error) {
|
||||
c, err := amqp.Dial(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return c, err
|
||||
}
|
||||
|
||||
func getChannel() (*amqp.Channel, error) {
|
||||
ch, err := conn.Channel()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = ch.ExchangeDeclare(
|
||||
exchange, // name
|
||||
"topic", // type
|
||||
true, // durable
|
||||
false, // auto-deleted
|
||||
false, // internal
|
||||
false, // no-wait
|
||||
nil, // arguments
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return ch, err
|
||||
}
|
||||
|
||||
func Init() {
|
||||
sec := setting.Cfg.Section("event_publisher")
|
||||
|
||||
if !sec.Key("enabled").MustBool(false) {
|
||||
return
|
||||
}
|
||||
|
||||
url = sec.Key("rabbitmq_url").String()
|
||||
exchange = sec.Key("exchange").String()
|
||||
bus.AddWildcardListener(eventListener)
|
||||
|
||||
if err := Setup(); err != nil {
|
||||
log.Fatal(4, "Failed to connect to notification queue: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Every connection should declare the topology they expect
|
||||
func Setup() error {
|
||||
c, err := getConnection()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
conn = c
|
||||
ch, err := getChannel()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
channel = ch
|
||||
|
||||
// listen for close events so we can reconnect.
|
||||
errChan := channel.NotifyClose(make(chan *amqp.Error))
|
||||
go func() {
|
||||
for e := range errChan {
|
||||
fmt.Println("connection to rabbitmq lost.")
|
||||
fmt.Println(e)
|
||||
fmt.Println("attempting to create new rabbitmq channel.")
|
||||
ch, err := getChannel()
|
||||
if err == nil {
|
||||
channel = ch
|
||||
break
|
||||
}
|
||||
|
||||
//could not create channel, so lets close the connection
|
||||
// and re-create.
|
||||
_ = conn.Close()
|
||||
|
||||
for err != nil {
|
||||
time.Sleep(2 * time.Second)
|
||||
fmt.Println("attempting to reconnect to rabbitmq.")
|
||||
err = Setup()
|
||||
}
|
||||
fmt.Println("Connected to rabbitmq again.")
|
||||
}
|
||||
}()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func publish(routingKey string, msgString []byte) {
|
||||
for {
|
||||
err := channel.Publish(
|
||||
exchange, //exchange
|
||||
routingKey, // routing key
|
||||
false, // mandatory
|
||||
false, // immediate
|
||||
amqp.Publishing{
|
||||
ContentType: "application/json",
|
||||
Body: msgString,
|
||||
},
|
||||
)
|
||||
if err == nil {
|
||||
return
|
||||
}
|
||||
// failures are most likely because the connection was lost.
|
||||
// the connection will be re-established, so just keep
|
||||
// retrying every 2seconds until we successfully publish.
|
||||
time.Sleep(2 * time.Second)
|
||||
fmt.Println("publish failed, retrying.")
|
||||
}
|
||||
}
|
||||
|
||||
func eventListener(event interface{}) error {
|
||||
wireEvent, err := events.ToOnWriteEvent(event)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
msgString, err := json.Marshal(wireEvent)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
routingKey := fmt.Sprintf("%s.%s", wireEvent.Priority, wireEvent.EventType)
|
||||
// this is run in a greenthread and we expect that publish will keep
|
||||
// retrying until the message gets sent.
|
||||
go publish(routingKey, msgString)
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user