Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
simple test on waitany
[simgrid.git] / teshsuite / s4u / comm-waitany / comm-waitany.cpp
1 #include <iostream>
2 #include <simgrid/s4u.hpp>
3 #include <stdlib.h>
4 #include <vector>
5
6 #define NUM_COMMS 10
7
8 XBT_LOG_NEW_DEFAULT_CATEGORY(mwe, "Minimum Working Example");
9
10 static void receiver()
11 {
12   simgrid::s4u::MailboxPtr mymailbox    = simgrid::s4u::Mailbox::byName("receiver_mailbox");
13   simgrid::s4u::MailboxPtr theirmailbox = simgrid::s4u::Mailbox::byName("sender_mailbox");
14
15   std::vector<simgrid::s4u::CommPtr> pending_comms;
16
17   XBT_INFO("Placing %d asynchronous recv requests", NUM_COMMS);
18   void* data = calloc(1, sizeof(int));
19   for (int i = 0; i < NUM_COMMS; i++) {
20     simgrid::s4u::CommPtr comm = simgrid::s4u::Comm::recv_async(mymailbox, &data);
21     pending_comms.push_back(comm);
22   }
23
24   XBT_INFO("Sleeping for 100 seconds...");
25   simgrid::s4u::this_actor::sleep_for(100.0);
26   XBT_INFO("Calling wait_any() for %ld pending comms", pending_comms.size());
27   std::vector<simgrid::s4u::CommPtr>::iterator ret_it =
28       simgrid::s4u::Comm::wait_any(pending_comms.begin(), pending_comms.end());
29   XBT_INFO("Counting the number of completed comms...");
30
31   int count = 0;
32   for (; ret_it != pending_comms.end(); count++, ret_it++)
33     ;
34
35   xbt_assert(count == 1, "wait_any() replied that %d comms have completed, which is broken!", count);
36 }
37
38 static void sender()
39 {
40   simgrid::s4u::MailboxPtr mymailbox    = simgrid::s4u::Mailbox::byName("sender_mailbox");
41   simgrid::s4u::MailboxPtr theirmailbox = simgrid::s4u::Mailbox::byName("receiver_mailbox");
42
43   void* data = calloc(1, sizeof(int));
44
45   for (int i = 0; i < NUM_COMMS; i++) {
46     XBT_INFO("Sending a message to the receiver");
47     simgrid::s4u::this_actor::send(theirmailbox, &data, 4);
48     XBT_INFO("Sleeping for 1000 seconds");
49     simgrid::s4u::this_actor::sleep_for(1000.0);
50   }
51 }
52
53 int main(int argc, char** argv)
54 {
55
56   simgrid::s4u::Engine* engine = new simgrid::s4u::Engine(&argc, argv);
57
58   if (argc < 2) {
59     std::cerr << "Usage: " << argv[0] << " <xml platform file>" << std::endl;
60     exit(1);
61   }
62
63   engine->loadPlatform(argv[1]);
64   simgrid::s4u::Host* host = simgrid::s4u::Host::by_name("Tremblay");
65
66   simgrid::s4u::Actor::createActor("Receiver", host, receiver);
67   simgrid::s4u::Actor::createActor("Sender", host, sender);
68
69   simgrid::s4u::Engine::instance()->run();
70
71   return 0;
72 }