PersistentStore.h

// Copyright 2008 Michael Marsh, University of Maryland.
//
// This file is part of pydtn.
//
// pydtn is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// pydtn is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with pydtn.  If not, see <http://www.gnu.org/licenses/>.
//
// The views and conclusions contained in the software and documentation
// are those of the authors and should not be interpreted as representing
// official policies, either expressed or implied, of the University
// of Maryland.
//
// pydtn extends and embeds the Python interpreter, which is
// Copyright 2001-2006 Python Software Foundation, All Rights Reserved,
// and is released under the PSF License Agreement.
//
// RANLUX random number generation uses the Boost library,
// Copyright 1994-2006 by various authors (details in individual files),
// which is released under the Boost Software License, Version 1.0.

#ifndef __WRAPPERS_HARNESS_PERSISTENT_STORE_H__
#define __WRAPPERS_HARNESS_PERSISTENT_STORE_H__

#include "config.h"

#include "simlpy/Time.h"
#include "dtn/PersistentBundleStore.h"
#include "dtn/ByteString.h"
#include <set>

class WrapNode;


class PersistentBundle
{
   public :
      PersistentBundle( DTN::Bundle* b );
      PersistentBundle( const DTN::ByteString& originator, uint32_t seqNum );
      virtual ~PersistentBundle();

      DTN::Bundle* bundle() const;

      bool operator<( const PersistentBundle& b ) const;

      const Time& timeAdded() const { return m_timeAdded; }

   private :
      DTN::Bundle*     m_bundle;
      DTN::ByteString  m_originator;
      uint32_t         m_seqNum;
      Time             m_timeAdded;
      Time             m_timeSent;
      unsigned int     m_numTries;
};

class PersistentStore : public DTN::PersistentBundleStore
{
   public :
      typedef std::set< PersistentBundle >  BundleList;
      typedef BundleList::key_type          key_type;

      PersistentStore( WrapNode* owner );
      virtual ~PersistentStore();

      BundleList::iterator begin();
      BundleList::iterator lower_bound( const key_type& x );
      BundleList::iterator upper_bound( const key_type& x );
      BundleList::iterator end();

      BundleList::const_iterator begin() const;
      BundleList::const_iterator lower_bound( const key_type& x ) const;
      BundleList::const_iterator upper_bound( const key_type& x ) const;
      BundleList::const_iterator end() const;

   protected :
      virtual DTN::BundlePointer p_addBundle( DTN::Bundle* b );
      virtual void p_deleteBundle( DTN::Bundle* b );

      virtual DTN::BundlePointer p_getPointer( const DTN::BundlePointer& p );

      virtual DTN::BundlePointer p_getPointer( const DTN::ByteString& sender,
                                               uint32_t seqNum );

      virtual bool p_validatePointer( const DTN::BundlePointer& p );

      virtual size_t p_bytesUsed() { return m_bytesUsed; }

      virtual size_t p_bundles() { return m_list.size(); }

      virtual void p_shrinkStore( size_t s );

   private :
      BundleList  m_list;
      size_t      m_bytesUsed;
      WrapNode*   m_wowner;
};

class PersistentStoreItr : public DTN::BundlePointerRepr
{
   public :
      PersistentStoreItr( PersistentStore& store,
                          PersistentStore::BundleList::iterator itr );
      virtual ~PersistentStoreItr();

      virtual bool operator==( const DTN::BundlePointerRepr& b );

      virtual DTN::BundlePointerRepr* next();

      virtual DTN::BundlePointerRepr* clone();

      virtual DTN::Bundle* bundle() const;

   private :
      PersistentStore&                       m_store;
      PersistentStore::BundleList::iterator  m_itr;
};


#endif /* __WRAPPERS_HARNESS_PERSISTENT_STORE_H__ */

---