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

新增获取客户端ip接口

Browse Source
This commit is contained in:
duanhf2012
2020-05-06 10:40:41 +08:00
parent 2afa611ef5
commit 0512b27a67
2 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
network/tcp_conn.go
View File

@@ -71,6 +71,10 @@ func (tcpConn *TCPConn) Close() {
tcpConn.closeFlag = true tcpConn.closeFlag = true
} }
func (tcpConn *TCPConn) GetRemoteIp() string {
return tcpConn.conn.RemoteAddr().String()
}
func (tcpConn *TCPConn) doWrite(b []byte) { func (tcpConn *TCPConn) doWrite(b []byte) {
if len(tcpConn.writeChan) == cap(tcpConn.writeChan) { if len(tcpConn.writeChan) == cap(tcpConn.writeChan) {
log.Debug("close conn: channel full") log.Debug("close conn: channel full")

10
sysservice/tcpservice.go
View File

@@ -187,3 +187,13 @@ func (slf *TcpService) Close(clientid uint64) {
return return
} }
func (slf *TcpService) GetClientIp(clientid uint64) string{
slf.mapClientLocker.Lock()
defer slf.mapClientLocker.Unlock()
pClient,ok := slf.mapClient[clientid]
if ok == false{
return ""
}
return pClient.tcpConn.GetRemoteIp()
}