inflight_requests.h

 
#pragma once
 
#include <map>
#include <memory>
#include <mutex>
#include <utility>
 
namespace ucxx {
 
class Request;
 
typedef std::
  map<std::shared_ptr<Request>, std::shared_ptr<Request>, std::owner_less<std::shared_ptr<Request>>>
    InflightRequestsMap;
 
typedef struct TrackedRequests {
  InflightRequestsMap _inflight{};   
  InflightRequestsMap _canceling{};  
  std::mutex _mutex{};               
  std::mutex
    _cancelMutex{};  
} TrackedRequests;
 
typedef std::unique_ptr<TrackedRequests> TrackedRequestsPtr;
 
class InflightRequests {
 private:
  TrackedRequestsPtr _trackedRequests{
    std::make_unique<TrackedRequests>()};  
  std::recursive_mutex _mutex{};           
 
  size_t dropCanceled();
 
 public:
  InflightRequests() = default;
 
  InflightRequests(const InflightRequests&)            = delete;
  InflightRequests& operator=(InflightRequests const&) = delete;
  InflightRequests(InflightRequests&& o)               = delete;
  InflightRequests& operator=(InflightRequests&& o)    = delete;
 
  ~InflightRequests();
 
  [[nodiscard]] size_t size();
 
  void insert(std::shared_ptr<Request> request);
 
  void merge(TrackedRequestsPtr trackedRequests);
 
  void remove(std::shared_ptr<Request> request);
 
  size_t cancelAll();
 
  [[nodiscard]] TrackedRequestsPtr release();
 
  [[nodiscard]] size_t getCancelingSize();
};
 
}  // namespace ucxx