Subscriber reconnect (#4776)

* Force infra/subscriber disconnection when is already connected Signed-off-by: Bruno Ledesma <[email protected]> * Force infra/subscriber disconnection when is already connected Signed-off-by: Bruno Ledesma <[email protected]> --------- Signed-off-by: Bruno Ledesma <[email protected]>
litmuschaos · Aug 14, 2024 · fd9cb3f · fd9cb3f
1 parent 3874a5b
commit fd9cb3f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/chaoscenter/graphql/server/graph/chaos_infrastructure.resolvers.go b/chaoscenter/graphql/server/graph/chaos_infrastructure.resolvers.go
diff --git a/chaoscenter/subscriber/pkg/requests/webhook.go b/chaoscenter/subscriber/pkg/requests/webhook.go
@@ -69,7 +69,7 @@ func (req *subscriberRequests) AgentConnect(infraData map[string]string) {
 	for {
 		_, message, err := c.ReadMessage()
 		if err != nil {
-			logrus.WithError(err).Fatal("Failed to read message")
+			logrus.WithError(err).Panic("Failed to read message")
 		}
 
 		var r types.RawData
@@ -87,6 +87,7 @@ func (req *subscriberRequests) AgentConnect(infraData map[string]string) {
 		}
 		if r.Payload.Errors != nil {
 			logrus.Error("Error response from the server : ", string(message))
+			panicWhen("ALREADY CONNECTED", message)
 			continue
 		}
 
@@ -97,6 +98,12 @@ func (req *subscriberRequests) AgentConnect(infraData map[string]string) {
 	}
 }
 
+func panicWhen(errorMessage string, message []byte) {
+	if strings.Contains(string(message), errorMessage) {
+		logrus.Panic("Server error: ", errorMessage)
+	}
+}
+
 func (req *subscriberRequests) RequestProcessor(infraData map[string]string, r types.RawData) error {
 	if strings.Index("kubeobject kubeobjects", strings.ToLower(r.Payload.Data.InfraConnect.Action.RequestType)) >= 0 {
 		KubeObjRequest := types.KubeObjRequest{