examples/cpp/plugin-link-load/s4u-plugin-link-load.cpp

   1 /* Copyright (c) 2007-2022. 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 "simgrid/plugins/load.h"
   7 #include "simgrid/s4u.hpp"
   8
   9 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this s4u example");
  10
  11 static void sender(const std::string& mailbox, uint64_t msg_size)
  12 {
  13   auto mbox = simgrid::s4u::Mailbox::by_name(mailbox);
  14   static int payload = 42;
  15   mbox->put(&payload, msg_size);
  16 }
  17
  18 static void receiver(const std::string& mailbox)
  19 {
  20   auto mbox = simgrid::s4u::Mailbox::by_name(mailbox);
  21   mbox->get<int>();
  22 }
  23
  24 static void run_transfer(simgrid::s4u::Host* src_host, simgrid::s4u::Host* dst_host, const std::string& mailbox,
  25                          unsigned long msg_size)
  26 {
  27   XBT_INFO("Launching the transfer of %lu bytes", msg_size);
  28   simgrid::s4u::Actor::create("sender", src_host, sender, mailbox, msg_size);
  29   simgrid::s4u::Actor::create("receiver", dst_host, receiver, mailbox);
  30 }
  31
  32 static void execute_load_test()
  33 {
  34   auto host0 = simgrid::s4u::Host::by_name("node-0.simgrid.org");
  35   auto host1 = simgrid::s4u::Host::by_name("node-1.simgrid.org");
  36
  37   simgrid::s4u::this_actor::sleep_for(1);
  38   run_transfer(host0, host1, "1", 1000 * 1000 * 1000);
  39
  40   simgrid::s4u::this_actor::sleep_for(10);
  41   run_transfer(host0, host1, "2", 1000 * 1000 * 1000);
  42   simgrid::s4u::this_actor::sleep_for(3);
  43   run_transfer(host0, host1, "3", 1000 * 1000 * 1000);
  44 }
  45
  46 static void show_link_load(const std::string& link_name, const simgrid::s4u::Link* link)
  47 {
  48   XBT_INFO("%s link load (cum, avg, min, max): (%g, %g, %g, %g)", link_name.c_str(), sg_link_get_cum_load(link),
  49            sg_link_get_avg_load(link), sg_link_get_min_instantaneous_load(link),
  50            sg_link_get_max_instantaneous_load(link));
  51 }
  52
  53 static void monitor()
  54 {
  55   auto link_backbone = simgrid::s4u::Link::by_name("cluster0_backbone");
  56   auto link_host0    = simgrid::s4u::Link::by_name("cluster0_link_0_UP");
  57   auto link_host1    = simgrid::s4u::Link::by_name("cluster0_link_1_DOWN");
  58
  59   XBT_INFO("Tracking desired links");
  60   sg_link_load_track(link_backbone);
  61   sg_link_load_track(link_host0);
  62   sg_link_load_track(link_host1);
  63
  64   show_link_load("Backbone", link_backbone);
  65   while (simgrid::s4u::Engine::get_clock() < 5) {
  66     simgrid::s4u::this_actor::sleep_for(1);
  67     show_link_load("Backbone", link_backbone);
  68   }
  69
  70   XBT_INFO("Untracking the backbone link");
  71   sg_link_load_untrack(link_backbone);
  72
  73   show_link_load("Host0_UP", link_host0);
  74   show_link_load("Host1_UP", link_host1);
  75
  76   XBT_INFO("Now resetting and probing host links each second.");
  77
  78   while (simgrid::s4u::Engine::get_clock() < 29) {
  79     sg_link_load_reset(link_host0);
  80     sg_link_load_reset(link_host1);
  81
  82     simgrid::s4u::this_actor::sleep_for(1);
  83
  84     show_link_load("Host0_UP", link_host0);
  85     show_link_load("Host1_UP", link_host1);
  86   }
  87 }
  88
  89 int main(int argc, char* argv[])
  90 {
  91   simgrid::s4u::Engine e(&argc, argv);
  92   sg_link_load_plugin_init();
  93
  94   xbt_assert(argc == 2, "Usage: %s platform_file\n\tExample: %s ../platforms/energy_platform.xml\n", argv[0], argv[0]);
  95   e.load_platform(argv[1]);
  96
  97   simgrid::s4u::Actor::create("load_test", e.host_by_name("node-42.simgrid.org"), execute_load_test);
  98   simgrid::s4u::Actor::create("monitor", e.host_by_name("node-51.simgrid.org"), monitor);
  99
 100   e.run();
 101
 102   XBT_INFO("Total simulation time: %.2f", simgrid::s4u::Engine::get_clock());
 103
 104   return 0;
 105 }