C++ WebSocket库对WebSocket消息的路由管理

在C++中，有多种WebSocket库可用于实现WebSocket通信。其中一个流行的库是websocketpp。为了实现WebSocket消息的路由管理，你需要设计一个路由器类，该类负责将收到的WebSocket消息分发给相应的处理函数。

以下是一个简单的示例，展示如何使用websocketpp库创建一个基本的WebSocket服务器，并实现路由管理：

1. 首先，安装websocketpp库。你可以使用包管理器（如vcpkg）或从源代码编译。

2. 创建一个名为WebSocketServer.h的头文件，定义一个WebSocketServer类：

#pragma once

#include <websocketpp/config/asio_no_tls.hpp>
#include <websocketpp/server.hpp>
#include<functional>
#include <unordered_map>

typedef websocketpp::server<websocketpp::config::asio> WebSocketServerType;

class WebSocketServer {
public:
    WebSocketServer(uint16_t port);
    void run();
    void addRoute(const std::string& route, std::function<void(const std::string&)> handler);

private:
    void onMessage(websocketpp::connection_hdl hdl, WebSocketServerType::message_ptr msg);
    void onOpen(websocketpp::connection_hdl hdl);
    void onClose(websocketpp::connection_hdl hdl);

    uint16_t m_port;
    WebSocketServerType m_server;
    std::unordered_map<std::string, std::function<void(const std::string&)>> m_routes;
};

3. 创建一个名为WebSocketServer.cpp的源文件，实现WebSocketServer类：

#include "WebSocketServer.h"
#include<iostream>

WebSocketServer::WebSocketServer(uint16_t port) : m_port(port) {
    m_server.init_asio();
    m_server.set_message_handler(std::bind(&WebSocketServer::onMessage, this, std::placeholders::_1, std::placeholders::_2));
    m_server.set_open_handler(std::bind(&WebSocketServer::onOpen, this, std::placeholders::_1));
    m_server.set_close_handler(std::bind(&WebSocketServer::onClose, this, std::placeholders::_1));
}

void WebSocketServer::run() {
    m_server.listen(m_port);
    m_server.start_accept();
    m_server.run();
}

void WebSocketServer::addRoute(const std::string& route, std::function<void(const std::string&)> handler) {
    m_routes[route] = handler;
}

void WebSocketServer::onMessage(websocketpp::connection_hdl hdl, WebSocketServerType::message_ptr msg) {
    auto it = m_routes.find(msg->get_payload());
    if (it != m_routes.end()) {
        it->second(msg->get_payload());
    } else {
        std::cout << "No handler found for message: "<< msg->get_payload()<< std::endl;
    }
}

void WebSocketServer::onOpen(websocketpp::connection_hdl hdl) {
    std::cout << "Connection opened"<< std::endl;
}

void WebSocketServer::onClose(websocketpp::connection_hdl hdl) {
    std::cout << "Connection closed"<< std::endl;
}

4. 创建一个名为main.cpp的主文件，使用WebSocketServer类：

#include "WebSocketServer.h"
#include<iostream>

int main() {
    WebSocketServer server(9002);

    server.addRoute("hello", [](const std::string& msg) {
        std::cout << "Received message: "<< msg<< std::endl;
    });

    server.run();

    return 0;
}

5. 编译并运行程序。现在，当你向服务器发送一个包含"hello"的WebSocket消息时，它将输出"Received message: hello"。

这个示例展示了如何使用websocketpp库创建一个简单的WebSocket服务器，并实现基本的路由管理。你可以根据需要扩展此示例，以支持更复杂的路由和消息处理。