BundleSelector.h

Go to the documentation of this file.

/*
 * BundleSelector.h
 *
 *  Created on: 22.01.2013
 *      Author: morgenro
 */

#ifndef BUNDLESELECTOR_H_
#define BUNDLESELECTOR_H_

#include "storage/BundleResult.h"
#include <ibrdtn/data/EID.h>
#include <ibrdtn/data/MetaBundle.h>
#include <ibrdtn/data/Number.h>
#include <ibrcommon/Exceptions.h>

namespace dtn
{
        namespace storage
        {
                class BundleSelectorException : public ibrcommon::Exception
                {
                public:
                        BundleSelectorException(std::string what = "Bundle query aborted.") throw() : ibrcommon::Exception(what)
                        {
                        }
                };

                class NoBundleFoundException : public dtn::MissingObjectException
                {
                public:
                        NoBundleFoundException(std::string what = "No bundle match the specified criteria.") throw() : dtn::MissingObjectException(what)
                        {
                        };
                };

                class BundleSelector
                {
                public:
                        virtual ~BundleSelector() {};

                        virtual dtn::data::Size limit() const throw () { return 1; };

                        virtual bool shouldAdd(const dtn::data::MetaBundle&) const throw (BundleSelectorException) { return false; };

                        virtual bool addIfSelected(BundleResult &result, const dtn::data::MetaBundle &bundle) const throw (BundleSelectorException) {
                                if (shouldAdd(bundle)) {
                                        result.put(bundle);
                                        return true;
                                }
                                return false;
                        };
                };
        }
}

#endif /* BUNDLESELECTOR_H_ */