123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746 |
- // Copyright (C) 2012 Vicente J. Botet Escriba
- //
- // 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)
- #define BOOST_THREAD_PROVIDES_SHARED_MUTEX_UPWARDS_CONVERSIONS
- #define BOOST_THREAD_PROVIDES_EXPLICIT_LOCK_CONVERSION
- #define BOOST_THREAD_PROVIDES_GENERIC_SHARED_MUTEX_ON_WIN
- #include <iostream>
- #include <boost/thread/mutex.hpp>
- #include <boost/thread/shared_mutex.hpp>
- #include <boost/thread/lock_algorithms.hpp>
- #include <boost/thread/thread_only.hpp>
- #include <vector>
- #if defined BOOST_THREAD_USES_CHRONO
- #include <boost/chrono/chrono_io.hpp>
- enum {reading, writing};
- int state = reading;
- #if 1
- boost::mutex&
- cout_mut()
- {
- static boost::mutex m;
- return m;
- }
- void
- print(const char* tag, unsigned count, char ch)
- {
- boost::lock_guard<boost::mutex> _(cout_mut());
- std::cout << tag << count << ch;
- }
- #elif 0
- boost::recursive_mutex&
- cout_mut()
- {
- static boost::recursive_mutex m;
- return m;
- }
- void print() {}
- template <class A0, class ...Args>
- void
- print(const A0& a0, const Args& ...args)
- {
- boost::lock_guard<boost::recursive_mutex> _(cout_mut());
- std::cout << a0;
- print(args...);
- }
- #else
- template <class A0, class A1, class A2>
- void
- print(const A0&, const A1& a1, const A2&)
- {
- assert(a1 > 10000);
- }
- #endif
- namespace S
- {
- boost::shared_mutex mut;
- void reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock_shared();
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- print("reader = ", count, '\n');
- }
- void writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock();
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- print("writer = ", count, '\n');
- }
- void try_reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared())
- {
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- }
- print("try_reader = ", count, '\n');
- }
- void try_writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock())
- {
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- }
- print("try_writer = ", count, '\n');
- }
- void try_for_reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- }
- print("try_for_reader = ", count, '\n');
- }
- void try_for_writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_for(boost::chrono::microseconds(5)))
- {
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- }
- print("try_for_writer = ", count, '\n');
- }
- void
- test_shared_mutex()
- {
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(reader);
- boost::thread t2(writer);
- boost::thread t3(reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(try_reader);
- boost::thread t2(try_writer);
- boost::thread t3(try_reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(try_for_reader);
- boost::thread t2(try_for_writer);
- boost::thread t3(try_for_reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- }
- }
- namespace U
- {
- boost::upgrade_mutex mut;
- void reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock_shared();
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- print("reader = ", count, '\n');
- }
- void writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock();
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- print("writer = ", count, '\n');
- }
- void try_reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared())
- {
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- }
- print("try_reader = ", count, '\n');
- }
- void try_writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock())
- {
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- }
- print("try_writer = ", count, '\n');
- }
- void try_for_reader()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- ++count;
- mut.unlock_shared();
- }
- }
- print("try_for_reader = ", count, '\n');
- }
- void try_for_writer()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_for(boost::chrono::microseconds(5)))
- {
- state = writing;
- assert(state == writing);
- state = reading;
- ++count;
- mut.unlock();
- }
- }
- print("try_for_writer = ", count, '\n');
- }
- void upgradable()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock_upgrade();
- assert(state == reading);
- ++count;
- mut.unlock_upgrade();
- }
- print("upgradable = ", count, '\n');
- }
- void try_upgradable()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_upgrade())
- {
- assert(state == reading);
- ++count;
- mut.unlock_upgrade();
- }
- }
- print("try_upgradable = ", count, '\n');
- }
- void try_for_upgradable()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_upgrade_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- ++count;
- mut.unlock_upgrade();
- }
- }
- print("try_for_upgradable = ", count, '\n');
- }
- void clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock_shared();
- assert(state == reading);
- if (mut.try_unlock_shared_and_lock())
- {
- state = writing;
- }
- else if (mut.try_unlock_shared_and_lock_upgrade())
- {
- assert(state == reading);
- mut.unlock_upgrade_and_lock();
- state = writing;
- }
- else
- {
- mut.unlock_shared();
- continue;
- }
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_upgrade();
- assert(state == reading);
- mut.unlock_upgrade_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- print("clockwise = ", count, '\n');
- }
- void counter_clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- mut.lock_upgrade();
- assert(state == reading);
- mut.unlock_upgrade_and_lock();
- assert(state == reading);
- state = writing;
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- print("counter_clockwise = ", count, '\n');
- }
- void try_clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared())
- {
- assert(state == reading);
- if (mut.try_unlock_shared_and_lock())
- {
- state = writing;
- }
- else if (mut.try_unlock_shared_and_lock_upgrade())
- {
- assert(state == reading);
- mut.unlock_upgrade_and_lock();
- state = writing;
- }
- else
- {
- mut.unlock_shared();
- continue;
- }
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_upgrade();
- assert(state == reading);
- mut.unlock_upgrade_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- }
- print("try_clockwise = ", count, '\n');
- }
- void try_for_clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_shared_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- if (mut.try_unlock_shared_and_lock_for(boost::chrono::microseconds(5)))
- {
- state = writing;
- }
- else if (mut.try_unlock_shared_and_lock_upgrade_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- mut.unlock_upgrade_and_lock();
- state = writing;
- }
- else
- {
- mut.unlock_shared();
- continue;
- }
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_upgrade();
- assert(state == reading);
- mut.unlock_upgrade_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- }
- print("try_for_clockwise = ", count, '\n');
- }
- void try_counter_clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_upgrade())
- {
- assert(state == reading);
- if (mut.try_unlock_upgrade_and_lock())
- {
- assert(state == reading);
- state = writing;
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- else
- {
- mut.unlock_upgrade();
- }
- }
- }
- print("try_counter_clockwise = ", count, '\n');
- }
- void try_for_counter_clockwise()
- {
- typedef boost::chrono::steady_clock Clock;
- unsigned count = 0;
- Clock::time_point until = Clock::now() + boost::chrono::seconds(3);
- while (Clock::now() < until)
- {
- if (mut.try_lock_upgrade_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- if (mut.try_unlock_upgrade_and_lock_for(boost::chrono::microseconds(5)))
- {
- assert(state == reading);
- state = writing;
- assert(state == writing);
- state = reading;
- mut.unlock_and_lock_shared();
- assert(state == reading);
- mut.unlock_shared();
- ++count;
- }
- else
- {
- mut.unlock_upgrade();
- }
- }
- }
- print("try_for_counter_clockwise = ", count, '\n');
- }
- void
- test_upgrade_mutex()
- {
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(reader);
- boost::thread t2(writer);
- boost::thread t3(reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(try_reader);
- boost::thread t2(try_writer);
- boost::thread t3(try_reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(try_for_reader);
- boost::thread t2(try_for_writer);
- boost::thread t3(try_for_reader);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(reader);
- boost::thread t2(writer);
- boost::thread t3(upgradable);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(reader);
- boost::thread t2(writer);
- boost::thread t3(try_upgradable);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- boost::thread t1(reader);
- boost::thread t2(writer);
- boost::thread t3(try_for_upgradable);
- t1.join();
- t2.join();
- t3.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- state = reading;
- boost::thread t1(clockwise);
- boost::thread t2(counter_clockwise);
- boost::thread t3(clockwise);
- boost::thread t4(counter_clockwise);
- t1.join();
- t2.join();
- t3.join();
- t4.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- {
- state = reading;
- boost::thread t1(try_clockwise);
- boost::thread t2(try_counter_clockwise);
- t1.join();
- t2.join();
- }
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- // {
- // state = reading;
- // boost::thread t1(try_for_clockwise);
- // boost::thread t2(try_for_counter_clockwise);
- // t1.join();
- // t2.join();
- // }
- // std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- }
- }
- namespace Assignment
- {
- class A
- {
- typedef boost::upgrade_mutex mutex_type;
- typedef boost::shared_lock<mutex_type> SharedLock;
- typedef boost::upgrade_lock<mutex_type> UpgradeLock;
- typedef boost::unique_lock<mutex_type> Lock;
- mutable mutex_type mut_;
- std::vector<double> data_;
- public:
- A(const A& a)
- {
- SharedLock _(a.mut_);
- data_ = a.data_;
- }
- A& operator=(const A& a)
- {
- if (this != &a)
- {
- Lock this_lock(mut_, boost::defer_lock);
- SharedLock that_lock(a.mut_, boost::defer_lock);
- boost::lock(this_lock, that_lock);
- data_ = a.data_;
- }
- return *this;
- }
- void swap(A& a)
- {
- Lock this_lock(mut_, boost::defer_lock);
- Lock that_lock(a.mut_, boost::defer_lock);
- boost::lock(this_lock, that_lock);
- data_.swap(a.data_);
- }
- void average(A& a)
- {
- assert(data_.size() == a.data_.size());
- assert(this != &a);
- Lock this_lock(mut_, boost::defer_lock);
- UpgradeLock share_that_lock(a.mut_, boost::defer_lock);
- boost::lock(this_lock, share_that_lock);
- for (unsigned i = 0; i < data_.size(); ++i)
- data_[i] = (data_[i] + a.data_[i]) / 2;
- SharedLock share_this_lock(boost::move(this_lock));
- Lock that_lock(boost::move(share_that_lock));
- a.data_ = data_;
- }
- };
- } // Assignment
- void temp()
- {
- using namespace boost;
- static upgrade_mutex mut;
- unique_lock<upgrade_mutex> ul(mut);
- shared_lock<upgrade_mutex> sl;
- sl = BOOST_THREAD_MAKE_RV_REF(shared_lock<upgrade_mutex>(boost::move(ul)));
- }
- int main()
- {
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- typedef boost::chrono::high_resolution_clock Clock;
- typedef boost::chrono::duration<double> sec;
- Clock::time_point t0 = Clock::now();
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- S::test_shared_mutex();
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- U::test_upgrade_mutex();
- std::cout << __FILE__ << "[" <<__LINE__ << "]" << std::endl;
- Clock::time_point t1 = Clock::now();
- std::cout << sec(t1 - t0) << '\n';
- return 0;
- }
- #else
- #error "This platform doesn't support Boost.Chrono"
- #endif
|