Modify socket exception disconnection

This commit is contained in:
LinkLeong 2023-03-20 09:02:06 +00:00
parent 338a7d55fc
commit c2d73e09d0

View File

@ -985,6 +985,7 @@ func init() {
func (c *Client) writePump() { func (c *Client) writePump() {
defer func() { defer func() {
c.handler.unregister <- c c.handler.unregister <- c
c.conn.Close() c.conn.Close()
}() }()
for { for {
@ -1011,6 +1012,7 @@ func (c *Client) readPump() {
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) { if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
log.Printf("error: %v", err) log.Printf("error: %v", err)
} }
c.handler.broadcast <- []byte(`{"type":"peer-left","peerId":"` + c.ID + `"}`)
break break
} }
// 要的话,推给广播中心,广播中心再推给每个用户 // 要的话,推给广播中心,广播中心再推给每个用户