C++通过ODBC操作Excle表

#include <windows.h>
#include <sqlext.h>
#include <stdio.h>

#ifdef  _MSC_VER
#pragma comment(lib,"odbc32.lib")
#define SQLLEN SQLINTEGER
#endif

enum errtype { ERRORFREE,  ERRACTION, XLSBUSY, TXTBUSY, TXTOPENERROR,
 SQLHANDLEENVERROR,  SQLSETENVERROR,    SQLHANDLEDBCERROR,
 SQLCONNECTERROR,    SQLALLOCSTMTERROR, SQLCREATEERROR,
 SQLSELECTERROR,     SQLBINDERROR       } ;

SQLHENV       henv;
SQLHDBC       hdbc;
SQLHSTMT      hstmt;

static void dbError( LPSTR lp, SQLSMALLINT handleType, SQLHANDLE handle)
{
 　　BYTE buf[250], sqlstate[15];
　　 SQLGetDiagRec( handleType,handle, 1, (SQLCHAR*)sqlstate, NULL,(SQLCHAR*)buf, sizeof(buf),NULL);
 　　//fprintf(stderr, "%s: %s SQLSTATE=%s\n",lp, buf, sqlstate);
}

static void dbCleanup(SQLHENV henv,SQLHDBC hdbc,SQLHSTMT hstmt)
{
 　　if (hstmt != SQL_NULL_HANDLE)

　　{
  　　　　SQLFreeStmt(hstmt, SQL_UNBIND);
  　　　　SQLFreeHandle(SQL_HANDLE_STMT,hstmt);
　　 }
 　　if (hdbc != SQL_NULL_HANDLE)

　　 {
  　　　　SQLDisconnect(hdbc);
  　　　　SQLFreeHandle(SQL_HANDLE_DBC,hdbc);
　　 }
　　 if (henv != SQL_NULL_HANDLE)

　　　　SQLFreeHandle(SQL_HANDLE_ENV,henv);
}

static void dbErrorCleanup( LPSTR lp,SQLSMALLINT handleType, SQLHENV henv,SQLHDBC hdbc,SQLHSTMT hstmt)
{
 　　SQLHANDLE handle;
 　　switch (handleType)

　　{
　　　　 case SQL_HANDLE_STMT:
　　　　　　  handle = hstmt;
　　　　  　　break;
　　　　 case SQL_HANDLE_DBC:
　　　　　　  handle = hdbc;
　　　　　　  break;
　　　　 default:
　　　　　　  handle = henv;
　　}
 　　//if (handle != SQL_NULL_HANDLE)

　　 dbError( lp, handleType, handle  );
 　　dbCleanup(henv, hdbc, hstmt );
}

static BOOL isFileOpen(const BYTE *fileName)
{
 　　HANDLE hFile;
 　　// 以非共享方式打开文件. 检查EXCEL文件当前是否在使用中...........
 　　hFile = CreateFile((LPCSTR)fileName,GENERIC_READ,0,NULL,OPEN_EXISTING,0,NULL);
　　 if (hFile == INVALID_HANDLE_VALUE)

　　{
  　　　　if (GetLastError() == ERROR_SHARING_VIOLATION)

　　　　　　return TRUE; // 使用中
  　　　　else return FALSE; // 未使用中,或者无此文件(现在可以创建)
 　　}

　　  else

　　 {
  　　　　CloseHandle(hFile);
 　　　　 return FALSE;
　　 }
}

int OpenExcel(BYTE *szExcelName, BYTE *szSheetName, BYTE *szFieldName[], BYTE *szFieldType[], int nColCount)
{
 　　SQLRETURN  retcode;
 　　BYTE       szSql[512], *pszSql=szSql;
　　 BYTE       szdatabase[256];
 　　int        rc, i;
 　　henv = hdbc = hstmt = SQL_NULL_HANDLE;
 　　if ( isFileOpen(szExcelName) )  return  XLSBUSY;
 　　// 准备环境 -------------------;
 　　retcode = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &henv);
　　 if (retcode != SQL_SUCCESS)
　　 {
　　　　  dbErrorCleanup( "SQLAllocHandle(ENV)",SQL_HANDLE_ENV,henv,hdbc,hstmt);
　　　　  return SQLHANDLEENVERROR;
　　 }
　　 retcode = SQLSetEnvAttr(henv,SQL_ATTR_ODBC_VERSION, (SQLPOINTER) SQL_OV_ODBC3,0);
　　 if (retcode != SQL_SUCCESS)
 　　{
　　　　  dbErrorCleanup( "SQLSetEnvAttr()",SQL_HANDLE_ENV,henv,hdbc,hstmt);
　　　　  return SQLSETENVERROR;
　　 }
　　 retcode = SQLAllocHandle(SQL_HANDLE_DBC, henv, &hdbc);
　　 if (retcode != SQL_SUCCESS)
　　 {
　　　　  dbErrorCleanup( "SQLAllocHandle(DBC)",SQL_HANDLE_ENV,henv,hdbc,hstmt);
　　　　  return SQLHANDLEDBCERROR;
　　 }
　　 // 与数据库建立连接 ------------------;
　　 sprintf((char*)szdatabase,"DRIVER=Microsoft Excel Driver (*.xls);CREATE_DB=%s;DBQ=%s;READONLY=FALSE;EXCLUSIVE=Yes;",
　　  szExcelName, szExcelName);
　　 fprintf(stderr, "DATABASE: %s\n", szdatabase);
　　 retcode = SQLDriverConnect(hdbc, NULL, (SQLCHAR *)szdatabase, (short) (strlen((char*)szdatabase)+1), NULL, 0, NULL, SQL_DRIVER_NOPROMPT);
　　 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
　　 {
　　　　  dbErrorCleanup( "SQLDriverConnect()",SQL_HANDLE_DBC,henv,hdbc,hstmt);
　　　　  return SQLCONNECTERROR;
　　 }
　　 retcode = SQLAllocHandle(SQL_HANDLE_STMT,hdbc, &hstmt);
　　 if (retcode != SQL_SUCCESS)
　　 {
　　　　  dbErrorCleanup( "SQLAllocHandle(STMT)",SQL_HANDLE_DBC,henv,hdbc,hstmt);
　　　　  return  SQLALLOCSTMTERROR;
　　 }
　　 // 构造执行 CREATE TABLE 语句 ----------------------;
　　 pszSql = szSql + sprintf((char*)szSql, "CREATE TABLE %s (", szSheetName);
　　 for (i=0; i<nColCount; ++i)
　　 {
　　　　  if (i) *pszSql++ = ',';
　　　　  pszSql += sprintf((char*)pszSql, "%s %s", szFieldName[i], szFieldType[i]);
　　 }
　　 *pszSql++ = ')';
　　 *pszSql   = '\0';
　　 fprintf(stderr, "STATEMENT: [%s]\n", szSql);
　　 retcode = SQLExecDirect(hstmt, (unsigned char*)szSql, SQL_NTS);
　　 if (retcode != SQL_SUCCESS)
　　 {
　　　　  char sqlState[15];
　　　　  SQLGetDiagField( SQL_HANDLE_STMT, hstmt,1,SQL_DIAG_SQLSTATE,(unsigned char*)sqlState,sizeof(sqlState),NULL);
　　  if ( lstrcmpiA((char*)sqlState,"42S01") == 0 )
　　  {
　　　　   // Already existing...
　　　　   dbErrorCleanup( " SQLExecDirect()",SQL_HANDLE_STMT,henv,hdbc,hstmt);
　　　　   return SQLCREATEERROR;
　　  }
 }
 
 return 0;
}

int ExecExcle(BYTE *szExcelName, BYTE *szSheetName, BYTE *szFieldName[], BYTE *szFieldType[], int nColCount)
{
　　 return 0;
}

int AppendExcel(BYTE *szSheetName, BYTE *szFieldName[], int nColCount, BYTE *szData)
{
　　 int   i, rc;
　　 BYTE  szSql[512], /**s, *p, *d,*/ *psz;
　　 //构造 INSERT INTO 语句 --------------------------------;
　　 psz = szSql + sprintf((char*)szSql, "INSERT INTO %s (", szSheetName);
　　 for (i=0; i<nColCount; ++i) {
　　  if (i) *psz++ = ',';
　　  psz += sprintf((char*)psz, "%s", szFieldName[i]);
　　 }
　　 psz += sprintf((char*)psz, ") VALUES (%s)", szData);
　　 fprintf(stderr, "STATEMENT: %s\n", szSql);
　　 rc = (int) SQLExecDirect(hstmt, (SQLCHAR*)szSql, SQL_NTS);
　　 if (rc != SQL_SUCCESS) {
　　  dbError( " SQLExecDirect()",SQL_HANDLE_STMT,hstmt);
　　 }
 　　return rc;
}

void CloseExcel()
{
　　 fprintf(stderr, "CleanUp: henv=%08X; hdbc=%08X, hstmt=%08X.\n", henv, hdbc, hstmt);
　　 dbCleanup(henv, hdbc, hstmt );
}

//------------------ Test data ---------------------------------------;
static BYTE szExcelName[] = "TestExcle.xls";          // Excel 文件名;
static BYTE szSheetNameA[] = "姓名";         // 工作表名;
static char *szFieldNameA[] = { "Name", "Age", "Sex" };    // 字段名;
static char *szFieldTypeA[] = { "TEXT", "NUMBER", "TEXT" };    // 字段类型;
static BYTE szSheetNameB[] = "备忘";         // 工作表名;
static char *szFieldNameB[] = { "日期", "时间", "事件" };   // 字段名;
static char *szFieldTypeB[] = { "TEXT", "TEXT", "TEXT" };   // 字段类型;

int main()
{
 　　if (OpenExcel(szExcelName, szSheetNameA, (unsigned char**)szFieldNameA, (unsigned char**)szFieldTypeA, 3)==0)
　　 {
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'Koby',33, 'man'");
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'James',44, 'man'" );
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'Antony',55, 'man'");
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'Jermy',66, 'woman'" );
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'Kevin',77, 'woman'");
　　　　  AppendExcel(szSheetNameA, (BYTE**)szFieldNameA, 3, (BYTE*)"'Marry',88, 'woman'");
　　　　  printf("Excel文件[%s]中的工作表[%s] 操作成功.\n",  szExcelName, szSheetNameA);
　　　　  CloseExcel();
　　 }
　　 if (OpenExcel(szExcelName, szSheetNameB, (unsigned char**)szFieldNameB, (unsigned char**)szFieldTypeB, 3)==0)
　　 {
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月1日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月2日','12:00','吃饭'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月3日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月4日','12:00','吃饭'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月5日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月6日','12:00','吃饭'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月7日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月8日','12:00','吃饭'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月9日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月10日','12:00','吃饭'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月11日','18:00','喝酒'");
　　　　  AppendExcel(szSheetNameB, (BYTE**)szFieldNameB, 3, (BYTE*)"'5月12日','12:00','吃饭'");
　　　　  printf("Excel文件[%s]中的工作表[%s] 操作成功.\n",  szExcelName, szSheetNameB);
　　　　  CloseExcel();
　　 }

　　 return 0;
}

 

转载于:https://www.cnblogs.com/TeaWater/archive/2012/07/20/2600734.html