channel.h

Go to the documentation of this file.

#pragma once

#include <functional>


namespace data_channel {

class DataChannel {
private:
  typedef std::function<void(const std::string &message)> MessageHandler;
  typedef std::function<void()> CloseHandler;

public:
  DataChannel(MessageHandler message_handler, CloseHandler close_handler)
      : send_message_(message_handler), close_channel_(close_handler) {}

  DataChannel(const DataChannel &other) = delete;
  DataChannel &operator=(const DataChannel &) = delete;

  ~DataChannel() { close_channel_(); }

  void SendMessage(const std::string &message) { send_message_(message); }

  void SetOnMessageHandler(std::function<void(const std::string &message)> handler) { on_message_ = handler; }

  void SetOnCloseHandler(std::function<void()> handler) { on_close_ = handler; }

  const MessageHandler &GetOnMessageHandler() { return on_message_; }

  const CloseHandler &GetOnCloseHandler() { return on_close_; }

private:
  MessageHandler send_message_;
  CloseHandler close_channel_;

  MessageHandler on_message_;
  CloseHandler on_close_;
};

}  // namespace data_channel