Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[sonar] Declare functions "const" in examples/ and teshsuite/.
[simgrid.git] / examples / s4u / dht-chord / s4u-dht-chord-node.cpp
index 8eda1ef..fbfdec6 100644 (file)
@@ -52,6 +52,8 @@ Node::Node(std::vector<std::string> args)
 
   // initialize my node
   id_                = std::stoi(args[1]);
+  XBT_DEBUG("Initialize node with id: %d", id_);
+  random.set_seed(id_);
   mailbox_           = simgrid::s4u::Mailbox::by_name(std::to_string(id_));
   next_finger_to_fix = 0;
   fingers_.resize(nb_bits, id_);
@@ -134,7 +136,7 @@ void Node::notifyAndQuit()
 void Node::randomLookup()
 {
   int res          = id_;
-  int random_index = simgrid::xbt::random::uniform_int(0, nb_bits - 1);
+  int random_index = random.uniform_int(0, nb_bits - 1);
   int random_id    = fingers_[random_index];
   XBT_DEBUG("Making a lookup request for id %d", random_id);
   if (random_id != id_)
@@ -362,7 +364,7 @@ void Node::notify(int predecessor_candidate_id)
 }
 
 /* Notifies a remote node that its predecessor may have changed. */
-void Node::remoteNotify(int notify_id, int predecessor_candidate_id)
+void Node::remoteNotify(int notify_id, int predecessor_candidate_id) const
 {
   ChordMessage* message = new ChordMessage(NOTIFY);
   message->request_id   = predecessor_candidate_id;
@@ -474,3 +476,77 @@ void Node::handleMessage(ChordMessage* message)
     delete message;
   }
 }
+
+void Node::operator()()
+{
+  simgrid::s4u::this_actor::sleep_for(start_time_);
+  if (known_id_ == -1) {
+    setPredecessor(-1); // -1 means that I have no predecessor
+    printFingerTable();
+    joined = true;
+  } else {
+    join(known_id_);
+  }
+
+  if (not joined)
+    return;
+  void* data                         = nullptr;
+  double now                         = simgrid::s4u::Engine::get_clock();
+  double next_stabilize_date         = start_time_ + PERIODIC_STABILIZE_DELAY;
+  double next_fix_fingers_date       = start_time_ + PERIODIC_FIX_FINGERS_DELAY;
+  double next_check_predecessor_date = start_time_ + PERIODIC_CHECK_PREDECESSOR_DELAY;
+  double next_lookup_date            = start_time_ + PERIODIC_LOOKUP_DELAY;
+  simgrid::s4u::CommPtr comm_receive = nullptr;
+  while (now < std::min(start_time_ + deadline_, MAX_SIMULATION_TIME)) {
+    if (comm_receive == nullptr)
+      comm_receive = mailbox_->get_async(&data);
+    bool comm_completed = true;
+    try {
+      if (not comm_receive->test())
+        comm_completed = false;
+    } catch (const simgrid::TimeoutException&) {
+      XBT_DEBUG("Caught a timeout, go ahead.");
+    }
+
+    if (comm_completed) {
+      if (data != nullptr) {
+        ChordMessage* message = static_cast<ChordMessage*>(data);
+        handleMessage(message);
+        data = nullptr;
+      }
+      comm_receive = nullptr;
+    } else {
+      // no task was received: make some periodic calls
+      if (now >= next_stabilize_date) {
+        stabilize();
+        next_stabilize_date = simgrid::s4u::Engine::get_clock() + PERIODIC_STABILIZE_DELAY;
+      } else if (now >= next_fix_fingers_date) {
+        fixFingers();
+        next_fix_fingers_date = simgrid::s4u::Engine::get_clock() + PERIODIC_FIX_FINGERS_DELAY;
+      } else if (now >= next_check_predecessor_date) {
+        checkPredecessor();
+        next_check_predecessor_date = simgrid::s4u::Engine::get_clock() + PERIODIC_CHECK_PREDECESSOR_DELAY;
+      } else if (now >= next_lookup_date) {
+        randomLookup();
+        next_lookup_date = simgrid::s4u::Engine::get_clock() + PERIODIC_LOOKUP_DELAY;
+      } else {
+        // nothing to do: sleep for a while
+        simgrid::s4u::this_actor::sleep_for(SLEEP_DELAY);
+      }
+    }
+
+    now = simgrid::s4u::Engine::get_clock();
+  }
+  if (comm_receive != nullptr) {
+    try {
+      if (comm_receive->test())
+        delete static_cast<ChordMessage*>(data);
+      else
+        comm_receive->cancel();
+    } catch (const simgrid::TimeoutException&) {
+      XBT_DEBUG("Caught a timeout for last message, nevermind.");
+    }
+  }
+  // leave the ring
+  leave();
+}