fix: Skip expected error from web socket (#9194)

determined-ai · Apr 22, 2024 · 3568f27 · 3568f27
1 parent 1b212ae
commit 3568f27
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/master/internal/stream/publisher.go b/master/internal/stream/publisher.go
@@ -178,7 +178,12 @@ func (ps *PublisherSet) streamHandler(
 	var startupMsg StartupMsg
 	err := socket.ReadJSON(&startupMsg)
 	if err != nil {
-		return fmt.Errorf("error while reading initial startup message: %s", err.Error())
+		if websocket.IsUnexpectedCloseError(
+			err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure,
+		) {
+			return fmt.Errorf("error while reading initial startup message: %s", err.Error())
+		}
+		return nil
 	}
 
 	// startups is where we collect StartupMsg