#ifndef __DBUtil__
#define __DBUtil__
#include <string>
#include "cocos2d.h"
#include "sqlite3.h"using namespace std;
USING_NS_CC;
class DBUtil{
public:/************************************************************封装 sqlite3操作************************************************************///用来创建一个db数据库 db为数据库的名字//  打开数据库static void initDB(const char *url);//用来判断表格是否存在// name:表示表名static bool tableIsExist(string table_name);//用来创建一个表名为name的表格，创建时会先匹配时否有该表的存在如果存在则不创建//创建表static void createTable(string sql,string table_name);//用来删除一张表名为name的表格，删除时会先匹配是否有该表的存在如果不存在则不执行删除操作//删除表名static void deleteTable(string sql,string table_name);//用来向表中插入一条数据//插入一条数据static void insertData(string sql);//用来向表中删除一条数据//删除一条数据static void deleteData(string sql);//用来向表中修改一条数据// 修改一条数据static void updateData(string sql);//获取表的全部数据// 返回记录的条数static int getAllData(string table_name,std::function<void(std::map<string,string>)> callback);//读取一条记录的信息/**  此方法是查询方法,pSender()*/static void getDataInfo(string sql,std::function<void(std::map<string,string>)> callback);//关闭打开的数据库static void closeDB();
};#endif

#include "DBUtil.h"sqlite3 *pDB = NULL;//数据库指针
char * errMsg = NULL;//错误信息
std::string sqlstr;//SQL指令
int result;//sqlite3_exec返回值
unsigned int _count=0;
std::map<unsigned int,std::function<void(std::map<string,string>)>> _keyMap;void DBUtil::initDB(const char *db)
{//打开一个数据库，如果该数据库不存在，则创建一个数据库文件result = sqlite3_open(db, &pDB);if( result != SQLITE_OK )log( "打开数据库失败，错误码:%d ，错误原因:%s\n" , result, errMsg );
}//tableIsExist的回调函数
int isExisted( void * para, int n_column, char ** column_value, char ** column_name )
{bool *isExisted_=(bool*)para;*isExisted_=(**column_value)!='0';return 0;
}bool DBUtil::tableIsExist( string table_name )
{if (pDB!=NULL){//判断表是否存在bool tableIsExisted;sqlstr = "select count(type) from sqlite_master where type='table' and name ='"+table_name+"'";result =sqlite3_exec(pDB,sqlstr.c_str(),isExisted,&tableIsExisted,&errMsg);return tableIsExisted;}return false;
}void DBUtil::createTable( string sql,string table_name )
{if (!tableIsExist(table_name)){//创建表，设置ID为主键，且自动增加result = sqlite3_exec(pDB,sql.c_str(),NULL,NULL,&errMsg);if( result != SQLITE_OK )log( "创建表失败，错误码:%d ，错误原因:%s\n" , result, errMsg );}}void DBUtil::deleteTable( string sql,string table_name )
{if (tableIsExist(table_name)){result = sqlite3_exec(pDB,sql.c_str(),NULL,NULL,&errMsg);if( result != SQLITE_OK )log( "创建表失败，错误码:%d ，错误原因:%s\n" , result, errMsg );}
}//@示例语句sqlstr=" insert into MyTable_1( name ) values ( 'data',..... ) ";
void DBUtil::insertData( string sql ){result = sqlite3_exec( pDB, sql.c_str() , NULL, NULL, &errMsg );if(result != SQLITE_OK )log( "插入记录失败，错误码:%d ，错误原因:%s\n" , result, errMsg );
}//@示例语句sqlstr="delete from MyTable_1 where ID = 2";
void DBUtil::deleteData( string sql )
{result=sqlite3_exec( pDB, sql.c_str() , NULL, NULL, &errMsg );if(result != SQLITE_OK )log( "插入记录失败，错误码:%d ，错误原因:%s\n" , result, errMsg );
}//@示例语句        sqlstr="update MyTable_1 set name='data' where ID = 3";
void DBUtil::updateData( string sql )
{result = sqlite3_exec( pDB, sql.c_str() , NULL, NULL, &errMsg );if(result != SQLITE_OK )log( "插入记录失败，错误码:%d ，错误原因:%s\n" , result, errMsg );
}//getDataCount的回调函数
int loadRecordCount( void *para, int n_column, char ** column_value, char ** column_name )
{std::map<string,string> data;int index;for( index = 0 ; index < n_column; index ++ ){data.insert(std::pair<string,string>( column_name[index], column_value[index] ));}auto num = *(unsigned int *)para;auto callback = _keyMap.at(num);if( callback != nullptr ){callback(data);}return 0;
}int DBUtil::getAllData(string table_name,std::function<void(std::map<string,string>)> callback)
{string sql = "select * from "+table_name;_count++;_keyMap.insert(std::pair<unsigned int,std::function<void(std::map<string,string>)>>(_count,callback));int count=sqlite3_exec( pDB, sql.c_str() , loadRecordCount, &_count, &errMsg );return count;
}//getDataInfo的回调函数
int loadRecord( void * para, int n_column, char ** column_value, char ** column_name )
{std::map<string,string> data;int index;for( index = 0 ; index < n_column; index ++ ){string str1 = column_name[index];if( column_value[index] == NULL ){log("DBUtil Get Data Value Failure! Table Name Is  '%s'",str1.c_str());continue;}string str2 = column_value[index];data.insert(std::pair<string,string>(str1,str2));}auto num = *(unsigned int *)para;auto callback = _keyMap.at(num);if( callback != nullptr ){_keyMap.erase(num);callback(data);}return 0;
}void DBUtil::getDataInfo( string sql,std::function<void(std::map<string,string>)> callback )
{_count++;_keyMap.insert(std::pair<unsigned int,std::function<void(std::map<string,string>)>>(_count,callback));sqlite3_exec( pDB, sql.c_str() , loadRecord, &_count, &errMsg );
}void DBUtil::closeDB()
{sqlite3_close(pDB);
}