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 1
   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;
  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   for (int i = 0; i < NUM_COMMS; i++) {
  25     XBT_INFO("Sleeping for 3 seconds (for the %dth time)...", i + 1);
  26     simgrid::s4u::this_actor::sleep_for(3.0);
  27     XBT_INFO("Calling wait_any() for %zu pending comms", pending_comms.size());
  28     std::vector<simgrid::s4u::CommPtr>::iterator ret_it =
  29         simgrid::s4u::Comm::wait_any(pending_comms.begin(), pending_comms.end());
  30     XBT_INFO("Counting the number of completed comms...");
  31
  32     int count = 0;
  33     for (; ret_it != pending_comms.end(); count++, ret_it++)
  34       ;
  35
  36     XBT_INFO("wait_any() replied that %d comms have completed", count);
  37     // xbt_assert(count == 1, "wait_any() replied that %d comms have completed, which is broken!", count);
  38   }
  39 }
  40
  41 static void sender()
  42 {
  43   simgrid::s4u::MailboxPtr mymailbox    = simgrid::s4u::Mailbox::byName("sender_mailbox");
  44   simgrid::s4u::MailboxPtr theirmailbox = simgrid::s4u::Mailbox::byName("receiver_mailbox");
  45
  46   void* data = (void*)"data";
  47
  48   for (int i = 0; i < NUM_COMMS; i++) {
  49     XBT_INFO("Sending a message to the receiver");
  50     simgrid::s4u::this_actor::send(theirmailbox, &data, 4);
  51     XBT_INFO("Sleeping for 1000 seconds");
  52     simgrid::s4u::this_actor::sleep_for(1000.0);
  53   }
  54 }
  55
  56 int main(int argc, char** argv)
  57 {
  58
  59   simgrid::s4u::Engine* engine = new simgrid::s4u::Engine(&argc, argv);
  60
  61   if (argc < 2) {
  62     std::cerr << "Usage: " << argv[0] << " <xml platform file>" << std::endl;
  63     exit(1);
  64   }
  65
  66   engine->loadPlatform(argv[1]);
  67   simgrid::s4u::Host* host = simgrid::s4u::Host::by_name("Tremblay");
  68
  69   simgrid::s4u::Actor::createActor("Receiver", host, receiver);
  70   simgrid::s4u::Actor::createActor("Sender", host, sender);
  71
  72   simgrid::s4u::Engine::instance()->run();
  73
  74   return 0;
  75 }