scheduler.hpp

Go to the documentation of this file.

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

#include "../common.hpp"
#include "monotonic_deadline_timer.hpp"

#include <set>

namespace ndn {
namespace util {
namespace scheduler {

typedef function<void()> EventCallback;

class EventInfo;

class EventId
{
public:
  EventId(std::nullptr_t = nullptr)
  {
  }

  explicit
  operator bool() const
  {
    return !this->operator!();
  }

  bool
  operator!() const;

  bool
  operator==(const EventId& other) const;

  bool
  operator!=(const EventId& other) const
  {
    return !this->operator==(other);
  }

  void
  reset()
  {
    m_info.reset();
  }

private:
  explicit
  EventId(const weak_ptr<EventInfo>& info)
    : m_info(info)
  {
  }

private:
  weak_ptr<EventInfo> m_info;

  friend class Scheduler;
  friend std::ostream& operator<<(std::ostream& os, const EventId& eventId);
};

std::ostream&
operator<<(std::ostream& os, const EventId& eventId);

class EventQueueCompare
{
public:
  bool
  operator()(const shared_ptr<EventInfo>& a, const shared_ptr<EventInfo>& b) const;
};

typedef std::multiset<shared_ptr<EventInfo>, EventQueueCompare> EventQueue;

class Scheduler : noncopyable
{
public:
  typedef EventCallback Event;

  explicit
  Scheduler(boost::asio::io_service& ioService);

  EventId
  scheduleEvent(const time::nanoseconds& after, const EventCallback& callback);

  void
  cancelEvent(const EventId& eventId);

  void
  cancelAllEvents();

private:
  void
  scheduleNext();

  void
  executeEvent(const boost::system::error_code& code);

private:
  monotonic_deadline_timer m_deadlineTimer;
  EventQueue m_queue;
  bool m_isEventExecuting;
};

} // namespace scheduler

using util::scheduler::Scheduler;

} // namespace util

// for backwards compatibility
using util::scheduler::Scheduler;
using util::scheduler::EventId;

} // namespace ndn

#endif // NDN_UTIL_SCHEDULER_HPP