123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824 |
- #ifndef BOOST_LOCKFREE_STACK_HPP_INCLUDED
- #define BOOST_LOCKFREE_STACK_HPP_INCLUDED
- #include <boost/assert.hpp>
- #include <boost/checked_delete.hpp>
- #include <boost/core/allocator_access.hpp>
- #include <boost/core/no_exceptions_support.hpp>
- #include <boost/integer_traits.hpp>
- #include <boost/static_assert.hpp>
- #include <boost/tuple/tuple.hpp>
- #include <boost/type_traits/is_copy_constructible.hpp>
- #include <boost/lockfree/detail/atomic.hpp>
- #include <boost/lockfree/detail/copy_payload.hpp>
- #include <boost/lockfree/detail/freelist.hpp>
- #include <boost/lockfree/detail/parameter.hpp>
- #include <boost/lockfree/detail/tagged_ptr.hpp>
- #include <boost/lockfree/lockfree_forward.hpp>
- #ifdef BOOST_HAS_PRAGMA_ONCE
- #pragma once
- #endif
- namespace boost {
- namespace lockfree {
- namespace detail {
- typedef parameter::parameters<boost::parameter::optional<tag::allocator>,
- boost::parameter::optional<tag::capacity>
- > stack_signature;
- }
- #ifdef BOOST_NO_CXX11_VARIADIC_TEMPLATES
- template <typename T, class A0, class A1, class A2>
- #else
- template <typename T, typename ...Options>
- #endif
- class stack
- {
- private:
- #ifndef BOOST_DOXYGEN_INVOKED
- BOOST_STATIC_ASSERT(boost::is_copy_constructible<T>::value);
- #ifdef BOOST_NO_CXX11_VARIADIC_TEMPLATES
- typedef typename detail::stack_signature::bind<A0, A1, A2>::type bound_args;
- #else
- typedef typename detail::stack_signature::bind<Options...>::type bound_args;
- #endif
- static const bool has_capacity = detail::extract_capacity<bound_args>::has_capacity;
- static const size_t capacity = detail::extract_capacity<bound_args>::capacity;
- static const bool fixed_sized = detail::extract_fixed_sized<bound_args>::value;
- static const bool node_based = !(has_capacity || fixed_sized);
- static const bool compile_time_sized = has_capacity;
- struct node
- {
- node(T const & val):
- v(val)
- {}
- typedef typename detail::select_tagged_handle<node, node_based>::handle_type handle_t;
- handle_t next;
- const T v;
- };
- typedef typename detail::extract_allocator<bound_args, node>::type node_allocator;
- typedef typename detail::select_freelist<node, node_allocator, compile_time_sized, fixed_sized, capacity>::type pool_t;
- typedef typename pool_t::tagged_node_handle tagged_node_handle;
-
- BOOST_STATIC_ASSERT((mpl::if_c<has_capacity,
- mpl::bool_<capacity - 1 < boost::integer_traits<boost::uint16_t>::const_max>,
- mpl::true_
- >::type::value));
- struct implementation_defined
- {
- typedef node_allocator allocator;
- typedef std::size_t size_type;
- };
- #endif
- BOOST_DELETED_FUNCTION(stack(stack const&))
- BOOST_DELETED_FUNCTION(stack& operator= (stack const&))
- public:
- typedef T value_type;
- typedef typename implementation_defined::allocator allocator;
- typedef typename implementation_defined::size_type size_type;
-
- bool is_lock_free (void) const
- {
- return tos.is_lock_free() && pool.is_lock_free();
- }
-
- stack(void):
- pool(node_allocator(), capacity)
- {
-
-
- BOOST_ASSERT(has_capacity);
- initialize();
- }
-
- template <typename U>
- explicit stack(typename boost::allocator_rebind<node_allocator, U>::type const & alloc):
- pool(alloc, capacity)
- {
- BOOST_STATIC_ASSERT(has_capacity);
- initialize();
- }
-
- explicit stack(allocator const & alloc):
- pool(alloc, capacity)
- {
-
-
- BOOST_ASSERT(has_capacity);
- initialize();
- }
-
- explicit stack(size_type n):
- pool(node_allocator(), n)
- {
-
-
- BOOST_ASSERT(!has_capacity);
- initialize();
- }
-
- template <typename U>
- stack(size_type n, typename boost::allocator_rebind<node_allocator, U>::type const & alloc):
- pool(alloc, n)
- {
- BOOST_STATIC_ASSERT(!has_capacity);
- initialize();
- }
-
- void reserve(size_type n)
- {
-
-
- BOOST_ASSERT(!has_capacity);
- pool.template reserve<true>(n);
- }
-
- void reserve_unsafe(size_type n)
- {
-
-
- BOOST_ASSERT(!has_capacity);
- pool.template reserve<false>(n);
- }
-
- ~stack(void)
- {
- detail::consume_noop consume_functor;
- (void)consume_all(consume_functor);
- }
- private:
- #ifndef BOOST_DOXYGEN_INVOKED
- void initialize(void)
- {
- tos.store(tagged_node_handle(pool.null_handle(), 0));
- }
- void link_nodes_atomic(node * new_top_node, node * end_node)
- {
- tagged_node_handle old_tos = tos.load(detail::memory_order_relaxed);
- for (;;) {
- tagged_node_handle new_tos (pool.get_handle(new_top_node), old_tos.get_tag());
- end_node->next = pool.get_handle(old_tos);
- if (tos.compare_exchange_weak(old_tos, new_tos))
- break;
- }
- }
- void link_nodes_unsafe(node * new_top_node, node * end_node)
- {
- tagged_node_handle old_tos = tos.load(detail::memory_order_relaxed);
- tagged_node_handle new_tos (pool.get_handle(new_top_node), old_tos.get_tag());
- end_node->next = pool.get_handle(old_tos);
- tos.store(new_tos, memory_order_relaxed);
- }
- template <bool Threadsafe, bool Bounded, typename ConstIterator>
- tuple<node*, node*> prepare_node_list(ConstIterator begin, ConstIterator end, ConstIterator & ret)
- {
- ConstIterator it = begin;
- node * end_node = pool.template construct<Threadsafe, Bounded>(*it++);
- if (end_node == NULL) {
- ret = begin;
- return make_tuple<node*, node*>(NULL, NULL);
- }
- node * new_top_node = end_node;
- end_node->next = NULL;
- BOOST_TRY {
-
- for (; it != end; ++it) {
- node * newnode = pool.template construct<Threadsafe, Bounded>(*it);
- if (newnode == NULL)
- break;
- newnode->next = new_top_node;
- new_top_node = newnode;
- }
- } BOOST_CATCH (...) {
- for (node * current_node = new_top_node; current_node != NULL;) {
- node * next = current_node->next;
- pool.template destruct<Threadsafe>(current_node);
- current_node = next;
- }
- BOOST_RETHROW;
- }
- BOOST_CATCH_END
- ret = it;
- return make_tuple(new_top_node, end_node);
- }
- #endif
- public:
-
- bool push(T const & v)
- {
- return do_push<false>(v);
- }
-
- bool bounded_push(T const & v)
- {
- return do_push<true>(v);
- }
- #ifndef BOOST_DOXYGEN_INVOKED
- private:
- template <bool Bounded>
- bool do_push(T const & v)
- {
- node * newnode = pool.template construct<true, Bounded>(v);
- if (newnode == 0)
- return false;
- link_nodes_atomic(newnode, newnode);
- return true;
- }
- template <bool Bounded, typename ConstIterator>
- ConstIterator do_push(ConstIterator begin, ConstIterator end)
- {
- node * new_top_node;
- node * end_node;
- ConstIterator ret;
- tie(new_top_node, end_node) = prepare_node_list<true, Bounded>(begin, end, ret);
- if (new_top_node)
- link_nodes_atomic(new_top_node, end_node);
- return ret;
- }
- public:
- #endif
-
- template <typename ConstIterator>
- ConstIterator push(ConstIterator begin, ConstIterator end)
- {
- return do_push<false, ConstIterator>(begin, end);
- }
-
- template <typename ConstIterator>
- ConstIterator bounded_push(ConstIterator begin, ConstIterator end)
- {
- return do_push<true, ConstIterator>(begin, end);
- }
-
- bool unsynchronized_push(T const & v)
- {
- node * newnode = pool.template construct<false, false>(v);
- if (newnode == 0)
- return false;
- link_nodes_unsafe(newnode, newnode);
- return true;
- }
-
- template <typename ConstIterator>
- ConstIterator unsynchronized_push(ConstIterator begin, ConstIterator end)
- {
- node * new_top_node;
- node * end_node;
- ConstIterator ret;
- tie(new_top_node, end_node) = prepare_node_list<false, false>(begin, end, ret);
- if (new_top_node)
- link_nodes_unsafe(new_top_node, end_node);
- return ret;
- }
-
- bool pop(T & ret)
- {
- return pop<T>(ret);
- }
-
- template <typename U>
- bool pop(U & ret)
- {
- BOOST_STATIC_ASSERT((boost::is_convertible<T, U>::value));
- detail::consume_via_copy<U> consumer(ret);
- return consume_one(consumer);
- }
-
- bool unsynchronized_pop(T & ret)
- {
- return unsynchronized_pop<T>(ret);
- }
-
- template <typename U>
- bool unsynchronized_pop(U & ret)
- {
- BOOST_STATIC_ASSERT((boost::is_convertible<T, U>::value));
- tagged_node_handle old_tos = tos.load(detail::memory_order_relaxed);
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!pool.get_pointer(old_tos))
- return false;
- node * new_tos_ptr = pool.get_pointer(old_tos_pointer->next);
- tagged_node_handle new_tos(pool.get_handle(new_tos_ptr), old_tos.get_next_tag());
- tos.store(new_tos, memory_order_relaxed);
- detail::copy_payload(old_tos_pointer->v, ret);
- pool.template destruct<false>(old_tos);
- return true;
- }
-
- template <typename Functor>
- bool consume_one(Functor & f)
- {
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return false;
- tagged_node_handle new_tos(old_tos_pointer->next, old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos)) {
- f(old_tos_pointer->v);
- pool.template destruct<true>(old_tos);
- return true;
- }
- }
- }
-
- template <typename Functor>
- bool consume_one(Functor const & f)
- {
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return false;
- tagged_node_handle new_tos(old_tos_pointer->next, old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos)) {
- f(old_tos_pointer->v);
- pool.template destruct<true>(old_tos);
- return true;
- }
- }
- }
-
- template <typename Functor>
- size_t consume_all(Functor & f)
- {
- size_t element_count = 0;
- while (consume_one(f))
- element_count += 1;
- return element_count;
- }
-
- template <typename Functor>
- size_t consume_all(Functor const & f)
- {
- size_t element_count = 0;
- while (consume_one(f))
- element_count += 1;
- return element_count;
- }
-
- template <typename Functor>
- size_t consume_all_atomic(Functor & f)
- {
- size_t element_count = 0;
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return 0;
- tagged_node_handle new_tos(pool.null_handle(), old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos))
- break;
- }
- tagged_node_handle nodes_to_consume = old_tos;
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_to_consume);
- f(node_pointer->v);
- element_count += 1;
- node * next_node = pool.get_pointer(node_pointer->next);
- if (!next_node) {
- pool.template destruct<true>(nodes_to_consume);
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_to_consume.get_next_tag());
- pool.template destruct<true>(nodes_to_consume);
- nodes_to_consume = next;
- }
- return element_count;
- }
-
- template <typename Functor>
- size_t consume_all_atomic(Functor const & f)
- {
- size_t element_count = 0;
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return 0;
- tagged_node_handle new_tos(pool.null_handle(), old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos))
- break;
- }
- tagged_node_handle nodes_to_consume = old_tos;
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_to_consume);
- f(node_pointer->v);
- element_count += 1;
- node * next_node = pool.get_pointer(node_pointer->next);
- if (!next_node) {
- pool.template destruct<true>(nodes_to_consume);
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_to_consume.get_next_tag());
- pool.template destruct<true>(nodes_to_consume);
- nodes_to_consume = next;
- }
- return element_count;
- }
-
- template <typename Functor>
- size_t consume_all_atomic_reversed(Functor & f)
- {
- size_t element_count = 0;
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return 0;
- tagged_node_handle new_tos(pool.null_handle(), old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos))
- break;
- }
- tagged_node_handle nodes_to_consume = old_tos;
- node * last_node_pointer = NULL;
- tagged_node_handle nodes_in_reversed_order;
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_to_consume);
- node * next_node = pool.get_pointer(node_pointer->next);
- node_pointer->next = pool.get_handle(last_node_pointer);
- last_node_pointer = node_pointer;
- if (!next_node) {
- nodes_in_reversed_order = nodes_to_consume;
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_to_consume.get_next_tag());
- nodes_to_consume = next;
- }
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_in_reversed_order);
- f(node_pointer->v);
- element_count += 1;
- node * next_node = pool.get_pointer(node_pointer->next);
- if (!next_node) {
- pool.template destruct<true>(nodes_in_reversed_order);
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_in_reversed_order.get_next_tag());
- pool.template destruct<true>(nodes_in_reversed_order);
- nodes_in_reversed_order = next;
- }
- return element_count;
- }
-
- template <typename Functor>
- size_t consume_all_atomic_reversed(Functor const & f)
- {
- size_t element_count = 0;
- tagged_node_handle old_tos = tos.load(detail::memory_order_consume);
- for (;;) {
- node * old_tos_pointer = pool.get_pointer(old_tos);
- if (!old_tos_pointer)
- return 0;
- tagged_node_handle new_tos(pool.null_handle(), old_tos.get_next_tag());
- if (tos.compare_exchange_weak(old_tos, new_tos))
- break;
- }
- tagged_node_handle nodes_to_consume = old_tos;
- node * last_node_pointer = NULL;
- tagged_node_handle nodes_in_reversed_order;
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_to_consume);
- node * next_node = pool.get_pointer(node_pointer->next);
- node_pointer->next = pool.get_handle(last_node_pointer);
- last_node_pointer = node_pointer;
- if (!next_node) {
- nodes_in_reversed_order = nodes_to_consume;
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_to_consume.get_next_tag());
- nodes_to_consume = next;
- }
- for(;;) {
- node * node_pointer = pool.get_pointer(nodes_in_reversed_order);
- f(node_pointer->v);
- element_count += 1;
- node * next_node = pool.get_pointer(node_pointer->next);
- if (!next_node) {
- pool.template destruct<true>(nodes_in_reversed_order);
- break;
- }
- tagged_node_handle next(pool.get_handle(next_node), nodes_in_reversed_order.get_next_tag());
- pool.template destruct<true>(nodes_in_reversed_order);
- nodes_in_reversed_order = next;
- }
- return element_count;
- }
-
- bool empty(void) const
- {
- return pool.get_pointer(tos.load()) == NULL;
- }
- private:
- #ifndef BOOST_DOXYGEN_INVOKED
- detail::atomic<tagged_node_handle> tos;
- static const int padding_size = BOOST_LOCKFREE_CACHELINE_BYTES - sizeof(tagged_node_handle);
- char padding[padding_size];
- pool_t pool;
- #endif
- };
- }
- }
- #endif
|