1 #include "artdaq/DAQdata/Globals.hh"
2 #define TRACE_NAME (app_name + "_RequestSender").c_str()
10 #include "artdaq/DAQrate/RequestSender.hh"
12 #include "artdaq-core/Core/StatisticsCollection.hh"
14 #include "artdaq/DAQrate/detail/RoutingPacket.hh"
15 #include "cetlib_except/exception.h"
19 : send_requests_(pset.get<bool>(
"send_requests", false))
21 , request_address_(pset.get<std::string>(
"request_address",
"227.128.12.26"))
22 , request_port_(pset.get<int>(
"request_port", 3001))
23 , request_delay_(pset.get<size_t>(
"request_delay_ms", 0) * 1000)
24 , request_shutdown_timeout_us_(pset.get<size_t>(
"request_shutdown_timeout_us", 100000))
26 , multicast_out_addr_(pset.get<std::string>(
"multicast_interface_ip", pset.get<std::string>(
"output_address",
"0.0.0.0")))
27 , request_mode_(detail::RequestMessageMode::Normal)
33 TLOG(TLVL_DEBUG) <<
"RequestSender CONSTRUCTOR";
36 auto rmConfig = pset.get<fhicl::ParameterSet>(
"routing_token_config", fhicl::ParameterSet());
37 send_routing_tokens_ = rmConfig.get<
bool>(
"use_routing_manager",
false);
38 token_port_ = rmConfig.get<
int>(
"routing_token_port", 35555);
39 token_address_ = rmConfig.get<std::string>(
"routing_manager_hostname",
"localhost");
41 TLOG(12) <<
"artdaq::RequestSender::RequestSender ctor - reader_thread_ initialized";
47 TLOG(TLVL_INFO) <<
"Shutting down RequestSender: Waiting for " << request_sending_.load() <<
" requests to be sent";
49 auto start_time = std::chrono::steady_clock::now();
51 while (request_sending_.load() > 0 && request_shutdown_timeout_us_ + request_delay_ > TimeUtils::GetElapsedTimeMicroseconds(start_time))
56 std::lock_guard<std::mutex> lk(request_mutex_);
57 std::lock_guard<std::mutex> lk2(request_send_mutex_);
59 TLOG(TLVL_INFO) <<
"Shutting down RequestSender: request_socket_: " << request_socket_ <<
", token_socket_: " << token_socket_;
60 if (request_socket_ != -1)
62 if (shutdown(request_socket_, 2) != 0 && errno == ENOTSOCK)
64 TLOG(TLVL_ERROR) <<
"Shutdown of request_socket_ resulted in ENOTSOCK. NOT Closing file descriptor!";
68 close(request_socket_);
72 if (token_socket_ != -1)
74 if (shutdown(token_socket_, 2) != 0 && errno == ENOTSOCK)
76 TLOG(TLVL_ERROR) <<
"Shutdown of token_socket_ resulted in ENOTSOCK. NOT Closing file descriptor!";
89 std::lock_guard<std::mutex> lk(request_mutex_);
95 void RequestSender::setup_requests_()
99 request_socket_ = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
100 if (request_socket_ < 0)
102 TLOG(TLVL_ERROR) <<
"I failed to create the socket for sending Data Requests! err=" << strerror(errno);
105 int sts =
ResolveHost(request_address_.c_str(), request_port_, request_addr_);
108 TLOG(TLVL_ERROR) <<
"Unable to resolve Data Request address, err=" << strerror(errno);
125 if (multicast_out_addr_ !=
"localhost" && multicast_out_addr_ !=
"0.0.0.0")
132 TLOG(TLVL_ERROR) <<
"Unable to determine the multicast interface address for " << multicast_out_addr_ <<
", err=" << strerror(errno);
135 char addr_str[INET_ADDRSTRLEN];
136 inet_ntop(AF_INET, &(addr), addr_str, INET_ADDRSTRLEN);
137 TLOG(TLVL_INFO) <<
"Successfully determined the multicast network interface for " << multicast_out_addr_ <<
": " << addr_str;
139 if (setsockopt(request_socket_, IPPROTO_IP, IP_MULTICAST_IF, &addr,
sizeof(addr)) == -1)
141 TLOG(TLVL_ERROR) <<
"Cannot set outgoing interface, err=" << strerror(errno);
146 if (setsockopt(request_socket_, SOL_SOCKET, SO_REUSEADDR, &yes,
sizeof(yes)) < 0)
148 TLOG(TLVL_ERROR) <<
"Unable to enable port reuse on request socket, err=" << strerror(errno);
151 if (setsockopt(request_socket_, IPPROTO_IP, IP_MULTICAST_LOOP, &yes,
sizeof(yes)) < 0)
153 TLOG(TLVL_ERROR) <<
"Unable to enable multicast loopback on request socket, err=" << strerror(errno);
156 if (setsockopt(request_socket_, SOL_SOCKET, SO_BROADCAST, &yes,
sizeof(yes)) == -1)
158 TLOG(TLVL_ERROR) <<
"Cannot set request socket to broadcast, err=" << strerror(errno);
164 void RequestSender::setup_tokens_()
166 if (send_routing_tokens_)
168 TLOG(TLVL_DEBUG) <<
"Creating Routing Token sending socket";
169 auto start_time = std::chrono::steady_clock::now();
170 while (token_socket_ < 0 && TimeUtils::GetElapsedTime(start_time) < 30)
172 token_socket_ =
TCPConnect(token_address_.c_str(), token_port_, 0,
sizeof(detail::RoutingToken));
173 if (token_socket_ < 0)
175 TLOG(TLVL_TRACE) <<
"Waited " << TimeUtils::GetElapsedTime(start_time) <<
" s for Routing Manager to open token socket";
179 if (token_socket_ < 0)
181 TLOG(TLVL_ERROR) <<
"I failed to create the socket for sending Routing Tokens! err=" << strerror(errno);
184 TLOG(TLVL_INFO) <<
"Routing Token sending socket created successfully for address " << token_address_;
188 void RequestSender::do_send_request_()
195 if (request_socket_ == -1)
200 TLOG(TLVL_TRACE) <<
"Waiting for " << request_delay_ <<
" microseconds.";
201 std::this_thread::sleep_for(std::chrono::microseconds(request_delay_));
203 TLOG(TLVL_TRACE) <<
"Creating RequestMessage";
204 detail::RequestMessage message;
205 message.setRank(my_rank);
206 message.setRunNumber(run_number_);
208 std::lock_guard<std::mutex> lk(request_mutex_);
209 for (
auto& req : active_requests_)
211 TLOG(12) <<
"Adding a request with sequence ID " << req.first <<
", timestamp " << req.second <<
" to request message";
212 message.addRequest(req.first, req.second);
214 TLOG(TLVL_TRACE) <<
"Setting mode flag in Message Header to " << request_mode_;
215 message.setMode(request_mode_);
217 char str[INET_ADDRSTRLEN];
218 inet_ntop(AF_INET, &(request_addr_.sin_addr), str, INET_ADDRSTRLEN);
219 std::lock_guard<std::mutex> lk2(request_send_mutex_);
220 TLOG(TLVL_TRACE) <<
"Sending request for " << message.size() <<
" events to multicast group " << str
221 <<
", port " << request_port_ <<
", interface " << multicast_out_addr_;
222 auto buf = message.GetMessage();
223 auto sts = sendto(request_socket_, &buf[0], buf.size(), 0,
reinterpret_cast<struct sockaddr*
>(&request_addr_),
sizeof(request_addr_));
224 if (sts < 0 || static_cast<size_t>(sts) != buf.size())
226 TLOG(TLVL_ERROR) <<
"Error sending request message err=" << strerror(errno) <<
"sts=" << sts;
227 request_socket_ = -1;
231 TLOG(TLVL_TRACE) <<
"Done sending request sts=" << sts;
235 void RequestSender::send_routing_token_(
int nSlots,
int run_number)
237 TLOG(TLVL_TRACE) <<
"send_routing_token_ called, send_routing_tokens_=" << std::boolalpha << send_routing_tokens_;
238 if (!send_routing_tokens_)
242 if (token_socket_ == -1)
246 detail::RoutingToken token;
247 token.header = TOKEN_MAGIC;
248 token.rank = my_rank;
249 token.new_slots_free = nSlots;
250 token.run_number = run_number;
252 TLOG(TLVL_TRACE) <<
"Sending RoutingToken to " << token_address_ <<
":" << token_port_;
254 while (sts <
sizeof(detail::RoutingToken))
256 auto res = send(token_socket_, reinterpret_cast<uint8_t*>(&token) + sts,
sizeof(detail::RoutingToken) - sts, 0);
259 TLOG(TLVL_WARNING) <<
"Error on token_socket, reconnecting";
260 close(token_socket_);
268 tokens_sent_ += nSlots;
269 TLOG(TLVL_TRACE) <<
"Done sending RoutingToken to " << token_address_ <<
":" << token_port_;
274 while (!initialized_)
278 if (!send_routing_tokens_)
282 boost::thread token([=] { send_routing_token_(nSlots, run_number); });
289 while (!initialized_)
299 std::lock_guard<std::mutex> lk(request_mutex_);
306 boost::thread request([=] { do_send_request_(); });
312 while (!initialized_)
318 std::lock_guard<std::mutex> lk(request_mutex_);
319 if (active_requests_.count(seqID) == 0u)
321 TLOG(12) <<
"Adding request for sequence ID " << seqID <<
" and timestamp " << timestamp <<
" to request list.";
322 active_requests_[seqID] = timestamp;
330 while (!initialized_)
334 std::lock_guard<std::mutex> lk(request_mutex_);
335 TLOG(12) <<
"Removing request for sequence ID " << seqID <<
" from request list.";
336 active_requests_.erase(seqID);
void RemoveRequest(Fragment::sequence_id_t seqID)
Remove a request from the request list.
int ResolveHost(char const *host_in, in_addr &addr)
Convert a string hostname to a in_addr suitable for socket communication.
End of Run mode (Used to end request processing on receiver)
int TCPConnect(char const *host_in, int dflt_port, int64_t flags=0, int sndbufsiz=0)
Connect to a host on a given port.
void SendRoutingToken(int nSlots, int run_number)
Send a RoutingToken message indicating that slots are available.
virtual ~RequestSender()
RequestSender Destructor.
void SendRequest(bool endOfRunOnly=false)
Send a request message containing all current requests.
int GetInterfaceForNetwork(char const *host_in, in_addr &addr)
Convert an IP address to the network address of the interface sharing the subnet mask.
RequestSender()=delete
Default Constructor is deleted.
void SetRequestMode(detail::RequestMessageMode mode)
Set the mode for RequestMessages. Used to indicate when RequestSender should enter "EndOfRun" mode...
RequestMessageMode
Mode used to indicate current run conditions to the request receiver.
void AddRequest(Fragment::sequence_id_t seqID, Fragment::timestamp_t timestamp)
Add a request to the request list.