server/DBConnPool.h

// !defined(AFX_DBCONNPOOL_H__42089E9A_FD85_4DD4_A973_64A4980332A5__INCLUDED_)  
// DBConnPool.h: interface for the DBConnPool class.
//
//////////////////////////////////////////////////////////////////////
#if !defined(AFX_DBCONNPOOL_H__42089E9A_FD85_4DD4_A973_64A4980332A5__INCLUDED_)
#define AFX_DBCONNPOOL_H__42089E9A_FD85_4DD4_A973_64A4980332A5__INCLUDED_
#define  SQL_BUFFER_LEN 1024
#import "C:/Program Files/Common Files/System/ado/msado15.dll" rename("EOF","ADOEOF")
#include <list>
#include <windows.h>
#include <string>
#include <vector>
//#include  "DBDefine.h"
using namespace ADODB;
using namespace std;

class RFIDAPI DBConnect
{
public:
	DBConnect(LPCSTR strDstAddress, 
		LPCSTR strUsername, 
		LPCSTR strPassword, 
		LPCSTR strDBName,
		BOOL &state);
	~DBConnect();
public:
	// 连接到数据库

	int Open(LPCTSTR strDstAddress, LPCTSTR strUsername, LPCTSTR strPassword, LPCTSTR strDBName);
	// 关闭数据库

	int Close();
	// 数据库是否已连接

	BOOL IsOpen() const;
public:
private:
	ADODB::_ConnectionPtr           _connection_ptr;   //ADO的数据库连接智能指针
	bool                            _isAdoValid;       //ADO环境是否已经初化成功标志量
	bool       m_bDBOpen;

	LPCSTR       _strDstAddress;    //数据源地址或服务名
	LPCSTR       _strUsername;      //数据库用户名
	LPCSTR       _strPassword;      //数据库密码
	LPCSTR       _strDBName;        //数据库名称
	void         VarientToString(_variant_t var, string& str);
	//对外公共接口
public:
	int          GetSubmitInfo(vector<SOAP_SUBMIT_SMS> &vecsoapSms);    

	int          InsertHistory(int id);
	int          UserLogin(LPCSTR name,LPCSTR password);
}; 
typedef std::list<DBConnect*> DBConnectList;
class  DBConnPool  
{
public:
	DBConnPool();
	virtual ~DBConnPool();
	// 获取实例指针
	static DBConnPool * Instanse();
	// 初始化所有连接
	int InitializeAllDBConnections();

	// 关闭所有连接
	void DestroyAllDBConnections();
	// 获取一个空闲连接
	DBConnect* GetAConnection();
	// 交还连接给空闲队列
	int RestoreAConnection(DBConnect* pDBEngine);
	void SetDBInfo(LPCSTR strDstAddress, LPCSTR strUsername, LPCSTR strPassword, LPCSTR strDBName,int minConn,int maxConn);
private:

	// 创建一个连接
	int InitializeAConnection();
	// 关闭一个连接
	void CloseAConnection(DBConnect* pDBEngine);
	// 停止工作线程
	void StopThread();

	// 判断是否需要停止
	BOOL IsNeedStop();
	BOOL IsNeedConnection();
	// 将守卫类作为连接池类的友元类
	friend class ConnGuard;
	// 唯一实例
	static DBConnPool *m_pInstanse;

	// 空闲数据库连接队列
	DBConnectList m_listIdleConnection;
	// 在使用的数据库连接
	DBConnectList m_listBusyConnection;

	// 队列保护的临界区
	CRITICAL_SECTION m_csIdleConnList;
	CRITICAL_SECTION m_csBusyConnList;

	// 可用连接总数的三个指标：最大、最小
	int m_nMaxCount;
	int m_nMinCount;

	//  // 数据库信息
	LPCSTR       _strDstAddress;    //数据源地址或服务名
	LPCSTR       _strUsername;      //数据库用户名
	LPCSTR       _strPassword;      //数据库密码
	LPCSTR       _strDBName;        //数据库名称

	// 维护线程
	HANDLE m_hMaintanceThread; // 线程句柄
	HANDLE m_hHaveData; // 信号

	BOOL m_bNeedStop; // 管理线程起停的标志位
	BOOL m_bNeedConnection; // 需要创建连接的标志
	static DWORD WINAPI thread_run( LPVOID pdata);
};
// 守卫类，利用构造和析构函数保证连接取出和归还必须成对，防止资源泄露
class  DBConnGuard
{
public:
	DBConnGuard(DBConnect*& DBConn)
	{
		DBConn = DBConnPool::Instanse()->GetAConnection();
		m_pDBConn = DBConn;
	}
	virtual ~DBConnGuard()
	{
		DBConnPool::Instanse()->RestoreAConnection(m_pDBConn);
	}
private:
	DBConnect *m_pDBConn;
};
RFIDAPI void  InitDBIterface(LPCSTR strDstAddress, LPCSTR strUsername, LPCSTR strPassword, LPCSTR strDBName,int minConn,int maxConn);
RFIDAPI DBConnect * GetAConnect();
#endif // !defined(AFX_DBCONNPOOL_H__42089E9A_FD85_4DD4_A973_64A4980332A5__INCLUDED_)