|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201 |
- #include "librf/librf.h"
-
- namespace librf
- {
- namespace detail
- {
- LIBRF_API void state_event_base_t::resume()
- {
- coroutine_handle<> handler = _coro;
- if (handler)
- {
- _coro = nullptr;
- _scheduler->del_final(this);
- handler.resume();
- }
- }
-
- LIBRF_API bool state_event_base_t::has_handler() const noexcept
- {
- return (bool)_coro;
- }
-
- LIBRF_API void state_event_t::on_cancel() noexcept
- {
- event_v2_impl** oldValue = _value.load(std::memory_order_acquire);
- if (oldValue != nullptr && _value.compare_exchange_strong(oldValue, nullptr, std::memory_order_acq_rel))
- {
- *oldValue = nullptr;
- _thandler.stop();
-
- this->_coro = nullptr;
- }
- }
-
- LIBRF_API bool state_event_t::on_notify(event_v2_impl* eptr)
- {
- event_v2_impl** oldValue = _value.load(std::memory_order_acquire);
- if (oldValue != nullptr && _value.compare_exchange_strong(oldValue, nullptr, std::memory_order_acq_rel))
- {
- *oldValue = eptr;
- _thandler.stop();
-
- assert(this->_scheduler != nullptr);
- if (this->_coro)
- this->_scheduler->add_generator(this);
-
- return true;
- }
- return false;
- }
-
- LIBRF_API bool state_event_t::on_timeout()
- {
- event_v2_impl** oldValue = _value.load(std::memory_order_acquire);
- if (oldValue != nullptr && _value.compare_exchange_strong(oldValue, nullptr, std::memory_order_acq_rel))
- {
- *oldValue = nullptr;
- _thandler.reset();
-
- assert(this->_scheduler != nullptr);
- if (this->_coro)
- this->_scheduler->add_generator(this);
-
- return true;
- }
- return false;
- }
-
-
-
-
- LIBRF_API void state_event_all_t::on_cancel() noexcept
- {
- intptr_t oldValue = _counter.load(std::memory_order_acquire);
- if (oldValue >= 0 && _counter.compare_exchange_strong(oldValue, -1, std::memory_order_acq_rel))
- {
- *_value = false;
- _thandler.stop();
-
- this->_coro = nullptr;
- }
- }
-
- LIBRF_API bool state_event_all_t::on_notify(event_v2_impl*)
- {
- intptr_t oldValue = _counter.load(std::memory_order_acquire);
- if (oldValue <= 0) return false;
-
- oldValue = _counter.fetch_add(-1, std::memory_order_acq_rel);
- if (oldValue == 1)
- {
- *_value = true;
- _thandler.stop();
-
- assert(this->_scheduler != nullptr);
- if (this->_coro)
- this->_scheduler->add_generator(this);
-
- return true;
- }
-
- return oldValue >= 1;
- }
-
- LIBRF_API bool state_event_all_t::on_timeout()
- {
- intptr_t oldValue = _counter.load(std::memory_order_acquire);
- if (oldValue >= 0 && _counter.compare_exchange_strong(oldValue, -1, std::memory_order_acq_rel))
- {
- *_value = false;
- _thandler.reset();
-
- assert(this->_scheduler != nullptr);
- if (this->_coro)
- this->_scheduler->add_generator(this);
-
- return true;
- }
- return false;
- }
-
-
-
- LIBRF_API event_v2_impl::event_v2_impl(bool initially) noexcept
- : _counter(initially ? 1 : 0)
- {
- }
-
- template<class _Ty, class _Ptr>
- static auto try_pop_list(intrusive_link_queue<_Ty, _Ptr>& list)
- {
- return list.try_pop();
- }
- template<class _Ptr>
- static _Ptr try_pop_list(std::list<_Ptr>& list)
- {
- if (!list.empty())
- {
- _Ptr ptr = list.front();
- list.pop_front();
- return ptr;
- }
- return nullptr;
- }
- template<class _Ty, class _Ptr>
- static void clear_list(intrusive_link_queue<_Ty, _Ptr>& list)
- {
- for (; list.try_pop() != nullptr;);
- }
- template<class _Ptr>
- static void clear_list(std::list<_Ptr>& list)
- {
- list.clear();
- }
-
- LIBRF_API event_v2_impl::~event_v2_impl()
- {
- clear_list(_wait_awakes);
- }
-
- LIBRF_API void event_v2_impl::signal_all() noexcept
- {
- scoped_lock<lock_type> lock_(_lock);
-
- _counter.store(0, std::memory_order_release);
-
- state_event_ptr state;
- for (; (state = try_pop_list(_wait_awakes)) != nullptr;)
- {
- (void)state->on_notify(this);
- }
- }
-
- LIBRF_API void event_v2_impl::signal() noexcept
- {
- scoped_lock<lock_type> lock_(_lock);
-
- state_event_ptr state;
- for (; (state = try_pop_list(_wait_awakes)) != nullptr;)
- {
- if (state->on_notify(this))
- return;
- }
-
- _counter.fetch_add(1, std::memory_order_acq_rel);
- }
- }
-
- LIBRF_API event_t::event_t(bool initially)
- :_event(std::make_shared<detail::event_v2_impl>(initially))
- {
- }
-
- LIBRF_API event_t::event_t(std::adopt_lock_t)
- {
- }
-
- LIBRF_API event_t::~event_t()
- {
- }
- }
|