httplib_websocket.h

Go to the documentation of this file.

#ifndef YAZE_APP_NET_NATIVE_HTTPLIB_WEBSOCKET_H_
#define YAZE_APP_NET_NATIVE_HTTPLIB_WEBSOCKET_H_
 
#include <atomic>
#include <memory>
#include <thread>
 
#include "app/net/websocket_interface.h"
 
// Forward declaration
namespace httplib {
class Client;
}
 
namespace yaze {
namespace net {
 
class HttpLibWebSocket : public IWebSocket {
 public:
  HttpLibWebSocket();
  ~HttpLibWebSocket() override;
 
  absl::Status Connect(const std::string& url) override;
 
  absl::Status Send(const std::string& message) override;
 
  absl::Status SendBinary(const uint8_t* data, size_t length) override;
 
  absl::Status Close(int code = 1000,
                    const std::string& reason = "") override;
 
  WebSocketState GetState() const override { return state_; }
 
  void OnMessage(MessageCallback callback) override {
    message_callback_ = callback;
  }
 
  void OnBinaryMessage(BinaryMessageCallback callback) override {
    binary_message_callback_ = callback;
  }
 
  void OnOpen(OpenCallback callback) override {
    open_callback_ = callback;
  }
 
  void OnClose(CloseCallback callback) override {
    close_callback_ = callback;
  }
 
  void OnError(ErrorCallback callback) override {
    error_callback_ = callback;
  }
 
 private:
  absl::Status ParseWebSocketUrl(const std::string& ws_url,
                                 std::string& http_url);
 
  void ReceiveLoop();
 
  void StopReceiveLoop();
 
  // HTTP client for fallback implementation
  std::shared_ptr<httplib::Client> client_;
 
  // Background receive thread
  std::thread receive_thread_;
  std::atomic<bool> stop_receive_;
 
  // Connection details
  std::string session_id_;
  std::string http_endpoint_;
};
 
}  // namespace net
}  // namespace yaze
 
#endif  // YAZE_APP_NET_NATIVE_HTTPLIB_WEBSOCKET_H_