Reconnect on closed connection

messenger.go

@@ -28,6 +28,7 @@ type Event struct {
 // Messenger is an interface for sending messages for different file events
 type Messenger interface {
 	SendMessage(string, []byte) error
+	IsConnClosed() bool
 }
 
 // AMQPMessenger is a Messenger that sends messages to a local AMQP broker
@@ -157,3 +158,7 @@ func (m *AMQPMessenger) createNewChannel() error {
 
 	return nil
 }
+
+func (m *AMQPMessenger) IsConnClosed() bool {
+	return m.connection.IsClosed()
+}

proxy.go

@@ -139,15 +139,28 @@ func (p *Proxy) allowedResponse(w http.ResponseWriter, r *http.Request) {
 			return
 		}
 
-		if err = p.messenger.SendMessage(p.fileIds[r.URL.Path], jsonMessage); err != nil {
-			log.Debug("error when sending message")
-			log.Error(err)
-		}
+		switch p.messenger.IsConnClosed() {
+		case true:
+			log.Errorln("connection is closed")
+			w.WriteHeader(http.StatusServiceUnavailable)
+
+			tlsBroker, _ := TLSConfigBroker(Conf)
+			m, err := NewAMQPMessenger(Conf.Broker, tlsBroker)
+			if err == nil {
+				p.messenger = m
+			}
 
-		log.Debugf("marking file %v as 'uploaded' in database", p.fileIds[r.URL.Path])
-		err = p.database.MarkFileAsUploaded(p.fileIds[r.URL.Path], username, string(jsonMessage))
-		if err != nil {
-			log.Error(err)
+		case false:
+			if err = p.messenger.SendMessage(p.fileIds[r.URL.Path], jsonMessage); err != nil {
+				log.Debug("error when sending message")
+				log.Error(err)
+			}
+
+			log.Debugf("marking file %v as 'uploaded' in database", p.fileIds[r.URL.Path])
+			err = p.database.MarkFileAsUploaded(p.fileIds[r.URL.Path], username, string(jsonMessage))
+			if err != nil {
+				log.Error(err)
+			}
 		}
 
 		delete(p.fileIds, r.URL.Path)