From f5dc94ddd76e0a274777184154fb6c40df12085a Mon Sep 17 00:00:00 2001 From: duanhf2012 Date: Thu, 12 Mar 2020 15:47:00 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=8D=95=E8=8E=B7=E6=89=80?= =?UTF-8?q?=E6=9C=89=E5=8D=8F=E8=AE=AE=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sysservice/tcpsocketpbservice.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/sysservice/tcpsocketpbservice.go b/sysservice/tcpsocketpbservice.go index 9e77158..6e4b60f 100644 --- a/sysservice/tcpsocketpbservice.go +++ b/sysservice/tcpsocketpbservice.go @@ -18,10 +18,12 @@ type TcpSocketPbService struct { disconnEvent EventHandler exceptMsgHandler ExceptMsgHandler + messageRecvHandler MessageRecvHandler } type MessageHandler func(clientid uint64,msgtype uint16,msg proto.Message) +type MessageRecvHandler func(clientid uint64,pPack *network.MsgBasePack) type EventHandler func(clientid uint64) type ExceptMsgHandler func(clientid uint64,pPack *network.MsgBasePack,err error) @@ -85,6 +87,10 @@ func (slf *TcpSocketPbService) RegExceptMessage(exceptMsgHandler ExceptMsgHandle slf.exceptMsgHandler = exceptMsgHandler } +func (slf *TcpSocketPbService) RegRecvMessage(msgHandler MessageRecvHandler){ + slf.messageRecvHandler = msgHandler +} + func (slf *TcpSocketPbService) OnConnected(pClient *network.SClient){ if slf.connEvent!=nil { @@ -126,6 +132,9 @@ func (slf *TcpSocketPbService) OnRecvMsg(pClient *network.SClient, pPack *networ info.msgHandler(pClient.GetId(),pPack.PackType, msg.(proto.Message)) return + }else if slf.messageRecvHandler!=nil { + slf.messageRecvHandler(pClient.GetId(),pPack) + return } slf.OnExceptMsg(pClient,pPack,errors.New("not found PackType"))