examples/cpp/activityset-waitallfor/s4u-activityset-waitallfor.cpp

   1 /* Copyright (c) 2010-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 #include "simgrid/s4u.hpp"
   7 #include <cstdlib>
   8 #include <iostream>
   9 #include <string>
  10 namespace sg4 = simgrid::s4u;
  11
  12 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_activity_waitallfor, "Messages specific for this s4u example");
  13
  14 static void bob()
  15 {
  16   sg4::Mailbox* mbox    = sg4::Mailbox::by_name("mbox");
  17   sg4::MessageQueue* mqueue = sg4::MessageQueue::by_name("mqueue");
  18   const sg4::Disk* disk = sg4::Host::current()->get_disks().front();
  19   std::string* payload;
  20   std::string* message;
  21
  22   XBT_INFO("Create my asynchronous activities");
  23   auto exec = sg4::this_actor::exec_async(5e9);
  24   auto comm = mbox->get_async(&payload);
  25   auto io   = disk->read_async(3e8);
  26   auto mess = mqueue->get_async(&message);
  27
  28   sg4::ActivitySet pending_activities({exec, comm, io, mess});
  29
  30   XBT_INFO("Wait for asynchronous activities to complete");
  31   while (not pending_activities.empty()) {
  32     try {
  33       pending_activities.wait_all_for(1);
  34     } catch (simgrid::TimeoutException& e) {
  35       XBT_INFO("Not all activities are terminated yet.");
  36     }
  37     while (auto completed_one = pending_activities.test_any()) {
  38       if (boost::dynamic_pointer_cast<sg4::Comm>(completed_one))
  39         XBT_INFO("Completed a Comm");
  40       if (boost::dynamic_pointer_cast<sg4::Mess>(completed_one))
  41         XBT_INFO("Completed a Mess");
  42       if (boost::dynamic_pointer_cast<sg4::Exec>(completed_one))
  43         XBT_INFO("Completed an Exec");
  44       if (boost::dynamic_pointer_cast<sg4::Io>(completed_one))
  45         XBT_INFO("Completed an I/O");
  46     }
  47   }
  48   XBT_INFO("Last activity is complete");
  49   delete payload;
  50   delete message;
  51 }
  52
  53 static void alice()
  54 {
  55   auto* payload = new std::string("Message");
  56   XBT_INFO("Send '%s'", payload->c_str());
  57   sg4::Mailbox::by_name("mbox")->put(payload, 6e8);
  58 }
  59
  60 static void carl()
  61 {
  62   sg4::this_actor::sleep_for(1.99);
  63   auto* payload = new std::string("Control Message");
  64   XBT_INFO("Send '%s'", payload->c_str());
  65   sg4::MessageQueue::by_name("mqueue")->put(payload);
  66 }
  67
  68 int main(int argc, char* argv[])
  69 {
  70   sg4::Engine e(&argc, argv);
  71
  72   e.load_platform(argv[1]);
  73
  74   sg4::Actor::create("bob", e.host_by_name("bob"), bob);
  75   sg4::Actor::create("alice", e.host_by_name("alice"), alice);
  76   sg4::Actor::create("carl", e.host_by_name("carl"), carl);
  77
  78   e.run();
  79
  80   return 0;
  81 }