From a3c36e76903f1abc3707a6781c8751f60429ae95 Mon Sep 17 00:00:00 2001 From: duanhf2012 Date: Fri, 14 Jan 2022 17:33:11 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=B9=B6=E5=85=BC=E5=AE=B9?= =?UTF-8?q?=E5=8E=9F=E5=A7=8B=E6=B6=88=E6=81=AF=E5=8F=91=E9=80=81=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sysservice/tcpservice/tcpservice.go | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/sysservice/tcpservice/tcpservice.go b/sysservice/tcpservice/tcpservice.go index a0fbd01..7732440 100644 --- a/sysservice/tcpservice/tcpservice.go +++ b/sysservice/tcpservice/tcpservice.go @@ -263,6 +263,7 @@ func (tcpService *TcpService) GetClientIp(clientid uint64) string{ return pClient.tcpConn.GetRemoteIp() } + func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{ tcpService.mapClientLocker.Lock() client,ok := tcpService.mapClient[clientId] @@ -272,9 +273,22 @@ func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{ } tcpService.mapClientLocker.Unlock() client.tcpConn.SetWriteDeadline(tcpService.WriteDeadline) - return client.tcpConn.WriteRawMsg(msg) + return client.tcpConn.WriteMsg(msg) } +func (tcpService *TcpService) SendRawData(clientId uint64,data []byte) error{ + tcpService.mapClientLocker.Lock() + client,ok := tcpService.mapClient[clientId] + if ok == false{ + tcpService.mapClientLocker.Unlock() + return fmt.Errorf("client %d is disconnect!",clientId) + } + tcpService.mapClientLocker.Unlock() + client.tcpConn.SetWriteDeadline(tcpService.WriteDeadline) + return client.tcpConn.WriteRawMsg(data) +} + + func (tcpService *TcpService) GetConnNum() int { tcpService.mapClientLocker.Lock() connNum := len(tcpService.mapClient)