一个用功能强大的ODBC API 函数访问数据库类 (转)

于 2007-12-11 14:15:48 发布
阅读量89 收藏
点赞数
文章标签： 数据库
本文介绍了一个使用ODBC API函数的数据库访问类，该类支持通过动态更改表信息来访问ODBC数据源的所有表，并提供了插入、更新、删除等操作。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

一个用功能强大的ODBC API 函数访问数据库类 (转)[@more@]

/*-- 专项技术学习　－数据库访问技术之OdbC api　 
  这个类 通过ODBC API 函数访问ODBC 数据源的数据表 　　　　　　　　　　　
  功能简介:能够帮定一个临时指定表,调用类的函数即可对该表　　　　
  插入、增加、修改、简单统计等等操作(注:通过动态改变表的信息:表名字、
  表字段总数、字段名字、字段类型、字段是否为数值、字段是否为主键。这样
  就可以访问ODBC　数据源的所有表了)　 
  另外提供一个函数　可以对该数据源的其他表（包括帮定的表）进行　
  删除、修改功能 
  LSZ 2001.08.25  */

/*头文件*/

#ifndef _WDC_DATABASE_H
#define _WDC_DATABASE_H

#include <sql.h>
#include

#define  MAX_FIELDS  500
#define  MAX_ERROR_MSG  1024
#define  MAX_DATA  500

class CWdcDatabase 
{
public:
 CWdcDatabase();
 virtual ~CWdcDatabase();
  BOOL Open(const CString& sDSN,
  const CString& sTable,
  const CString& sUserId,
  const CString& sPassword)  ; /*打开一个ODBC数据源数据库*/

public:
 BOOL m_bOpen; /*标志数据库是否打开*/
 BOOL m_bBrackets; /*标识表名、字段名是否用方扩号扩起来*/
 int m_nError; /*ODBC错误返回代码*/
 CString m_sQuery; /*操作ODBC 数据源公用的语句，用于方便构造*/
 CString m_sErrMsg; /*ODBC错误返回信息*/
 CString m_sDSN; /*ODBC 数据源的 DSN*/
 CString m_sTable;  /*临时绑定的表名*/

public:
 void DropHstmtselect(); /*确保公共查询的语句句柄释放*/
 long GetRecordCount(CString &sSel); /*取得符合条件的记录数*/
 BOOL NextMatchingRecord(); /*查询符合条件的下一条记录*/
 int SelectMatchingRecords(CString& sWhereContext); /*查询符合条件的记录*/
 BOOL UpdateRecord(); /*更新临时绑定的表的一条记录*/
 BOOL AddRecord();  /*增加临时绑定表一条新记录*/
 BOOL DeleteRecord(CString &sWhereDelete); /*删除临时绑定表符合条件记录*/
 BOOL Execute(const CString  &sText); /*执行一个SQL语句，注:可以执行存储过程*/
 void Close();  /*关闭数据库*/
public:
 int m_nFields;  /*临时绑定表字段的总数*/
 CString m_sfieldName[MAX_FIELDS]; /*临时绑定表字段名字*/
 CString m_sFieldValue[MAX_FIELDS];/*临时绑定表字段值*/
 BOOL m_bKeyField[MAX_FIELDS];/*临时绑定表字段是否为主键*/
 BOOL m_bNumeric[MAX_FIELDS];/*临时绑定表字段是否为数值类型*/

private:
 CString Quote(CString &sQuote); /*字符串替换*/
 RETCODE rc;  /*ODBC API 涵数执行返回码*/
 HSTMT hstmt, hstmt_select; /*公共语句句柄*/
 unsigned char szDSN[200]; /*连接数据源的 DSN */ 
 unsigned char szID[200]; /*连接数据源的 UID */ 
 unsigned char szPassword[200]; /*连接数据源的 PWD*/
 unsigned char szQuery[3000]; /*公共执行的语句句柄*/
 unsigned char szTemp[1000]; /*公共使用的字符串*/
 CString sTemp ; /*公共使用的字符串*/
 HENV henv;  /*环境句柄*/
 HDBC hdbc;  /*ODBC 连接句柄*/
};

#endif

/*实现文件*/

#include "stdafx.h"
#include "WdcDatabase.h"
#include
#include

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

CWdcDatabase::CWdcDatabase()
{
 m_nError = 0;
 m_bOpen = false;
 m_nFields = 0;
 m_bBrackets = false;
}

CWdcDatabase::~CWdcDatabase()
{
  if(m_bOpen)
 {
 Close() ;
 }
}

BOOL CWdcDatabase::Open(const CString& sDSN,
 const CString& sTable,
 const CString& sUserId,
 const CString& sPassword)
{
  if(m_bOpen)
 {
 Close() ;
 }
 
 hstmt_select = NULL ;
 RETCODE rc ;
 CString stemp ;

 memset(szDSN,0x00,200) ;
  memset(szID,0x00,200) ;
 memset(szPassword,0x00,200) ;
 /*Open Database */
 SQLAllocEnv(&henv) ;
 SQLAllocConnect(henv,&hdbc) ;

 strcpy((char*)szDSN,sDSN) ;

 stemp = sUserId ;
 strcpy((char*)szID,stemp) ;

  stemp = sPassword ;
 strcpy((char*)szPassword,stemp) ;

 m_sDSN  = sDSN ;
 m_sTable = sTable ;

 rc = SQLConnect(hdbc,
 szDSN,
 SQL_NTS,
 szID,
 SQL_NTS,
 szPassword,
 SQL_NTS);

 m_nError = (int) rc ;

 if((rc == SQL_SUCCESS ) ||(rc == SQL_SUCCESS_WITH_INFO))
 {
  m_bOpen = TRUE ;
  TRACE0("Open database Succeed !n") ;
  return TRUE  ;
 }
  else
 {
  m_bOpen = FALSE ;
  TRACE0("Open Database Failed !n") ;
  return FALSE ;
 }

}

void CWdcDatabase::Close()
{
  if(m_bOpen)
 {
 if(hstmt_select)
 {
 SQLFreeStmt(hstmt_select,SQL_DROP) ;
 hstmt_select = NULL ;
 }

 SQLDiSCOnnect(hdbc) ;  /*Dis  hdbc*/
 SQLFreeConnect(hdbc) ;  /*Free  Odbc*/
 SQLFreeEnv(henv) ;  /*Free  henv*/ 
 m_bOpen = false;
 TRACE0("Close database Succeed !n") ;
 }
}

BOOL CWdcDatabase::Execute(const CString &sSQL)
{
 RETCODE rc ;
  unsigned char szSQLSTATE[6] ;
 SDWORD nErr ;
 unsigned char msg[MAX_ERROR_MSG + 1] ;
 SWORD cbmsg ;

 m_sErrMsg = "" ;
 m_sQuery = sSQL ;

 if(!m_bOpen)
 return FALSE ;

 SQLAllocStmt(hdbc,&hstmt) ;
 memset((char*)szQuery,0x00,sizeof(szQuery)) ;
 strcpy((char*)szQuery,m_sQuery) ;

 rc = SQLExecDirect(hstmt,
  szQuery,
  SQL_NTS) ;

  m_nError = (int) rc ;
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  {
  SQLTransact(henv,hdbc,SQL_COMMIT);
  SQLFreeStmt(hstmt,SQL_DROP) ;
  return TRUE ;
  }
  else
  {
   m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg),
  &cbmsg) == SQL_SUCCESS)
  m_sErrMsg = "tt" + (CString)msg + "n" ;
  SQLFreeStmt(hstmt,SQL_DROP) ;
  return FALSE ;
  }

}

/*带入条件为 Where 以后的条件*/
BOOL CWdcDatabase::DeleteRecord(CString &sWhereDelete)
{
  RETCODE rc  ;
  unsigned char szSQLSTATE[6] ;
  SDWORD nErr ;
  unsigned char msg[MAX_ERROR_MSG+1] ;
  SWORD cbmsg ;

  CString sTemp ;

  if(!m_bOpen)
  return FALSE ;

  m_sQuery = "" ;
  m_sErrMsg = "" ;
 
  /*开始构造删除语句*/
  m_sQuery.Format("Delete From %s ",m_sTable) ;

/* int ff =0 ;
  for(int f= 0 ;f  {
  if(m_bKeyField[f]) /*如果是表的主键
  {
  if(ff>0) /*如果主键超过2个,要把条件连接起来
  m_sQuery += " AND " ;
  ff++ ;

  if (m_bNumeric[f])
 sTemp.Format("%s=%s",m_sFieldName[f],m_sFieldValue[f]) ;
  else
 sTemp.Format("%s=%s",m_sFieldName[f],Quote(m_sFieldValue[f])) ;
 
  m_sQuery += sTemp ;
  }
  }
*/
  if(sWhereDelete !="")
  m_sQuery +=" Where " + sWhereDelete ;

  TRACE0(m_sQuery) ;
 memset((char*)szQuery,0x00,sizeof(szQuery)) ;
  strcpy((char*)szQuery,m_sQuery) ;
  SQLAllocStmt(hdbc,&hstmt) ;
  rc = SQLExecDirect(hstmt,
  szQuery,
  SQL_NTS)  ;

 m_nError = (int) rc ;
  if(rc ==SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
 { /*如果执行成功*/
  SQLTransact(henv,hdbc,SQL_COMMIT);
 SQLFreeStmt(hstmt,SQL_DROP) ;
 return TRUE ;
 }
 else
 {
 m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg),
  &cbmsg) == SQL_SUCCESS)
  m_sErrMsg = "tt" + (CString) msg + "n" ;
  SQLFreeStmt(hstmt,SQL_DROP) ;
  return FALSE ;
 }

}

CString CWdcDatabase::Quote(CString &sText)
{
  CString sResult="" ;
  CString sChar  ;
  int iTextLen = sText.GetLength() ;

  for(int pos = 0 ;pos  {
  sChar = sText.Mid(pos,1) ;
  if (sChar == "'")
  sChar = "''" ;
  sResult += sChar ;
  }

  return  CString("'" + sResult + "'") ;

}

BOOL CWdcDatabase::AddRecord()
{
  unsigned char szSQLSTATE[6] ;
 SDWORD nErr ;
 unsigned char msg[MAX_ERROR_MSG+1] ;
 SWORD cbmsg ;

 CString sTemp ;

 if(!m_bOpen)
 return FALSE ;

  m_sQuery.Format("Insert Into %s(",m_sTable) ;
 
 int ff=0 ;
 for(int f=0 ;f {
  if(ff>0)
  m_sQuery += "," ;
  ff ++ ;
  m_sQuery += m_sFieldName[f] ;
 }
 
 m_sQuery += ") values(" ;
 ff = 0 ;
  for(f=0 ;f {
 if(ff>0)
 m_sQuery += "," ;

 ff ++ ;
 /*---如果数据不为空*/
    if(m_sFieldValue[f]!="")
  {
  if(m_bNumeric[f])
  sTemp.Format("%s",m_sFieldValue[f]) ;
  else
  sTemp.Format("%s",Quote(m_sFieldValue[f])) ;
  }
  else
  { /*如果数据为空*/
  sTemp.Format("%s","NULL") ;
  }

 m_sQuery += sTemp ;
 }

 m_sQuery += ")" ;

 memset((char*)szQuery,0x00,sizeof(szQuery)) ;
 strcpy((char*)szQuery,m_sQuery) ;
 SQLAllocStmt(hdbc,&hstmt) ;
 rc = SQLExecDirect(hstmt,
  szQuery,
  SQL_NTS) ;

 m_nError = (int) rc ;
 if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
 {
  SQLTransact(henv,hdbc,SQL_COMMIT);
  SQLFreeStmt(hstmt,SQL_DROP) ;
 hstmt=NULL ;
 return TRUE ;
 }
 else
 {
 m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg),
  &cbmsg) == SQL_SUCCESS )

  m_sErrMsg += "tt" + (CString) msg +"n" ;
  SQLFreeStmt(hstmt,SQL_DROP) ;
   return FALSE ;
 }
}
/*更新表的指定纪录*/
BOOL CWdcDatabase::UpdateRecord()
{
  unsigned char szSQLSTATE[6] ;
  SDWORD nErr ;
  unsigned char msg[MAX_ERROR_MSG + 1] ;
  SWORD cbmsg ;

  RETCODE rc ;
  CString sTemp ;
 
  if(!m_bOpen)
  return FALSE ;

  m_sQuery.Format("Update %s Set ",m_sTable) ;
 
  int ff = 0 ;
  for(int f= 0 ;f  {
  if (ff>0)
  m_sQuery += "," ;
 
  ff ++ ;

  if(m_sFieldValue[f]!="")  /*数据为空*/
  {
  if(m_bNumeric[f])
  sTemp.Format("%s=%s",m_sFieldName[f],m_sFieldValue[f]) ;
  else
  sTemp.Format("%s=%s",m_sFieldName[f],Quote(m_sFieldValue[f])) ;
  }
  else
  sTemp.Format("%s=%s",m_sFieldName[f],NULL) ;

  m_sQuery += sTemp ;
  }
  m_sQuery += " Where " ;

  ff = 0 ;
  for(f=0 ;f  {
  if(m_bKeyField[f]) /*如果为主键*/
  {
  if(ff>0)
  m_sQuery += "," ;
  ff ++ ;
  if(m_bNumeric[f])
  sTemp.Format("%s=%s",m_sFieldName[f],m_sFieldValue[f]) ;
  else
  sTemp.Format("%s=%s",m_sFieldName[f],Quote(m_sFieldValue[f])) ;

  m_sQuery += sTemp ;
  }
  }

  memset((char*)szQuery,0x00,sizeof(szQuery)) ;
  strcpy((char*)szQuery,m_sQuery) ;
  SQLAllocStmt(hdbc,
  &hstmt) ;
  rc = SQLExecDirect(hstmt,
  szQuery,
  SQL_NTS) ;
  m_nError = (int) rc ;

  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  { 
  SQLTransact(henv,hdbc,SQL_COMMIT);
  SQLFreeStmt(hstmt,SQL_DROP) ;
  return TRUE ;
  }
  else
  { 
 m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg) ,
  &cbmsg ) == SQL_SUCCESS )
  m_sErrMsg += "tt" +(CString) msg + "n" ;
 
  SQLFreeStmt(hstmt,SQL_DROP) ;
  return FALSE ;
  }

}

/* 查找纪录---
返回值
  1： 即没有错误发生的找到纪录
  -1： 在查找过程中出现错误
  0： 查找过程当中没有错误出现，但是没有找到适合条件的纪录，即没有返回纪录  */
int CWdcDatabase::SelectMatchingRecords(CString &sWhereContext)
{
  unsigned char szSQLSTATE[6];
  SDWORD nErr ;
  unsigned char msg[MAX_ERROR_MSG+1] ;
  SWORD cbmsg ;
  CString sTemp ;
 
  char szData[MAX_DATA] ;
  SDWORD cbData ;
  RETCODE rc ;

  if(!m_bOpen)
  return -1 ;

  sTemp="" ;
  int ff=0 ;
  for(int j=0;j  {
  if(ff>0)
  sTemp += "," ;
  ff ++ ;
  sTemp += m_sFieldName[j] ;
  }

  if(sWhereContext == "")
  m_sQuery.Format("Select %s from %s",sTemp,m_sTable) ;
  else
  m_sQuery.Format("Select %s From %s Where %s",sTemp,m_sTable,sWhereContext) ;
 
  SQLAllocStmt(hdbc,&hstmt_select) ;
  memset((char*)szQuery,0x00,sizeof(szQuery)) ;
  strcpy((char*)szQuery,m_sQuery) ;

  rc = SQLExecDirect(hstmt_select,
  szQuery,
  SQL_NTS) ;
  m_nError = (int) rc ;

  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  {
  rc = SQLFetch(hstmt_select );
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  {  /*执行成功，并且有数据*/
  for(int f = 0 ; f {
  rc = SQLGetData(hstmt_select,
  f+1 ,
  SQL_C_CHAR,
  szData,
  sizeof(szData) ,
  &cbData ) ;
 
  if(cbData != SQL_NULL_DATA)
 {  /*如果数据不为空*/
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
    m_sFieldValue[f] = CString(szData) ;
  else
  m_sFieldValue[f] = "" ;
 }
 else
 m_sFieldValue[f] = "" ;

 } 

  return 1 ;  /*执行成功，并且有数据*/
  }
  else if(rc == 100)
  {  /*查找成功，但是没有数据*/
  return 0 ;
  }
  }
    m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt_select,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg),
  &cbmsg) == SQL_SUCCESS )
  m_sErrMsg += "tt" +CString(msg) + "n" ;
 
  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;
  return -1 ;
}

BOOL CWdcDatabase::NextMatchingRecord()
{
  unsigned char szSQLSTATE[6] ;
  SDWORD nErr ;
  unsigned char msg[MAX_ERROR_MSG+1] ;
  SWORD cbmsg ;

  char szData[MAX_DATA] ;
  SDWORD cbData ;
  RETCODE rc ;

  if(!m_bOpen)
  return FALSE ;

  rc = SQLFetch(hstmt_select) ;
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  {
  for(int f =0 ;f  {
  rc = SQLGetData(hstmt_select,
  f+1 ,
  SQL_C_CHAR,
  szData,
  sizeof(szData),
  &cbData) ;
  if(cbData !=SQL_NULL_DATA)
 { /*如果数据不为空*/
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
    m_sFieldValue[f] = (CString) szData ;
  else
  m_sFieldValue[f] = "" ;
 }
 else
 m_sFieldValue[f] = "" ;

  } 
  return TRUE ;
  }

 m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt_select,
  szSQLSTATE,
  &nErr,
  msg,
  sizeof(msg),
  &cbmsg )==SQL_SUCCESS )
 
  m_sErrMsg += "tt" +CString (msg) + "n" ;
  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;

  return FALSE ;
}
/*统计符合条件的纪录总数
  返回值
  -1: 执行出错
  >0: 执行成功,并且返回统计结果 */
long CWdcDatabase::GetRecordCount(CString &sSel)
{
  unsigned char szSQLSTATE[6] ;
  SDWORD nErr ;
  unsigned char ErrMsg[500] ;
  SWORD cbmsg;
  long lRet ;  /*返回值*/

  unsigned char lpSelect[1000];
  memset((char*)lpSelect,0x00,sizeof(lpSelect)) ;
  if(sSel=="")
  sTemp.Format("Select count(*) from %s ",m_sTable) ;
  else
  sTemp.Format("Select Count(*) From %s Where %s",m_sTable,sSel) ;

  if(!m_bOpen)
  return -1  ;
  strcpy((char*)lpSelect,sTemp) ;
  SQLAllocStmt(hdbc,&hstmt_select) ;
  rc=SQLExecDirect(hstmt_select,
  lpSelect,
  SQL_NTS) ;
  if(rc==SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  { /* SQLExecDirect True*/
  SQLBindCol(hstmt_select,1,SQL_INTEGER,&lRet,sizeof(int),NULL) ;
  rc = SQLFetch(hstmt_select) ;
  if(rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO)
  { /*有数据*/
  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;
  return lRet ;
  }
  else if (rc == 100)
  { /*没有数据*/
  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;
  return 0 ;
  }
  else if (rc < 0)
  { /*出错*/
  goto end ;
  }
  }
  else
  { /* SQLExecDirect FALSE */
  goto end ;
  }

end:
  {
  m_sErrMsg = "" ;
  while(SQLError(0,
  0,
  hstmt_select,
  szSQLSTATE,
  &nErr,
  ErrMsg,
  sizeof(ErrMsg),
  &cbmsg)==SQL_SUCCESS)
  m_sErrMsg += "tt" + CString(ErrMsg) +"n" ; 

  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;
  return -1 ;
  }
}
/*非常有用函数,用在确保语句句柄释放*/
void CWdcDatabase::DropHstmtSelect()
{
 if(hstmt_select !=NULL)
 {
  SQLFreeStmt(hstmt_select,SQL_DROP) ;
  hstmt_select = NULL ;
 }
}

 


来自 “ ITPUB博客 ” ，链接：http://blog.itpub.net/10752043/viewspace-991456/，如需转载，请注明出处，否则将追究法律责任。

转载于:http://blog.itpub.net/10752043/viewspace-991456/

csd3176
关注
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值