python_future_task_collector.h

 
#pragma once
 
#include <mutex>
#include <vector>
 
#include <Python.h>
 
namespace ucxx {
 
namespace python {
 
class PythonFutureTaskCollector {
 public:
  std::vector<PyObject*> _toCollect{};  
  std::mutex _mutex{};                  
 
  [[nodiscard]] static PythonFutureTaskCollector& get();
 
  void push(PyObject* handle);
 
  void collect();
 
 private:
  PythonFutureTaskCollector();
 
 public:
  PythonFutureTaskCollector(const PythonFutureTaskCollector&)            = delete;
  PythonFutureTaskCollector& operator=(PythonFutureTaskCollector const&) = delete;
  PythonFutureTaskCollector(PythonFutureTaskCollector&& o)               = delete;
  PythonFutureTaskCollector& operator=(PythonFutureTaskCollector&& o)    = delete;
 
  ~PythonFutureTaskCollector();
};
 
}  // namespace python
 
}  // namespace ucxx