Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-03-17 15:17:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

支持捕获所有协议接口

Browse Source
This commit is contained in:
duanhf2012
2020-03-12 15:47:00 +08:00
parent 7724a11063
commit f5dc94ddd7
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sysservice/tcpsocketpbservice.go
View File

@@ -18,10 +18,12 @@ type TcpSocketPbService struct {
disconnEvent EventHandler disconnEvent EventHandler
exceptMsgHandler ExceptMsgHandler exceptMsgHandler ExceptMsgHandler
messageRecvHandler MessageRecvHandler
} }
type MessageHandler func(clientid uint64,msgtype uint16,msg proto.Message) type MessageHandler func(clientid uint64,msgtype uint16,msg proto.Message)
type MessageRecvHandler func(clientid uint64,pPack *network.MsgBasePack)
type EventHandler func(clientid uint64) type EventHandler func(clientid uint64)
type ExceptMsgHandler func(clientid uint64,pPack *network.MsgBasePack,err error) type ExceptMsgHandler func(clientid uint64,pPack *network.MsgBasePack,err error)
@@ -85,6 +87,10 @@ func (slf *TcpSocketPbService) RegExceptMessage(exceptMsgHandler ExceptMsgHandle
slf.exceptMsgHandler = exceptMsgHandler slf.exceptMsgHandler = exceptMsgHandler
} }
func (slf *TcpSocketPbService) RegRecvMessage(msgHandler MessageRecvHandler){
slf.messageRecvHandler = msgHandler
}
func (slf *TcpSocketPbService) OnConnected(pClient *network.SClient){ func (slf *TcpSocketPbService) OnConnected(pClient *network.SClient){
if slf.connEvent!=nil { 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)) info.msgHandler(pClient.GetId(),pPack.PackType, msg.(proto.Message))
return return
}else if slf.messageRecvHandler!=nil {
slf.messageRecvHandler(pClient.GetId(),pPack)
return
} }
slf.OnExceptMsg(pClient,pPack,errors.New("not found PackType")) slf.OnExceptMsg(pClient,pPack,errors.New("not found PackType"))