From 02a78a2bde0412ec8f0f4210a8901bddb61b1b5b Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Mon, 29 Jan 2024 22:32:14 +0800
Subject: [PATCH] 1

---
 server/gameserver/roomob.cc | 6 ++++++
 server/gameserver/roomob.h  | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/server/gameserver/roomob.cc b/server/gameserver/roomob.cc
index a29ce04a..e16eae05 100644
--- a/server/gameserver/roomob.cc
+++ b/server/gameserver/roomob.cc
@@ -28,3 +28,9 @@ Player* RoomOb::GetByAccountId(const std::string& account_id)
     auto itr = id_hash_.find(account_id);
     return itr != id_hash_.end() ? itr->second : nullptr;
 }
+
+Player* RoomOb::GetBySocket(int socket_handle)
+{
+    auto itr = socket_hash_.find(socket_handle);
+    return itr != socket_hash_.end() ? itr->second : nullptr;
+}
diff --git a/server/gameserver/roomob.h b/server/gameserver/roomob.h
index 16b759f7..51655e70 100644
--- a/server/gameserver/roomob.h
+++ b/server/gameserver/roomob.h
@@ -13,8 +13,10 @@ class RoomOb : public std::enable_shared_from_this<RoomOb>
     void UnInit();
 
     Player* GetByAccountId(const std::string& account_id);
+    Player* GetBySocket(int socket_handle);
 
 private:
     Room* room_ = nullptr;
     std::map<std::string, Player*> id_hash_;
+    std::map<int, Player*> socket_hash_;
 };