DHTDataStorage.h

Go to the documentation of this file.

//
// Copyright (C) 2007 Institut fuer Telematik, Universitaet Karlsruhe (TH)
//
// This program 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 2
// of the License, or (at your option) any later version.
//
// This program 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 this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
//

#ifndef __DHTDATASTORAGE_H_
#define __DHTDATASTORAGE_H_

#include <set>
#include <vector>
#include <map>
#include <sstream>

#include <omnetpp.h>

#include <NodeHandle.h>
#include <InitStages.h>
#include <BinaryValue.h>
#include <NodeVector.h>
#include <CommonMessages_m.h>

typedef std::map<BinaryValue, NodeVector> SiblingVoteMap;

struct DhtDataEntry
{
    BinaryValue value;
    uint32_t kind;
    uint32_t id;
    cMessage* ttlMessage;
    bool is_modifiable;
    NodeHandle sourceNode;
    bool responsible; //is this node responsible for this key ?
    friend std::ostream& operator<<(std::ostream& Stream, const DhtDataEntry entry);
    SiblingVoteMap siblingVote;
};

typedef std::vector<std::pair<OverlayKey, DhtDataEntry> > DhtDataVector;
typedef std::vector<DhtDumpEntry> DhtDumpVector;
typedef std::multimap<OverlayKey, DhtDataEntry> DhtDataMap;

class DHTDataStorage : public cSimpleModule
{
  public:

    virtual int numInitStages() const
    {
        return MAX_STAGE_APP + 1;
    }
    virtual void initialize(int stage);
    virtual void handleMessage(cMessage* msg);

    virtual uint32_t getSize();

    virtual void clear();

    DhtDataEntry* getDataEntry(const OverlayKey& key,
                               uint32_t kind, uint32_t id);


    virtual DhtDataVector* getDataVector(const OverlayKey& key,
                                         uint32_t kind = 0,
                                         uint32_t id = 0);

    virtual const NodeHandle& getSourceNode(const OverlayKey& key,
                                            uint32_t kind, uint32_t id);

    virtual const bool isModifiable(const OverlayKey& key,
                                    uint32_t kind, uint32_t id);

    virtual const DhtDataMap::iterator begin();

    virtual const DhtDataMap::iterator end();


    virtual DhtDataEntry* addData(const OverlayKey& key, uint32_t kind,
                                  uint32_t id,
                                  BinaryValue value, cMessage* ttlMessage,
                                  bool is_modifiable=true,
                                  NodeHandle sourceNode=NodeHandle::UNSPECIFIED_NODE,
                                  bool responsible=true);

    virtual void removeData(const OverlayKey& key, uint32_t kind, uint32_t id);

    void display();

    DhtDumpVector* dumpDht(const OverlayKey& key = OverlayKey::UNSPECIFIED_KEY,
                           uint32_t kind = 0, uint32_t id = 0);

  protected:
    DhtDataMap dataMap; 
    void updateDisplayString();

    void updateTooltip();
};

#endif