worker_progress_thread.h

 
#pragma once
 
#include <functional>
#include <memory>
#include <mutex>
#include <thread>
 
#include <ucxx/delayed_submission.h>
 
namespace ucxx {
 
typedef std::function<void(void)> SignalWorkerFunction;
 
typedef std::function<void(void*)> ProgressThreadStartCallback;
 
typedef void* ProgressThreadStartCallbackArg;
 
class WorkerProgressThread {
 private:
  std::thread _thread{};                                       
  std::shared_ptr<bool> _stop{std::make_shared<bool>(false)};  
  bool _pollingMode{false};  
  SignalWorkerFunction _signalWorkerFunction{
    nullptr};  
  ProgressThreadStartCallback _startCallback{
    nullptr};  
  ProgressThreadStartCallbackArg _startCallbackArg{
    nullptr};  
  std::shared_ptr<DelayedSubmissionCollection> _delayedSubmissionCollection{
    nullptr};  
 
  static void progressUntilSync(
    std::function<bool(void)> progressFunction,
    std::shared_ptr<bool> stop,
    std::function<void(void)> setThreadId,
    ProgressThreadStartCallback startCallback,
    ProgressThreadStartCallbackArg startCallbackArg,
    std::shared_ptr<DelayedSubmissionCollection> delayedSubmissionCollection);
 
 public:
  WorkerProgressThread()                                       = default;
  WorkerProgressThread(const WorkerProgressThread&)            = delete;
  WorkerProgressThread& operator=(WorkerProgressThread const&) = delete;
  WorkerProgressThread(WorkerProgressThread&& o)               = default;
  WorkerProgressThread& operator=(WorkerProgressThread&& o)    = default;
 
  WorkerProgressThread(const bool pollingMode,
                       std::function<bool(void)> progressFunction,
                       std::function<void(void)> signalWorkerFunction,
                       std::function<void(void)> setThreadId,
                       ProgressThreadStartCallback startCallback,
                       ProgressThreadStartCallbackArg startCallbackArg,
                       std::shared_ptr<DelayedSubmissionCollection> delayedSubmissionCollection);
 
  ~WorkerProgressThread();
 
  [[nodiscard]] bool pollingMode() const;
 
  [[nodiscard]] std::thread::id getId() const;
 
  [[nodiscard]] bool isRunning() const;
 
  void stop();
};
 
}  // namespace ucxx