C++ WebSocket庫如何管理心跳檢測

在C++中，使用WebSocket庫進行心跳檢測通常需要以下幾個步驟：

1. 建立連接：首先，你需要使用WebSocket庫（如websocketpp、libwebsockets等）創建一個客戶端或服務器端的WebSocket連接。這些庫通常提供了創建和管理WebSocket連接的API。

2. 設置心跳間隔：在創建WebSocket連接時，你可以設置心跳檢測的時間間隔。這通常是通過庫提供的配置選項來實現的。例如，在websocketpp庫中，你可以設置set_ping_interval和set_pong_timeout來控制心跳檢測的時間間隔。

3. 發送心跳：在連接建立后，你需要定期發送心跳消息。這可以通過庫提供的API來實現，例如websocketpp庫中的send_ping方法。你可以使用定時器（如boost::asio::deadline_timer）來定期調用這個方法。

4. 處理心跳響應：當收到對方發送的心跳響應（如PONG消息）時，你需要更新連接的最后活動時間。這可以通過庫提供的回調函數來實現，例如websocketpp庫中的on_pong事件。在這個回調函數中，你可以記錄當前時間作為最后活動時間。

5. 檢測連接超時：你需要定期檢查連接是否超時。這可以通過比較當前時間和最后活動時間來實現。如果連接超時，你可以關閉連接并觸發相應的事件（如重新連接或報告錯誤）。

下面是一個使用websocketpp庫進行心跳檢測的簡單示例：

#include <websocketpp/config/asio_no_tls_client.hpp>
#include <websocketpp/client.hpp>
#include<iostream>
#include<chrono>
#include<thread>

typedef websocketpp::client<websocketpp::config::asio_client> client;

using websocketpp::lib::placeholders::_1;
using websocketpp::lib::placeholders::_2;
using websocketpp::lib::bind;

// 定義心跳間隔和超時時間
const int PING_INTERVAL = 10; // 10秒
const int PONG_TIMEOUT = 30; // 30秒

// 心跳定時器
websocketpp::lib::shared_ptr<websocketpp::lib::asio::steady_timer> ping_timer;

// 記錄最后活動時間
std::chrono::time_point<std::chrono::steady_clock> last_activity;

void on_open(client* c, websocketpp::connection_hdl hdl) {
    // 開始發送心跳
    ping_timer->expires_from_now(std::chrono::seconds(PING_INTERVAL));
    ping_timer->async_wait(bind(&on_ping_timer, c, hdl, _1));
}

void on_pong(client* c, websocketpp::connection_hdl hdl, std::string data) {
    // 更新最后活動時間
    last_activity = std::chrono::steady_clock::now();
}

void on_ping_timer(client* c, websocketpp::connection_hdl hdl, const boost::system::error_code& ec) {
    if (ec) {
        return;
    }

    // 發送心跳
    c->send_ping(hdl);

    // 檢查連接超時
    auto now = std::chrono::steady_clock::now();
    auto duration = std::chrono::duration_cast<std::chrono::seconds>(now - last_activity).count();
    if (duration > PONG_TIMEOUT) {
        std::cout << "Connection timeout"<< std::endl;
        c->close(hdl, websocketpp::close::status::normal, "");
        return;
    }

    // 繼續發送心跳
    ping_timer->expires_from_now(std::chrono::seconds(PING_INTERVAL));
    ping_timer->async_wait(bind(&on_ping_timer, c, hdl, _1));
}

int main() {
    client c;

    try {
        // 初始化WebSocket客戶端
        c.init_asio();
        c.set_ping_interval(0); // 禁用內置的心跳檢測
        c.set_pong_timeout(0);
        c.set_open_handler(bind(&on_open, &c, _1));
        c.set_pong_handler(bind(&on_pong, &c, _1, _2));

        // 連接到WebSocket服務器
        websocketpp::lib::error_code ec;
        client::connection_ptr con = c.get_connection("ws://example.com", ec);
        if (ec) {
            std::cout << "Could not create connection: " << ec.message()<< std::endl;
            return 1;
        }

        c.connect(con);

        // 啟動事件循環
        c.run();
    } catch (websocketpp::exception const& e) {
        std::cout << e.what()<< std::endl;
    }

    return 0;
}

請注意，這個示例僅用于演示目的，實際使用時可能需要根據你的需求進行修改。