worker_builder.h

#pragma once
 
#include <memory>
#include <utility>
 
#include <ucxx/buffer.h>
 
namespace ucxx {
 
// Forward declarations
class Context;
class Worker;

class WorkerBuilder final {
 public:
  explicit WorkerBuilder(std::shared_ptr<Context> context);

  ~WorkerBuilder();

  WorkerBuilder(const WorkerBuilder& other);
  WorkerBuilder& operator=(const WorkerBuilder& other);
  WorkerBuilder(WorkerBuilder&&) noexcept;
  WorkerBuilder& operator=(WorkerBuilder&&) noexcept;

  operator std::shared_ptr<Worker>();

  WorkerBuilder& delayedSubmission(bool enable = true);

  WorkerBuilder& pythonFuture(bool enable = true);

  WorkerBuilder& requestAttributes(bool enable = true);

  WorkerBuilder& cudaBufferType(BufferType bufferType);

  [[nodiscard]] std::shared_ptr<Worker> build();
 
 private:
  struct Impl;
  std::unique_ptr<Impl> _impl;
};

[[nodiscard]] inline WorkerBuilder workerBuilder(std::shared_ptr<Context> context)
{
  return WorkerBuilder(std::move(context));
}
 
}  // namespace ucxx