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