register_inflight_request.h

#pragma once
 
#include <memory>
#include <tuple>
#include <utility>
 
#include <ucxx/endpoint.h>
#include <ucxx/request.h>
#include <ucxx/worker.h>
 
namespace ucxx {
 
class Component;
 
namespace detail {
 
inline void registerInflightRequest(std::shared_ptr<Component> component,
                                    std::shared_ptr<Request> req)
{
  if (auto ep = std::dynamic_pointer_cast<Endpoint>(component))
    std::ignore = ep->registerInflightRequest(std::move(req));
  else if (auto wk = std::dynamic_pointer_cast<Worker>(component))
    std::ignore = wk->registerInflightRequest(std::move(req));
}
 
}  // namespace detail
 
}  // namespace ucxx