examples/s4u/app-token-ring/s4u_app-token-ring.cpp

   1 /* Copyright (c) 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 #include "xbt/str.h"
   7 #include <simgrid/s4u.hpp>
   8 #include <string>
   9 #include <unordered_map>
  10 #include <vector>
  11
  12 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_app_token_ring, "Messages specific for this s4u example");
  13 // FIXME: The following duplicates the content of s4u::Host
  14 extern std::unordered_map<std::string, simgrid::s4u::Host*> host_list;
  15
  16 class RelayRunner {
  17 public:
  18   size_t task_comm_size = 1000000; /* The token is 1MB long*/
  19   simgrid::s4u::MailboxPtr my_mailbox;
  20   simgrid::s4u::MailboxPtr neighbor_mailbox;
  21   unsigned int rank      = 0;
  22   explicit RelayRunner() = default;
  23
  24   void operator()()
  25   {
  26     rank = xbt_str_parse_int(simgrid::s4u::this_actor::name().c_str(),
  27                              "Any process of this example must have a numerical name, not %s");
  28     my_mailbox = simgrid::s4u::Mailbox::byName((std::to_string(rank)).c_str());
  29     if (rank + 1 == host_list.size())
  30       /* The last process, which sends the token back to rank 0 */
  31       neighbor_mailbox = simgrid::s4u::Mailbox::byName("0");
  32     else
  33       /* The others processes send to their right neighbor (rank+1) */
  34       neighbor_mailbox = simgrid::s4u::Mailbox::byName((std::to_string(rank + 1)).c_str());
  35
  36     if (rank == 0) {
  37       /* The root process (rank 0) first sends the token then waits to receive it back */
  38       XBT_INFO("Host \"%u\" send 'Token' to Host \"%s\"", rank, neighbor_mailbox->name());
  39       simgrid::s4u::this_actor::send(neighbor_mailbox, xbt_strdup("Token"), task_comm_size);
  40       char* res = static_cast<char*>(simgrid::s4u::this_actor::recv(my_mailbox));
  41       XBT_INFO("Host \"%u\" received \"%s\"", rank, res);
  42     } else {
  43       char* res = static_cast<char*>(simgrid::s4u::this_actor::recv(my_mailbox));
  44       XBT_INFO("Host \"%u\" received \"%s\"", rank, res);
  45       XBT_INFO("Host \"%u\" send 'Token' to Host \"%s\"", rank, neighbor_mailbox->name());
  46       simgrid::s4u::this_actor::send(neighbor_mailbox, res, task_comm_size);
  47     }
  48   }
  49 };
  50
  51 int main(int argc, char** argv)
  52 {
  53   simgrid::s4u::Engine* e = new simgrid::s4u::Engine(&argc, argv);
  54   xbt_assert(argc > 1, "Usage: %s platform.xml\n", argv[0]);
  55   e->loadPlatform(argv[1]);
  56
  57   XBT_INFO("Number of hosts '%lu'", host_list.size());
  58
  59   int id = 0;
  60   for (auto h : host_list) {
  61     simgrid::s4u::Host* host = h.second;
  62     /* - Give a unique rank to each host and create a @ref relay_runner process on each */
  63     simgrid::s4u::Actor::createActor((std::to_string(id)).c_str(), host, RelayRunner());
  64     id++;
  65   }
  66   e->run();
  67   XBT_INFO("Simulation time %g", e->getClock());
  68   return 0;
  69 }