pending-interest-table.hpp

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

#include <ndn-cpp/face.hpp>

namespace ndn {

class PendingInterestTable {
public:
  class Entry {
  public:
    Entry
      (uint64_t pendingInterestId,
       const ptr_lib::shared_ptr<const Interest>& interest, const OnData& onData,
       const OnTimeout& onTimeout, const OnNetworkNack& onNetworkNack)
    : pendingInterestId_(pendingInterestId), interest_(interest), onData_(onData),
      onTimeout_(onTimeout), onNetworkNack_(onNetworkNack), isRemoved_(false)
    {
    }

    uint64_t
    getPendingInterestId() { return pendingInterestId_; }

    const ptr_lib::shared_ptr<const Interest>&
    getInterest() { return interest_; }

    const OnData&
    getOnData() { return onData_; }

    const OnNetworkNack&
    getOnNetworkNack() { return onNetworkNack_; }

    void
    setIsRemoved() { isRemoved_ = true; }

    bool
    getIsRemoved() { return isRemoved_; }

    void
    callTimeout();

  private:
    ptr_lib::shared_ptr<const Interest> interest_;
    uint64_t pendingInterestId_;  
    const OnData onData_;
    const OnTimeout onTimeout_;
    const OnNetworkNack onNetworkNack_;
    bool isRemoved_;
  };

  ptr_lib::shared_ptr<Entry>
  add(uint64_t pendingInterestId,
      const ptr_lib::shared_ptr<const Interest>& interestCopy,
      const OnData& onData, const OnTimeout& onTimeout,
      const OnNetworkNack& onNetworkNack);

  void
  extractEntriesForExpressedInterest
    (const Name& name, std::vector<ptr_lib::shared_ptr<Entry> > &entries);

  void
  extractEntriesForNackInterest
    (const Interest& interest, std::vector<ptr_lib::shared_ptr<Entry> > &entries);

  void
  removePendingInterest(uint64_t pendingInterestId);

  bool
  removeEntry(const ptr_lib::shared_ptr<Entry>& pendingInterest);

private:
  std::vector<ptr_lib::shared_ptr<Entry> > table_;
  std::vector<uint64_t> removeRequests_;
};

}

#endif