Changeset 6835 for source/ariba/overlay/BaseOverlay.cpp

Timestamp:

Nov 5, 2009, 10:18:18 AM (16 years ago)

Author:

mies

Message:

added replace flag

File:

• source/ariba/overlay/BaseOverlay.cpp (modified) (2 diffs)

source/ariba/overlay/BaseOverlay.cpp

@@ -2108 +2108 @@
         switch (msg->getType()) {
         case OverlayMsg::typeDHTPut: {
+                if (dhtMsg->doReplace()) dht->remove(dhtMsg->getKey());
                 BOOST_FOREACH( Data value, dhtMsg->getValues() )
                                         dht->put(dhtMsg->getKey(), value, dhtMsg->getTTL() );
@@ -2145 +2146 @@
 
 /// put a value to the DHT with a ttl given in seconds
-void BaseOverlay::dhtPut( const Data& key, const Data& value, int ttl ) {
+void BaseOverlay::dhtPut( const Data& key, const Data& value, int ttl, bool replace ) {
 
         logging_info("DHT: putting key=" <<  key
                         << " value=" << value
                         << " ttl=" << ttl
+                        << " replace=" << replace
         );
 
+
         // put into local data store (for refreshes)
-        localDHT->put(key,value,ttl);
+        if (replace) localDHT->remove(key);
+        localDHT->put(key, value, ttl);
 
         // calculate hash
         NodeID dest = NodeID::sha1(key.getBuffer(), key.getLength() / 8);
         DHTMessage dhtmsg(key,value);
+        dhtmsg.setReplace(replace);
         dhtmsg.setTTL(ttl);