websockets_endpoint.hpp

Go to the documentation of this file.

 
#ifndef PLOTLY_DETAILS_WEBSOCKETS_ENDPOINT_HPP
#define PLOTLY_DETAILS_WEBSOCKETS_ENDPOINT_HPP
 
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <functional>
#include <mutex>
#include <queue>
#include <string>
#include <string_view>
#include <thread>
#include <unordered_map>
 
namespace plotly::detail {

class WebsocketEndpointInterface {
public:
  WebsocketEndpointInterface();

  virtual ~WebsocketEndpointInterface();

  WebsocketEndpointInterface(const WebsocketEndpointInterface &) = delete;

  auto operator=(const WebsocketEndpointInterface &)
      -> WebsocketEndpointInterface & = delete;

  [[nodiscard]] virtual auto
  waitConnection(std::chrono::milliseconds timeout) const -> bool = 0;

  [[nodiscard]] virtual auto isConnected() const -> bool = 0;

  virtual auto send(const std::string_view &message) -> bool = 0;

  using callback_t = std::function<void(const std::string_view &)>;

  virtual void registerCallback(const std::string_view &eventName,
                                callback_t callback) = 0;

  virtual void unregisterCallback(const std::string_view &eventName) = 0;

  virtual void stop() = 0;

  [[nodiscard]] virtual auto getName() const -> std::string = 0;
};

class WebsocketEndpointImpl : public WebsocketEndpointInterface {
protected:
  std::atomic<bool> running{false};

  std::thread serviceThread;

  std::mutex recvMessagesMutex;

  std::condition_variable recvMessagesCv;

  std::queue<std::string> recvMessages;

  std::mutex callbackMutex;

  std::unordered_map<std::string, callback_t> callbacks;

  std::thread callbackExecutorThread;

  virtual void serviceLoop() = 0;

  void callbackExecutorLoop();

  void handleMessage(const std::string &message);

  void startCallbackExecutor();
 
public:
  void stopCallbackExecutor();

  void registerCallback(const std::string_view &eventName,
                        callback_t callback) override;

  void unregisterCallback(const std::string_view &eventName) override;
};
} // namespace plotly::detail
#endif