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

Timestamp:

Nov 2, 2009, 1:31:19 AM (14 years ago)

Author:

mies

Message:

Changed Data to Message conversion constructor in Message to explicit
Fixed some general bugs in Data: operator<<
Fixed bug in DHTMessage: allow unspecified key/values
Added local DHT message delivery
Adapted sources to work with gcc 4.4.1

File:

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

source/ariba/overlay/BaseOverlay.cpp

@@ -727 +727 @@
         sideport(&SideportListener::DEFAULT), started(false), counter(0) {
         dht = new DHT();
+        localDHT = new DHT();
 }
 
@@ -1913 +1914 @@
         if (msg->getType()==OverlayMsg::typeDHTData) {
                 const ServiceID& service = msg->getService();
-                logging_debug( "Received DHT data for service " << service.toString() );
+                logging_info( "Received DHT data for service " << service.toString() );
 
                 // delegate data message
@@ -1940 +1941 @@
 
         case OverlayMsg::typeDHTGet: {
+                logging_info("DHT-Get: key=" << dhtMsg->getKey() );
                 vector<Data> vect = dht->get(dhtMsg->getKey());
+                BOOST_FOREACH(const Data& d, vect)
+                        logging_info("DHT-Get: value=" << d);
                 OverlayMsg omsg(*msg);
                 omsg.swapRoles();
@@ -1946 +1950 @@
                 DHTMessage dhtmsg(dhtMsg->getKey(), vect);
                 omsg.encapsulate(&dhtmsg);
-                this->send(&omsg, omsg.getDestinationNode());
+                dhtSend(&omsg, omsg.getDestinationNode());
                 break;
         }
@@ -1969 +1973 @@
 /// put a value to the DHT with a ttl given in seconds
 void BaseOverlay::dhtPut( const Data& key, const Data& value, int ttl ) {
+
+        logging_info("DHT: putting key=" <<  key
+                        << " value=" << value
+                        << " ttl=" << ttl
+        );
+
+        // put into local data store (for refreshes)
+        localDHT->put(key,value,ttl);
+
         // calculate hash
         NodeID dest = NodeID::sha1(key.getBuffer(), key.getLength() / 8);
-        OverlayMsg msg(OverlayMsg::typeDHTPut);
         DHTMessage dhtmsg(key,value);
         dhtmsg.setTTL(ttl);
-        msg.setDestinationNode(dest);
+
+        OverlayMsg msg(OverlayMsg::typeDHTPut);
         msg.encapsulate( &dhtmsg );
-        send(&msg, dest);
+        dhtSend(&msg, dest);
 }
 
 /// removes a key value pair from the DHT
 void BaseOverlay::dhtRemove( const Data& key, const Data& value ) {
+        // remove from local data store
+        localDHT->remove(key,value);
+
         // calculate hash
         NodeID dest = NodeID::sha1(key.getBuffer(), key.getLength() / 8);
+        DHTMessage dhtmsg(key,value);
+
+        // send message
         OverlayMsg msg(OverlayMsg::typeDHTRemove);
-        DHTMessage dhtmsg(key,value);
-        msg.setDestinationNode(dest);
         msg.encapsulate( &dhtmsg );
-        send(&msg, dest);
+        dhtSend(&msg, dest);
 }
 
@@ -1994 +2011 @@
         // calculate hash
         NodeID dest = NodeID::sha1(key.getBuffer(), key.getLength() / 8);
+        DHTMessage dhtmsg(key);
+
+        // send message
         OverlayMsg msg(OverlayMsg::typeDHTRemove);
-        DHTMessage dhtmsg(key);
-        msg.setDestinationNode(dest);
         msg.encapsulate( &dhtmsg );
-        send(&msg, dest);
+        dhtSend(&msg, dest);
 }
 
 /// requests data stored using key
 void BaseOverlay::dhtGet( const Data& key, const ServiceID& service ) {
+        logging_info("DHT: trying to resolve key=" <<
+                        key << " for service=" << service.toString() );
+
         // calculate hash
         NodeID dest = NodeID::sha1(key.getBuffer(), key.getLength() / 8);
+        DHTMessage dhtmsg(key);
+
+        // send message
         OverlayMsg msg(OverlayMsg::typeDHTGet);
-        DHTMessage dhtmsg(key);
-        msg.setDestinationNode(dest);
         msg.setService(service);
         msg.encapsulate( &dhtmsg );
-        send(&msg, dest);
-}
+        dhtSend(&msg, dest);
+}
+
+void BaseOverlay::dhtSend( OverlayMsg* msg, const NodeID& dest ) {
+        logging_info("DHT: sending message with key=" << dest.toString() );
+        msg->setSourceNode(this->nodeId);
+        msg->setDestinationNode(dest);
+
+        // local storage? yes-> put into DHT directly
+        if (overlayInterface->isClosestNodeTo(msg->getDestinationNode())) {
+                Data d = data_serialize(msg);
+                Message* m2 = new Message(d);
+                OverlayMsg* m3 = m2->decapsulate<OverlayMsg>();
+                handleDHTMessage(m3);
+                delete m2;
+                return;
+        }
+
+        // send message "normally"
+        send(msg, dest);
+}
+
 
 }} // namespace ariba, overlay