commit

alberliu · May 7, 2022 · 443fea1 · 443fea1
1 parent 2e8eb07
commit 443fea1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/codec.go b/codec.go
@@ -55,7 +55,7 @@ func (d *headerLenDecoder) Decode(c *Conn) error {
 			return nil
 		}
 
-		c.server.handler.OnMessage(c, body)
+		c.OnMessage(body)
 	}
 }
 

diff --git a/conn.go b/conn.go
@@ -105,6 +105,11 @@ func (c *Conn) CloseRead() error {
 	return nil
 }
 
+// OnMessage 消息处理
+func (c *Conn) OnMessage(bytes []byte) {
+	c.server.handler.OnMessage(c, bytes)
+}
+
 // GetData 获取数据
 func (c *Conn) GetData() interface{} {
 	return c.data
-Original file line number
+Diff line change
@@ Expand Up / @@ -55,7 +55,7 @@ func (d *headerLenDecoder) Decode(c *Conn) error { @@
     			return nil
     		}
-    		c.server.handler.OnMessage(c, body)
+    		c.OnMessage(body)
     	}
     }
@@ Expand Down @@