asio fiber_manager

2026-02-18 14:02:18 +00:00 · 2014-03-23 20:17:18 +01:00
parent ab91ab52b8
commit a0b4d61ea3
2 changed files with 346 additions and 0 deletions
--- a/include/boost/fiber/asio/fiber_manager.hpp
+++ b/include/boost/fiber/asio/fiber_manager.hpp
@@ -0,0 +1,113 @@
+//          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)
+
+#ifndef BOOST_FIBERS_ASIO_FIBER_MANAGER_H
+#define BOOST_FIBERS_ASIO_FIBER_MANAGER_H
+
+#include <boost/assert.hpp>
+#include <boost/config.hpp>
+#include <boost/scoped_ptr.hpp>
+#include <boost/thread/lock_types.hpp> 
+#include <boost/utility.hpp>
+
+#include <boost/fiber/detail/config.hpp>
+#include <boost/fiber/detail/main_fiber.hpp>
+#include <boost/fiber/detail/spinlock.hpp>
+#include <boost/fiber/detail/waiting_queue.hpp>
+#include <boost/fiber/detail/worker_fiber.hpp>
+#include <boost/fiber/fiber.hpp>
+
+#ifdef BOOST_HAS_ABI_HEADERS
+#  include BOOST_ABI_PREFIX
+#endif
+
+# if defined(BOOST_MSVC)
+# pragma warning(push)
+# pragma warning(disable:4251 4275)
+# endif
+
+namespace boost {
+namespace fibers {
+namespace asio {
+
+struct sched_algorithm
+{
+    virtual ~sched_algorithm() {}
+
+    virtual void awakened( detail::worker_fiber *) = 0;
+
+    virtual detail::worker_fiber * pick_next() = 0;
+
+    virtual void priority( detail::worker_fiber *, int) BOOST_NOEXCEPT = 0;
+};
+
+struct fiber_manager : private noncopyable
+{
+    fiber_manager() BOOST_NOEXCEPT;
+
+    virtual ~fiber_manager() BOOST_NOEXCEPT;
+
+    void set_sched_algo( sched_algorithm * algo) {
+        sched_algo_ = algo;
+        def_algo_.reset();
+    }
+
+    void spawn( detail::worker_fiber *);
+
+    void priority( detail::worker_fiber * f, int prio) BOOST_NOEXCEPT
+    { sched_algo_->priority( f, prio); }
+
+    template< typename Rep, typename Period >
+    void wait_interval( chrono::duration< Rep, Period > const& wait_interval) const BOOST_NOEXCEPT
+    { wait_interval_ = wait_interval; }
+
+    template< typename Rep, typename Period >
+    chrono::duration< Rep, Period > wait_interval() const BOOST_NOEXCEPT
+    { return wait_interval_; }
+
+    void join( detail::worker_fiber *);
+
+    detail::worker_fiber * active() BOOST_NOEXCEPT
+    { return active_fiber_; }
+
+    virtual void run();
+
+    void wait( unique_lock< detail::spinlock > &);
+    bool wait_until( clock_type::time_point const&,
+                     unique_lock< detail::spinlock > &);
+    template< typename Rep, typename Period >
+    bool wait_for( chrono::duration< Rep, Period > const& timeout_duration,
+                   unique_lock< detail::spinlock > & lk)
+    { return wait_until( clock_type::now() + timeout_duration, lk); }
+
+    void yield();
+
+protected:
+    typedef detail::waiting_queue   wqueue_t;
+
+    scoped_ptr< sched_algorithm >   def_algo_;
+    sched_algorithm             *   sched_algo_;
+    wqueue_t                        wqueue_;
+
+    void resume_( detail:: worker_fiber *);
+
+private:
+    clock_type::duration            wait_interval_;
+    detail::worker_fiber        *   active_fiber_;
+
+    clock_type::time_point next_wakeup_();
+};
+
+}}}
+
+# if defined(BOOST_MSVC)
+# pragma warning(pop)
+# endif
+
+#ifdef BOOST_HAS_ABI_HEADERS
+#  include BOOST_ABI_SUFFIX
+#endif
+
+#endif // BOOST_FIBERS_ASIO_FIBER_MANAGER_H