restart replication if RabbitMQ connection closed

https://github.com/google/go-cloud/issues/2958
4 years ago · 0cfed8c3cb
3 changed files with 4 additions and 0 deletions
--- a/docker/compose/notification.toml
+++ b/docker/compose/notification.toml
--- a/docker/compose/replication.toml
+++ b/docker/compose/replication.toml
--- a/weed/replication/sub/notification_gocdk_pub_sub.go
+++ b/weed/replication/sub/notification_gocdk_pub_sub.go
@ -113,6 +113,10 @@ func (k *GoCDKPubSubInput) Initialize(configuration util.Configuration, prefix s
 func (k *GoCDKPubSubInput) ReceiveMessage() (key string, message *filer_pb.EventNotification, onSuccessFn func(), onFailureFn func(), err error) {
 	msg, err := k.sub.Receive(context.Background())
 	if err != nil {
+		var conn *amqp.Connection
+		if k.sub.As(&conn) && conn.IsClosed() {
+			glog.Fatalln(err)
+		}
 		return
 	}
 	onFailureFn = func() {