1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
/* Copyright (c) 2012-2025. The SimGrid Team. All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
* under the terms of the license (GNU LGPL) which comes with this package. */
#include "s4u-tracker.hpp"
#include <algorithm>
XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_bt_tracker, "Messages specific for the tracker");
namespace sg4 = simgrid::s4u;
Tracker::Tracker(std::vector<std::string> args)
{
// Checking arguments
xbt_assert(args.size() == 2, "Wrong number of arguments for the tracker.");
// Retrieving end time
try {
deadline = std::stod(args[1]);
} catch (const std::invalid_argument&) {
throw std::invalid_argument("Invalid deadline:" + args[1]);
}
xbt_assert(deadline > 0, "Wrong deadline supplied");
mailbox = sg4::Mailbox::by_name(TRACKER_MAILBOX);
XBT_INFO("Tracker launched.");
}
void Tracker::operator()()
{
sg4::CommPtr comm = nullptr;
TrackerQuery* query = nullptr;
while (sg4::Engine::get_clock() < deadline) {
if (comm == nullptr)
comm = mailbox->get_async<TrackerQuery>(&query);
if (comm->test()) {
// Retrieve the data sent by the peer.
xbt_assert(query != nullptr);
// Add the peer to our peer list, if not already known.
known_peers.emplace(query->getPeerId());
// Sending back peers to the requesting peer
auto* answer = new TrackerAnswer(TRACKER_QUERY_INTERVAL);
std::set<int>::iterator next_peer;
int nb_known_peers = static_cast<int>(known_peers.size());
int max_tries = std::min(MAXIMUM_PEERS, nb_known_peers);
int tried = 0;
while (tried < max_tries) {
do {
next_peer = known_peers.begin();
std::advance(next_peer, random.uniform_int(0, nb_known_peers - 1));
} while (answer->getPeers().find(*next_peer) != answer->getPeers().end());
answer->addPeer(*next_peer);
tried++;
}
query->getReturnMailbox()->put_init(answer, TRACKER_COMM_SIZE)->detach();
delete query;
comm = nullptr;
} else {
sg4::this_actor::sleep_for(1);
}
}
XBT_INFO("Tracker is leaving");
}
|
