fix: add "server closed" as connection error (#3355)

add "server closed" as connection error
kubeshop · Nov 14, 2023 · 4d13812 · 4d13812
1 parent 33783c4
commit 4d13812
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/agent/client/client.go b/agent/client/client.go
@@ -193,7 +193,21 @@ func (c *Client) handleDisconnectionError(inputErr error) (bool, error) {
 }
 
 func isConnectionError(err error) bool {
-	return err != nil && strings.Contains(err.Error(), "connection refused")
+	if err == nil {
+		return false
+	}
+
+	possibleErrors := []string{
+		"connection refused",
+		"server closed",
+	}
+	for _, possibleErr := range possibleErrors {
+		if strings.Contains(err.Error(), possibleErr) {
+			return true
+		}
+	}
+
+	return false
 }
 
 func isEndOfFileError(err error) bool {