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 }