artdaq  v2_03_02
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Pages
RoutingMasterPolicy.cc
1 #include "artdaq/Application/Routing/RoutingMasterPolicy.hh"
2 #include "fhiclcpp/ParameterSet.h"
3 
5  : next_sequence_id_(0)
6  , tokens_()
7  , max_token_count_(0)
8 {
9  auto receiver_ranks = ps.get<std::vector<int>>("receiver_ranks");
10  receiver_ranks_.insert(receiver_ranks.begin(), receiver_ranks.end());
11 }
12 
13 void artdaq::RoutingMasterPolicy::AddReceiverToken(int rank, unsigned new_slots_free)
14 {
15  if (!receiver_ranks_.count(rank)) return;
16  TRACE(10, "RoutingMasterPolicy::AddReceiverToken BEGIN");
17  std::unique_lock<std::mutex> lk(tokens_mutex_);
18  for (unsigned i = 0; i < new_slots_free; ++i)
19  {
20  tokens_.push_back(rank);
21  }
22  if (tokens_.size() > max_token_count_) max_token_count_ = tokens_.size();
23  TRACE(10, "RoutingMasterPolicy::AddReceiverToken END");
24 }
25 
26 std::unique_ptr<std::deque<int>> artdaq::RoutingMasterPolicy::getTokensSnapshot()
27 {
28  TRACE(10, "RoutingMasterPolicy::getTokensSnapshot BEGIN");
29  std::unique_lock<std::mutex> lk(tokens_mutex_);
30  auto out = std::make_unique<std::deque<int>>(tokens_);
31  tokens_.clear();
32  TRACE(10, "RoutingMasterPolicy::getTokensSnapshot END");
33  return out;
34 }
35 
36 void artdaq::RoutingMasterPolicy::addUnusedTokens(std::unique_ptr<std::deque<int>> tokens)
37 {
38  std::unique_lock<std::mutex> lk(tokens_mutex_);
39  for (auto token = tokens.get()->rbegin(); token != tokens.get()->rend(); ++token)
40  {
41  tokens_.push_front(*token);
42  }
43  if (tokens_.size() > max_token_count_) max_token_count_ = tokens_.size();
44 }
void AddReceiverToken(int rank, unsigned new_slots_free)
Add a token to the token list.
std::unique_ptr< std::deque< int > > getTokensSnapshot()
Gets the current token list, used for building Routing Tables.
void addUnusedTokens(std::unique_ptr< std::deque< int >> tokens)
If necessary, return unused tokens to the token list, for subsequent updates.
RoutingMasterPolicy(fhicl::ParameterSet ps)
RoutingMasterPolicy Constructor.