mirror of
https://github.com/boostorg/fiber.git
synced 2026-02-11 23:52:29 +00:00
56 lines
1.3 KiB
C++
56 lines
1.3 KiB
C++
|
|
// Copyright Oliver Kowalke 2013.
|
|
// Distributed under the Boost Software License, Version 1.0.
|
|
// (See accompanying file LICENSE_1_0.txt or copy at
|
|
// http://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
#include "workstealing_round_robin.hpp"
|
|
|
|
#include <boost/assert.hpp>
|
|
|
|
#ifdef BOOST_HAS_ABI_HEADERS
|
|
# include BOOST_ABI_PREFIX
|
|
#endif
|
|
|
|
void
|
|
workstealing_round_robin::awakened( boost::fibers::detail::fiber_handle & f)
|
|
{
|
|
std::unique_lock< std::mutex > lk( mtx_);
|
|
BOOST_ASSERT( f->is_ready() );
|
|
rqueue_.push_back( f);
|
|
}
|
|
|
|
boost::fibers::detail::fiber_handle
|
|
workstealing_round_robin::pick_next()
|
|
{
|
|
std::unique_lock< std::mutex > lk( mtx_);
|
|
boost::fibers::detail::fiber_handle f;
|
|
if ( ! rqueue_.empty() )
|
|
{
|
|
f = rqueue_.front();
|
|
rqueue_.pop_front();
|
|
BOOST_ASSERT( f->is_ready() );
|
|
}
|
|
return f;
|
|
}
|
|
|
|
boost::fibers::fiber
|
|
workstealing_round_robin::steal()
|
|
{
|
|
std::unique_lock< std::mutex > lk( mtx_);
|
|
for ( boost::fibers::detail::fiber_handle f : rqueue_)
|
|
{
|
|
if ( ! f->thread_affinity() )
|
|
{
|
|
rqueue_.remove( f);
|
|
BOOST_ASSERT( f->is_ready() );
|
|
return boost::fibers::fiber( f);
|
|
}
|
|
}
|
|
return boost::fibers::fiber();
|
|
}
|
|
|
|
#ifdef BOOST_HAS_ABI_HEADERS
|
|
# include BOOST_ABI_SUFFIX
|
|
#endif
|