From 0512b27a67779fd5958943d69d6433e1d934c774 Mon Sep 17 00:00:00 2001
From: duanhf2012 <boyce@atbc.com>
Date: Wed, 6 May 2020 10:40:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=8E=B7=E5=8F=96=E5=AE=A2?=
 =?UTF-8?q?=E6=88=B7=E7=AB=AFip=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 network/tcp_conn.go      |  4 ++++
 sysservice/tcpservice.go | 10 ++++++++++
 2 files changed, 14 insertions(+)

diff --git a/network/tcp_conn.go b/network/tcp_conn.go
index 26c2318..668bf92 100644
--- a/network/tcp_conn.go
+++ b/network/tcp_conn.go
@@ -71,6 +71,10 @@ func (tcpConn *TCPConn) Close() {
 	tcpConn.closeFlag = true
 }
 
+func (tcpConn *TCPConn) GetRemoteIp() string {
+	return tcpConn.conn.RemoteAddr().String()
+}
+
 func (tcpConn *TCPConn) doWrite(b []byte) {
 	if len(tcpConn.writeChan) == cap(tcpConn.writeChan) {
 		log.Debug("close conn: channel full")
diff --git a/sysservice/tcpservice.go b/sysservice/tcpservice.go
index 427b9e3..ba6bd70 100644
--- a/sysservice/tcpservice.go
+++ b/sysservice/tcpservice.go
@@ -187,3 +187,13 @@ func (slf *TcpService) Close(clientid uint64) {
 	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()
+}