SQLite C++操作类

前端之家收集整理的这篇文章主要介绍了SQLite C++操作类前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
为了方便sqlite的使用,封装了一个sqlite的C++类,同时支持ANSI 和UNICODE编码。代码如下:
 
文件sqlite.h)
[cpp] view plaincopy
/********************************************************************
filename: sqlite.h
created: 2012-11-05
author: firehood
 
purpose: sqlite数据库操作类
*********************************************************************/
#pragma once
#include <windows.h>
#include "..\sqlite\sqlite3.h"
#pragma comment(lib,"sqlite.lib")
 
typedef BOOL (WINAPI *QueryCallback) (void *para,int n_column,char **column_value,char **column_name);
 
typedef enum _sqlITE_DATATYPE
{
sqlITE_DATATYPE_INTEGER = sqlITE_INTEGER,
sqlITE_DATATYPE_FLOAT = sqlITE_FLOAT,
sqlITE_DATATYPE_TEXT = sqlITE_TEXT,
sqlITE_DATATYPE_BLOB = sqlITE_BLOB,
sqlITE_DATATYPE_NULL= sqlITE_NULL,
}sqlITE_DATATYPE;
 
class sqlite;
 
class sqliteDataReader
{
public:
sqliteDataReader(sqlite3_stmt *pStmt);
~sqliteDataReader();
public:
// 读取一行数据
BOOL Read();
// 关闭Reader,读取结束后调用
void Close();
// 总的列数
int ColumnCount(void);
// 获取某列的名称
LPCTSTR GetName(int nCol);
// 获取某列的数据类型
sqlITE_DATATYPE GetDataType(int nCol);
// 获取某列的值(字符串)
LPCTSTR GetStringValue(int nCol);
// 获取某列的值(整形)
int GetIntValue(int nCol);
// 获取某列的值(长整形)
long GetInt64Value(int nCol);
// 获取某列的值(浮点形)
double GetFloatValue(int nCol);
// 获取某列的值(二进制数据)
const BYTE* GetBlobValue(int nCol,int &nLen);
private:
sqlite3_stmt *m_pStmt;
};
 
class sqliteCommand
{
public:
sqliteCommand(sqlite* psqlite);
sqliteCommand(sqlite* psqlite,LPCTSTR lpsql);
~sqliteCommand();
public:
// 设置命令
BOOL SetCommandText(LPCTSTR lpsql);
// 绑定参数(index为要绑定参数的序号,从1开始)
BOOL BindParam(int index,LPCTSTR szValue);
BOOL BindParam(int index,const int nValue);
BOOL BindParam(int index,const double dValue);
BOOL BindParam(int index,const unsigned char* blobValue,int nLen);
// 执行命令
BOOL Excute();
// 清除命令(命令不再使用时需调用该接口清除)
void Clear();
private:
sqlite *m_psqlite;
sqlite3_stmt *m_pStmt;
};
 
class sqlite
{
public:
sqlite(void);
~sqlite(void);
public:
// 打开数据库
BOOL Open(LPCTSTR lpDbFlie);
void Close();
 
// 执行非查询操作(更新或删除
BOOL ExcuteNonQuery(LPCTSTR lpsql);
BOOL ExcuteNonQuery(sqliteCommand* pCmd);
 
// 查询
sqliteDataReader ExcuteQuery(LPCTSTR lpsql);
// 查询(回调方式)
BOOL ExcuteQuery(LPCTSTR lpsql,QueryCallback pCallBack);
 
// 开始事务
BOOL BeginTransaction();
// 提交事务
BOOL CommitTransaction();
// 回滚事务
BOOL RollbackTransaction();
 
// 获取上一条错误信息
LPCTSTR GetLastErrorMsg();
public:
friend class sqliteCommand;
private:
sqlite3 *m_db;
};
 
文件sqlite.cpp)
[cpp] view plaincopy
/********************************************************************
filename: sqlite.cpp
created: 2012-11-05
author: firehood
 
purpose: sqlite数据库操作类
*********************************************************************/
#include "sqlite.h"
 
const char* WcharToUtf8(const wchar_t *pwStr)
{
if (pwStr == NULL)
{
return NULL;
}
 
int len = WideCharToMultiByte(CP_UTF8,pwStr,-1,NULL,NULL);
if (len <= 0)
{
return NULL;
}
char *pStr = new char[len];
WideCharToMultiByte(CP_UTF8,pStr,len,NULL);
return pStr;
}
 
const wchar_t* Utf8ToWchar(const char *pStr)
{
if (pStr == NULL)
{
return NULL;
}
 
int len = MultiByteToWideChar(CP_UTF8,0);
if (len <= 0)
{
return NULL;
}
wchar_t *pwStr = new wchar_t[len];
MultiByteToWideChar(CP_UTF8,len);
return pwStr;
}
 
sqlite::sqlite(void):
m_db(NULL)
{
}
 
sqlite::~sqlite(void)
{
Close();
}
 
BOOL sqlite::Open(LPCTSTR lpDbFlie)
{
if(lpDbFlie == NULL)
{
return FALSE;
}
#ifdef UNICODE
if(sqlite3_open16(lpDbFlie,&m_db) != sqlITE_OK)
#else
if(sqlite3_open(lpDbFlie,&m_db) != sqlITE_OK)
#endif
{
return FALSE;
}
return TRUE;
}
 
void sqlite::Close()
{
if(m_db)
{
sqlite3_close(m_db);
m_db = NULL;
}
}
 
BOOL sqlite::ExcuteNonQuery(LPCTSTR lpsql)
{
if(lpsql == NULL)
{
return FALSE;
}
sqlite3_stmt* stmt;
#ifdef UNICODE
if(sqlite3_prepare16_v2(m_db,lpsql,&stmt,NULL) != sqlITE_OK)
#else
if(sqlite3_prepare_v2(m_db,NULL) != sqlITE_OK)
#endif
{
return FALSE;
}
sqlite3_step(stmt);
 
return (sqlite3_finalize(stmt) == sqlITE_OK) ? TRUE : FALSE ;
}
 
BOOL sqlite::ExcuteNonQuery(sqliteCommand* pCmd)
{
if(pCmd == NULL)
{
return FALSE;
}
return pCmd->Excute();
}
 
// 查询(回调方式)
BOOL sqlite::ExcuteQuery(LPCTSTR lpsql,QueryCallback pCallBack)
{
if(lpsql == NULL || pCallBack == NULL)
{
return FALSE;
}
char *errmsg = NULL;
#ifdef UNICODE
const char *szsql = WcharToUtf8(lpsql);
if(sqlite3_exec(m_db,szsql,pCallBack,&errmsg) != sqlITE_OK)
{
delete[] szsql;
return FALSE;
}
delete[] szsql;
#else
if(sqlite3_exec(m_db,&errmsg) != sqlITE_OK)
{
return FALSE;
}
#endif
return TRUE;
}
 
// 查询
sqliteDataReader sqlite::ExcuteQuery(LPCTSTR lpsql)
{
if(lpsql == NULL)
{
return FALSE;
}
sqlite3_stmt* stmt;
#ifdef UNICODE
if(sqlite3_prepare16_v2(m_db,NULL) != sqlITE_OK)
#endif
{
return FALSE;
}
return sqliteDataReader(stmt);
}
 
// 开始事务
BOOL sqlite::BeginTransaction()
{
char * errmsg = NULL;
if(sqlite3_exec(m_db,"BEGIN TRANSACTION;",&errmsg) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
// 提交事务
BOOL sqlite::CommitTransaction()
{
char * errmsg = NULL;
if(sqlite3_exec(m_db,"COMMIT TRANSACTION;;",&errmsg) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
// 回滚事务
BOOL sqlite::RollbackTransaction()
{
char * errmsg = NULL;
if(sqlite3_exec(m_db,"ROLLBACK TRANSACTION;",&errmsg) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
// 获取上一条错误信息
LPCTSTR sqlite::GetLastErrorMsg()
{
#ifdef UNICODE
return (LPCTSTR)sqlite3_errmsg16(m_db);
#else
return sqlite3_errmsg(m_db);
#endif
}
 
 
 
sqliteDataReader::sqliteDataReader(sqlite3_stmt *pStmt):
m_pStmt(pStmt)
{
 
}
 
sqliteDataReader::~sqliteDataReader()
{
Close();
}
 
// 读取一行数据
BOOL sqliteDataReader::Read()
{
if(m_pStmt == NULL)
{
return FALSE;
}
if(sqlite3_step(m_pStmt) != sqlITE_ROW)
{
return FALSE;
}
return TRUE;
}
 
// 关闭Reader,读取结束后调用
void sqliteDataReader::Close()
{
if(m_pStmt)
{
sqlite3_finalize(m_pStmt);
m_pStmt = NULL;
}
}
 
// 总的列数
int sqliteDataReader::ColumnCount(void)
{
return sqlite3_column_count(m_pStmt);
}
 
// 获取某列的名称
LPCTSTR sqliteDataReader::GetName(int nCol)
{
#ifdef UNICODE
return (LPCTSTR)sqlite3_column_name16(m_pStmt,nCol);
#else
return (LPCTSTR)sqlite3_column_name(m_pStmt,nCol);
#endif
}
 
// 获取某列的数据类型
sqlITE_DATATYPE sqliteDataReader::GetDataType(int nCol)
{
return (sqlITE_DATATYPE)sqlite3_column_type(m_pStmt,nCol);
}
 
// 获取某列的值(字符串)
LPCTSTR sqliteDataReader::GetStringValue(int nCol)
{
#ifdef UNICODE
return (LPCTSTR)sqlite3_column_text16(m_pStmt,nCol);
#else
return (LPCTSTR)sqlite3_column_text(m_pStmt,nCol);
#endif
}
 
// 获取某列的值(整形)
int sqliteDataReader::GetIntValue(int nCol)
{
return sqlite3_column_int(m_pStmt,nCol);
}
 
// 获取某列的值(长整形)
long sqliteDataReader::GetInt64Value(int nCol)
{
return (long)sqlite3_column_int64(m_pStmt,nCol);
}
 
// 获取某列的值(浮点形)
double sqliteDataReader::GetFloatValue(int nCol)
{
return sqlite3_column_double(m_pStmt,nCol);
}
 
// 获取某列的值(二进制数据)
const BYTE* sqliteDataReader::GetBlobValue(int nCol,int &nLen)
{
nLen = sqlite3_column_bytes(m_pStmt,nCol);
return (const BYTE*)sqlite3_column_blob(m_pStmt,nCol);
}
 
sqliteCommand::sqliteCommand(sqlite* psqlite):
m_psqlite(psqlite),
m_pStmt(NULL)
{
}
 
sqliteCommand::sqliteCommand(sqlite* psqlite,LPCTSTR lpsql):
m_psqlite(psqlite),
m_pStmt(NULL)
{
SetCommandText(lpsql);
}
 
sqliteCommand::~sqliteCommand()
{
 
}
 
BOOL sqliteCommand::SetCommandText(LPCTSTR lpsql)
{
#ifdef UNICODE
if(sqlite3_prepare16_v2(m_psqlite->m_db,&m_pStmt,NULL) != sqlITE_OK)
#else
if(sqlite3_prepare_v2(m_psqlite->m_db,NULL) != sqlITE_OK)
#endif
{
return FALSE;
}
return TRUE;
}
 
BOOL sqliteCommand::BindParam(int index,LPCTSTR szValue)
{
#ifdef UNICODE
if(sqlite3_bind_text16(m_pStmt,index,szValue,sqlITE_TRANSIENT) != sqlITE_OK)
#else
if(sqlite3_bind_text(m_pStmt,sqlITE_TRANSIENT) != sqlITE_OK)
#endif
{
return FALSE;
}
return TRUE;
}
 
BOOL sqliteCommand::BindParam(int index,const int nValue)
{
if(sqlite3_bind_int(m_pStmt,nValue) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
BOOL sqliteCommand::BindParam(int index,const double dValue)
{
if(sqlite3_bind_double(m_pStmt,dValue) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
BOOL sqliteCommand::BindParam(int index,const unsigned char* blobBuf,int nLen)
{
if(sqlite3_bind_blob(m_pStmt,blobBuf,nLen,NULL) != sqlITE_OK)
{
return FALSE;
}
return TRUE;
}
 
BOOL sqliteCommand::Excute()
{
sqlite3_step(m_pStmt);
 
return (sqlite3_reset(m_pStmt) == sqlITE_OK) ? TRUE : FALSE ;
}
 
void sqliteCommand::Clear()
{
if(m_pStmt)
{
sqlite3_finalize(m_pStmt);
}
}
 
使用方法
通过sqlite类操作数据库的基本代码如下:
[cpp] view plaincopy
void sqliteOperate()
{
TCHAR *szDbPath = _T("Book.db");
 
::DeleteFile(szDbPath);
 
sqlite sqlite;
 
// 打开或创建数据库
//******************************************************
if(!sqlite.Open(szDbPath))
{
_tprintf(_T("%s\n"),sqlite.GetLastErrorMsg());
return;
}
//******************************************************
 
// 创建数据库
//******************************************************
TCHAR sql[512] = {0};
_stprintf(sql,_T("%s"),
_T("CREATE TABLE [Book] (")
_T("[id] INTEGER NOT NULL PRIMARY KEY,")
_T("[name] NVARCHAR(20),")
_T("[author] NVARCHAR(20),")
_T("[catagory_id] INTEGER REFERENCES [Category]([id]),")
_T("[abstruct] NVARCHAR(100),")
_T("[path] NVARCHAR(50),")
_T("[image] BLOB);")
_T("CREATE INDEX [Book_id] ON [Book] ([id]);")
);
if(!sqlite.ExcuteNonQuery(sql))
{
printf("Create database table Failed...\n");
}
//******************************************************
 
// 插入数据【普通方式】
DWORD dwBeginTick = GetTickCount();
//******************************************************
// 当一次性插入多条记录时候,采用事务的方式,提高效率
sqlite.BeginTransaction();
// 批量插入数据
for(int i=0;i<1000;i++)
{
memset(sql,sizeof(sql));
_stprintf(sql,_T("insert into Book(name,author,catagory_id) values(‘红高粱%d‘,‘莫言‘,1)"),i);
if(!sqlite.ExcuteNonQuery(sql))
{
_tprintf(_T("%s\n"),sqlite.GetLastErrorMsg());
break;
}
}
// 提交事务
sqlite.CommitTransaction();
printf("Insert Data Take %dMS...\n",GetTickCount()-dwBeginTick);
//******************************************************
 
 
// 插入数据【通过参数绑定的方式,提交批量数据时,比上面的普通模式效率更高(提高约45%),同时可支持插入二进制数据】
dwBeginTick = GetTickCount();
//******************************************************
// 当一次性插入多条记录时候,采用事务的方式,提高效率
sqlite.BeginTransaction();
memset(sql,catagory_id,image) values(?,‘韩寒‘,?,?)"));
sqliteCommand cmd(&sqlite,sql);
// 批量插入数据
for(int i=0;i<1000;i++)
{
TCHAR strValue[16] = {0};
_stprintf(strValue,_T("他的国%d"),i);
// 绑定第一个参数(name字段值)
cmd.BindParam(1,strValue);
// 绑定第二个参数(catagory_id字段值)
cmd.BindParam(2,20);
BYTE imageBuf[] = {0xff,0xff,0xff};
// 绑定第三个参数(image字段值,二进制数据)
cmd.BindParam(3,imageBuf,sizeof(imageBuf));
if(!sqlite.ExcuteNonQuery(&cmd))
{
_tprintf(_T("%s\n"),sqlite.GetLastErrorMsg());
break;
}
}
// 清空cmd
cmd.Clear();
// 提交事务
sqlite.CommitTransaction();
printf("Insert Data Take %dMS...\n",GetTickCount()-dwBeginTick);
//******************************************************
 
// 查询
dwBeginTick = GetTickCount();
//******************************************************
memset(sql,_T("select * from Book where name = ‘他的国345‘"));
 
sqliteDataReader Reader = sqlite.ExcuteQuery(sql);
 
int index = 0;
int len = 0;
while(Reader.Read())
{
_tprintf( _T("***************【第%d条记录】***************\n"),++index);
_tprintf( _T("字段名:%s 字段值:%d\n"),Reader.GetName(0),Reader.GetIntValue(0));
_tprintf( _T("字段名:%s 字段值:%s\n"),Reader.GetName(1),Reader.GetStringValue(1));
_tprintf( _T("字段名:%s 字段值:%s\n"),Reader.GetName(2),Reader.GetStringValue(2));
_tprintf( _T("字段名:%s 字段值:%d\n"),Reader.GetName(3),Reader.GetIntValue(3));
_tprintf( _T("字段名:%s 字段值:%s\n"),Reader.GetName(4),Reader.GetStringValue(4));
// 读取图片二进制文件
const BYTE *ImageBuf = Reader.GetBlobValue(6,len);
_tprintf( _T("*******************************************\n"));
}
Reader.Close();
printf("Query Take %dMS...\n",GetTickCount()-dwBeginTick);
//******************************************************
 
sqlite.Close();
}
 
运行结果
Insert Data Take 645MS...
Insert Data Take 229MS...
***************【第1条记录】***************
字段名:id 字段值:1346
字段名:name 字段值:他的国345
字段名:author 字段值:韩寒
字段名:catagory_id 字段值:20
字段名:abstruct 字段值:(null)
*******************************************
Query Take 63MS...

猜你在找的Sqlite相关文章