client.h

Go to the documentation of this file.

#pragma once

#include <cstdint>
#include <string>
#include <vector>

#include "boost/asio.hpp"

class Clients;
class Node;

class Client : public std::enable_shared_from_this<Client> {
 public:
  Client(Node* node, Clients* clients, boost::asio::ip::tcp::socket socket);
  ~Client();

  void start();
  void stop();

  void publish(const std::string& channel, const std::string& msg);

 private:
  Node& node;
  Clients& clients;

  boost::asio::ip::tcp::socket socket;

  uint8_t temp[1024 * 1024];
  std::vector<uint8_t> buffer;
  std::vector<uint8_t> writing;
  std::vector<uint8_t> pending;
  std::mutex publish_mutex;
  const uint MAX_PENDING_SIZE = 1073741824;  // 1GB

  void handle_read(const boost::system::error_code& ec, std::size_t length);
  void handle_write(const boost::system::error_code& ec);

  void handle_register_desc();
  void handle_add_writer();
  void handle_add_reader();
  void handle_publish();

  uint32_t get32le(size_t offset) const;
};