-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add client command
- Loading branch information
Showing
11 changed files
with
421 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| #include "client_map.h" | ||
| #include "log.h" | ||
|
|
||
| namespace kiwi { | ||
|
|
||
| uint32_t ClientMap::GetAllClientInfos(std::vector<ClientInfo>& results) { | ||
| // client info string type: ip, port, fd. | ||
| std::shared_lock<std::shared_mutex> client_map_lock(client_map_mutex_); | ||
| for (auto& [id, client_weak] : clients_) { | ||
| if (auto client = client_weak.lock()) { | ||
| results.emplace_back(client->GetClientInfo()); | ||
| } | ||
| } | ||
| return results.size(); | ||
| } | ||
|
|
||
| bool ClientMap::AddClient(int id, std::weak_ptr<PClient> client) { | ||
| std::unique_lock client_map_lock(client_map_mutex_); | ||
| if (clients_.find(id) == clients_.end()) { | ||
| clients_.insert({id, client}); | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| ClientInfo ClientMap::GetClientsInfoById(int id) { | ||
| std::shared_lock client_map_lock(client_map_mutex_); | ||
| if (auto it = clients_.find(id); it != clients_.end()) { | ||
| if (auto client = it->second.lock(); client) { | ||
| return client->GetClientInfo(); | ||
| } | ||
| } | ||
| ERROR("Client with ID {} not found in GetClientsInfoById", id); | ||
| return ClientInfo::invalidClientInfo; | ||
| } | ||
|
|
||
| bool ClientMap::RemoveClientById(int id) { | ||
| std::unique_lock client_map_lock(client_map_mutex_); | ||
| if (auto it = clients_.find(id); it != clients_.end()) { | ||
| clients_.erase(it); | ||
| INFO("Removed client with ID {}", id); | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| bool ClientMap::KillAllClients() { | ||
| std::vector<std::shared_ptr<PClient>> clients_to_close; | ||
| { | ||
| std::shared_lock<std::shared_mutex> client_map_lock(client_map_mutex_); | ||
| for (auto& [id, client_weak] : clients_) { | ||
| if (auto client = client_weak.lock()) { | ||
| clients_to_close.push_back(client); | ||
| } | ||
| } | ||
| } | ||
| for (auto& client : clients_to_close) { | ||
| client->Close(); | ||
| } | ||
| return true; | ||
| } | ||
|
|
||
| bool ClientMap::KillClientByAddrPort(const std::string& addr_port) { | ||
| std::shared_ptr<PClient> client_to_close; | ||
| { | ||
| std::shared_lock<std::shared_mutex> client_map_lock(client_map_mutex_); | ||
| for (auto& [id, client_weak] : clients_) { | ||
| if (auto client = client_weak.lock()) { | ||
| std::string client_ip_port = client->PeerIP() + ":" + std::to_string(client->PeerPort()); | ||
| if (client_ip_port == addr_port) { | ||
| client_to_close = client; | ||
| break; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| if (client_to_close) { | ||
| client_to_close->Close(); | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| bool ClientMap::KillClientById(int client_id) { | ||
| std::shared_ptr<PClient> client_to_close; | ||
| { | ||
| std::shared_lock<std::shared_mutex> client_map_lock(client_map_mutex_); | ||
| if (auto it = clients_.find(client_id); it != clients_.end()) { | ||
| if (auto client = it->second.lock()) { | ||
| client_to_close = client; | ||
| } | ||
| } | ||
| } | ||
| if (client_to_close) { | ||
| INFO("Closing client with ID {}", client_id); | ||
| client_to_close->Close(); | ||
| INFO("Client with ID {} closed", client_id); | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| } // namespace kiwi |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,41 @@ | ||
| #pragma once | ||
|
|
||
| #include <map> | ||
| #include <memory> | ||
| #include <shared_mutex> | ||
| #include <string> | ||
| #include "client.h" | ||
|
|
||
| namespace kiwi { | ||
| class ClientMap { | ||
| private: | ||
| ClientMap() = default; | ||
| // 禁用复制构造函数和赋值运算符 | ||
|
|
||
| private: | ||
| std::map<int, std::weak_ptr<PClient>> clients_; | ||
| std::shared_mutex client_map_mutex_; | ||
|
|
||
| public: | ||
| static ClientMap& getInstance() { | ||
| static ClientMap instance; | ||
| return instance; | ||
| } | ||
|
|
||
| ClientMap(const ClientMap&) = delete; | ||
| ClientMap& operator=(const ClientMap&) = delete; | ||
|
|
||
| // client info function | ||
| kiwi::ClientInfo GetClientsInfoById(int id); | ||
| uint32_t GetAllClientInfos(std::vector<ClientInfo>& results); | ||
|
|
||
| bool AddClient(int id, std::weak_ptr<PClient>); | ||
|
|
||
| bool RemoveClientById(int id); | ||
|
|
||
| bool KillAllClients(); | ||
| bool KillClientById(int client_id); | ||
| bool KillClientByAddrPort(const std::string& addr_port); | ||
| }; | ||
|
|
||
| } // namespace kiwi |
Oops, something went wrong.