C++ WebSocket庫如何處理WebSocket子協議

在C++中，處理WebSocket子協議通常需要使用一個支持子協議的WebSocket庫。這里我們以websocketpp庫為例，介紹如何處理WebSocket子協議。

首先，確保已經安裝了websocketpp庫。如果沒有安裝，可以參考官方文檔進行安裝：https://github.com/zaphoyd/websocketpp

接下來，我們創(chuàng)建一個簡單的WebSocket服務器，支持一個自定義的子協議"my_subprotocol"。

#include<iostream>
#include <websocketpp/config/asio_no_tls.hpp>
#include <websocketpp/server.hpp>

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

void on_open(server* s, websocketpp::connection_hdl hdl) {
    std::cout << "Connection opened"<< std::endl;
}

void on_message(server* s, websocketpp::connection_hdl hdl, server::message_ptr msg) {
    std::cout << "Received message: "<< msg->get_payload()<< std::endl;
    s->send(hdl, msg->get_payload(), msg->get_opcode());
}

int main() {
    server s;

    // 設置子協議
    s.set_validate_handler([](websocketpp::connection_hdl hdl) -> bool {
        auto con = s.get_con_from_hdl(hdl);
        std::string protocol = con->get_request_header("Sec-WebSocket-Protocol");
        if (protocol == "my_subprotocol") {
            con->append_header("Sec-WebSocket-Protocol", protocol);
            return true;
        }
        return false;
    });

    s.set_open_handler(std::bind(&on_open, &s, std::placeholders::_1));
    s.set_message_handler(std::bind(&on_message, &s, std::placeholders::_1, std::placeholders::_2));

    s.init_asio();
    s.listen(9002);
    s.start_accept();
    s.run();

    return 0;
}

在這個示例中，我們首先創(chuàng)建了一個WebSocket服務器，并設置了驗證處理器validate_handler。這個處理器會檢查客戶端請求的子協議，如果子協議是"my_subprotocol"，則接受連接并添加相應的響應頭。

然后，我們設置了連接打開處理器open_handler和消息處理器message_handler，分別用于處理連接打開和接收到的消息事件。

最后，我們初始化了服務器的異步IO服務，開始監(jiān)聽端口9002，并啟動服務器。

要測試這個服務器，可以使用支持WebSocket子協議的客戶端進行連接。例如，在JavaScript中，可以使用以下代碼創(chuàng)建一個連接：

const socket = new WebSocket('ws://localhost:9002', 'my_subprotocol');

socket.addEventListener('open', () => {
    console.log('Connection opened');
    socket.send('Hello, WebSocket!');
});

socket.addEventListener('message', (event) => {
    console.log('Received message:', event.data);
});

這段代碼會創(chuàng)建一個連接到本地9002端口的WebSocket，并使用"my_subprotocol"子協議。當連接打開時，它會發(fā)送一條消息"Hello, WebSocket!"，并在接收到服務器的回復時打印消息內容。