|
3. DBConnection.h
#pragma once
#include <comutil.h>
#pragma comment(lib,"comsupp.lib")
class CDBConnectionImpl ;
class CDBRecordSet ;
class AFX_EXT_CLASS CDBConnection
{
friend class CDBRecordSet ;
public:
CDBConnection(void);
virtual ~CDBConnection(void);
/*
* 设置数据库连接参数
* 参数：
* [in] szServerName -> 数据库所在机器的机器名或者IP地址
* [in] szBaseName -> 数据库实例名称
* [in] szUser -> 数据库登陆用户名
* [in] szPasswd -> 登陆密码
*/
void SetParameters(LPCTSTR szServerName,LPCTSTR szBaseName,LPCTSTR szUser,LPCTSTR szPasswd) ;
//是否连接已经打开
BOOL IsOpened(void) ;
//打开连接
BOOL Open(void) ;
//开始事务
BOOL BeginTrans(void) ;
//提交事务
BOOL CommitTrans(void) ;
//回滚事务
BOOL RollbackTrans(void) ;
/*
* 执行SQL语句，例如 insert , update 等,不返回记录
* 参数：
* [in] szSQL -> SQL语句
* [out] pnRecordsAffected -> the number of records that the operation affected.
*/
BOOL Execute(LPCTSTR szSQL,int* pnRecordsAffected = NULL) ;
/*
* 执行SQL语句，并返回记录集
* 参数：
* [in] szSQL -> SQL语句, select ......
* [out] recordSet -> 本次查询得到的记录集
*/
BOOL OpenRecordSet(LPCTSTR szSQL,CDBRecordSet& rRecordSet) ;
//关闭连接
BOOL Close(void) ;
/*
* 设置错误信息
* 参数：
* [in] szErrMsg -> 错误信息
* [in] szSourceFile -> 发生错误的源文件
* [in] nLine -> 发生错误的行号
*/
void SetErrorMessage(LPCTSTR szErrMsg,const char* szSourceFile=NULL,int nLine=0);
//获取错误信息
LPCTSTR GetErrorMessage(void) const;
//获取数据库访问的错误信息
LPCTSTR GetDBConnectionErrorMsg(void) ;
public:
//clone 方法，clone后两个对象具有相同的数据库连接参数，但是各自占有独立的数据库连接，
//clone 以后，各自有独立的操作，不会相互影响
BOOL CloneTo(CDBConnection& toConnection) ;
//copy 后，两个对象使用共同的连接，实际只使用用一个相同的数据库连接
//对数据库连接的操作会相互影响，需要注意。
//源的close/open会关闭所有副本的连接，但是副本的close/open不会影响其他连接
//copy 构造函数
CDBConnection(const CDBConnection& rConnection) ;
//赋值运算符
CDBConnection& operator= (const CDBConnection& rConnection);
protected:
//生成数据库连接字符串
void FormatDataSource(void) ;
//获取数据库连接接口
BOOL GetConnectionIDispatch(_variant_t& vActiveConnection) ;
protected:
//数据库连接
CDBConnectionImpl* m_pConnImpl ;
//数据库所在机器的机器名或者IP地址
CString m_strServerName ;
//数据库实例名称
CString m_strBaseName ;
//数据库登陆用户名
CString m_strDBUser ;
//登陆密码
CString m_strPasswd ;
//数据库连接字符串
CString m_strDataSource ;
//数据库连接错误
CString m_strConnErrorMsg ;
};
4. DBConnection.cpp
#include "StdAfx.h"
#include ".\dbconnection.h"
#include "DBConnectionImpl.h"
#include "DBErrorMsgDefs.h"
#include "DBRecordSet.h"
//ADO _ConnectionPtr
#define DBCONNECTIONPtr m_pConnImpl->GetConnection()
CDBConnection::CDBConnection(void)
{
m_pConnImpl = new CDBConnectionImpl() ;
}
CDBConnection::~CDBConnection(void)
{
if (m_pConnImpl != NULL)
{
delete m_pConnImpl ;
m_pConnImpl = NULL ;
}
}
//copy 构造函数
CDBConnection::CDBConnection(const CDBConnection& rConnection):
m_pConnImpl(NULL)
{
//调用默认构造函数
*this = rConnection ;
}
//赋值运算符
CDBConnection& CDBConnection::operator= (const CDBConnection& rConnection)
{
if(this == &rConnection)
{
return *this ;
}
m_strServerName = rConnection.m_strServerName ;
m_strBaseName = rConnection.m_strBaseName ;
m_strDBUser = rConnection.m_strDBUser ;
m_strPasswd = rConnection.m_strPasswd ;
m_strDataSource = rConnection.m_strDataSource ;
if(rConnection.m_pConnImpl == NULL)
{
if(m_pConnImpl != NULL)
{
delete m_pConnImpl ;
m_pConnImpl = NULL ;
}
}
else
{
if(m_pConnImpl == NULL)
{
m_pConnImpl = new CDBConnectionImpl() ;
}
*m_pConnImpl = *rConnection.m_pConnImpl ; //copy it
}
return *this ;
}
void CDBConnection::SetParameters(LPCTSTR szServerName,LPCTSTR szBaseName,LPCTSTR szUser,LPCTSTR szPasswd)
{
m_strServerName = _T("") ;
m_strBaseName = _T("");
m_strDBUser = _T("");
m_strPasswd = _T("") ;
if(szServerName)
{
m_strServerName = szServerName ;
}
if (szBaseName)
{
m_strBaseName = szBaseName ;
}
if(szUser)
{
m_strDBUser = szUser ;
}
if(szPasswd)
{
m_strPasswd = szPasswd ;
}
FormatDataSource() ;
}
void CDBConnection::FormatDataSource(void)
{
m_strDataSource.Format(_T("Provider=''''sqloledb'''';Data Source=''''%s'''';Initial Catalog=''''%s'''';User Id=''''%s'''';Password=''''%s'''';"),
(LPCTSTR)m_strServerName,
(LPCTSTR)m_strBaseName,
(LPCTSTR)m_strDBUser,
(LPCTSTR)m_strPasswd
) ;
}
//是否连接已经打开
BOOL CDBConnection::IsOpened(void)
{
if (!m_pConnImpl->isValid())
{
SetErrorMessage(CONNCTION_INVALIDHANDLE,__FILE__,__LINE__) ;
return FALSE ;
}
return (DBCONNECTIONPtr->GetState() == adStateOpen);
}
//打开连接
BOOL CDBConnection::Open(void)
{
try
{
if (!m_pConnImpl->isValid())
{
SetErrorMessage(CONNCTION_INVALIDHANDLE,__FILE__,__LINE__) ;
return FALSE ;
}
if (m_pConnImpl != NULL)
{
if(!m_pConnImpl->GetAutoClose())
{
//删除从其他对象复制的连接，防止连接冲突
delete m_pConnImpl ;
m_pConnImpl = new CDBConnectionImpl() ;
}
}
//check if is already opened
if(DBCONNECTIONPtr->GetState() != adStateClosed)
{
//if is opened,close it first
DBCONNECTIONPtr->Close() ;
}
DBCONNECTIONPtr->Errors->Clear();
DBCONNECTIONPtr->CursorLocation = adUseClient;
DBCONNECTIONPtr->Open(_bstr_t((LPCTSTR)m_strDataSource),_bstr_t(_T("")),_bstr_t(_T("")),adConnectUnspecified);
return (DBCONNECTIONPtr->GetState() == adStateOpen);
}
catch (_com_error &e)
{
SetErrorMessage((LPCTSTR)e.Description(),__FILE__,__LINE__) ;
}
catch (...)
{
SetErrorMessage(EXCEPTION_UNKNOWN,__FILE__,__LINE__) ;
}
return FALSE ;
}
//关闭连接
BOOL CDBConnection::Close(void)
{
try
{
if (!m_pConnImpl->isValid())
{
SetErrorMessage(CONNCTION_INVALIDHANDLE,__FILE__,__LINE__) ;
return FALSE ;
}
if (m_pConnImpl != NULL)
{
if(!m_pConnImpl->GetAutoClose())
{
//删除从其他对象复制的连接，防止连接冲突
delete m_pConnImpl ;
m_pConnImpl = new CDBConnectionImpl() ;
}
}
HRESULT hr = S_OK ;
if(DBCONNECTIONPtr->GetState() == adStateOpen)
{
hr = DBCONNECTIONPtr->Close() ;
}
return (SUCCEEDED(hr)) ;
}
catch (_com_error &e)
{
SetErrorMessage((LPCTSTR)e.Description(),__FILE__,__LINE__) ;
}
catch (...)
{
SetErrorMessage(EXCEPTION_UNKNOWN,__FILE__,__LINE__) ;
}
return FALSE ;
}
//开始事务
BOOL CDBConnection::BeginTrans(void)
{
try
{
if (!m_pConnImpl->isValid())
{
SetErrorMessage(CONNCTION_INVALIDHANDLE,__FILE__,__LINE__) ;
return FALSE ;
}
DBCONNECTIONPtr->Errors->Clear();
if (DBCONNECTIONPtr->GetState() != adStateClosed)
{
DBCONNECTIONPtr->BeginTrans() ;
return TRUE ;
}
}
catch (_com_error &e)
{
SetErrorMessage((LPCTSTR)e.Description(),__FILE__,__LINE__) ;
}
[1] [2] 下一页 | 
