123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062 |
- #ifndef BOOST_THREAD_V2_SHARED_MUTEX_HPP
- #define BOOST_THREAD_V2_SHARED_MUTEX_HPP
- // shared_mutex.hpp
- //
- // Copyright Howard Hinnant 2007-2010.
- // Copyright Vicente J. Botet Escriba 2012.
- //
- // 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)
- /*
- <shared_mutex> synopsis
- namespace boost
- {
- namespace thread_v2
- {
- class shared_mutex
- {
- public:
- shared_mutex();
- ~shared_mutex();
- shared_mutex(const shared_mutex&) = delete;
- shared_mutex& operator=(const shared_mutex&) = delete;
- // Exclusive ownership
- void lock();
- bool try_lock();
- template <class Rep, class Period>
- bool try_lock_for(const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock();
- // Shared ownership
- void lock_shared();
- bool try_lock_shared();
- template <class Rep, class Period>
- bool
- try_lock_shared_for(const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_shared();
- };
- class upgrade_mutex
- {
- public:
- upgrade_mutex();
- ~upgrade_mutex();
- upgrade_mutex(const upgrade_mutex&) = delete;
- upgrade_mutex& operator=(const upgrade_mutex&) = delete;
- // Exclusive ownership
- void lock();
- bool try_lock();
- template <class Rep, class Period>
- bool try_lock_for(const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock();
- // Shared ownership
- void lock_shared();
- bool try_lock_shared();
- template <class Rep, class Period>
- bool
- try_lock_shared_for(const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_shared();
- // Upgrade ownership
- void lock_upgrade();
- bool try_lock_upgrade();
- template <class Rep, class Period>
- bool
- try_lock_upgrade_for(
- const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_upgrade();
- // Shared <-> Exclusive
- bool try_unlock_shared_and_lock();
- template <class Rep, class Period>
- bool
- try_unlock_shared_and_lock_for(
- const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_unlock_shared_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_and_lock_shared();
- // Shared <-> Upgrade
- bool try_unlock_shared_and_lock_upgrade();
- template <class Rep, class Period>
- bool
- try_unlock_shared_and_lock_upgrade_for(
- const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_unlock_shared_and_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_upgrade_and_lock_shared();
- // Upgrade <-> Exclusive
- void unlock_upgrade_and_lock();
- bool try_unlock_upgrade_and_lock();
- template <class Rep, class Period>
- bool
- try_unlock_upgrade_and_lock_for(
- const boost::chrono::duration<Rep, Period>& rel_time);
- template <class Clock, class Duration>
- bool
- try_unlock_upgrade_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- void unlock_and_lock_upgrade();
- };
- } // thread_v2
- } // boost
- */
- #include <boost/thread/detail/config.hpp>
- #include <boost/thread/mutex.hpp>
- #include <boost/thread/condition_variable.hpp>
- #include <boost/thread/mutex.hpp>
- #include <boost/chrono.hpp>
- #include <climits>
- #include <boost/system/system_error.hpp>
- #define BOOST_THREAD_INLINE inline
- namespace boost {
- namespace thread_v2 {
- class shared_mutex
- {
- typedef ::boost::mutex mutex_t;
- typedef ::boost::condition_variable cond_t;
- typedef unsigned count_t;
- mutex_t mut_;
- cond_t gate1_;
- cond_t gate2_;
- count_t state_;
- static const count_t write_entered_ = 1U << (sizeof(count_t)*CHAR_BIT - 1);
- static const count_t n_readers_ = ~write_entered_;
- public:
- BOOST_THREAD_INLINE shared_mutex();
- BOOST_THREAD_INLINE ~shared_mutex();
- #ifndef BOOST_NO_CXX11_DELETED_FUNCTIONS
- shared_mutex(shared_mutex const&) = delete;
- shared_mutex& operator=(shared_mutex const&) = delete;
- #else // BOOST_NO_CXX11_DELETED_FUNCTIONS
- private:
- shared_mutex(shared_mutex const&);
- shared_mutex& operator=(shared_mutex const&);
- public:
- #endif // BOOST_NO_CXX11_DELETED_FUNCTIONS
- // Exclusive ownership
- BOOST_THREAD_INLINE void lock();
- BOOST_THREAD_INLINE bool try_lock();
- template <class Rep, class Period>
- bool try_lock_for(const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_lock_until(boost::chrono::steady_clock::now() + rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock();
- // Shared ownership
- BOOST_THREAD_INLINE void lock_shared();
- BOOST_THREAD_INLINE bool try_lock_shared();
- template <class Rep, class Period>
- bool
- try_lock_shared_for(const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_lock_shared_until(boost::chrono::steady_clock::now() +
- rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_shared();
- #if defined BOOST_THREAD_USES_DATETIME
- bool timed_lock(system_time const& timeout);
- template<typename TimeDuration>
- bool timed_lock(TimeDuration const & relative_time)
- {
- return timed_lock(get_system_time()+relative_time);
- }
- bool timed_lock_shared(system_time const& timeout);
- template<typename TimeDuration>
- bool timed_lock_shared(TimeDuration const & relative_time)
- {
- return timed_lock_shared(get_system_time()+relative_time);
- }
- #endif
- };
- template <class Clock, class Duration>
- bool
- shared_mutex::try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ & write_entered_)
- {
- while (true)
- {
- boost::cv_status status = gate1_.wait_until(lk, abs_time);
- if ((state_ & write_entered_) == 0)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- state_ |= write_entered_;
- if (state_ & n_readers_)
- {
- while (true)
- {
- boost::cv_status status = gate2_.wait_until(lk, abs_time);
- if ((state_ & n_readers_) == 0)
- break;
- if (status == boost::cv_status::timeout)
- {
- state_ &= ~write_entered_;
- return false;
- }
- }
- }
- return true;
- }
- template <class Clock, class Duration>
- bool
- shared_mutex::try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & write_entered_) || (state_ & n_readers_) == n_readers_)
- {
- while (true)
- {
- boost::cv_status status = gate1_.wait_until(lk, abs_time);
- if ((state_ & write_entered_) == 0 &&
- (state_ & n_readers_) < n_readers_)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- return true;
- }
- #if defined BOOST_THREAD_USES_DATETIME
- bool shared_mutex::timed_lock(system_time const& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ & write_entered_)
- {
- while (true)
- {
- bool status = gate1_.timed_wait(lk, abs_time);
- if ((state_ & write_entered_) == 0)
- break;
- if (!status)
- return false;
- }
- }
- state_ |= write_entered_;
- if (state_ & n_readers_)
- {
- while (true)
- {
- bool status = gate2_.timed_wait(lk, abs_time);
- if ((state_ & n_readers_) == 0)
- break;
- if (!status)
- {
- state_ &= ~write_entered_;
- return false;
- }
- }
- }
- return true;
- }
- bool shared_mutex::timed_lock_shared(system_time const& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ & write_entered_)
- {
- while (true)
- {
- bool status = gate1_.timed_wait(lk, abs_time);
- if ((state_ & write_entered_) == 0)
- break;
- if (!status )
- return false;
- }
- }
- state_ |= write_entered_;
- if (state_ & n_readers_)
- {
- while (true)
- {
- bool status = gate2_.timed_wait(lk, abs_time);
- if ((state_ & n_readers_) == 0)
- break;
- if (!status)
- {
- state_ &= ~write_entered_;
- return false;
- }
- }
- }
- return true;
- }
- #endif
- class upgrade_mutex
- {
- typedef boost::mutex mutex_t;
- typedef boost::condition_variable cond_t;
- typedef unsigned count_t;
- mutex_t mut_;
- cond_t gate1_;
- cond_t gate2_;
- count_t state_;
- static const unsigned write_entered_ = 1U << (sizeof(count_t)*CHAR_BIT - 1);
- static const unsigned upgradable_entered_ = write_entered_ >> 1;
- static const unsigned n_readers_ = ~(write_entered_ | upgradable_entered_);
- public:
- BOOST_THREAD_INLINE upgrade_mutex();
- BOOST_THREAD_INLINE ~upgrade_mutex();
- #ifndef BOOST_CXX11_NO_DELETED_FUNCTIONS
- upgrade_mutex(const upgrade_mutex&) = delete;
- upgrade_mutex& operator=(const upgrade_mutex&) = delete;
- #else // BOOST_CXX11_NO_DELETED_FUNCTIONS
- private:
- upgrade_mutex(const upgrade_mutex&);
- upgrade_mutex& operator=(const upgrade_mutex&);
- public:
- #endif // BOOST_CXX11_NO_DELETED_FUNCTIONS
- // Exclusive ownership
- BOOST_THREAD_INLINE void lock();
- BOOST_THREAD_INLINE bool try_lock();
- template <class Rep, class Period>
- bool try_lock_for(const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_lock_until(boost::chrono::steady_clock::now() + rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock();
- // Shared ownership
- BOOST_THREAD_INLINE void lock_shared();
- BOOST_THREAD_INLINE bool try_lock_shared();
- template <class Rep, class Period>
- bool
- try_lock_shared_for(const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_lock_shared_until(boost::chrono::steady_clock::now() +
- rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_shared();
- // Upgrade ownership
- BOOST_THREAD_INLINE void lock_upgrade();
- BOOST_THREAD_INLINE bool try_lock_upgrade();
- template <class Rep, class Period>
- bool
- try_lock_upgrade_for(
- const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_lock_upgrade_until(boost::chrono::steady_clock::now() +
- rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_upgrade();
- // Shared <-> Exclusive
- BOOST_THREAD_INLINE bool try_unlock_shared_and_lock();
- template <class Rep, class Period>
- bool
- try_unlock_shared_and_lock_for(
- const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_unlock_shared_and_lock_until(
- boost::chrono::steady_clock::now() + rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_unlock_shared_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_and_lock_shared();
- // Shared <-> Upgrade
- BOOST_THREAD_INLINE bool try_unlock_shared_and_lock_upgrade();
- template <class Rep, class Period>
- bool
- try_unlock_shared_and_lock_upgrade_for(
- const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_unlock_shared_and_lock_upgrade_until(
- boost::chrono::steady_clock::now() + rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_unlock_shared_and_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_upgrade_and_lock_shared();
- // Upgrade <-> Exclusive
- BOOST_THREAD_INLINE void unlock_upgrade_and_lock();
- BOOST_THREAD_INLINE bool try_unlock_upgrade_and_lock();
- template <class Rep, class Period>
- bool
- try_unlock_upgrade_and_lock_for(
- const boost::chrono::duration<Rep, Period>& rel_time)
- {
- return try_unlock_upgrade_and_lock_until(
- boost::chrono::steady_clock::now() + rel_time);
- }
- template <class Clock, class Duration>
- bool
- try_unlock_upgrade_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time);
- BOOST_THREAD_INLINE void unlock_and_lock_upgrade();
- #if defined BOOST_THREAD_USES_DATETIME
- inline bool timed_lock(system_time const& abs_time);
- template<typename TimeDuration>
- bool timed_lock(TimeDuration const & relative_time)
- {
- return timed_lock(get_system_time()+relative_time);
- }
- inline bool timed_lock_shared(system_time const& abs_time);
- template<typename TimeDuration>
- bool timed_lock_shared(TimeDuration const & relative_time)
- {
- return timed_lock_shared(get_system_time()+relative_time);
- }
- inline bool timed_lock_upgrade(system_time const& abs_time);
- template<typename TimeDuration>
- bool timed_lock_upgrade(TimeDuration const & relative_time)
- {
- return timed_lock_upgrade(get_system_time()+relative_time);
- }
- #endif
- };
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ & (write_entered_ | upgradable_entered_))
- {
- while (true)
- {
- boost::cv_status status = gate1_.wait_until(lk, abs_time);
- if ((state_ & (write_entered_ | upgradable_entered_)) == 0)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- state_ |= write_entered_;
- if (state_ & n_readers_)
- {
- while (true)
- {
- boost::cv_status status = gate2_.wait_until(lk, abs_time);
- if ((state_ & n_readers_) == 0)
- break;
- if (status == boost::cv_status::timeout)
- {
- state_ &= ~write_entered_;
- return false;
- }
- }
- }
- return true;
- }
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_lock_shared_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & write_entered_) || (state_ & n_readers_) == n_readers_)
- {
- while (true)
- {
- boost::cv_status status = gate1_.wait_until(lk, abs_time);
- if ((state_ & write_entered_) == 0 &&
- (state_ & n_readers_) < n_readers_)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- return true;
- }
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & (write_entered_ | upgradable_entered_)) ||
- (state_ & n_readers_) == n_readers_)
- {
- while (true)
- {
- boost::cv_status status = gate1_.wait_until(lk, abs_time);
- if ((state_ & (write_entered_ | upgradable_entered_)) == 0 &&
- (state_ & n_readers_) < n_readers_)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= upgradable_entered_ | num_readers;
- return true;
- }
- #if defined BOOST_THREAD_USES_DATETIME
- bool upgrade_mutex::timed_lock(system_time const& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ & (write_entered_ | upgradable_entered_))
- {
- while (true)
- {
- bool status = gate1_.timed_wait(lk, abs_time);
- if ((state_ & (write_entered_ | upgradable_entered_)) == 0)
- break;
- if (!status)
- return false;
- }
- }
- state_ |= write_entered_;
- if (state_ & n_readers_)
- {
- while (true)
- {
- bool status = gate2_.timed_wait(lk, abs_time);
- if ((state_ & n_readers_) == 0)
- break;
- if (!status)
- {
- state_ &= ~write_entered_;
- return false;
- }
- }
- }
- return true;
- }
- bool upgrade_mutex::timed_lock_shared(system_time const& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & write_entered_) || (state_ & n_readers_) == n_readers_)
- {
- while (true)
- {
- bool status = gate1_.timed_wait(lk, abs_time);
- if ((state_ & write_entered_) == 0 &&
- (state_ & n_readers_) < n_readers_)
- break;
- if (!status)
- return false;
- }
- }
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- return true;
- }
- bool upgrade_mutex::timed_lock_upgrade(system_time const& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & (write_entered_ | upgradable_entered_)) ||
- (state_ & n_readers_) == n_readers_)
- {
- while (true)
- {
- bool status = gate1_.timed_wait(lk, abs_time);
- if ((state_ & (write_entered_ | upgradable_entered_)) == 0 &&
- (state_ & n_readers_) < n_readers_)
- break;
- if (!status)
- return false;
- }
- }
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= upgradable_entered_ | num_readers;
- return true;
- }
- #endif
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_unlock_shared_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ != 1)
- {
- while (true)
- {
- boost::cv_status status = gate2_.wait_until(lk, abs_time);
- if (state_ == 1)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- state_ = write_entered_;
- return true;
- }
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_unlock_shared_and_lock_upgrade_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & (write_entered_ | upgradable_entered_)) != 0)
- {
- while (true)
- {
- boost::cv_status status = gate2_.wait_until(lk, abs_time);
- if ((state_ & (write_entered_ | upgradable_entered_)) == 0)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- state_ |= upgradable_entered_;
- return true;
- }
- template <class Clock, class Duration>
- bool
- upgrade_mutex::try_unlock_upgrade_and_lock_until(
- const boost::chrono::time_point<Clock, Duration>& abs_time)
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if ((state_ & n_readers_) != 1)
- {
- while (true)
- {
- boost::cv_status status = gate2_.wait_until(lk, abs_time);
- if ((state_ & n_readers_) == 1)
- break;
- if (status == boost::cv_status::timeout)
- return false;
- }
- }
- state_ = write_entered_;
- return true;
- }
- //////
- // shared_mutex
- shared_mutex::shared_mutex()
- : state_(0)
- {
- }
- shared_mutex::~shared_mutex()
- {
- boost::lock_guard<mutex_t> _(mut_);
- }
- // Exclusive ownership
- void
- shared_mutex::lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- while (state_ & write_entered_)
- gate1_.wait(lk);
- state_ |= write_entered_;
- while (state_ & n_readers_)
- gate2_.wait(lk);
- }
- bool
- shared_mutex::try_lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ == 0)
- {
- state_ = write_entered_;
- return true;
- }
- return false;
- }
- void
- shared_mutex::unlock()
- {
- boost::lock_guard<mutex_t> _(mut_);
- state_ = 0;
- gate1_.notify_all();
- }
- // Shared ownership
- void
- shared_mutex::lock_shared()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- while ((state_ & write_entered_) || (state_ & n_readers_) == n_readers_)
- gate1_.wait(lk);
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- }
- bool
- shared_mutex::try_lock_shared()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- count_t num_readers = state_ & n_readers_;
- if (!(state_ & write_entered_) && num_readers != n_readers_)
- {
- ++num_readers;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- return true;
- }
- return false;
- }
- void
- shared_mutex::unlock_shared()
- {
- boost::lock_guard<mutex_t> _(mut_);
- count_t num_readers = (state_ & n_readers_) - 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- if (state_ & write_entered_)
- {
- if (num_readers == 0)
- gate2_.notify_one();
- }
- else
- {
- if (num_readers == n_readers_ - 1)
- gate1_.notify_one();
- }
- }
- // upgrade_mutex
- upgrade_mutex::upgrade_mutex()
- : gate1_(),
- gate2_(),
- state_(0)
- {
- }
- upgrade_mutex::~upgrade_mutex()
- {
- boost::lock_guard<mutex_t> _(mut_);
- }
- // Exclusive ownership
- void
- upgrade_mutex::lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- while (state_ & (write_entered_ | upgradable_entered_))
- gate1_.wait(lk);
- state_ |= write_entered_;
- while (state_ & n_readers_)
- gate2_.wait(lk);
- }
- bool
- upgrade_mutex::try_lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ == 0)
- {
- state_ = write_entered_;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock()
- {
- boost::lock_guard<mutex_t> _(mut_);
- state_ = 0;
- gate1_.notify_all();
- }
- // Shared ownership
- void
- upgrade_mutex::lock_shared()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- while ((state_ & write_entered_) || (state_ & n_readers_) == n_readers_)
- gate1_.wait(lk);
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- }
- bool
- upgrade_mutex::try_lock_shared()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- count_t num_readers = state_ & n_readers_;
- if (!(state_ & write_entered_) && num_readers != n_readers_)
- {
- ++num_readers;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock_shared()
- {
- boost::lock_guard<mutex_t> _(mut_);
- count_t num_readers = (state_ & n_readers_) - 1;
- state_ &= ~n_readers_;
- state_ |= num_readers;
- if (state_ & write_entered_)
- {
- if (num_readers == 0)
- gate2_.notify_one();
- }
- else
- {
- if (num_readers == n_readers_ - 1)
- gate1_.notify_one();
- }
- }
- // Upgrade ownership
- void
- upgrade_mutex::lock_upgrade()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- while ((state_ & (write_entered_ | upgradable_entered_)) ||
- (state_ & n_readers_) == n_readers_)
- gate1_.wait(lk);
- count_t num_readers = (state_ & n_readers_) + 1;
- state_ &= ~n_readers_;
- state_ |= upgradable_entered_ | num_readers;
- }
- bool
- upgrade_mutex::try_lock_upgrade()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- count_t num_readers = state_ & n_readers_;
- if (!(state_ & (write_entered_ | upgradable_entered_))
- && num_readers != n_readers_)
- {
- ++num_readers;
- state_ &= ~n_readers_;
- state_ |= upgradable_entered_ | num_readers;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock_upgrade()
- {
- {
- boost::lock_guard<mutex_t> _(mut_);
- count_t num_readers = (state_ & n_readers_) - 1;
- state_ &= ~(upgradable_entered_ | n_readers_);
- state_ |= num_readers;
- }
- gate1_.notify_all();
- }
- // Shared <-> Exclusive
- bool
- upgrade_mutex::try_unlock_shared_and_lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ == 1)
- {
- state_ = write_entered_;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock_and_lock_shared()
- {
- {
- boost::lock_guard<mutex_t> _(mut_);
- state_ = 1;
- }
- gate1_.notify_all();
- }
- // Shared <-> Upgrade
- bool
- upgrade_mutex::try_unlock_shared_and_lock_upgrade()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (!(state_ & (write_entered_ | upgradable_entered_)))
- {
- state_ |= upgradable_entered_;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock_upgrade_and_lock_shared()
- {
- {
- boost::lock_guard<mutex_t> _(mut_);
- state_ &= ~upgradable_entered_;
- }
- gate1_.notify_all();
- }
- // Upgrade <-> Exclusive
- void
- upgrade_mutex::unlock_upgrade_and_lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- count_t num_readers = (state_ & n_readers_) - 1;
- state_ &= ~(upgradable_entered_ | n_readers_);
- state_ |= write_entered_ | num_readers;
- while (state_ & n_readers_)
- gate2_.wait(lk);
- }
- bool
- upgrade_mutex::try_unlock_upgrade_and_lock()
- {
- boost::unique_lock<mutex_t> lk(mut_);
- if (state_ == (upgradable_entered_ | 1))
- {
- state_ = write_entered_;
- return true;
- }
- return false;
- }
- void
- upgrade_mutex::unlock_and_lock_upgrade()
- {
- {
- boost::lock_guard<mutex_t> _(mut_);
- state_ = upgradable_entered_ | 1;
- }
- gate1_.notify_all();
- }
- } // thread_v2
- } // boost
- namespace boost {
- //using thread_v2::shared_mutex;
- using thread_v2::upgrade_mutex;
- typedef thread_v2::upgrade_mutex shared_mutex;
- }
- #endif
|