webSocket模块组件

对应的头文件

cpp 复制代码
#include <websocketpp/config/asio_no_tls.hpp>
#include <websocketpp/server.hpp>
#include <nlohmann/json.hpp>
#include <boost/asio.hpp>
#include <queue>
#include <functional>
#include <memory>
#include <mutex>

using json = nlohmann::json;
using namespace websocketpp;

namespace asio = boost::asio;
using asio::io_context;

WebSocketServer类的实现

cpp 复制代码
// WebSocket服务器类
class WebSocketServer {
public:
    using server_type = server<config::asio>;
    WebSocketServer(asio::io_context& io_context)
        : m_server(io_context),
          m_io_context(io_context) {}

    void start(uint16_t port);
    void stop();
    void handle_message(server_type::connection_hdl hdl, server_type::message_ptr msg);

private:
    server_type m_server;
    asio::io_context& m_io_context;
};

MessageHandler类的实现:

cpp 复制代码
// 消息处理器类
class MessageHandler {
public:
    static void process_message(const json& message, std::function<void(json)> callback);
};

ConnectionManager类的实现:

cpp 复制代码
// 连接管理器类
class ConnectionManager {
public:
    void on_open(server_type::connection_hdl hdl);
    void on_close(server_type::connection_hdl hdl);

private:
    std::mutex m_mutex;
    std::unordered_map<server_type::connection_hdl, bool> m_connections;
};

TaskQueue类的实现:

cpp 复制代码
// 异步任务队列类
class TaskQueue {
public:
    void enqueue(std::function<void()> task);
    void run();

private:
    std::queue<std::function<void()>> m_queue;
    std::mutex m_mutex;
    std::condition_variable m_condition;
};

类函数的定义

cpp 复制代码
void WebSocketServer::start(uint16_t port) {
    m_server.init_asio();
    m_server.set_access_channels(websocketpp::log::alevel::all);
    m_server.clear_access_channels(websocketpp::log::alevel::frame_payload);
    m_server.listen(port);
    m_server.set_open_handler(std::bind(&ConnectionManager::on_open, &m_connection_manager, std::placeholders::_1));
    m_server.set_close_handler(std::bind(&ConnectionManager::on_close, &m_connection_manager, std::placeholders::_1));
    m_server.set_message_handler(std::bind(&WebSocketServer::handle_message, this, std::placeholders::_1, std::placeholders::_2));
    m_server.start_accept();
    m_io_context.run();
}

void WebSocketServer::stop() {
    m_io_context.stop();
}

void WebSocketServer::handle_message(server_type::connection_hdl hdl, server_type::message_ptr msg) {
    try {
        auto j = json::parse(msg->get_payload());
        MessageHandler::process_message(j, [this, hdl](const json& response) {
            m_server.send(hdl, response.dump(), websocketpp::frame::opcode::text);
        });
    } catch (std::exception& e) {
        // 错误处理
    }
}

void MessageHandler::process_message(const json& message, std::function<void(json)> callback) {
    // 根据消息类型处理消息
    if (message["type"] == "request") {
        // 调用相应的业务逻辑
        // ...

        // 假设我们得到了响应数据
        json response = {{"type", "response"}, {"data", "Hello World!"}};
        callback(response);
    }
}

void ConnectionManager::on_open(server_type::connection_hdl hdl) {
    std::lock_guard<std::mutex> lock(m_mutex);
    m_connections[hdl] = true;
}

void ConnectionManager::on_close(server_type::connection_hdl hdl) {
    std::lock_guard<std::mutex> lock(m_mutex);
    m_connections.erase(hdl);
}

void TaskQueue::enqueue(std::function<void()> task) {
    std::lock_guard<std::mutex> lock(m_mutex);
    m_queue.push(task);
    m_condition.notify_one();
}

void TaskQueue::run() {
    std::unique_lock<std::mutex> lock(m_mutex);
    while (true) {
        if (!m_queue.empty()) {
            auto task = std::move(m_queue.front());
            m_queue.pop();
            lock.unlock();
            task();
            lock.lock();
        } else {
            m_condition.wait(lock);
        }
    }
}
相关推荐
浮游本尊4 分钟前
React 18.x 学习计划 - 第五天:React状态管理
前端·学习·react.js
-睡到自然醒~9 分钟前
[go 面试] 前端请求到后端API的中间件流程解析
前端·中间件·面试
你什么冠军?13 分钟前
云计算与服务器概述
运维·服务器·云计算
洛卡卡了17 分钟前
Sentry 都不想接,这锅还让我背?这xx工作我不要了!
前端·架构
UNbuff_017 分钟前
Linux top 命令使用说明
linux·运维·服务器
咖啡の猫21 分钟前
Vue 实例生命周期
前端·vue.js·okhttp
JNU freshman35 分钟前
vue 之 import 的语法
前端·javascript·vue.js
剑亦未配妥36 分钟前
Vue 2 响应式系统常见问题与解决方案(包含_demo以下划线开头命名的变量导致响应式丢失问题)
前端·javascript·vue.js
凉柚ˇ39 分钟前
Vue图片压缩方案
前端·javascript·vue.js
慧一居士39 分钟前
vue 中 directive 作用,使用场景和使用示例
前端