38 lines
1.7 KiB
C++
38 lines
1.7 KiB
C++
#pragma once
|
|
#include "Graph_Session.hpp"
|
|
#include <drogon/WebSocketController.h>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <string>
|
|
namespace aethera::web {
|
|
class Graph_WebSocket final : public std::enable_shared_from_this<Graph_WebSocket> {
|
|
public:
|
|
Graph_WebSocket(drogon::WebSocketConnectionPtr connection, std::shared_ptr<Graph_Session> session);
|
|
~Graph_WebSocket();
|
|
Graph_WebSocket(const Graph_WebSocket&) = delete;
|
|
Graph_WebSocket& operator=(const Graph_WebSocket&) = delete;
|
|
using Callback = std::function<void(std::string_view)>;
|
|
void start();
|
|
void receive(std::string_view message);
|
|
void close();
|
|
void register_callback(std::string key, Callback callback);
|
|
void unregister_callback(std::string_view key);
|
|
void send(std::string_view key, std::string_view frame);
|
|
private:
|
|
struct Private;
|
|
std::unique_ptr<Private> d; /* Drogon 连接与图会话绑定。 */
|
|
};
|
|
class Graph_WebSocket_Controller final : public drogon::WebSocketController<Graph_WebSocket_Controller, false> {
|
|
public:
|
|
explicit Graph_WebSocket_Controller(std::shared_ptr<Graph_Registry> registry);
|
|
void handleNewMessage(const drogon::WebSocketConnectionPtr& connection, std::string&& message, const drogon::WebSocketMessageType& type) override;
|
|
void handleNewConnection(const drogon::HttpRequestPtr& request, const drogon::WebSocketConnectionPtr& connection) override;
|
|
void handleConnectionClosed(const drogon::WebSocketConnectionPtr& connection) override;
|
|
WS_PATH_LIST_BEGIN
|
|
WS_ADD_PATH_VIA_REGEX("^/ws/graphs/[^/]+$");
|
|
WS_PATH_LIST_END
|
|
private:
|
|
std::shared_ptr<Graph_Registry> registry; /* 所有图 WebSocket 共用的会话注册表。 */
|
|
};
|
|
}
|