Adding returns when an error occurred

ystv · Dec 27, 2023 · 2d4179b · 2d4179b
1 parent 8a7d4ab
commit 2d4179b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/views/websocket.go b/server/views/websocket.go
@@ -99,6 +99,7 @@ func (v *Views) Websocket(c echo.Context) error {
 				v.cache.Delete(responseTransporter.Server.String())
 				v.cache.Delete(responseTransporter.Server.String() + internalChannelNameAppend)
 				loop = false
+				return nil
 			}
 
 			_, msg, err = ws.ReadMessage()
@@ -109,6 +110,7 @@ func (v *Views) Websocket(c echo.Context) error {
 				v.cache.Delete(responseTransporter.Server.String())
 				v.cache.Delete(responseTransporter.Server.String() + internalChannelNameAppend)
 				loop = false
+				return nil
 			}
 			internalChannel <- msg
 			log.Printf("Message received from \"%s\": %s", responseTransporter.Server, msg)
@@ -121,6 +123,7 @@ func (v *Views) Websocket(c echo.Context) error {
 				v.cache.Delete(responseTransporter.Server.String())
 				v.cache.Delete(responseTransporter.Server.String() + internalChannelNameAppend)
 				loop = false
+				return nil
 			}
 			var msgType int
 			msgType, msg, err = ws.ReadMessage()
@@ -131,6 +134,7 @@ func (v *Views) Websocket(c echo.Context) error {
 				v.cache.Delete(responseTransporter.Server.String())
 				v.cache.Delete(responseTransporter.Server.String() + internalChannelNameAppend)
 				loop = false
+				return nil
 			}
 		}
 	}