Logo AND Algorithmique Numérique Distribuée

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