Crow项目中的WebSocket随机数生成器实现解析

概述

在Crow C++ Web框架中实现WebSocket服务时，开发者经常需要处理实时数据推送的场景。本文将以一个WebSocket随机数生成器为例，深入探讨如何正确实现周期性数据推送功能，并解决多客户端连接管理中的常见问题。

基础实现与问题

初始实现尝试在WebSocket的onopen回调中使用无限循环发送随机数：

CROW_WEBSOCKET_ROUTE(app, "/")
    .onopen([&](crow::websocket::connection& conn) {
        while(true) {
            int randomNumber = generateRandomNumber(1, 100);
            json jsonData = {{"randomNumber", randomNumber}};
            conn.send_text(jsonData.dump());
            std::this_thread::sleep_for(thread_sleep_periode);
        }
    })

这种方法存在明显问题：循环会阻塞WebSocket事件循环，导致服务器无法处理其他连接或消息。

线程池解决方案

正确的做法是使用异步线程处理数据生成和发送：

asio::thread_pool our_thread_pool(max_threads);

void workerThread(crow::websocket::connection& conn) {
    std::random_device rd;
    std::mt19937 gen(rd());
    std::uniform_int_distribution<int> distribution(1, 100);
    
    while (!shouldExit) {
        int randomNumber = distribution(gen);
        json jsonData = {{"randomNumber", randomNumber}};
        conn.send_text(jsonData.dump());
        std::this_thread::sleep_for(1000ms);
    }
}

CROW_WEBSOCKET_ROUTE(app, "/")
    .onopen([&](crow::websocket::connection& conn) {
        asio::post(our_thread_pool, [&conn]() { workerThread(conn); });
    })

这里使用了ASIO的线程池来管理工作线程，避免了阻塞主事件循环。

多客户端连接管理

初始线程池方案存在一个关键问题：使用全局shouldExit标志会导致所有客户端连接同时终止。解决方案是为每个连接维护独立的状态：

struct ClientData {
    std::atomic<bool> shouldExit{false};
};

std::unordered_map<crow::websocket::connection*, ClientData> clientDataMap;

void workerThread(crow::websocket::connection& conn, ClientData& clientData) {
    // ... 生成和发送随机数的逻辑
    while (!clientData.shouldExit) {
        // 工作代码
    }
}

.onopen([&](crow::websocket::connection& conn) {
    ClientData& clientData = clientDataMap[&conn];
    clientData.shouldExit = false;
    asio::post(our_thread_pool, [&conn, &clientData]() {
        workerThread(conn, clientData);
    });
})
.onclose([&](crow::websocket::connection& conn, const std::string& reason) {
    ClientData& clientData = clientDataMap[&conn];
    clientData.shouldExit = true;
    clientDataMap.erase(&conn);
})

服务器关闭时的资源清理

为确保服务器关闭时能正确清理所有资源，应在main函数结束时终止所有工作线程：

int main() {
    // ... 服务器初始化代码
    
    app.bindaddr("127.0.0.1").port(8080).multithreaded().run();
    
    // 清理代码
    for (auto &clientData : clientDataMap) {
        clientData.second.shouldExit = true;
    }
    our_thread_pool.join();
    return 0;
}

最佳实践总结

1. 避免阻塞事件循环：永远不要在WebSocket回调中执行耗时操作或无限循环
2. 使用线程池：ASIO的线程池是管理后台任务的理想选择
3. 独立连接状态：为每个连接维护独立的状态和控制标志
4. 资源清理：确保在服务器关闭时正确释放所有资源
5. 线程安全：使用原子变量或适当的同步机制保护共享数据

这种模式不仅适用于随机数生成器，也可应用于任何需要向WebSocket客户端推送周期性数据的场景，如实时监控、游戏状态更新等。