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