inflight_requests.h

 
#pragma once
 
#include <map>
#include <memory>
#include <mutex>
#include <utility>
 
namespace ucxx {
 
class Request;
 
typedef std::map<const Request* const, std::shared_ptr<Request>> InflightRequestsMap;
 
typedef std::unique_ptr<InflightRequestsMap> InflightRequestsMapPtr;
 
typedef struct TrackedRequests {
  InflightRequestsMapPtr _inflight;   
  InflightRequestsMapPtr _canceling;  
 
  TrackedRequests()
    : _inflight(std::make_unique<InflightRequestsMap>()),
      _canceling(std::make_unique<InflightRequestsMap>())
  {
  }
} TrackedRequests;
 
typedef std::unique_ptr<TrackedRequests> TrackedRequestsPtr;
 
class InflightRequests {
 private:
  TrackedRequestsPtr _trackedRequests{
    std::make_unique<TrackedRequests>()};  
  std::mutex _mutex{};  
  std::mutex
    _cancelMutex{};  
 
  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();
 
  size_t size();
 
  void insert(std::shared_ptr<Request> request);
 
  void merge(TrackedRequestsPtr trackedRequests);
 
  void remove(const Request* const request);
 
  size_t cancelAll();
 
  TrackedRequestsPtr release();
 
  size_t getCancelingSize();
};
 
}  // namespace ucxx