CDBConnPool.h 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323
  1. #ifndef CDBConnPool_h__
  2. #define CDBConnPool_h__
  3. /**
  4. * @brief
  5. 数据库连接池类
  6. * @version
  7. V 1.0.0
  8. * @author
  9. 王益俊
  10. * @date
  11. 创建时间: 2018-04-17\n
  12. * @note
  13. 2018-04-17 创建类。\n
  14. * @warning
  15. * @bug
  16. */
  17. #pragma once
  18. #include <boost/atomic.hpp>
  19. #include <boost/thread/thread.hpp>
  20. #include <boost/serialization/singleton.hpp>
  21. #include <list>
  22. #include <string>
  23. #include <mutex>
  24. #include "CDBCommon.h"
  25. #include "CDBConnect.h"
  26. namespace YADB
  27. {
  28. class CDBConnPool
  29. {
  30. private:
  31. std::list<CDBConnect*> __BusyConnList;//正在使用的数据库连接列表
  32. std::list<CDBConnect*> __IdleConnList;//没有使用的数据库连接列表
  33. _DB_POOL_SETTING_ __Setting;//数据库连接池设置
  34. std::mutex __mtx;//互斥量
  35. boost::atomic<bool> __Running{ true };//线程是否运行的标识
  36. boost::atomic<bool> __IsExited{ false };//线程是否已退出的标识
  37. CDBConnect* __pAsyncDBConn;//异步执行线程用的数据库连接
  38. private:
  39. /**
  40. * @brief
  41. 创建一个空闲连接函数。
  42. * @param [out] std::string& Error 错误信息\n
  43. * @param [in] bool IsTemp 创建的连接是否是临时的\n
  44. * @return 返回创建的连接
  45. * @return >0 成功\n
  46. * @return ==0 失败\n
  47. * @note
  48. * @warning
  49. * @bug
  50. */
  51. CDBConnect* __CreateIdleConn( std::string& Error, bool IsTemp = false );
  52. /**
  53. * @brief
  54. 创建异步执行线程所用连接函数。
  55. * @param 无\n
  56. * @return 无\n
  57. * @note
  58. * @warning
  59. * @bug
  60. */
  61. void __CreateAsyncThrdConn();
  62. /**
  63. * @brief
  64. ??????创建异步执行线程所用连接函数。
  65. * @param 无\n
  66. * @return 无\n
  67. * @note
  68. * @warning
  69. * @bug
  70. */
  71. void __DestroyAsyncThrdConn();
  72. /**
  73. * @brief
  74. 启动异步执行线程函数。
  75. * @param 无\n
  76. * @return 无
  77. * @note
  78. * @warning
  79. * @bug
  80. */
  81. void __StartAsyncThrd();
  82. /**
  83. * @brief
  84. 停止异步执行线程函数。
  85. * @param 无\n
  86. * @return 无
  87. * @note
  88. * @warning
  89. * @bug
  90. */
  91. void __StopAsyncThrd();
  92. protected:
  93. /**
  94. * @brief
  95. 异步执行线程函数。
  96. * @param [in] CDBConnPool* pOwner 线程所属对象\n
  97. * @return 无\n
  98. * @note
  99. * @warning
  100. * @bug
  101. */
  102. void _AsyncThreadFunc( CDBConnPool* pOwner );//线程函数
  103. public:
  104. CDBConnPool();
  105. ~CDBConnPool();
  106. /**
  107. * @brief
  108. 创建数据库连接池函数。
  109. * @param [in] const _DB_POOL_SETTING_& Setting 数据库连接池设置\n
  110. * @param [out] std::string& Error 错误信息\n
  111. * @return 返回创建数据库连接池是否成功
  112. * @return true 成功
  113. * @return false 失败
  114. * @note
  115. * @warning
  116. * @bug
  117. */
  118. bool Create( const _DB_POOL_SETTING_& Setting, std::string& Error );
  119. /**
  120. * @brief
  121. 关闭线程池函数。
  122. * @param 无\n
  123. * @return 无\n
  124. * @note
  125. * @warning
  126. * @bug
  127. */
  128. void Close();
  129. /**
  130. * @brief
  131. 从连接池中获得一个数据库连接函数。
  132. * @param [out] std::string& Error 错误信息\n
  133. * @return 返回获得一个数据库连接是否成功
  134. * @return 0 失败\n
  135. * @return !=0 获得的数据库连接\n
  136. * @note
  137. * @warning
  138. * @bug
  139. */
  140. CDBConnect* GetDBConnect( std::string& Error );
  141. /**
  142. * @brief
  143. 归还一个数据库连接到连接池中函数。
  144. * @param [in] CDBConnect* pConn 数据库连接\n
  145. * @return 无\n
  146. * @note
  147. * @warning
  148. * @bug
  149. */
  150. void GiveBack( CDBConnect* pConn );
  151. /**
  152. * @brief
  153. 把SQL语句加入异步执行队列函数。
  154. * @param [in] const std::string& strSQL 要执行的SQL语句\n
  155. * @return 加入异步执行队列是否成功\n
  156. * @return true成功\n
  157. * @return false失败\n
  158. * @note
  159. * @warning
  160. * @bug
  161. */
  162. bool PushAsync( const std::string& strSQL );
  163. /**
  164. * @brief
  165. 执行SQL语句返回结果集函数。
  166. * @param [in] const char *szSql SQL语句\n
  167. * @param [out] CDBResultSet& DBRes 结果集\n
  168. * @param [out] std::string& Error 错误信息\n
  169. * @return 返回查询是否成功
  170. * @return true 成功\n
  171. * @return false 失败\n
  172. * @note
  173. * @warning
  174. * @bug
  175. */
  176. bool Query( const char *szSql, CDBResultSet& DBRes, std::string& Error );
  177. /**
  178. * @brief
  179. 执行SQL语句返回原生结果集函数。
  180. * @param [in] const char *szSql SQL语句\n
  181. * @param [out] std::string& Error 错误信息\n
  182. * @return 返回查询的结果集
  183. * @note
  184. * @warning
  185. * @bug
  186. */
  187. MYSQL_RES* Query( const char *szSql, std::string& Error );
  188. /**
  189. * @brief
  190. 执行SQL语句函数。
  191. * @param [in] const char *szSql SQL语句\n
  192. * @param [out] std::string& Error 错误信息\n
  193. * @return 返回影响到的记录数量
  194. * @return -1 失败\n
  195. * @return >=0 成功\n
  196. * @note
  197. * @warning
  198. * @bug
  199. */
  200. my_ulonglong ExecuteSql( const char *szSql, std::string& Error );
  201. /**
  202. * @brief
  203. 执行SQL语句并获得ID的函数。
  204. * @param [in] const char *szSql SQL语句\n
  205. * @param [out] std::string& Error 错误信息\n
  206. * @return 返回获得的ID
  207. * @return <=0 失败\n
  208. * @return >0 成功\n
  209. * @note
  210. * @warning
  211. * @bug
  212. */
  213. my_ulonglong ExecuteSqlID( const char *szSql, std::string& Error );
  214. };
  215. }
  216. //单件相关定义
  217. typedef boost::serialization::singleton<YADB::CDBConnPool> singleton_CDBConnPool;
  218. #define sDBConnPool singleton_CDBConnPool::get_mutable_instance()
  219. #endif // CDBConnPool_h__