recipients_handler.hh

Go to the documentation of this file.

#ifndef RECIPIENTS_HANDLER_HH
#define RECIPIENTS_HANDLER_HH

#include "recipient.hh"

#include <string>
#include <vector>

#include <boost/asio.hpp>

/**
 * \author cpellegrino
 */

boost::asio::ip::tcp::endpoint make_endpoint(const std::string& address);

class RecipientsHandler
{
  typedef std::vector<Recipient*> container_t;

  boost::asio::io_service m_service;
  const size_t m_circbuff_size;
  container_t m_recipients;

 public:

  RecipientsHandler(size_t circbuf_size = 10)
    :
    m_circbuff_size(circbuf_size)
  {}

  ~RecipientsHandler()
  {
    wipe();
  }

  bool add(const std::string& id)
  {
    if (find(make_endpoint(id)) == m_recipients.end())
    {
      m_recipients.push_back(
          new Recipient(
              m_service,
              make_endpoint(id),
              m_circbuff_size));
      return true;
    }

    return false;
  }

  bool remove(const std::string& id)
  {
    container_t::iterator it = find(make_endpoint(id));

    if (it != m_recipients.end()) {
      const Recipient* to_be_erased = *it;

      m_recipients.erase(it);

      // write on disk to_be_erased->m_cbuffer. If necessary...

      delete to_be_erased;
      return true;
    }

    return false;
  }

  bool send(const Frame& data)
  {
    bool status = false;
    if (m_recipients.size())
    {
      const size_t n = data.getSeqNumber() % m_recipients.size();
      status = m_recipients[n]->sendIfPossible(data);
    }
    return status;
  }

  void wipe()
  {
    for (size_t i = 0; i < m_recipients.size(); ++i)
    {
      delete m_recipients[i];
    }
    m_recipients.clear();
  }

 private:

  container_t::iterator find(const boost::asio::ip::tcp::endpoint& ep)
  {
    container_t::iterator it = m_recipients.begin();
    for (container_t::const_iterator et = m_recipients.end(); it != et; ++it)
    {
      if ((*it)->m_endpoint == ep)
      {
        break;
      }
    }

    return it;
  }

};

#endif // RECIPIENTS_HANDLER_HPP