FiberManager.h

Go to the documentation of this file.

 
/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#pragma once
#include <boost/fiber/algo/shared_work.hpp>
#include <boost/fiber/condition_variable.hpp>
#include <boost/fiber/fiber.hpp>
#include <boost/fiber/mutex.hpp>
#include <boost/version.hpp>
#if ( BOOST_VERSION >= 108400 )
#  include <boost/fiber/stack_allocator_wrapper.hpp>
#endif
 
class FiberManager {
 
public:
  FiberManager( int n_threads );
  ~FiberManager();
 
  template <typename F>
  void schedule( F&& func ) {
    boost::fibers::fiber( boost::fibers::launch::post, std::forward<F>( func ) ).detach();
    for ( auto* p : m_schedAlgoList ) { p->notify(); }
  }
 
private:
  using SchedAlgo = boost::fibers::algo::shared_work;
  boost::fibers::condition_variable m_shuttingDown_cv{};
  boost::fibers::mutex              m_shuttingDown_mtx{};
  std::vector<std::thread>          m_threads{};
  std::vector<SchedAlgo*>           m_schedAlgoList{};
};