examples/cpp/mess-wait/s4u-mess-wait.cpp

   1 /* Copyright (c) 2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 /* This example shows how to use simgrid::s4u::this_actor::wait() to wait for a given communication.
   7  *
   8  * As for the other asynchronous examples, the sender initiate all the messages it wants to send and
   9  * pack the resulting simgrid::s4u::CommPtr objects in a vector. All messages thus occurs concurrently.
  10  *
  11  * The sender then loops until there is no ongoing communication.
  12  */
  13
  14 #include "simgrid/s4u.hpp"
  15 #include <cstdlib>
  16 #include <iostream>
  17 #include <string>
  18 namespace sg4 = simgrid::s4u;
  19
  20 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_mess_wait, "Messages specific for this s4u example");
  21
  22 static void sender(int messages_count)
  23 {
  24   sg4::MessageQueue* mqueue = sg4::MessageQueue::by_name("control");
  25
  26   sg4::this_actor::sleep_for(0.5);
  27
  28   for (int i = 0; i < messages_count; i++) {
  29     std::string msg_content = "Message " + std::to_string(i);
  30     // Copy the data we send: the 'msg_content' variable is not a stable storage location.
  31     // It will be destroyed when this actor leaves the loop, ie before the receiver gets the data
  32     auto* payload = new std::string(msg_content);
  33
  34     /* Create a control message and put it in the message queue */
  35     sg4::MessPtr mess = mqueue->put_async(payload);
  36     XBT_INFO("Send '%s' to '%s'", msg_content.c_str(), mqueue->get_cname());
  37     mess->wait();
  38   }
  39
  40   /* Send message to let the receiver know that it should stop */
  41   XBT_INFO("Send 'finalize' to 'receiver'");
  42   mqueue->put(new std::string("finalize"), 0);
  43 }
  44
  45 /* Receiver actor expects 1 argument: its ID */
  46 static void receiver()
  47 {
  48   sg4::MessageQueue* mqueue = sg4::MessageQueue::by_name("control");
  49
  50   sg4::this_actor::sleep_for(1);
  51
  52   XBT_INFO("Wait for my first message");
  53   for (bool cont = true; cont;) {
  54     std::string* received;
  55     sg4::MessPtr mess = mqueue->get_async<std::string>(&received);
  56
  57     sg4::this_actor::sleep_for(0.1);
  58     mess->wait();
  59
  60     XBT_INFO("I got a '%s'.", received->c_str());
  61     if (*received == "finalize")
  62       cont = false; // If it's a finalize message, we're done.
  63     delete received;
  64   }
  65 }
  66
  67 int main(int argc, char* argv[])
  68 {
  69   sg4::Engine e(&argc, argv);
  70
  71   e.load_platform(argv[1]);
  72
  73   sg4::Actor::create("sender", e.host_by_name("Tremblay"), sender, 3);
  74   sg4::Actor::create("receiver", e.host_by_name("Fafard"), receiver);
  75
  76   e.run();
  77
  78   return 0;
  79 }