FiberManager.h

Go to the documentation of this file.

 
/***********************************************************************************\
* (c) Copyright 1998-2026 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/fixedsize_stack.hpp>
#include <boost/fiber/mutex.hpp>
 
#include <thread>
 
class FiberManager {
 
public:
  FiberManager( int n_threads, std::size_t stackSize = 16ULL * 1024 * 1024 );
  ~FiberManager();

  template <typename F>
  void schedule( F&& func ) {
    // Add a new fiber invoking the function to the shared work queue, then notify the workers in case they were
    // sleeping.
    boost::fibers::fiber( boost::fibers::launch::post, std::allocator_arg, StackAlloc( m_stackSize ),
                          std::forward<F>( func ) )
        .detach();
    for ( auto* p : m_schedAlgoList ) { p->notify(); }
  }
 
private:
  using StackAlloc = boost::fibers::fixedsize_stack;
  using SchedAlgo  = boost::fibers::algo::shared_work;
  boost::fibers::condition_variable m_shuttingDown_cv{};
  boost::fibers::mutex              m_shuttingDown_mtx{};
  std::vector<std::jthread>         m_threads{};
  std::vector<SchedAlgo*>           m_schedAlgoList{};

  std::size_t m_stackSize;
};