首次提交

2026-06-16 11:18:22 +08:00
commit 068ef188e7
201 changed files with 57503 additions and 0 deletions
@@ -0,0 +1,115 @@
+#include "WebSocket_Manager.h"
+#include "../global_include.h"
+#include "Global.h"
+#include <drogon/HttpAppFramework.h>
+#include <drogon/WebSocketController.h>
+#include <unordered_set>
+class DSP_Ws_Mgr : public drogon::WebSocketController<DSP_Ws_Mgr> {
+public:
+  static DSP_Ws_Mgr *instance() { return inst_; }
+  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;
+  DSP_Ws_Mgr();
+protected:
+  static DSP_Ws_Mgr *inst_;
+};
+DSP_Ws_Mgr *DSP_Ws_Mgr::inst_ = nullptr;
+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)  {
+  std::lock_guard<std::mutex> g(mtx);
+  connections.erase(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)  {
+  std::lock_guard<std::mutex> g(mtx);
+  for (const auto &coon: connections) {
+    func(coon);
+  }
+}
+void WebSocket_Manager::push_data(const char *buf, uint64_t len)  {
+  std::lock_guard<std::mutex> g(mtx);
+  for (const auto &conn: connections) {
+    conn->send(buf, len, drogon::WebSocketMessageType::Binary);
+  }
+}
+size_t WebSocket_Manager::get_websocket_connect_num()  {
+  std::lock_guard<std::mutex> g(mtx);
+  return connections.size();
+}
+
+std::shared_ptr<WebSocket_Manager> Ws_Mgr::register_ws(std::string path) {
+  auto it = map.find(path);
+  if (it != map.end()) {
+    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);
+  map[path] = cur;
+  return cur;
+}
+
+std::shared_ptr<WebSocket_Manager> Ws_Mgr::get_ws(std::string path) {
+  auto it = map.find(path);
+  if (it != map.end()) {
+    return it->second;
+  }
+  return nullptr;
+}
+
+
+void DSP_Ws_Mgr::handleNewMessage(const drogon::WebSocketConnectionPtr &,
+                                  std::string &&,
+                                  const drogon::WebSocketMessageType &) {
+
+}
+void DSP_Ws_Mgr::handleConnectionClosed(
+const drogon::WebSocketConnectionPtr &conn)  {
+  auto &s = conn->getContextRef<WebSocket_Connection>();
+  auto  rp = s.type;
+  auto ws = Ws_Mgr::instance()->get_ws(rp);
+  //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());
+  ws->remove(conn);
+}
+void DSP_Ws_Mgr::handleNewConnection(
+    const drogon::HttpRequestPtr &req,
+    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";
+
+  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;
+}