ae/ECAP_Server
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

重新设置代码格式

Browse Source
This commit is contained in:
种太阳的热带美短
2026-06-25 15:44:25 +08:00
parent 6a87cf52a8
commit 26b5c8a54d
123 changed files with 34632 additions and 35952 deletions
Download Patch File Download Diff File Expand all files Collapse all files
module/Local_Server/server/WebSocket_Manager.cpp
+34 -35
View File
@@ -10,10 +10,14 @@ public:
WS_PATH_LIST_BEGIN
WS_ADD_PATH_VIA_REGEX(R"(^/ws/.*$)", drogon::Get);
WS_PATH_LIST_END
void handleNewMessage(const drogon::WebSocketConnectionPtr &, std::string &&, const drogon::WebSocketMessageType &) override;
void handleConnectionClosed(const drogon::WebSocketConnectionPtr & conn) override;
void handleNewConnection(const drogon::HttpRequestPtr & req, const drogon::WebSocketConnectionPtr & conn) override;
void handleNewMessage(const drogon::WebSocketConnectionPtr &, std::string &&,
const drogon::WebSocketMessageType &) override;
void
handleConnectionClosed(const drogon::WebSocketConnectionPtr &conn) override;
void handleNewConnection(const drogon::HttpRequestPtr &req,
const drogon::WebSocketConnectionPtr &conn) override;
DSP_Ws_Mgr();
protected:
static DSP_Ws_Mgr *inst_;
};
@@ -22,30 +26,31 @@ struct WebSocket_Connection {
std::string type;
};
WebSocket_Manager::WebSocket_Manager(std::string path) : path(std::move(path)) {}
void WebSocket_Manager::remove(const drogon::WebSocketConnectionPtr &conn) {
WebSocket_Manager::WebSocket_Manager(std::string path)
: path(std::move(path)) {}
void WebSocket_Manager::remove(const drogon::WebSocketConnectionPtr &conn) {
std::lock_guard<std::mutex> g(mtx);
connections.erase(conn);
}
void WebSocket_Manager::add(const drogon::WebSocketConnectionPtr &conn) {
void WebSocket_Manager::add(const drogon::WebSocketConnectionPtr &conn) {
std::lock_guard<std::mutex> g(mtx);
connections.insert(conn);
}
void WebSocket_Manager::visit(
const std::function<void(const drogon::WebSocketConnectionPtr &conn)>
&func) {
&func) {
std::lock_guard<std::mutex> g(mtx);
for (const auto &coon: connections) {
for (const auto &coon : connections) {
func(coon);
}
}
void WebSocket_Manager::push_data(const char *buf, uint64_t len) {
void WebSocket_Manager::push_data(const char *buf, uint64_t len) {
std::lock_guard<std::mutex> g(mtx);
for (const auto &conn: connections) {
for (const auto &conn : connections) {
conn->send(buf, len, drogon::WebSocketMessageType::Binary);
}
}
size_t WebSocket_Manager::get_websocket_connect_num() {
size_t WebSocket_Manager::get_websocket_connect_num() {
std::lock_guard<std::mutex> g(mtx);
return connections.size();
}
@@ -56,7 +61,8 @@ std::shared_ptr<WebSocket_Manager> Ws_Mgr::register_ws(std::string path) {
Psc::fail_fast_core_dump("register_ws:" + VAR_STR_1(path) + "重复创建！");
return it->second;
}
std::shared_ptr<WebSocket_Manager> cur = std::make_shared<WebSocket_Manager>(path);
std::shared_ptr<WebSocket_Manager> cur =
std::make_shared<WebSocket_Manager>(path);
map[path] = cur;
return cur;
}
@@ -69,47 +75,40 @@ std::shared_ptr<WebSocket_Manager> Ws_Mgr::get_ws(std::string path) {
return nullptr;
}
void DSP_Ws_Mgr::handleNewMessage(const drogon::WebSocketConnectionPtr &,
std::string &&,
const drogon::WebSocketMessageType &) {
}
const drogon::WebSocketMessageType &) {}
void DSP_Ws_Mgr::handleConnectionClosed(
const drogon::WebSocketConnectionPtr &conn) {
const drogon::WebSocketConnectionPtr &conn) {
auto &s = conn->getContextRef<WebSocket_Connection>();
auto rp = s.type;
auto rp = s.type;
auto ws = Ws_Mgr::instance()->get_ws(rp);
//LOG_DEBUG
// LOG_DEBUG
std::ostringstream oss;
oss
<< "已有连接数:" << ws->get_websocket_connect_num() << " "
<< "关闭websocket连接:" << rp << "\n"
<< "local_addr:" << inet_address_to_string(conn->localAddr()) << "\n"
<< "peer_addr:" << inet_address_to_string(conn->peerAddr()) << "\n";
dsp_logger->c_debug("websocket", {},oss.str());
oss << "已有连接数:" << ws->get_websocket_connect_num() << " "
<< "关闭websocket连接:" << rp << "\n"
<< "local_addr:" << inet_address_to_string(conn->localAddr()) << "\n"
<< "peer_addr:" << inet_address_to_string(conn->peerAddr()) << "\n";
dsp_logger->c_debug("websocket", {}, oss.str());
ws->remove(conn);
}
void DSP_Ws_Mgr::handleNewConnection(
const drogon::HttpRequestPtr &req,
const drogon::WebSocketConnectionPtr &conn) {
const drogon::WebSocketConnectionPtr &conn) {
auto rpt = req->path();
auto rp = rpt.substr(std::string("/ws").size());
auto ws = Ws_Mgr::instance()->get_ws(rp);
std::ostringstream oss;
oss
<< "已有连接数:" << ws->get_websocket_connect_num() << " "
<< "新建websocket连接:" << rp << " "
<< "local_addr:" << inet_address_to_string(conn->localAddr()) << " "
<< "peer_addr:" << inet_address_to_string(conn->peerAddr()) << "\n";
oss << "已有连接数:" << ws->get_websocket_connect_num() << " "
<< "新建websocket连接:" << rp << " "
<< "local_addr:" << inet_address_to_string(conn->localAddr()) << " "
<< "peer_addr:" << inet_address_to_string(conn->peerAddr()) << "\n";
dsp_logger->c_debug("websocket", {},oss.str());
dsp_logger->c_debug("websocket", {}, oss.str());
WebSocket_Connection s;
s.type = rp;
conn->setContext(std::make_shared<WebSocket_Connection>(std::move(s)));
ws->add(conn);
}
DSP_Ws_Mgr::DSP_Ws_Mgr() {
inst_ = this;
}
DSP_Ws_Mgr::DSP_Ws_Mgr() { inst_ = this; }