Skip to content

Commit 5db30b3

Browse files
authored
Handle WebSocket Closing Error (#16)
* Improve error handling for websocket messages Handle websocket read errors with connection state check. * Apply suggestion
1 parent 9e76029 commit 5db30b3

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

psynetrpc.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,10 @@ package rlapi
33
import (
44
"context"
55
"encoding/json"
6+
"errors"
67
"fmt"
78
"log/slog"
9+
"net"
810
"strings"
911
"sync"
1012
"time"
@@ -199,7 +201,11 @@ func (p *PsyNetRPC) readMessages() {
199201
for {
200202
_, message, err := p.wsConn.ReadMessage()
201203
if err != nil {
202-
p.logger.Error("failed to read websocket message", slog.Any("err", err))
204+
if errors.Is(err, net.ErrClosed) {
205+
p.logger.Debug("websocket closed", slog.Any("err", err))
206+
} else {
207+
p.logger.Error("failed to read websocket message", slog.Any("err", err))
208+
}
203209
break
204210
}
205211

0 commit comments

Comments
 (0)