Update file count (IceWhaleTech#965)

UJStudio · Mar 20, 2023 · 5c2c3b5 · 5c2c3b5
1 parent a0dc582
commit 5c2c3b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/route/v1/file.go b/route/v1/file.go
@@ -985,6 +985,7 @@ func init() {
 func (c *Client) writePump() {
 	defer func() {
 		c.handler.unregister <- c
+
 		c.conn.Close()
 	}()
 	for {
@@ -1011,6 +1012,7 @@ func (c *Client) readPump() {
 			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
 				log.Printf("error: %v", err)
 			}
+			c.handler.broadcast <- []byte(`{"type":"peer-left","peerId":"` + c.ID + `"}`)
 			break
 		}
 		// 要的话，推给广播中心，广播中心再推给每个用户