ADO封装类头文件和定义

头文件 ADO.h：

// ADO.h: interface for the CADO class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_ADO_H__EE0962C4_DC53_489D_981F_B112A39F9C1B__INCLUDED_)
#define AFX_ADO_H__EE0962C4_DC53_489D_981F_B112A39F9C1B__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000


#import "C:\Program Files\Common Files\System\ado\msado15.dll" no_namespace rename("EOF","adoEOF")


class CADOConnection
{
private:
    static void InitADO();
    static void UnInitADO();
protected:
    _ConnectionPtr m_Connection;
public:
    BOOL IsOpen();
    _ConnectionPtr GetConnection();
    CString GetSQLConStr(CString IP,CString DBName);
    BOOL Open(CString ConStr);
    CADOConnection();
    virtual ~CADOConnection();

};

class CADODataSet
{
protected:
    _RecordsetPtr m_DataSet;
    CADOConnection *m_Connection;
public:
    BOOL Open(CString SQLStr,int LockType);
    void Delete();
    int GetRecordNo();
    void move(int nIndex);
    void Save();
    void SetFieldValue(CString FieldName,_variant_t Value);
    void AddNew();
    BOOL Next();
    FieldsPtr GetFields();
    int GetRecordCount();
    void SetConnection(CADOConnection *pCon);
    BOOL Open(CString SQLStr);

    CADODataSet();
    virtual ~CADODataSet();
private:
    BOOL IsOpen();
};


CADOConnection * GetConnection();
#endif // !defined(AFX_ADO_H__EE0962C4_DC53_489D_981F_B112A39F9C1B__INCLUDED_)

 

定义 ADO.CPP :

 

// ADO.cpp: implementation of the CADO class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "ADO.h"
#include <comdef.h>

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
int ConCount = 0;
CADOConnection g_Connection;//全局数据库连接对象

CADOConnection * GetConnection()
{
    return &g_Connection;
}

CADOConnection::CADOConnection()
{
    InitADO();
    m_Connection.CreateInstance("ADODB.Connection");
}

CADOConnection::~CADOConnection()
{
    if (IsOpen())
        m_Connection->Close();
    m_Connection = NULL;
    UnInitADO();
}

void CADOConnection::InitADO()
{
        if (ConCount++ == 0)
            CoInitialize(NULL);
};
void CADOConnection::UnInitADO()
{
        if (--ConCount == 0)
             CoUninitialize();
};

BOOL CADOConnection::Open(CString ConStr)
{
    if (IsOpen())
        m_Connection->Close();
    m_Connection->Open((_bstr_t)ConStr,"","",adModeUnknown);
    return IsOpen();
}

CString CADOConnection::GetSQLConStr(CString IP, CString DBName)
{
    CString Str;
    Str.Format("Provider=SQLOLEDB.1;Persist Security Info=False;User ID=sa;Initial Catalog=%s;Data Source=%s",DBName,IP);
    return Str;
}


_ConnectionPtr CADOConnection::GetConnection()
{
    return m_Connection;
}

BOOL CADOConnection::IsOpen()
{
    long State;
    m_Connection->get_State(&State);
    if (State == adStateOpen)
        return true;
    return false;
}

////////////////////////////////////////
CADODataSet::CADODataSet()
{
    m_DataSet.CreateInstance("ADODB.Recordset");
}

CADODataSet::~CADODataSet()
{
    if (IsOpen())
        m_DataSet->Close();
    m_DataSet = NULL;
    m_Connection = NULL;
}

void CADODataSet::SetConnection(CADOConnection *pCon)
{
    m_Connection = pCon;
}

int CADODataSet::GetRecordCount()
{
    if (IsOpen())
        return m_DataSet->GetRecordCount();
    else
        return 0;
}

BOOL CADODataSet::Open(CString SQLStr)
{
    if (IsOpen())
        m_DataSet->Close();
    //*/
    m_DataSet->Open(_bstr_t(SQLStr),
            _variant_t((IDispatch*)g_Connection.GetConnection(), true),
            adOpenKeyset, adLockOptimistic, adCmdText);
    return IsOpen();
    //*/

}

BOOL CADODataSet::IsOpen()
{
    long State;
    m_DataSet->get_State(&State);
    if (State == adStateOpen)
        return true;

    return false;
}

FieldsPtr CADODataSet::GetFields()
{
    return m_DataSet->GetFields();
}

BOOL CADODataSet::Next()
{
    if (m_DataSet->adoEOF)
        return false;
    m_DataSet->MoveNext();
    return true;
}

void CADODataSet::AddNew()
{
    m_DataSet->AddNew();
}

void CADODataSet::SetFieldValue(CString FieldName, _variant_t Value)
{
    m_DataSet->PutCollect((_bstr_t)FieldName,Value);
}

void CADODataSet::Save()
{
    m_DataSet->Update();
}

void CADODataSet::move(int nIndex)
{
    m_DataSet->MoveFirst();
    m_DataSet->Move(nIndex);
}

int CADODataSet::GetRecordNo()
{
    return m_DataSet->AbsolutePosition;
}

void CADODataSet::Delete()
{
    m_DataSet->Delete(adAffectCurrent);
}

BOOL CADODataSet::Open(CString SQLStr, int LockType)
{
    if (IsOpen())
        m_DataSet->Close();
    //*/
    m_DataSet->Open(_bstr_t(SQLStr),
            _variant_t((IDispatch*)g_Connection.GetConnection(), true),
            adOpenKeyset,(LockTypeEnum) LockType, adCmdText);
    return IsOpen();
}