sqlite数据库包装类

sqlite 是 D. Richard Hipp 用 C 语言编写的开源嵌入式数据库引擎。它是完全独立的，不具有外部依赖性。我下面提供一个sqlite数据库包装类，供读者在相应的场合使用。

sqlite的官方网站下载地址是 http://www.sqlite.org/download.html，有兴趣的读者可以到这上面去下载最新的版本。

同时，要关注sqlite处理语句与标准sql稍有不同的几个地方（比如Top语句等），具体的可以去网上搜索。

（一）头文件 h

#ifndef _Q_MEMORY_DB_H
#define _Q_MEMORY_DB_H

#ifdef __cplusplus
extern "C" {
#endif
#ifdef _WIN32
#pragma comment(lib,"lib/sqlite3.lib")
#include "include/sqlite3.h"
#else
#include <sqlite3.h>
#define sqlITE_THREADSAFE 1//支持多线程
#endif

#ifdef __cplusplus
}
#endif

#define BS_LINE_TABLE_NAME "T_BL"
#define BS_LINE_DROP_sql "drop table T_BL;"
#define BS_LINE_CREAT_sql "create table T_BL(sqlITE_ID integer primary key,/
LON real,/
LAT real,/
CELLID integer,/
LAC integer); "
#define BS_LINE_INSERT_sql "insert into T_BL values(?,?,?); "
#define BS_LINE_CREATE_INDEX_CI_LAC_sql "create index idx_cellid_lac on T_BL(CELLID,LAC);"

class CMemDB
{
private:
bool InMemory;
int nTimeOut;
bool eof;
sqlite3 *db;
sqlite3_stmt* stmt;
int sql_code;
int total_rows,total_cols;
void Finalize();
void Close();
bool ExistTable(CString tbName);
public:

enum MemDBFieldType{ft_INT = 1,ft_FLOAT,ft_TEXT,ft_BLOB,ft_NULL};

CMemDB(int ATimeOut=60000);
virtual ~CMemDB();

void CreateMemoryDB();
const char* GetErrMsg();
//使用sqlite3_stmt
//执行带绑定参数的sql语句
void Preparesql(CString szsql);
bool Bind(int nCol,const char* szText);
bool Bind(int nCol,const int nValue);
bool Bind(int nCol,const double dwValue);
bool Bind(int nCol,const BYTE* vBlob,int nLen);
bool BindNULL(int nCol);
int Exec();
//执行sql语句(有返回结果)
int Execsql(CString szsql);
//直接执行(无返回结果)
int ExecCallBacksql(CString szsql,sqlite3_callback func = 0,void* para= NULL);//带回调函数的sql
//直接执行END
//字段操作
int FindIndex(CString szField);
const char* FieldName(int nCol);
const int GetFieldType(CString szField);
const int GetFieldType(int nCol);
bool FieldIsNULL(CString szField);
bool FieldIsNULL(int nCol);
int GetIntFieldByName(CString szField,int nDefault=0);
int GetIntFieldByIndex(int nCol,int nDefault=0);
double GetDoubleFieldByName(CString szField,double fDefault=0.00);
double GetDoubleFieldByIndex(int nCol,double fDefault=0.00);
const char* GetTextFieldByName(CString szField,const char* sDefault="");
const char* GetTextFieldByIndex(int nCol,const char* sDefault="");
const BYTE* GetBlobFieldByName(CString szField,int *nLen);
const BYTE* GetBlobFieldByIndex(int nCol,int *nLen);
//行列操作
bool Eof();
void Reset();
int Next();
int GetCols();
//使用sqlite3_stmt END
//Other Operation
bool BeginTran();
bool CommitTran();
bool RollbackTran();
void setTimeOut(int ATimeOut);
void Interrupt();
sqlite_int64 GetLastRowID();
//特殊应用
protected:
CString CurTableName;

public:
//针对基站和DT的拉线
bool FindBLKey(CString ACIValue,CString ALACValue);
void CreateBLTable();

//读取shape文件的dbf(主要是点数据的文本内容，线和面一般不做处理)
bool FindKey(CString szID);
CString GetValue(CString oid,int nFieldIndex);
void AddDBFromFile(CString dbFileName);

};
#endif

（二）实现文件

#include "stdafx.h"
#include "kml_baseclass.h"
#include "MapFileReader.h"
#include "MemDB.h"

CMemDB::CMemDB(int ATimeOut)
{
CurTableName = "";
db = NULL; stmt = NULL;eof = false;
InMemory = false;
total_cols = -1;total_rows = -1;;
sql_code = sqlite3_open(":memory:",&db);
if (sql_code == sqlITE_OK)
InMemory = true;
setTimeOut(ATimeOut);
}
CMemDB::~CMemDB()
{
Close();
}

void CMemDB::CreateMemoryDB()
{
Close();
sql_code = sqlite3_open(":memory:",&db);
if (sql_code == sqlITE_OK)
InMemory = true;
}
void CMemDB::Finalize()
{
if (stmt)
{
sql_code = sqlite3_finalize(stmt);
stmt = NULL;
}
}
void CMemDB::Close()
{
Finalize();
if (db)
{
sqlite3_close(db);
db = NULL;
}
InMemory = false;
}

bool CMemDB::ExistTable(CString tbName)
{
bool ret = false;
if (db)
{
if (Execsql("select count(*) from sqlite_master where type='table' and name='"+tbName+"'") != -1)
{
if (GetIntFieldByIndex(0)==1)
{
ret =true;
}
}
}
return ret;
}

const char* CMemDB::GetErrMsg()
{
return db?sqlite3_errmsg(db):"";
}

int CMemDB::ExecCallBacksql(CString szsql,sqlite3_callback func,void* para)
{
int ret = -1;
char* szMsg = NULL;
if (db)
{
sql_code = sqlite3_exec(db,AnsiToUtf8(szsql),func,para,&szMsg);
if (sql_code == sqlITE_OK)
ret = 0;
if (szMsg != NULL)
{
sqlite3_free(szMsg);
szMsg = NULL;
}
}
return ret;
}
void CMemDB::Preparesql(CString szsql)
{
Finalize();
if ((sql_code = sqlite3_prepare(db,-1,&stmt,0)) != sqlITE_OK)
{
stmt = NULL;
}
}
bool CMemDB::Bind(int nCol,const char* szText)
{
bool bRet = false;
if (stmt != NULL)
{
if ((sql_code =sqlite3_bind_text(stmt,nCol+1,szText,sqlITE_TRANSIENT)) == sqlITE_OK)
bRet = true;
}
return bRet;
}
bool CMemDB::Bind(int nCol,const int nValue)
{
bool bRet = false;
if (stmt != NULL)
{
if ((sql_code =sqlite3_bind_int(stmt,nValue)) == sqlITE_OK)
bRet = true;
}
return bRet;
}
bool CMemDB::Bind(int nCol,const double dwValue)
{
bool bRet = false;
if (stmt != NULL)
{
if ((sql_code =sqlite3_bind_double(stmt,dwValue)) == sqlITE_OK)
bRet = true;
}
return bRet;
}
bool CMemDB::Bind(int nCol,int nLen)
{
bool bRet = false;
if (stmt != NULL)
{
if ((sql_code = sqlite3_bind_blob(stmt,(const void*)vBlob,nLen,sqlITE_TRANSIENT)) == sqlITE_OK)
bRet = true;
}
return bRet;
}
bool CMemDB::BindNULL(int nCol)
{
bool bRet = false;
if (stmt != NULL)
{
if ((sql_code =sqlite3_bind_null(stmt,nCol+1)) == sqlITE_OK)
bRet = true;
}
return bRet;
}
int CMemDB::Exec()
{
int ret = -1;
if (stmt != NULL && db != NULL)
{
if ((sql_code = sqlite3_step(stmt)) == sqlITE_DONE)
{
ret = sqlite3_changes(db);
}
//sql_code = sqlite3_reset(stmt);
}
return ret;
}
int CMemDB::Execsql(CString szsql)
{
int ret = -1;
if (db)
{
Preparesql(szsql);
if (stmt)
{
sql_code = sqlite3_step(stmt);
if (sql_code == sqlITE_DONE)
{
eof = true;
//ret = 0;
}else if (sql_code == sqlITE_ROW)
{
eof = false;
ret = 0;
}else {
Finalize();
total_rows = 0;
total_cols = 0;
}
if (stmt)
{
total_cols = sqlite3_column_count(stmt);
}
}
}
return ret;
}

int CMemDB::FindIndex(CString szField)
{
int index = -1;
CString tmpstr,tmpfield;
if (stmt != NULL)
{
tmpfield = szField;
tmpfield.TrimLeft();
tmpfield.TrimRight();
if (tmpfield != "")
{
for(int i = 0;i<total_cols;i++)
{
const char* sztmp = sqlite3_column_name(stmt,i);
tmpstr = sztmp;
tmpstr.MakeUpper();
if ( sztmp != 0 && strcmp((LPCTSTR)tmpfield,(LPCTSTR)tmpstr)==0)
{
index = i;
break;
}
}
}
}
return index;
}
const char* CMemDB::FieldName(int nCol)
{
if (!stmt) return NULL;
if (!(nCol <0 || nCol > total_cols - 1))
return sqlite3_column_name(stmt,nCol);
else return NULL;
}
const int CMemDB::GetFieldType(CString szField)
{
int index = -1;
if (stmt == NULL)
return index;
index = FindIndex(szField);
if (index == -1)
return index;
else
return sqlite3_column_type(stmt,index);
}
const int CMemDB::GetFieldType(int nCol)
{
int index = -1;
if (stmt == NULL)
return index;
if (nCol <0 || nCol > total_cols - 1)
return index;
else
return sqlite3_column_type(stmt,nCol);
}
bool CMemDB::FieldIsNULL(CString szField)
{
int index;
bool bRet = true;
if ((index = GetFieldType(szField)) != -1)
bRet = (index == sqlITE_NULL);
return bRet;
}
bool CMemDB::FieldIsNULL(int nCol)
{
int index;
bool bRet = true;
if ((index = GetFieldType(nCol)) != -1)
bRet = (index == sqlITE_NULL);
return bRet;
}
int CMemDB::GetIntFieldByName(CString szField,int nDefault)
{
if (!stmt)
return nDefault;
else{
int index = FindIndex(szField);
if (index == -1) return nDefault;
if (FieldIsNULL(szField)) return nDefault;
return sqlite3_column_int(stmt,index);
}
}
int CMemDB::GetIntFieldByIndex(int nCol,int nDefault)
{
if (!stmt)
return nDefault;
else{
if (FieldIsNULL(nCol)) return nDefault;
return sqlite3_column_int(stmt,nCol);
}
}
double CMemDB::GetDoubleFieldByName(CString szField,double fDefault)
{
if (!stmt)
return fDefault;
else{
int index = FindIndex(szField);
if (index == -1) return fDefault;
if (FieldIsNULL(szField)) return fDefault;
return sqlite3_column_double(stmt,index);
}
}
double CMemDB::GetDoubleFieldByIndex(int nCol,double fDefault)
{
if (!stmt)
return fDefault;
else{
if (FieldIsNULL(nCol)) return fDefault;
return sqlite3_column_double(stmt,nCol);
}
}
const char* CMemDB::GetTextFieldByName(CString szField,const char* sDefault)
{
if (!stmt)
return sDefault;
else{
int index = FindIndex(szField);
if (index == -1) return sDefault;
if (FieldIsNULL(szField)) return sDefault;
return (const char*)sqlite3_column_text(stmt,index);
}
}
const char* CMemDB::GetTextFieldByIndex(int nCol,const char* sDefault)
{
if (!stmt)
return sDefault;
else{
if (FieldName(nCol) == NULL) return sDefault;
if (FieldIsNULL(nCol)) return sDefault;
return (const char*)sqlite3_column_text(stmt,nCol);
}
}
const BYTE* CMemDB::GetBlobFieldByName(CString szField,int *nLen)
{
BYTE* pData = 0;
*nLen = 0;
if (!FieldIsNULL(szField))
{
int index = FindIndex(szField);
if (index != -1)
{
*nLen = sqlite3_column_bytes(stmt,index);
pData = (BYTE*)sqlite3_column_blob(stmt,index);
}
}
return pData;
}
const BYTE* CMemDB::GetBlobFieldByIndex(int nCol,int *nLen)
{
BYTE* pData = 0;
*nLen = 0;
if (!FieldIsNULL(nCol))
{
*nLen = sqlite3_column_bytes(stmt,nCol);
pData = (BYTE*)sqlite3_column_blob(stmt,nCol);
}
return pData;
}
bool CMemDB::Eof()
{
return eof;
}
void CMemDB::Reset()
{
if(stmt)
{
eof = !(sqlite3_reset(stmt)==sqlITE_OK);
}
}
int CMemDB::Next()
{
int ret = -1;
if (stmt)
{
sql_code = sqlite3_step(stmt);
if (sql_code == sqlITE_DONE)
{
eof = true;
ret = 0;
}else if (sql_code == sqlITE_ROW)
{
eof = false;
ret = 0;
}else {
Finalize();
}
}
return ret;
}
int CMemDB::GetCols()
{
return total_cols;
}
bool CMemDB::BeginTran()
{
return (ExecCallBacksql("begin transaction;")!=-1);
}
bool CMemDB::CommitTran()
{
return (ExecCallBacksql("commit transaction;")!=-1);
}
bool CMemDB::RollbackTran()
{
return (ExecCallBacksql("rollback transaction;")!=-1);
}
void CMemDB::setTimeOut(int ATimeOut)
{
if (db)
{
nTimeOut = ATimeOut;
sqlite3_busy_timeout(db,nTimeOut);
}
}
void CMemDB::Interrupt()
{
if (db)
{
sqlite3_interrupt(db);
}
}

sqlite_int64 CMemDB::GetLastRowID()
{
return sqlite3_last_insert_rowid(db);
}

void CMemDB::CreateBLTable()
{
if (ExistTable(BS_LINE_TABLE_NAME))
{
ExecCallBacksql(BS_LINE_DROP_sql);
}
ExecCallBacksql(BS_LINE_CREAT_sql);
ExecCallBacksql(BS_LINE_CREATE_INDEX_CI_LAC_sql);
CurTableName = BS_LINE_TABLE_NAME;
}

bool CMemDB::FindBLKey(CString ACIValue,CString ALACValue)
{
CString szsql;
bool ret = false;
if (db && CurTableName.IsEmpty() == false)
{
szsql = "select * from " + CurTableName + " where CELLID="+ACIValue+" and LAC="+ALACValue + " limit 0,1";
if (Execsql(szsql) == 0)
ret = true;
}
return ret;
}

bool CMemDB::FindKey(CString szID)
{
CString szsql;
bool ret = false;
if (db && CurTableName.IsEmpty() == false)
{
szsql = "select * from " + CurTableName + " where sqlITE_ID ="+szID;
if (Execsql(szsql) == 0)
ret = true;
}
return ret;
}

CString CMemDB::GetValue(CString oid,int nFieldIndex)
{
CString ret = "";
if (FindKey(oid))
{
switch(GetFieldType(nFieldIndex))
{
case CMemDB::ft_TEXT: {
ret = GetTextFieldByIndex(nFieldIndex);
}break;
case CMemDB::ft_INT: ret = IntToStr(GetIntFieldByIndex(nFieldIndex)); break;
case CMemDB::ft_FLOAT:ret = FloatToStr(GetDoubleFieldByIndex(nFieldIndex)); break;
default:ret = "";break;
}
}
return ret;
}

void CMemDB::AddDBFromFile(CString dbFileName)
{
int i,j;
int NumRecs,FieldCount;
CString fname,ext;
CString dbSelectsql,dbCreatesql,dbInsertsql;
CDBFFileReader* dbFile;
if (dbFileName.IsEmpty()==false)
{
if (!InMemory)
CreateMemoryDB();
dbFile = new CDBFFileReader();
if (dbFile->Open(dbFileName))
{
fname = ExtractFileName(dbFileName);
ext = ExtractFileExt(dbFileName);
fname = fname.Mid(0,fname.GetLength() - ext.GetLength());
FieldCount = dbFile->GetFieldCount();
dbSelectsql = "select * from "+fname+";";
if (!ExistTable(fname))
{
dbCreatesql = "create table "+fname+" (sqlITE_ID integer primary key ";
dbInsertsql = "insert into "+fname;
for(i=0;i<FieldCount;i++)
{
switch(dbFile->GetFieldTypeByIndex(i))
{
case 'C':{
dbCreatesql += ","+dbFile->GetFieldNameByIndex(i)+" char("+IntToStr(dbFile->GetFieldLenByIndex(i))+")";
}break;
case 'F':{
dbCreatesql += ","+dbFile->GetFieldNameByIndex(i)+" real ";
}break;
case 'N':{
dbCreatesql += ","+dbFile->GetFieldNameByIndex(i)+" integer ";
}break;
case 'D':{
dbCreatesql += ","+dbFile->GetFieldNameByIndex(i)+" char(8)";//date
}break;
case 'L':{
dbCreatesql += ","+dbFile->GetFieldNameByIndex(i)+" char(1)";
}break;
case 'M':
default:break;
}
}
dbCreatesql += ");";
dbInsertsql+=" values(?";
for(i=0;i<FieldCount;i++)
dbInsertsql+=",?";
dbInsertsql += ");";
ExecCallBacksql(dbCreatesql);
}
CurTableName = fname;

j = 0;NumRecs = dbFile->GetRecordCount();BeginTran();while(j<NumRecs){Preparesql(dbInsertsql);Bind(0,j+1);for(i=0;i<FieldCount;i++){CString str = Trim(dbFile->ReadAttr(j+1,i));switch(dbFile->GetFieldTypeByIndex(i)){case 'C':case 'D':case 'L':{Bind(i+1,(LPCTSTR)str);}break;case 'F':{Bind(i+1,StrToFloat(str));}break;case 'N':{Bind(i+1,StrToInt(str));}break;case 'M':default:break;} }Exec();j++;if (j%1000==0){CommitTran();BeginTran();}}CommitTran();Execsql(dbSelectsql);}delete dbFile;}}

sqlite数据库包装类

猜你在找的Sqlite相关文章