thread_safe_list.h 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354
  1. /*
  2. * (1)这个实现要求构建工具支持C++11的atomic mutex condition_veriable功能。这是C++11的基础特性,一般2011年以后的C++编译器都能支持。 例如,visual studio 2012以上。
  3. (2)这个类的实现中有两处使用了unique_lock而不是lock_guard,这是data_cond.wait所需要的,unique_lock是lock_guard的增强版。
  4. 通过std::move的使用(前提是我们实现的类型T定义了移动构造函数和移动赋值函数),能利用移动语义带来的性能优势。
  5. 使用shared_ptr<T>返回元素,用户无需释放元素的内存。
  6. 原文链接:https://blog.csdn.net/weixin_41855721/article/details/81703818
  7. 增加了一些功能函数,
  8. 补充了注释说明
  9. termination_list
  10. // 在退出状态下,所有的功能函数不可用,返回false或者null。
  11. // wait_and_pop不会阻塞。让其直接通过,通过后直接return,不允许做其他的。
  12. pop系列函数
  13. //(1)没有调用termination时,每调用一次出队一个元素,直到队列为空本方法阻塞线程。
  14. //(2)在调用了termination后,本方法永不阻塞,如果原本已经处于阻塞状态,解除阻塞状态。
  15. //(3)返回true时,value值有效。返回false时,value值无效。调用了termination且队列为空时返回false.
  16. 注注注注注意了:模板类不支持分离编译。 模板类的实现必须放在头文件
  17. 为了方便阅读和编程规范,依然将声明和实现分开,就像是把cpp文件的代码复制到h文件的尾部。
  18. 如果将实现放到cpp里面,那么就要为cpp文件加 ifndef define endif 防止重定义。
  19. 然后在调用方include包含cpp文件,但是这样不好。
  20. thread_safe_queue 就是在 Thread_safe_queue 的基础上修改的,
  21. * */
  22. #ifndef __THREAD_SAFE_LIST_H__
  23. #define __THREAD_SAFE_LIST_H__
  24. #include <list>
  25. #include <atomic>
  26. #include <mutex>
  27. #include <condition_variable>
  28. template<class T>
  29. class Thread_safe_list
  30. {
  31. public:
  32. Thread_safe_list();
  33. Thread_safe_list(const Thread_safe_list& other);
  34. ~Thread_safe_list();
  35. //(1)没有调用termination时,每调用一次出队一个元素,直到队列为空本方法阻塞线程。
  36. //(2)在调用了termination后,本方法永不阻塞,如果原本已经处于阻塞状态,解除阻塞状态。
  37. //(3)返回true时,value值有效。返回false时,value值无效。调用了termination且队列为空时返回false.
  38. //等待并弹出数据,成功弹出则返回true
  39. // 队列为空则无限等待,termination终止队列,则返回false
  40. bool wait_and_pop(T& value);
  41. //尝试弹出数据,成功弹出则返回true
  42. //队列为空 或者 termination终止队列,返回false
  43. bool try_pop(T& value);
  44. //等待并弹出数据,成功弹出则返回true
  45. // 队列为空则无限等待,termination终止队列,则返回false
  46. std::shared_ptr<T> wait_and_pop();
  47. //尝试弹出数据,成功弹出则返回true
  48. //队列为空 或者 termination终止队列,返回false
  49. std::shared_ptr<T> try_pop();
  50. //插入一项,并唤醒一个线程,
  51. //如果成功插入,则返回true, 失败则返回false
  52. //注:只能唤醒一个线程,防止多线程误判empty()
  53. bool push(T new_value);
  54. //清除队列,只是将队列的实例抛出。T是实例内存,系统自动回收的。
  55. bool clear();
  56. //清除队列,抛出之后还要delete指针。T是动态内存,需要手动回收的。
  57. bool clear_and_delete();
  58. public:
  59. //判空
  60. bool empty();
  61. //获取队列大小
  62. size_t size();
  63. //设置队列为退出状态。并唤醒所有的线程,使其通过wait
  64. // 在退出状态下,所有的功能函数不可用,必须直接返回false或者null。
  65. // wait_and_pop不会阻塞。让其直接通过,通过后直接return,不允许做其他的。
  66. void termination_list();
  67. //唤醒队列,恢复所有的功能函数。wait_and_pop会继续阻塞。
  68. void wake_list();
  69. //获取退出状态
  70. bool get_termination_flag();
  71. //判断是否可以直接通过wait, m_data_list不为空或者m_termination终止时都可以通过等待。
  72. bool is_pass();
  73. public:
  74. std::mutex m_mutex; //队列的锁
  75. std::list<std::shared_ptr<T>> m_data_list; //队列数据,使用智能指针shared_ptr
  76. std::condition_variable m_data_cond; //条件变量
  77. std::atomic<bool> m_termination_flag; //终止标志位
  78. private:
  79. };
  80. template<class T>
  81. Thread_safe_list<T>::Thread_safe_list()
  82. {
  83. m_termination_flag = false;
  84. }
  85. template<class T>
  86. Thread_safe_list<T>::Thread_safe_list(const Thread_safe_list& other)
  87. {
  88. std::unique_lock<std::mutex> lock_this(m_mutex);
  89. std::unique_lock<std::mutex> lock_other(other.m_mutex);
  90. m_data_list = other.data_list;
  91. m_termination_flag = other.m_termination_flag;
  92. }
  93. template<class T>
  94. Thread_safe_list<T>::~Thread_safe_list()
  95. {
  96. //析构时,终止队列,让线程通过等待,方便线程推出。
  97. termination_list();
  98. }
  99. //(1)没有调用termination时,每调用一次出队一个元素,直到队列为空本方法阻塞线程。
  100. //(2)在调用了termination后,本方法永不阻塞,如果原本已经处于阻塞状态,解除阻塞状态。
  101. //(3)返回true时,value值有效。返回false时,value值无效。调用了termination且队列为空时返回false.
  102. //等待并弹出数据,成功弹出则返回true
  103. // 队列为空则无限等待,termination终止队列,则返回false
  104. template<class T>
  105. bool Thread_safe_list<T>::wait_and_pop(T& value)
  106. {
  107. if ( m_termination_flag )
  108. {
  109. return false;
  110. }
  111. else
  112. {
  113. std::unique_lock<std::mutex> lk(m_mutex);
  114. //无限等待,一直阻塞,除非有新的数据加入或者终止队列
  115. m_data_cond.wait(lk, [this]
  116. { return ((!m_data_list.empty()) || m_termination_flag); });
  117. if (m_termination_flag)
  118. {
  119. return false;
  120. }
  121. else
  122. {
  123. // value = move(*m_data_list.front());
  124. value = (*m_data_list.front());
  125. m_data_list.pop_front();
  126. return true;
  127. }
  128. }
  129. }
  130. //尝试弹出数据,成功弹出则返回true
  131. //队列为空 或者 termination终止队列,返回false
  132. template<class T>
  133. bool Thread_safe_list<T>::try_pop(T& value)
  134. {
  135. if ( m_termination_flag )
  136. {
  137. return false;
  138. }
  139. else
  140. {
  141. std::unique_lock<std::mutex> lk(m_mutex);
  142. if (m_data_list.empty())
  143. {
  144. return false;
  145. }
  146. else
  147. {
  148. // value = move(*m_data_list.front());
  149. value = (*m_data_list.front());
  150. m_data_list.pop();
  151. return true;
  152. }
  153. }
  154. }
  155. //等待并弹出数据,成功弹出则返回true
  156. // 队列为空则无限等待,termination终止队列,则返回false
  157. template<class T>
  158. std::shared_ptr<T> Thread_safe_list<T>::wait_and_pop()
  159. {
  160. if ( m_termination_flag )
  161. {
  162. return NULL;
  163. }
  164. else
  165. {
  166. std::unique_lock<std::mutex> lk(m_mutex);
  167. //无限等待,一直阻塞,除非有新的数据加入或者终止队列
  168. m_data_cond.wait(lk, [this]
  169. { return ((!m_data_list.empty()) || m_termination_flag); });
  170. if (m_termination_flag)
  171. {
  172. return NULL;
  173. }
  174. else
  175. {
  176. std::shared_ptr<T> res = m_data_list.front();
  177. m_data_list.pop();
  178. return res;
  179. }
  180. }
  181. }
  182. //尝试弹出数据,成功弹出则返回true
  183. //队列为空 或者 termination终止队列,返回false
  184. template<class T>
  185. std::shared_ptr<T> Thread_safe_list<T>::try_pop()
  186. {
  187. if ( m_termination_flag )
  188. {
  189. return NULL;
  190. }
  191. else
  192. {
  193. std::unique_lock<std::mutex> lk(m_mutex);
  194. if (m_data_list.empty())
  195. {
  196. return NULL;
  197. }
  198. else
  199. {
  200. std::shared_ptr<T> res = m_data_list.front();
  201. m_data_list.pop();
  202. return res;
  203. }
  204. }
  205. }
  206. //插入一项,并唤醒一个线程,
  207. //如果成功插入,则返回true, 失败则返回false
  208. //注:只能唤醒一个线程,防止多线程误判empty()
  209. template<class T>
  210. bool Thread_safe_list<T>::push(T new_value)
  211. {
  212. if (m_termination_flag)
  213. {
  214. return false;
  215. }
  216. else
  217. {
  218. // std::shared_ptr<T> data(std::make_shared<T>(move(new_value)));
  219. std::shared_ptr<T> data(std::make_shared<T>((new_value)));
  220. std::unique_lock<std::mutex> lk(m_mutex);
  221. m_data_list.push_back(data);
  222. m_data_cond.notify_one();
  223. return true;
  224. }
  225. }
  226. //清除队列,只是将队列的实例抛出。T是实例内存,系统自动回收的。
  227. template<class T>
  228. bool Thread_safe_list<T>::clear()
  229. {
  230. std::unique_lock<std::mutex> lk(m_mutex);
  231. while (!m_data_list.empty())
  232. {
  233. m_data_list.pop_front();
  234. }
  235. return true;
  236. }
  237. //清除队列,抛出之后还要delete指针。T是动态内存,需要手动回收的。
  238. template<class T>
  239. bool Thread_safe_list<T>::clear_and_delete()
  240. {
  241. std::unique_lock<std::mutex> lk(m_mutex);
  242. while (!m_data_list.empty())
  243. {
  244. T res = NULL;
  245. // res = move(*m_data_list.front());
  246. res = (*m_data_list.front());
  247. m_data_list.pop_front();
  248. if(res != NULL)
  249. {
  250. delete(res);
  251. }
  252. }
  253. return true;
  254. }
  255. //判空
  256. template<class T>
  257. bool Thread_safe_list<T>::empty()
  258. {
  259. std::unique_lock<std::mutex> lk(m_mutex);
  260. return m_data_list.empty();
  261. }
  262. //获取队列大小
  263. template<class T>
  264. size_t Thread_safe_list<T>::size()
  265. {
  266. std::unique_lock<std::mutex> lk(m_mutex);
  267. return m_data_list.size();
  268. }
  269. //设置队列为退出状态。并唤醒所有的线程,使其通过wait
  270. // 在退出状态下,所有的功能函数不可用,必须直接返回false或者null。
  271. // wait_and_pop不会阻塞。让其直接通过,通过后直接return,不允许做其他的。
  272. template<class T>
  273. void Thread_safe_list<T>::termination_list()
  274. {
  275. std::unique_lock<std::mutex> lk(m_mutex);
  276. m_termination_flag = true;
  277. m_data_cond.notify_all();
  278. }
  279. //唤醒队列,恢复所有的功能函数。wait_and_pop会继续阻塞。
  280. template<class T>
  281. void Thread_safe_list<T>::wake_list()
  282. {
  283. std::unique_lock<std::mutex> lk(m_mutex);
  284. m_termination_flag = false;
  285. m_data_cond.notify_all();
  286. }
  287. //获取退出状态
  288. template<class T>
  289. bool Thread_safe_list<T>::get_termination_flag()
  290. {
  291. return m_termination_flag;
  292. }
  293. //判断是否可以直接通过wait, m_data_list不为空或者m_termination终止时都可以通过等待。
  294. template<class T>
  295. bool Thread_safe_list<T>::is_pass()
  296. {
  297. return (!m_data_list.empty() || m_termination_flag);
  298. }
  299. #endif //__THREAD_SAFE_LIST_H__