84 lines
2.4 KiB
C++
84 lines
2.4 KiB
C++
#include "server-core.h"
|
|
|
|
// https://wizardforcel.gitbooks.io/wxwidgets-book/content/133.html
|
|
|
|
CServerCore::CServerCore()
|
|
{
|
|
}
|
|
|
|
CServerCore::~CServerCore()
|
|
{
|
|
}
|
|
|
|
bool CServerCore::Init(const wxString& ip, unsigned short port)
|
|
{
|
|
wxIPV4address addr;
|
|
addr.Service(port);
|
|
server_ = std::make_unique<wxSocketServer>(addr, wxSOCKET_NOWAIT);
|
|
|
|
if (!server_->IsOk()) {
|
|
return false;
|
|
}
|
|
|
|
server_id_ = wxNewId();
|
|
server_->SetEventHandler(*this, server_id_);
|
|
server_->SetNotify(wxSOCKET_CONNECTION_FLAG); // 1. 设置监听的事件类型
|
|
server_->Notify(true); // 2. 启用事件通知
|
|
|
|
Bind(wxEVT_SOCKET, &CServerCore::OnServerEvent, this, server_id_);
|
|
return true;
|
|
}
|
|
|
|
int CServerCore::Run()
|
|
{
|
|
wxEventLoop loop;
|
|
return loop.Run();
|
|
}
|
|
|
|
void CServerCore::OnServerEvent(wxSocketEvent& event)
|
|
{
|
|
if (event.GetSocketEvent() == wxSOCKET_CONNECTION) {
|
|
auto* cli_ptr = server_->Accept(true);
|
|
if (!cli_ptr) {
|
|
return;
|
|
}
|
|
std::shared_ptr<wxSocketBase> cli(cli_ptr, [](wxSocketBase* ptr) { ptr->Destroy(); });
|
|
wxIPV4address peer_addr;
|
|
wxString id;
|
|
if (cli->GetPeer(peer_addr)) {
|
|
id.append(peer_addr.IPAddress());
|
|
id.append(":");
|
|
id.append(std::to_string(peer_addr.Service()));
|
|
wxLogMessage("New connection from %s", id);
|
|
}
|
|
|
|
bool can = false;
|
|
|
|
{
|
|
std::unique_lock<std::shared_mutex> lock(cli_mut_);
|
|
if (client_cache_.size() >= 100) {
|
|
can = false;
|
|
cli->Close();
|
|
} else {
|
|
auto cache = std::make_shared<ClientCache>();
|
|
cache->socket = cli;
|
|
cache->last_recv_time = std::chrono::high_resolution_clock::now();
|
|
auto now = wxDateTime::Now();
|
|
cache->online_time = wxString::Format("%s.%03ld", now.Format("%Y-%m-%d %H:%M:%S"), now.GetMillisecond());
|
|
client_cache_.emplace(id, cache);
|
|
can = true;
|
|
}
|
|
}
|
|
|
|
if (!can) {
|
|
std::this_thread::sleep_for(std::chrono::minutes(1));
|
|
} else {
|
|
client_thread_.emplace(id, std::thread(&CServerCore::th_client_thread, this, cli, id));
|
|
}
|
|
}
|
|
}
|
|
|
|
void CServerCore::th_client_thread(const std::shared_ptr<wxSocketBase>& socket, const wxString& id)
|
|
{
|
|
}
|