if else chain to switch

Signed-off-by: Steve Coffman <steve@khanacademy.org>
99designs · May 26, 2022 · 64b882c · 64b882c
1 parent a7323ba
commit 64b882c
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/client/websocket.go b/client/websocket.go
@@ -50,7 +50,11 @@ func (p *Client) WebsocketOnce(query string, resp interface{}, options ...Option
 	return sock.Next(&resp)
 }
 
-func (p *Client) WebsocketWithPayload(query string, initPayload map[string]interface{}, options ...Option) *Subscription {
+func (p *Client) WebsocketWithPayload(
+	query string,
+	initPayload map[string]interface{},
+	options ...Option,
+) *Subscription {
 	r, err := p.newRequest(query, options...)
 	if err != nil {
 		return errorSubscription(fmt.Errorf("request: %w", err))
@@ -116,11 +120,12 @@ func (p *Client) WebsocketWithPayload(query string, initPayload map[string]inter
 					return err
 				}
 				if op.Type != dataMsg {
-					if op.Type == connectionKaMsg {
+					switch op.Type {
+					case connectionKaMsg:
 						continue
-					} else if op.Type == errorMsg {
+					case errorMsg:
 						return fmt.Errorf(string(op.Payload))
-					} else {
+					default:
 						return fmt.Errorf("expected data message, got %#v", op)
 					}
 				}