worker.h

#pragma once
 
#include <functional>
#include <memory>
#include <mutex>
#include <queue>
#include <thread>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/python/future.h>
#include <ucxx/python/notifier.h>
#include <ucxx/worker.h>
 
namespace ucxx {
 
namespace python {

class Worker : public ::ucxx::Worker {
 private:
  Worker(std::shared_ptr<Context> context,
         const bool enableDelayedSubmission = false,
         const bool enableFuture            = false);
 
 public:
  Worker()                         = delete;
  Worker(const Worker&)            = delete;
  Worker& operator=(Worker const&) = delete;
  Worker(Worker&& o)               = delete;
  Worker& operator=(Worker&& o)    = delete;

  friend std::shared_ptr<::ucxx::Worker> createWorker(std::shared_ptr<Context> context,
                                                      const bool enableDelayedSubmission,
                                                      const bool enableFuture);

  void populateFuturesPool() override;

  void clearFuturesPool() override;

  [[nodiscard]] std::shared_ptr<::ucxx::Future> getFuture() override;

  [[nodiscard]] RequestNotifierWaitState waitRequestNotifier(uint64_t periodNs) override;

  void runRequestNotifier() override;

  void stopRequestNotifierThread() override;
};
 
}  // namespace python
 
}  // namespace ucxx