websocket_client.hpp

Go to the documentation of this file.

 
#ifndef CAMCOM_TEST_WEBSOCKET_CLIENT_HPP
#define CAMCOM_TEST_WEBSOCKET_CLIENT_HPP
 
#include <string>
#include <vector>
#include <functional>
#include <mutex>
#include <condition_variable>
#include <atomic>
#include <memory>
 
namespace camcom::test {

class WebSocketClient {
public:
    using FrameCallback = std::function<void(const std::vector<uint8_t>&, bool)>;
 
    WebSocketClient();
    ~WebSocketClient();
 
    // Connect to WebSocket server
    bool connect(const std::string& host, int port, const std::string& path = "/ws");
 
    // Disconnect from server
    void disconnect();
 
    // Check if connected
    bool isConnected() const;
 
    // Send binary frame
    bool send(const std::vector<uint8_t>& data);
 
    // Send text frame
    bool sendText(const std::string& text);
 
    // Receive frame with timeout (blocking)
    bool receive(std::vector<uint8_t>& data, int timeout_ms = 1000);
 
    // Set frame callback (non-blocking receive)
    void setFrameCallback(FrameCallback callback);
 
    // Start background receive loop
    void startReceiveLoop();
 
    // Stop background receive loop
    void stopReceiveLoop();
 
    // Get received frames (for testing)
    std::vector<std::vector<uint8_t>> getReceivedFrames();
 
    // Clear received frames
    void clearReceivedFrames();
 
    // Get connection stats
    size_t getFramesReceived() const;
    size_t getBytesReceived() const;
 
    // Get last error message
    std::string getLastError() const;
 
private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
} // namespace camcom::test
 
#endif // CAMCOM_TEST_WEBSOCKET_CLIENT_HPP