Merge pull request #8 from hongcha98/fix-onmessage

fix(onmessage): trigger
a-wing · Aug 6, 2023 · 663a12d · 663a12d
2 parents 8adbd5b + 5537b50
commit 663a12d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -76,13 +76,15 @@ func (c *Client) readPump() {
 			c.Err = err
 			break
 		}
-		c.worker.broadcast <- Message{
+		msg := Message{
 			Name: c.Name,
 			Room: c.Room,
 			Code: code,
 			Data: data,
 			conn: c.conn,
 		}
+		c.worker.server.onMessage(&msg)
+		c.worker.broadcast <- msg
 	}
 }
 

diff --git a/worker.go b/worker.go
@@ -70,7 +70,6 @@ func (w *worker) run(ctx context.Context) {
 				return
 			}
 		case message := <-w.broadcast:
-			w.server.onMessage(&message)
 			for client := range w.clients {
 				if w.server.config.Local || message.conn != client.conn {
 					select {