From b7f3eb70505d8b112174feeefd9117dae7336fec Mon Sep 17 00:00:00 2001
From: lwch <lwch748@gmail.com>
Date: Tue, 24 Aug 2021 15:49:08 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9log?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 code/server/handler/client.go  | 2 +-
 code/server/handler/handler.go | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/code/server/handler/client.go b/code/server/handler/client.go
index 082a632..7107032 100644
--- a/code/server/handler/client.go
+++ b/code/server/handler/client.go
@@ -52,7 +52,7 @@ func (c *client) run() {
 			return
 		}
 		c.updated = time.Now()
-		c.parent.onMessage(c.c, msg)
+		c.parent.onMessage(c, c.c, msg)
 	}
 }
 
diff --git a/code/server/handler/handler.go b/code/server/handler/handler.go
index 29b9f06..14dd9db 100644
--- a/code/server/handler/handler.go
+++ b/code/server/handler/handler.go
@@ -126,7 +126,7 @@ func (h *Handler) readHandshake(c *network.Conn) (string, error) {
 }
 
 // onMessage forward message
-func (h *Handler) onMessage(conn *network.Conn, msg *network.Msg) {
+func (h *Handler) onMessage(from *client, conn *network.Conn, msg *network.Msg) {
 	if msg.GetXType() == network.Msg_keepalive {
 		return
 	}
@@ -147,7 +147,7 @@ func (h *Handler) onMessage(conn *network.Conn, msg *network.Msg) {
 		logging.Error("client %s not found", to)
 		return
 	}
-	logging.Info("link: %s, to: %s, conn: %s, cli: %s, type: %s", linkID, to, conn.RemoteAddr().String(), cli.id, msg.GetXType().String())
+	logging.Info("link: %s, from: %p, to: %p", linkID, from, cli)
 	h.msgHook(msg, cli)
 	cli.writeMessage(msg)
 }