Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
please sonar and codefactor
[simgrid.git] / examples / s4u / async-wait / s4u-async-wait.cpp
index d712f92..95b883d 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (c) 2010-2017. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2010-2020. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 #include <iostream>
 #include <string>
 
-XBT_LOG_NEW_DEFAULT_CATEGORY(msg_async_wait, "Messages specific for this s4u example");
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_async_wait, "Messages specific for this s4u example");
 
-class Sender {
-  long messages_count;  /* - number of tasks */
-  long receivers_count; /* - number of receivers */
-  double msg_size;      /* - communication cost in bytes */
-
-public:
-  explicit Sender(std::vector<std::string> args)
-  {
-    xbt_assert(args.size() == 4, "Expecting 3 parameters from the XML deployment file but got %zu", args.size());
-    messages_count  = std::stol(args[1]);
-    msg_size        = std::stod(args[2]);
-    receivers_count = std::stol(args[3]);
-  }
-  void operator()()
-  {
-    std::vector<simgrid::s4u::CommPtr> pending_comms;
-
-    /* Start dispatching all messages to receivers, in a round robin fashion */
-    for (int i = 0; i < messages_count; i++) {
-
-      std::string mboxName          = std::string("receiver-") + std::to_string(i % receivers_count);
-      simgrid::s4u::MailboxPtr mbox = simgrid::s4u::Mailbox::byName(mboxName);
-      std::string msgName = std::string("Message ") + std::to_string(i);
-      std::string* payload          = new std::string(msgName); // copy the data we send:
-                                                                // 'msgName' is not a stable storage location
-      XBT_INFO("Send '%s' to '%s'", msgName.c_str(), mboxName.c_str());
-      /* Create a communication representing the ongoing communication */
-      simgrid::s4u::CommPtr comm = mbox->put_async(payload, msg_size);
-      /* Add this comm to the vector of all known comms */
-      pending_comms.push_back(comm);
+static void sender(int argc, char** argv)
+{
+  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::stod(argv[2]); /* - communication cost in bytes */
+  double sleep_start_time = 5.0;
+  double sleep_test_time  = 0;
+
+  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);
+
+  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);
+
+    /* Create a communication representing the ongoing communication and then */
+    simgrid::s4u::CommPtr comm = mbox->put_async(payload, msg_size);
+    XBT_INFO("Send '%s' to '%s'", msg_content.c_str(), mbox->get_cname());
+
+    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();
     }
+  }
 
-    /* Start sending messages to let the workers know that they should stop */
-    for (int i = 0; i < receivers_count; i++) {
-      std::string mboxName          = std::string("receiver-") + std::to_string(i % receivers_count);
-      simgrid::s4u::MailboxPtr mbox = simgrid::s4u::Mailbox::byName(mboxName);
-      std::string* payload          = new std::string("finalize"); // Make a copy of the data we will send
+  /* Send message to let the receiver know that it should stop */
+  XBT_INFO("Send 'finalize' to 'receiver'");
+  mbox->put(new std::string("finalize"), 0);
+}
 
-      simgrid::s4u::CommPtr comm = mbox->put_async(payload, 0);
-      pending_comms.push_back(comm);
-      XBT_INFO("Send 'finalize' to 'receiver-%ld'", i % receivers_count);
-    }
-    XBT_INFO("Done dispatching all messages");
+/* Receiver actor expects 1 argument: its ID */
+static void receiver(XBT_ATTRIB_UNUSED int argc, XBT_ATTRIB_UNUSED char** argv)
+{
+  double sleep_start_time = 1.0;
+  double sleep_test_time  = 0.1;
 
-    /* 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();             // we could provide a timeout as a parameter
-      pending_comms.pop_back(); // remove it from the list
-    }
+  simgrid::s4u::Mailbox* mbox = simgrid::s4u::Mailbox::by_name("receiver");
 
-    XBT_INFO("Goodbye now!");
-  }
-};
+  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);
 
-/* Receiver actor expects 1 argument: its ID */
-class Receiver {
-  simgrid::s4u::MailboxPtr mbox;
-
-public:
-  explicit Receiver(std::vector<std::string> args)
-  {
-    xbt_assert(args.size() == 2, "Expecting one parameter from the XML deployment file but got %zu", args.size());
-    std::string mboxName = std::string("receiver-") + args[1];
-    mbox                 = simgrid::s4u::Mailbox::byName(mboxName);
-  }
+  XBT_INFO("Wait for my first message");
+  for (bool cont = true; cont;) {
+    void* payload;
+    simgrid::s4u::CommPtr comm = mbox->get_async(&payload);
 
-  void operator()()
-  {
-    XBT_INFO("Wait for my first message");
-    for (bool done = false; not done;) {
-      std::string* received = static_cast<std::string*>(mbox->get());
-      XBT_INFO("I got a '%s'.", received->c_str());
-      done = (*received == "finalize"); // If it's a finalize message, we're done
-      // Receiving the message was all we were supposed to do
-      delete received;
+    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();
     }
+
+    const 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.
+    delete received;
   }
-};
+}
 
 int main(int argc, char *argv[])
 {
   xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n", argv[0]);
 
   simgrid::s4u::Engine e(&argc, argv);
-  e.registerFunction<Sender>("sender");
-  e.registerFunction<Receiver>("receiver");
+  e.register_function("sender", &sender);
+  e.register_function("receiver", &receiver);
 
-  e.loadPlatform(argv[1]);
-  e.loadDeployment(argv[2]);
+  e.load_platform(argv[1]);
+  e.load_deployment(argv[2]);
   e.run();
 
   return 0;