搭建 WebSocket 服务的核心步骤

在 C++ 中实现 WebSocket 服务通常需要依赖第三方库，常用的库包括 Boost.Beast、WebSocket++ 或 uWebSockets。以下以 Boost.Beast 为例，介绍 WebSocket 服务的搭建方法。

安装依赖
Boost.Beast 是 Boost 的一部分，建议使用 Boost 1.70+ 版本。安装方法：

sudo apt-get install libboost-all-dev  # Ubuntu/Debian
brew install boost                     # macOS

创建 WebSocket 服务器
以下是一个基于 Boost.Beast 的最小化 WebSocket 服务器实现：

#include <boost/beast.hpp>
#include <boost/asio.hpp>
#include <iostream>

namespace beast = boost::beast;
namespace http = beast::http;
namespace websocket = beast::websocket;
namespace net = boost::asio;
using tcp = net::ip::tcp;

void handle_session(websocket::stream<beast::tcp_stream> ws) {
    try {
        ws.accept();  // 完成 WebSocket 握手
        beast::flat_buffer buffer;
        while (true) {
            ws.read(buffer);  // 读取客户端消息
            auto msg = beast::buffers_to_string(buffer.data());
            std::cout << "Received: " << msg << std::endl;
            ws.text(ws.got_text());
            ws.write(buffer.data());  // 回显消息
            buffer.consume(buffer.size());
        }
    } catch (beast::system_error const& e) {
        if (e.code() != websocket::error::closed) {
            std::cerr << "WebSocket error: " << e.what() << std::endl;
        }
    }
}

int main() {
    net::io_context ioc;
    tcp::acceptor acceptor(ioc, {tcp::v4(), 8080});
    while (true) {
        tcp::socket socket(ioc);
        acceptor.accept(socket);
        websocket::stream<beast::tcp_stream> ws(std::move(socket));
        std::thread(handle_session, std::move(ws)).detach();
    }
}

常见踩坑及解决方案

1. 跨域问题
默认情况下，浏览器会阻止跨域 WebSocket 连接，可以在握手阶段设置 Access-Control-Allow-Origin 头部：

ws.set_option(websocket::stream_base::decorator(
    [](websocket::response_type& res) {
        res.set(http::field::access_control_allow_origin, "*");
    }
));

2. 线程安全问题
Boost.Beast 的 io_context 是非线程安全的，多线程环境下建议使用 strand 同步操作：

net::strand<net::io_context::executor_type> strand(net::make_strand(ioc));
ws.async_accept(net::bind_executor(strand, [](beast::error_code ec) {
    if (ec) { /* 错误处理 */ }
}));

3. 心跳检测与超时
WebSocket 长时间无数据交互可能导致连接断开，可以通过 ping/pong 机制维持连接：

ws.set_option(websocket::stream_base::ping_callback(
    [](auto&&...) { std::cout << "Ping received" << std::endl; }
));

性能优化建议

• 使用异步模型：async_read 和 async_write 可避免线程阻塞，提升并发能力。
• 减少内存拷贝：boost::beast::flat_buffer 优化数据缓冲管理。
• 负载均衡：在高并发场景下，可采用多线程 io_context 或结合 Nginx 反向代理。

测试与调试

推荐使用 wscat 工具测试 WebSocket 服务：

npm install -g wscat
wscat -c ws://localhost:8080

通过以上步骤，可以在 C++ 中快速搭建高性能 WebSocket 服务，并规避常见问题。