atomic_count.hpp 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. //
  2. // detail/atomic_count.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
  6. //
  7. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. #ifndef ASIO_DETAIL_ATOMIC_COUNT_HPP
  11. #define ASIO_DETAIL_ATOMIC_COUNT_HPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include "asio/detail/config.hpp"
  16. #if !defined(ASIO_HAS_THREADS)
  17. // Nothing to include.
  18. #else // !defined(ASIO_HAS_THREADS)
  19. # include <atomic>
  20. #endif // !defined(ASIO_HAS_THREADS)
  21. namespace asio {
  22. namespace detail {
  23. #if !defined(ASIO_HAS_THREADS)
  24. typedef long atomic_count;
  25. inline void increment(atomic_count& a, long b) { a += b; }
  26. inline void decrement(atomic_count& a, long b) { a -= b; }
  27. inline void ref_count_up(atomic_count& a) { ++a; }
  28. inline bool ref_count_down(atomic_count& a) { return --a == 0; }
  29. #else // !defined(ASIO_HAS_THREADS)
  30. typedef std::atomic<long> atomic_count;
  31. inline void increment(atomic_count& a, long b) { a += b; }
  32. inline void decrement(atomic_count& a, long b) { a -= b; }
  33. inline void ref_count_up(atomic_count& a)
  34. {
  35. a.fetch_add(1, std::memory_order_relaxed);
  36. }
  37. inline bool ref_count_down(atomic_count& a)
  38. {
  39. if (a.fetch_sub(1, std::memory_order_release) == 1)
  40. {
  41. std::atomic_thread_fence(std::memory_order_acquire);
  42. return true;
  43. }
  44. return false;
  45. }
  46. #endif // !defined(ASIO_HAS_THREADS)
  47. } // namespace detail
  48. } // namespace asio
  49. #endif // ASIO_DETAIL_ATOMIC_COUNT_HPP