Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
massive cleanups in the s4u-async-waitall example
[simgrid.git] / examples / s4u / async-waitall / s4u_async-waitall.cpp
1 /* Copyright (c) 2010-2017. 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 asynchronous communications.
7  *
8  * The sender initiate all the messages it wants to send, and then block for their completion.
9  * All messages thus occurs concurrently.
10  *
11  * On the receiver side, the reception is synchronous.
12  *
13  * TODO: this example is supposed to test the waitall function, but this is not ported to s4u yet.
14  *
15  */
16
17 #include "simgrid/s4u.hpp"
18 #include "xbt/str.h"
19 #include <cstdlib>
20 #include <iostream>
21
22 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_async_waitall, "Messages specific for this msg example");
23
24 class sender {
25   long messages_count;
26   long receivers_count;
27   double msg_size; /* in bytes */
28
29 public:
30   explicit sender(std::vector<std::string> args)
31 {
32   xbt_assert(args.size()== 4, "This function expects 5 parameters from the XML deployment file");
33   messages_count  = std::stol(args[1]);
34   msg_size        = std::stod(args[2]);
35   receivers_count = std::stol(args[3]);
36
37 }
38 void operator()()
39 {
40   simgrid::s4u::CommPtr* comms = new simgrid::s4u::CommPtr[messages_count + receivers_count];
41
42   /* Start dispatching all messages to receivers, in a round robin fashion */
43   for (int i = 0; i < messages_count; i++) {
44
45     std::string mboxName          = std::string("receiver-") + std::to_string(i % receivers_count);
46     simgrid::s4u::MailboxPtr mbox = simgrid::s4u::Mailbox::byName(mboxName);
47     std::string msgName           = std::string("Message ") + std::to_string(i);
48     char* payload = xbt_strdup(msgName.c_str()); // copy the data we send: 'msgName' is not a stable storage location
49
50     XBT_INFO("Send '%s' to '%s'", msgName.c_str(), mboxName.c_str());
51     comms[i] = mbox->put_async((void*)payload, msg_size);
52   }
53   /* Start sending messages to let the workers know that they should stop */
54   for (int i = 0; i < receivers_count; i++) {
55     std::string mbox_name         = std::string("receiver-") + std::to_string(i % receivers_count);
56     simgrid::s4u::MailboxPtr mbox = simgrid::s4u::Mailbox::byName(mbox_name);
57     char* payload                 = xbt_strdup("finalize"); // Make a copy of the data we will send
58
59     comms[i + messages_count] = mbox->put_async((void*)payload, 0);
60     XBT_INFO("Send 'finalize' to 'receiver-%ld'", i % receivers_count);
61   }
62   XBT_INFO("Done dispatching all messages");
63
64   /* Now that all message exchanges were initiated, this loop waits for the termination of them all */
65   for (int i = 0; i < messages_count + receivers_count; i++)
66     comms[i]->wait();
67
68   delete [] comms;
69   XBT_INFO("Goodbye now!");
70 }
71 };
72
73 class receiver {
74   simgrid::s4u::MailboxPtr mbox;
75
76 public:
77   explicit receiver(std::vector<std::string> args)
78 {
79   xbt_assert(args.size() == 2, "This function expects 2 parameters from the XML deployment file but got %zu",
80              args.size());
81   int id = xbt_str_parse_int(args[1].c_str(), "Any process of this example must have a numerical name, not %s");
82   std::string mbox_name = std::string("receiver-") + std::to_string(id);
83   mbox                  = simgrid::s4u::Mailbox::byName(mbox_name);
84 }
85 void operator()()
86 {
87   XBT_INFO("Wait for my first message");
88   while (1) {
89     char* received = static_cast<char*>(mbox->get());
90     XBT_INFO("I got a '%s'.", received);
91     if (std::strcmp(received, "finalize") == 0) { /* If it's a finalize message, we're done */
92       xbt_free(received);
93       break;
94     }
95     /*  - Otherwise receiving the message was all we were supposed to do */
96     xbt_free(received);
97   }
98 }
99 };
100
101 int main(int argc, char *argv[])
102 {
103   simgrid::s4u::Engine* e = new simgrid::s4u::Engine(&argc, argv);
104
105   xbt_assert(argc > 2, "Usage: %s platform_file deployment_file\n", argv[0]);
106
107   e->registerFunction<sender>("sender");
108   e->registerFunction<receiver>("receiver");
109
110   e->loadPlatform(argv[1]);
111   e->loadDeployment(argv[2]); 
112   e->run();
113
114   return 0;
115 }