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{};     
  bool _stop{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,
    const bool& stop,
    ProgressThreadStartCallback startCallback,
    ProgressThreadStartCallbackArg startCallbackArg,
    std::shared_ptr<DelayedSubmissionCollection> delayedSubmissionCollection);
 
 public:
  WorkerProgressThread() = delete;
 
  WorkerProgressThread(const bool pollingMode,
                       std::function<bool(void)> progressFunction,
                       std::function<void(void)> signalWorkerFunction,
                       ProgressThreadStartCallback startCallback,
                       ProgressThreadStartCallbackArg startCallbackArg,
                       std::shared_ptr<DelayedSubmissionCollection> delayedSubmissionCollection);
 
  ~WorkerProgressThread();
 
  bool pollingMode() const;
 
  std::thread::id getId() const;
};
 
}  // namespace ucxx