Updated panic defer function to send message to frontend

internal/usecases/consume/handlemessage.go

@@ -25,13 +25,6 @@ HandleMessage gets called when a message is received
 */
 func (s *Service) HandleMessage(msg *broker.Message) {
 
-	// Defer a panic intercepting function
-	defer func() {
-		if err := recover(); err != nil {
-			log.Printf("PANIC: %e", err.(error))
-		}
-	}()
-
 	// Grab the headers from the message
 	sessionID, clientID, queryID, err := getHeaders(&msg.Headers)
 	if err != nil {
@@ -40,6 +33,15 @@ func (s *Service) HandleMessage(msg *broker.Message) {
 	}
 	s.publishStatus("Received", sessionID, queryID)
 
+	// Defer a panic intercepting function
+	defer func(sessionID, queryID *string) {
+		if err := recover(); err != nil {
+			log.Printf("PANIC: %e", err.(error))
+
+			s.publishStatus("Critical Error", sessionID, queryID)
+		}
+	}(sessionID, queryID)
+
 	// Unmarshall the incoming message into an IncomingJSONQuery object
 	JSONQuery, err := query.UnmarshalJSON(&msg.Body)
 	if err != nil {