summaryrefslogtreecommitdiffstats
path: root/Utilities/std/cm/shared_mutex
blob: 2ac9447dc7593221b1564490d1aa58493ffa722a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
// -*-c++-*-
// vim: set ft=cpp:

/* Distributed under the OSI-approved BSD 3-Clause License.  See accompanying
   file Copyright.txt or https://cmake.org/licensing for details.  */
#ifndef cm_shared_mutex
#define cm_shared_mutex

#if __cplusplus >= 201402L || defined(_MSVC_LANG) && _MSVC_LANG >= 201402L
#  define CMake_HAVE_CXX_SHARED_LOCK
#endif
#if __cplusplus >= 201703L || defined(_MSVC_LANG) && _MSVC_LANG >= 201703L
#  define CMake_HAVE_CXX_SHARED_MUTEX
#endif

#if defined(CMake_HAVE_CXX_SHARED_LOCK)
#  include <shared_mutex> // IWYU pragma: export
#endif
#if !defined(CMake_HAVE_CXX_SHARED_MUTEX)
#  include "cm_uv.h"
#endif

namespace cm {
#if defined(CMake_HAVE_CXX_SHARED_MUTEX)
using std::shared_mutex;
#else
class shared_mutex
{
  uv_rwlock_t _M_;

public:
  using native_handle_type = uv_rwlock_t*;

  shared_mutex() { uv_rwlock_init(&_M_); }
  ~shared_mutex() { uv_rwlock_destroy(&_M_); }

  shared_mutex(shared_mutex const&) = delete;
  shared_mutex& operator=(shared_mutex const&) = delete;

  void lock() { uv_rwlock_wrlock(&_M_); }
  bool try_lock() { return uv_rwlock_trywrlock(&_M_) == 0; }
  void unlock() { uv_rwlock_wrunlock(&_M_); }

  void lock_shared() { uv_rwlock_rdlock(&_M_); }
  void unlock_shared() { uv_rwlock_rdunlock(&_M_); }

  native_handle_type native_handle() { return &_M_; }
};
#endif

#if defined(CMake_HAVE_CXX_SHARED_LOCK)
using std::shared_lock;
#else
template <typename T>
class shared_lock
{
  T& _mutex;

public:
  using mutex_type = T;

  shared_lock(T& m)
    : _mutex(m)
  {
    _mutex.lock_shared();
  }

  ~shared_lock() { _mutex.unlock_shared(); }

  shared_lock(shared_lock const&) = delete;
  shared_lock& operator=(shared_lock const&) = delete;
};
#endif
}

#endif