Skip to content

Commit 355346a

Browse files
committed
Fix disconnection notification.
1 parent 45a705e commit 355346a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pkg/websocket/conn.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package websocket
22

33
import (
44
"errors"
5+
"net"
56
"sync"
67
"time"
78

@@ -46,7 +47,11 @@ func (conn *WebSocketConn) ReadMessage() {
4647
if err != nil {
4748
logger.Warnf("Got error: %v", err)
4849
if c, k := err.(*websocket.CloseError); k {
49-
conn.Emit("error", c.Code, c.Text)
50+
conn.Emit("close", c.Code, c.Text)
51+
} else {
52+
if c, k := err.(*net.OpError); k {
53+
conn.Emit("close", 1008, c.Error())
54+
}
5055
}
5156
close(stop)
5257
break

0 commit comments

Comments
 (0)