1 /* Copyright (c) 2012-2023. The SimGrid Team. All rights reserved. */
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. */
6 #include "s4u-tracker.hpp"
9 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_bt_tracker, "Messages specific for the tracker");
10 namespace sg4 = simgrid::s4u;
12 Tracker::Tracker(std::vector<std::string> args)
15 xbt_assert(args.size() == 2, "Wrong number of arguments for the tracker.");
16 // Retrieving end time
18 deadline = std::stod(args[1]);
19 } catch (const std::invalid_argument&) {
20 throw std::invalid_argument("Invalid deadline:" + args[1]);
22 xbt_assert(deadline > 0, "Wrong deadline supplied");
24 mailbox = sg4::Mailbox::by_name(TRACKER_MAILBOX);
26 XBT_INFO("Tracker launched.");
29 void Tracker::operator()()
31 sg4::CommPtr comm = nullptr;
32 TrackerQuery* query = nullptr;
33 while (sg4::Engine::get_clock() < deadline) {
35 comm = mailbox->get_async<TrackerQuery>(&query);
37 // Retrieve the data sent by the peer.
38 xbt_assert(query != nullptr);
40 // Add the peer to our peer list, if not already known.
41 known_peers.emplace(query->getPeerId());
43 // Sending back peers to the requesting peer
44 auto* answer = new TrackerAnswer(TRACKER_QUERY_INTERVAL);
45 std::set<int>::iterator next_peer;
46 int nb_known_peers = static_cast<int>(known_peers.size());
47 int max_tries = std::min(MAXIMUM_PEERS, nb_known_peers);
49 while (tried < max_tries) {
51 next_peer = known_peers.begin();
52 std::advance(next_peer, random.uniform_int(0, nb_known_peers - 1));
53 } while (answer->getPeers().find(*next_peer) != answer->getPeers().end());
54 answer->addPeer(*next_peer);
57 query->getReturnMailbox()->put_init(answer, TRACKER_COMM_SIZE)->detach();
62 sg4::this_actor::sleep_for(1);
65 XBT_INFO("Tracker is leaving");