Comm.h

Go to the documentation of this file.

#ifndef HYPERTABLE_COMMENGINE_H
#define HYPERTABLE_COMMENGINE_H

#include "Common/Mutex.h"
#include "Common/ReferenceCount.h"

#include "CommAddress.h"
#include "CommBuf.h"
#include "ConnectionHandlerFactory.h"
#include "DispatchHandler.h"
#include "HandlerMap.h"


namespace Hypertable {

  class Comm : public ReferenceCount {
  public:
    static Comm *instance() {
      ScopedLock lock(ms_mutex);

      if (!ms_instance)
        ms_instance = new Comm();

      return ms_instance;
    }

    static void destroy();

    int connect(const CommAddress &addr,
                DispatchHandlerPtr &default_handler);

    int connect(const CommAddress &addr, const CommAddress &local_addr,
                DispatchHandlerPtr &default_handler);


    int set_alias(const InetAddr &addr, const InetAddr &alias);

    int add_proxy(const String &proxy, const InetAddr &addr);

    void get_proxy_map(ProxyMapT &proxy_map);

    bool wait_for_proxy_load(Timer &timer);

    void listen(const CommAddress &addr, ConnectionHandlerFactoryPtr &chf);

    void listen(const CommAddress &addr, ConnectionHandlerFactoryPtr &chf,
                DispatchHandlerPtr &default_handler);

    int send_request(const CommAddress &addr, uint32_t timeout_ms,
                     CommBufPtr &cbuf, DispatchHandler *response_handler);

    int send_response(const CommAddress &addr, CommBufPtr &cbuf);

    int get_local_address(const CommAddress &addr, CommAddress &local_addr);

    void create_datagram_receive_socket(CommAddress &addr, int tos,
                                        DispatchHandlerPtr &handler);

    int send_datagram(const CommAddress &addr, const CommAddress &send_addr,
                      CommBufPtr &cbuf);

    int set_timer(uint32_t duration_millis, DispatchHandler *handler);

    int set_timer_absolute(boost::xtime expire_time, DispatchHandler *handler);

    int close_socket(const CommAddress &addr);

  private:
    Comm();     // prevent non-singleton usage
    ~Comm();

    static Comm *ms_instance;

    int send_request(IOHandlerDataPtr &data_handler, uint32_t timeout_ms,
                     CommBufPtr &cbuf, DispatchHandler *response_handler);

    int connect_socket(int sd, const CommAddress &addr,
                       DispatchHandlerPtr &default_handler);

    static atomic_t ms_next_request_id;

    static Mutex   ms_mutex;
    HandlerMapPtr  m_handler_map;
    ReactorPtr     m_timer_reactor;
  };

} // namespace Hypertable

#endif // HYPERTABLE_COMMENGINE_H