Makes linter happy

ksysoev · Oct 30, 2023 · 615ec82 · 615ec82
1 parent 7eb3b55
commit 615ec82
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/ws/ws.go b/pkg/ws/ws.go
@@ -106,8 +106,12 @@ func NewWS(url string, opts Options) (*Connection, error) {
 			err = websocket.Message.Receive(ws, &msg)
 			if err != nil {
 				if opts.WaitForResp >= 0 {
+					// If we are waiting for single response and connection is closed
+					// we just return from the function
 					return
-				} else if err.Error() == "EOF" {
+				}
+
+				if err.Error() == "EOF" {
 					color.New(color.FgRed).Println("Connection closed by the server")
 				} else {
 					color.New(color.FgRed).Println("Fail read from connection: ", err)