websocket-transport.hpp

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
#ifndef NFD_DAEMON_FACE_WEBSOCKET_TRANSPORT_HPP
#define NFD_DAEMON_FACE_WEBSOCKET_TRANSPORT_HPP

#include "transport.hpp"
#include "websocketpp.hpp"
#include "core/scheduler.hpp"

namespace nfd {
namespace face {

class WebSocketTransportCounters : public virtual Transport::Counters
{
public:
  PacketCounter nOutPings;

  PacketCounter nInPongs;
};

class WebSocketTransport final : public Transport
                               , protected virtual WebSocketTransportCounters
{
public:
  typedef WebSocketTransportCounters Counters;

  WebSocketTransport(websocketpp::connection_hdl hdl,
                     websocket::Server& server,
                     time::milliseconds pingInterval);

  virtual const Counters&
  getCounters() const override;

  void
  receiveMessage(const std::string& msg);

  void
  handlePong();

  void
  handlePongTimeout();

protected:
  virtual void
  beforeChangePersistency(ndn::nfd::FacePersistency newPersistency) final;

  virtual void
  doClose() final;

private:
  virtual void
  doSend(Transport::Packet&& packet) final;

  void
  schedulePing();

  void
  sendPing();

  void
  processErrorCode(const websocketpp::lib::error_code& error);

private:
  websocketpp::connection_hdl m_handle;
  websocket::Server& m_server;
  time::milliseconds m_pingInterval;
  scheduler::ScopedEventId m_pingEventId;
};

inline const WebSocketTransport::Counters&
WebSocketTransport::getCounters() const
{
  return *this;
}

} // namespace face
} // namespace nfd

#endif // NFD_DAEMON_FACE_WEBSOCKET_TRANSPORT_HPP