ExclusiveHolder.tlh 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. //-----------------------------------------------------------------------------
  2. // 用排他锁实现的并发共享数据的模板类.
  3. // 不是 dll, 没有版本号, 因此要手工编写版本号
  4. // V1, 2013-09-03, 5.1.1 中发布, 使用 Synchronize.hpp
  5. // V2, 2018-03-02, 7.0 中发布, 改用 std::recursive_mutex 及 std::lock_guard
  6. // V3, 2020-09-30, 7.3 中发布, 删除 XDWHolder, 改用 std::unique_ptr. 同时修改了文件名
  7. // V4, 2022-09-30, LockHolder 中, 删除 std::unique_ptr, 直接用 std::lock_guard < std::recursive_mutex > m_Lock;
  8. // 这样可以避免一次 new / delete
  9. //-----------------------------------------------------------------------------
  10. #pragma once
  11. #include <memory>
  12. #include <mutex>
  13. //-----------------------------------------------------------------------------
  14. // ExclusiveHolder.tlh
  15. // 允许互斥访问一个指针对象
  16. //-----------------------------------------------------------------------------
  17. template <typename T>
  18. class ExclusiveHolder
  19. {
  20. protected:
  21. T * m_pObject;
  22. public:
  23. ExclusiveHolder ();
  24. explicit ExclusiveHolder (T * object);
  25. virtual ~ExclusiveHolder ();
  26. private:
  27. // 禁止拷贝构造函数
  28. ExclusiveHolder (const ExclusiveHolder & h) = delete;
  29. // 禁止复制
  30. ExclusiveHolder & operator = (const ExclusiveHolder & h) = delete;
  31. public:
  32. void Attach (T * object);
  33. T * Detach ();
  34. ExclusiveHolder & operator = (T * object);
  35. bool IsEmpty () const;
  36. void Release ();
  37. std::recursive_mutex & GetMutex () { return m_Mutex; }
  38. protected:
  39. mutable std::recursive_mutex m_Mutex;
  40. public:
  41. class LockHolder
  42. {
  43. public:
  44. LockHolder (const ExclusiveHolder * holder)
  45. : m_Lock { std::lock_guard < std::recursive_mutex > (holder->m_Mutex) }
  46. , m_Holder (holder)
  47. {
  48. assert (m_Holder);
  49. }
  50. ~LockHolder () = default;
  51. LockHolder (LockHolder && from)
  52. {
  53. m_Lock.swap (from.m_Lock);
  54. m_Holder = from.m_Holder;
  55. }
  56. private:
  57. // 可以对同一个 ExclusiveHolder 对象反复调用 Lock, 但是 LockHolder 还是不要拷贝构造了吧
  58. // 禁止拷贝构造函数
  59. LockHolder (const LockHolder & h) = delete;
  60. // 禁止复制
  61. LockHolder & operator = (const LockHolder & h) = delete;
  62. public:
  63. T * As ()
  64. {
  65. return m_Holder->m_pObject;
  66. }
  67. const T * As () const
  68. {
  69. return m_Holder->m_pObject;
  70. }
  71. template <typename C> C * AS ()
  72. {
  73. return m_Holder->m_pObject;
  74. }
  75. template <typename C> const C * AS () const
  76. {
  77. return m_Holder->m_pObject;
  78. }
  79. T * operator -> ()
  80. {
  81. return As ();
  82. }
  83. const T * operator -> () const
  84. {
  85. return As ();
  86. }
  87. T & operator * ()
  88. {
  89. return *As ();
  90. }
  91. const T & operator * () const
  92. {
  93. return *As ();
  94. }
  95. operator bool () const
  96. {
  97. return As ();
  98. }
  99. protected:
  100. std::lock_guard < std::recursive_mutex > m_Lock;
  101. const ExclusiveHolder * m_Holder;
  102. };
  103. public:
  104. inline LockHolder Lock ()
  105. {
  106. return LockHolder (this);
  107. }
  108. inline LockHolder Lock () const
  109. {
  110. return LockHolder (this);
  111. }
  112. template <typename FuncType, typename ... Args>
  113. auto Action (FuncType && action, Args && ... args) -> typename std::invoke_result <FuncType, T *, Args && ...>::type
  114. {
  115. LockHolder holder (this);
  116. return action (holder.As (), args...);
  117. }
  118. #if 0
  119. template <typename FuncType, typename ... Args>
  120. auto Action (FuncType && action, Args && ... args) const -> typename std::invoke_result <FuncType, const T *, Args && ...>::type
  121. {
  122. LockHolder holder (this);
  123. return action (holder.As (), args...);
  124. }
  125. #endif
  126. };