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