Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
rework this example too
[simgrid.git] / examples / s4u / async-wait / s4u-async-wait.cpp
index 03da3fd..1270435 100644 (file)
@@ -20,60 +20,67 @@ XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_async_wait, "Messages specific for this s4u exa
 
 static void sender(int argc, char** argv)
 {
-  xbt_assert(argc == 4, "Expecting 3 parameters from the XML deployment file but got %d", argc);
+  xbt_assert(argc == 3, "Expecting 2 parameters from the XML deployment file but got %d", argc);
   long messages_count  = std::stol(argv[1]); /* - number of tasks */
-  double msg_size      = std::stol(argv[2]); /* - communication cost in bytes */
-  long receivers_count = std::stod(argv[3]); /* - number of receivers */
+  double msg_size         = std::stod(argv[2]); /* - communication cost in bytes */
+  double sleep_start_time = 5.0;
+  double sleep_test_time  = 0;
 
-  /* Vector in which we store all ongoing communications */
-  std::vector<simgrid::s4u::CommPtr> pending_comms;
+  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name("receiver");
 
-  /* Make a vector of the mailboxes to use */
-  std::vector<simgrid::s4u::Mailbox*> mboxes;
-  for (int i = 0; i < receivers_count; i++)
-    mboxes.push_back(simgrid::s4u::Mailbox::by_name(std::string("receiver-") + std::to_string(i)));
+  XBT_INFO("sleep_start_time : %f , sleep_test_time : %f", sleep_start_time, sleep_test_time);
+  simgrid::s4u::this_actor::sleep_for(sleep_start_time);
 
-  /* Start dispatching all messages to receivers, in a round robin fashion */
   for (int i = 0; i < messages_count; i++) {
     std::string msg_content = std::string("Message ") + std::to_string(i);
     // Copy the data we send: the 'msg_content' variable is not a stable storage location.
     // It will be destroyed when this actor leaves the loop, ie before the receiver gets the data
     std::string* payload = new std::string(msg_content);
 
-    XBT_INFO("Send '%s' to '%s'", msg_content.c_str(), mboxes[i % receivers_count]->get_cname());
+    XBT_INFO("Send '%s' to '%s'", msg_content.c_str(), mbox->get_cname());
 
-    /* Create a communication representing the ongoing communication, and store it in pending_comms */
-    simgrid::s4u::CommPtr comm = mboxes[i % receivers_count]->put_async(payload, msg_size);
-    pending_comms.push_back(comm);
-  }
-
-  /* Start sending messages to let the workers know that they should stop */
-  for (int i = 0; i < receivers_count; i++) {
-    XBT_INFO("Send 'finalize' to 'receiver-%d'", i);
-    simgrid::s4u::CommPtr comm = mboxes[i]->put_async(new std::string("finalize"), 0);
-    pending_comms.push_back(comm);
-  }
-  XBT_INFO("Done dispatching all messages");
+    /* Create a communication representing the ongoing communication and then */
+    simgrid::s4u::CommPtr comm = mbox->put_async(payload, msg_size);
 
-  /* Now that all message exchanges were initiated, wait for their completion, in order of creation. */
-  while (not pending_comms.empty()) {
-    simgrid::s4u::CommPtr comm = pending_comms.back();
-    comm->wait();
-    pending_comms.pop_back(); // remove it from the list
+    if (sleep_test_time > 0) {   /* - "test_time" is set to 0, wait */
+      while (not comm->test()) { /* - Call test() every "sleep_test_time" otherwise */
+        simgrid::s4u::this_actor::sleep_for(sleep_test_time);
+      }
+    } else {
+      comm->wait();
+    }
   }
 
-  XBT_INFO("Goodbye now!");
+  /* Send message to let the receiver know that it should stop */
+  XBT_INFO("Send 'finalize' to 'receiver'");
+  mbox->put(new std::string("finalize"), 0);
 }
 
 /* Receiver actor expects 1 argument: its ID */
 static void receiver(int argc, char** argv)
 {
-  xbt_assert(argc == 2, "Expecting one parameter from the XML deployment file but got %d", argc);
-  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name(std::string("receiver-") + argv[1]);
+  double sleep_start_time = 1.0;
+  double sleep_test_time  = 0.1;
+
+  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name("receiver");
+
+  XBT_INFO("sleep_start_time : %f , sleep_test_time : %f", sleep_start_time, sleep_test_time);
+  simgrid::s4u::this_actor::sleep_for(sleep_start_time);
 
   XBT_INFO("Wait for my first message");
   for (bool cont = true; cont;) {
-    const std::string* received = static_cast<std::string*>(mbox->get());
+    void* payload;
+    simgrid::s4u::CommPtr comm = mbox->get_async(&payload);
+
+    if (sleep_test_time > 0) {   /* - "test_time" is set to 0, wait */
+      while (not comm->test()) { /* - Call test() every "sleep_test_time" otherwise */
+        simgrid::s4u::this_actor::sleep_for(sleep_test_time);
+      }
+    } else {
+      comm->wait();
+    }
+
+    std::string* received = static_cast<std::string*>(payload);
     XBT_INFO("I got a '%s'.", received->c_str());
     if (*received == "finalize")
       cont = false; // If it's a finalize message, we're done.