Helper function to read a complete message from a WebSocket
(ctx context.Context, conn *websocket.Conn)
| 698 | |
| 699 | // Helper function to read a complete message from a WebSocket |
| 700 | func readWebSocketMessage(ctx context.Context, conn *websocket.Conn) ([]byte, error) { |
| 701 | |
| 702 | var message []byte |
| 703 | if err := websocket.Message.Receive(conn, &message); err != nil { |
| 704 | if err != io.EOF { |
| 705 | base.WarnfCtx(ctx, "Error reading initial websocket message: %v", err) |
| 706 | return nil, err |
| 707 | } |
| 708 | } |
| 709 | return message, nil |
| 710 | |
| 711 | } |
no test coverage detected