C++ WebSocket库如何处理WebSocket子协议

在C++中，处理WebSocket子协议通常需要使用一个支持子协议的WebSocket库。这里我们以websocketpp库为例，介绍如何处理WebSocket子协议。

首先，确保已经安装了websocketpp库。如果没有安装，可以参考官方文档进行安装：https://github.com/zaphoyd/websocketpp

接下来，我们创建一个简单的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;
}

在这个示例中，我们首先创建了一个WebSocket服务器，并设置了验证处理器validate_handler。这个处理器会检查客户端请求的子协议，如果子协议是"my_subprotocol"，则接受连接并添加相应的响应头。

然后，我们设置了连接打开处理器open_handler和消息处理器message_handler，分别用于处理连接打开和接收到的消息事件。

最后，我们初始化了服务器的异步IO服务，开始监听端口9002，并启动服务器。

要测试这个服务器，可以使用支持WebSocket子协议的客户端进行连接。例如，在JavaScript中，可以使用以下代码创建一个连接：

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);
});

这段代码会创建一个连接到本地9002端口的WebSocket，并使用"my_subprotocol"子协议。当连接打开时，它会发送一条消息"Hello, WebSocket!"，并在接收到服务器的回复时打印消息内容。