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服務器，并實現基本的路由管理。你可以根據需要擴展此示例，以支持更復雜的路由和消息處理。