平臺服務和腳本服務接口對后端PostgreSQL數據庫的使用目前采用短暫連接方式，造成多次調用服務時頻繁連接和斷開數據庫，效率很低。
如果共享數據庫連接，則會造成多線程訪問數據庫時的事務沖突，故必須采用連接池來管理對數據庫的并發訪問，某一線程連接到數據庫使用完畢后，不斷開數據庫連接，而是把連接歸還給連接池。
另一線程訪問數據庫時會首先向連接池申請已經存在的連接，如果連接池中沒有空閑連接，或者申請到得連接已經超時失效，再建立新的連接，使用完畢后同樣歸還到連接池。
這樣連接池中的連接數會隨著線程壓力的增加逐漸增長，直到所有的線程同時工作，達到最多連接數。
由于一個線程可能同時申請多個連接，故連接數可能會大于線程數。連接池在程序結束時銷毀全部連接，或者線程在申請到的某一連接失效時銷毀該連接。
1
2 class dbConnection
3 {
4 friend class dbConnectionPool;
5 bool is_using;
6
7 public:
8 pqxx::connection pqxx_conn;
9
10 public:
11 dbConnection(string opt)
12 : pqxx_conn(opt),is_using(false)
13 {}
14 };
15
16 typedef boost::shared_ptr<dbConnection> dbConnection_ptr;
17
18 class dbConnectionPool
19 {
20 string m_opt;
21 size_t m_max_num;
22 std::vector<dbConnection_ptr> m_pool;
23 cppx::thread_mutex m_mutex;
24
25 public:
26 dbConnectionPool(string opt,size_t max_num)
27 : m_opt(opt),m_max_num(max_num)
28 {}
29
30 dbConnection_ptr GetConnection(void){
31 ACE_DEBUG((LM_DEBUG,"(%t) LINE %d : %C\n", __LINE__ ,__FUNCTION__));
32
33 cppx::scoped_lock lock(m_mutex);
34 foreach(dbConnection_ptr ptr,m_pool){
35 if( ptr->is_using == false ){
36 ptr->is_using = true;
37
38 if( ptr->pqxx_conn.is_open() ){
39 ACE_DEBUG((LM_DEBUG,"(%t) 找到連接池空閑連接。\n"));
40 return ptr;
41 }
42 }
43 }
44
45 ACE_DEBUG((LM_DEBUG,"(%t) 沒有空閑連接，創建新連接。[%d]\n",m_pool.size()));
46
47 // 沒有空閑連接，創建新連接
48 dbConnection_ptr ptr(new dbConnection(m_opt));
49 ptr->is_using = true;
50
51 // 找到一個失效的連接，用新連接覆蓋
52 bool cover = false;
53 for(size_t idx = 0; idx < m_pool.size(); idx++){
54 dbConnection_ptr ptr2 = m_pool[idx];
55 if( ptr2->is_using == false && ptr2->pqxx_conn.is_open() == false ){
56 m_pool[idx] = ptr;
57 cover = true;
58 ACE_DEBUG((LM_DEBUG,"(%t) 覆蓋失效的連接[%d]。\n",idx));
59 break;
60 }
61 }
62 if( !cover ){
63 m_pool.push_back(ptr);
64 }
65
66 return ptr;
67 }
68
69 void ReleaseConnection(dbConnection_ptr ptr_){
70 ACE_DEBUG((LM_DEBUG,"(%t) LINE %d : %C\n", __LINE__ ,__FUNCTION__));
71 ACE_DEBUG((LM_DEBUG,"(%t) 連接使用完畢，歸還到連接池。[%d]\n",m_pool.size()));
72
73 cppx::scoped_lock lock(m_mutex);
74 ptr_->is_using = false;
75 }
76 };
77
78 class dbConnectionUser
79 {
80 dbConnectionPool & pool;
81
82 public:
83 dbConnection_ptr conn;
84
85 public:
86 dbConnectionUser(dbConnectionPool & pool_) : pool(pool_) {
87 conn = pool.GetConnection();
88 }
89 ~dbConnectionUser(void){
90 pool.ReleaseConnection(conn);
91 }
92 };
93
使用連接池的方法，只要在建立數據庫連接的類中增加連接池成員即可：
1
2 struct ALEE_PlatformService_iPqxx::pimpl_t
3 {
4 #if defined(USE_POOL_DBLINK) && (USE_POOL_DBLINK == 1)
5
6 public:
7 dbConnectionPool m_pool;
8
9 #else
10
為了方便，定義如下的宏：
1 #define DB_CONNECTION() \
2 dbConnectionUser user(pimpl_->m_pool); \
3 if( !user.conn || !user.conn->pqxx_conn.is_open() ) return false; \
4 pqxx::work X(user.conn->pqxx_conn);
5
需要調用數據庫查詢時，只需要把上面的宏插入try塊中：
1
2 try{
3 DB_CONNECTION();
4
5 fuStringCommand cmd;
6 cmd = pimpl_->m_sqls["apx_get_moi_attr_enum"];
7 cmd << X.quote(moi_hash)
8 << X.quote(attr_code);
9
10 result = new defs::xml_row;
11 return pimpl_->QueryRecord(X, cmd, *result);
12 }
13 catch_pqxx_error();
14
一個簡單的數據庫連接池就完成了，由于多線程并發時動態申請數據庫連接，既可以充分發揮數據庫并發的好處有避免了共享數據庫連接造成事務沖突。完全滿足簡單的數據庫查詢需要。
等燈。等燈。 Cppx Inside
posted on 2011-02-28 13:57
風雷九州 閱讀(5754)
評論(0) 編輯 收藏 引用