pit.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "ndn/common.hpp"
#include "ndn/interest.hpp"
#include "ndn/name.hpp"
#include <functional>
 
namespace ndn {
 
constexpr size_t PIT_MAX_ENTRIES = 50;
 
constexpr size_t PIT_MAX_FACES_PER_ENTRY = 5;
 
enum class PitInsertResult : uint8_t {
    New,         
    Aggregated,  
    Duplicate,   
    Full,        
};
 
class PitEntry {
public:
    const Name& name() const { return name_; }
 
    uint32_t nonce() const { return nonce_; }
 
    TimeMs expireTime() const { return expireTime_; }
 
    size_t faceCount() const { return numFaces_; }
 
    FaceId face(size_t index) const;
 
    bool hasFace(FaceId faceId) const;
 
    bool addFace(FaceId faceId);
private:
    friend class Pit;
    Name name_;                                                    
    uint32_t nonce_ = 0;                                           
    TimeMs expireTime_ = 0;                                        
    std::array<FaceId, PIT_MAX_FACES_PER_ENTRY> incomingFaces_{};  
    uint8_t numFaces_ = 0;                                         
    bool inUse_ = false;                                           
};
 
class Pit {
public:
    PitInsertResult insert(const Interest& interest, FaceId incomingFace,
                           PitEntry** outEntry = nullptr);
 
    PitEntry* find(const Name& name);
 
    const PitEntry* find(const Name& name) const;
 
    void remove(PitEntry* entry);
 
    void remove(const Name& name);
    using TimeoutCallback = std::function<void(const PitEntry&)>;
 
    void processTimeouts(TimeMs now, TimeoutCallback callback = nullptr);
    size_t size() const { return size_; }
 
    size_t capacity() const { return PIT_MAX_ENTRIES; }
 
    struct Stats {
        uint32_t insertions = 0;    
        uint32_t aggregations = 0;  
        uint32_t duplicates = 0;    
        uint32_t timeouts = 0;      
    };
 
    const Stats& stats() const { return stats_; }
private:
    std::array<PitEntry, PIT_MAX_ENTRIES> entries_{};  
    size_t size_ = 0;                                  
    Stats stats_{};                                    
};
 
}  // namespace ndn