MemoryBundleStorage.h

Go to the documentation of this file.

/*
 * MemoryBundleStorage.h
 *
 * Copyright (C) 2011 IBR, TU Braunschweig
 *
 * Written-by: Johannes Morgenroth <morgenroth@ibr.cs.tu-bs.de>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#ifndef MEMORYBUNDLESTORAGE_H_
#define MEMORYBUNDLESTORAGE_H_

#include "Component.h"
#include "core/BundleCore.h"
#include "core/TimeEvent.h"
#include "storage/BundleStorage.h"
#include "core/Node.h"
#include "core/EventReceiver.h"

#include <ibrdtn/data/Bundle.h>
#include <ibrdtn/data/BundleList.h>

namespace dtn
{
        namespace storage
        {
                class MemoryBundleStorage : public BundleStorage, public dtn::core::EventReceiver<dtn::core::TimeEvent>, public dtn::daemon::IntegratedComponent, public BundleList::Listener
                {
                        static const std::string TAG;

                public:
                        MemoryBundleStorage(const dtn::data::Length maxsize = 0);
                        virtual ~MemoryBundleStorage();

                        virtual void store(const dtn::data::Bundle &bundle);

                        virtual bool contains(const dtn::data::BundleID &id);

                        virtual dtn::data::MetaBundle info(const dtn::data::BundleID &id);

                        virtual dtn::data::Bundle get(const dtn::data::BundleID &id);

                        virtual void get(const BundleSelector &cb, BundleResult &result) throw (NoBundleFoundException, BundleSelectorException);

                        virtual const eid_set getDistinctDestinations();

                        void remove(const dtn::data::BundleID &id);

                        void clear();

                        bool empty();

                        dtn::data::Size count();

                        void releaseCustody(const dtn::data::EID &custodian, const dtn::data::BundleID &id);

                        void raiseEvent(const dtn::core::TimeEvent &evt) throw ();

                        virtual const std::string getName() const;

                protected:
                        virtual void componentUp() throw ();
                        virtual void componentDown() throw ();

                        virtual void eventBundleExpired(const dtn::data::MetaBundle &b) throw ();

                private:
                        ibrcommon::Mutex _bundleslock;

                        typedef std::set<dtn::data::Bundle> bundle_list;
                        bundle_list _bundles;
                        dtn::data::BundleList _list;

                        void __erase(const bundle_list::iterator &iter);

                        struct CMP_BUNDLE_PRIORITY
                        {
                                bool operator() (const dtn::data::MetaBundle& lhs, const dtn::data::MetaBundle& rhs) const
                                {
                                        if (lhs.getPriority() > rhs.getPriority())
                                                return true;

                                        if (lhs.getPriority() != rhs.getPriority())
                                                return false;

                                        return lhs < rhs;
                                }
                        };

                        typedef std::set<dtn::data::MetaBundle, CMP_BUNDLE_PRIORITY> prio_bundle_set;
                        prio_bundle_set _priority_index;

                        typedef std::map<dtn::data::BundleID, dtn::data::Length> size_map;
                        size_map _bundle_lengths;
                };
        }
}

#endif /* MEMORYBUNDLESTORAGE_H_ */