From a35add02b6e6207f34b750b7e36243beb000bb84 Mon Sep 17 00:00:00 2001
From: azw <vcl_delphi@126.com>
Date: Mon, 17 Apr 2023 06:37:54 +0000
Subject: [PATCH] 1

---
 server/wsproxy/GCListener.cc | 14 +++++++++++++-
 server/wsproxy/GCListener.h  |  2 ++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/server/wsproxy/GCListener.cc b/server/wsproxy/GCListener.cc
index 7c758be..fe1aca0 100644
--- a/server/wsproxy/GCListener.cc
+++ b/server/wsproxy/GCListener.cc
@@ -228,7 +228,19 @@ long long GCListener::GetSentBytesNum()
     return tcp_listener_->sent_bytes_num;
 }
 
+std::shared_ptr<KcpSession> GCListener::GetKcpSessionByRemoteKey(long long key)
+{
+    auto itr = kcp_session_addr_hash_.find(key);
+    return itr != kcp_session_addr_hash_.end() ? itr->second : nullptr;
+}
+
+std::shared_ptr<KcpSession> GCListener::GetKcpSessionBySocketHandle(int socket_handle)
+{
+    auto itr = kcp_session_handle_hash_.find(socket_handle);
+    return itr != kcp_session_handle_hash_.end() ? itr->second : nullptr;
+}
+
 void GCListener::ProcUdpPacket(a8::UdpPacket* pkt)
 {
-
+    long long key = pkt->GetRemoteKey();
 }
diff --git a/server/wsproxy/GCListener.h b/server/wsproxy/GCListener.h
index 2fb8106..7a3798f 100644
--- a/server/wsproxy/GCListener.h
+++ b/server/wsproxy/GCListener.h
@@ -37,6 +37,8 @@ class GCListener : public a8::Singleton<GCListener>
     long long GetSendNodeNum();
     long long GetSentBytesNum();
 
+    std::shared_ptr<KcpSession> GetKcpSessionByRemoteKey(long long key);
+    std::shared_ptr<KcpSession> GetKcpSessionBySocketHandle(int socket_handle);
     void ProcUdpPacket(a8::UdpPacket* pkt);
 
 private: