group.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <functional>
#include <mutex>
#include <optional>
#include <thread>
#include <vector>
 
namespace fur::mt {
 
class Runner {
  std::mutex& _mutex;
  bool& _should_terminate;
 
 public:
  Runner(std::mutex& mutex, bool& _should_terminate);
 
  bool alive();
};
 
template <typename State>
class ThreadGroup {
  using ThreadFn = std::function<void(Runner, State&, size_t)>;
 
  std::vector<std::thread> _threads;
  std::vector<State> _states;
  ThreadFn _thread_fn;
  std::mutex _mutex;
  bool _should_terminate;
 
 public:
  ThreadGroup();
  virtual ~ThreadGroup();
 
  //===========================================================================
  // This object is not copyable and not movable because of the mutex
 
  ThreadGroup(ThreadGroup&) = delete;
  ThreadGroup& operator=(ThreadGroup&) = delete;
  ThreadGroup(ThreadGroup&&) noexcept = delete;
  ThreadGroup& operator=(ThreadGroup&&) noexcept = delete;
 
  //===========================================================================
 
  void launch(ThreadFn fn, int64_t max_worker_threads = 0);
 
  void terminate();
 
  std::vector<State>& get_states();
 
  [[nodiscard]] int64_t get_worker_count() const;
 
 private:
  void thread_main(int64_t index);
};
 
}  // namespace fur::mt
 
#include <mt/group.inl>