#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); }