C++ 快速搭建 WebSocket 服务

使用 C++ 搭建 WebSocket 服务需要依赖网络库和 WebSocket 协议实现。推荐使用 Boost.Beast，它是基于 Boost.Asio 的高性能 HTTP/WebSocket 库。

环境准备

确保已安装 Boost 库（1.70 或更高版本）。若未安装，可通过以下命令安装：

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

基础 WebSocket 服务代码示例

以下是一个最小化的 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 = boost::asio::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& se) {
        if (se.code() != websocket::error::closed)
            std::cerr << "Error: " << se.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);
        std::thread(&handle_session, websocket::stream<beast::tcp_stream>(std::move(socket))).detach();
    }
    return 0;
}

---

常见问题与解决方法

连接断开问题

• 原因：未正确处理 WebSocket 协议的控制帧（如 Ping/Pong）。
• 修复：在循环中主动发送 Pong 响应：

ws.set_option(websocket::stream_base::timeout::suggested(beast::role_type::server));
ws.control_callback([&](websocket::frame_type kind, beast::string_view payload) {
    if (kind == websocket::frame_type::ping)
        ws.pong(payload);
});

多线程竞争

• 现象：并发读写时可能崩溃。
• 解决：为每个会话分配独立的 io_context 或使用 strand 同步：

net::strand<net::io_context::executor_type> strand(ioc.get_executor());
ws.async_read(buffer, net::bind_executor(strand, [&](...) { ... }));

性能优化

• 使用 beast::multi_buffer 替代 flat_buffer 减少内存拷贝。
• 启用 TCP_NODELAY 减少延迟：

beast::get_lowest_layer(ws).socket().set_option(tcp::no_delay(true));

---

进阶实践

集成 JSON 处理

结合 RapidJSON 或 nlohmann/json 处理 WebSocket 消息：

#include <nlohmann/json.hpp>
nlohmann::json j = nlohmann::json::parse(beast::buffers_to_string(buffer.data()));

SSL 支持

通过 Boost.Beast 实现 WSS（WebSocket Secure）：

websocket::stream<beast::ssl_stream<tcp::socket>> wss{ioc, ctx};

---

调试技巧

• 使用 Wireshark 抓包分析协议交互。
• 开启 Boost.Beast 日志：

beast::get_lowest_layer(ws).socket().set_option(boost::asio::socket_base::debug(true));

通过以上方法可以快速构建稳定的 C++ WebSocket 服务，同时规避常见问题。