inflight_requests.h

 
#pragma once
 
#include <memory>
#include <mutex>
#include <unordered_set>
#include <vector>
 
namespace ucxx {
 
class Request;
 
struct TrackedRequests {
  std::vector<std::shared_ptr<Request>> inflight{};   
  std::vector<std::shared_ptr<Request>> canceling{};  
};
 
class InflightRequests {
 private:
  std::unordered_set<std::shared_ptr<Request>> _inflight{};
  std::unordered_set<std::shared_ptr<Request>> _canceling{};
 
  std::mutex _mutex{};
 
 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(const std::shared_ptr<Request>& request);
 
  void merge(TrackedRequests&& trackedRequests);
 
  void remove(const std::shared_ptr<Request>& request);
 
  size_t cancelAll();
 
  [[nodiscard]] TrackedRequests release();
 
  [[nodiscard]] size_t getCancelingSize();
};
 
}  // namespace ucxx