Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-05-08 23:07:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化并兼容原始消息发送接口

Browse Source
This commit is contained in:
duanhf2012
2022-01-14 17:33:11 +08:00
parent e676587b4d
commit a3c36e7690
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
sysservice/tcpservice/tcpservice.go
View File

@@ -263,6 +263,7 @@ func (tcpService *TcpService) GetClientIp(clientid uint64) string{
return pClient.tcpConn.GetRemoteIp() return pClient.tcpConn.GetRemoteIp()
} }
func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{ func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{
tcpService.mapClientLocker.Lock() tcpService.mapClientLocker.Lock()
client,ok := tcpService.mapClient[clientId] client,ok := tcpService.mapClient[clientId]
@@ -272,9 +273,22 @@ func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{
} }
tcpService.mapClientLocker.Unlock() tcpService.mapClientLocker.Unlock()
client.tcpConn.SetWriteDeadline(tcpService.WriteDeadline) 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 { func (tcpService *TcpService) GetConnNum() int {
tcpService.mapClientLocker.Lock() tcpService.mapClientLocker.Lock()
connNum := len(tcpService.mapClient) connNum := len(tcpService.mapClient)