添加sqlite模块

FaceDetection/FaceDetection.cpp

@@ -21,6 +21,7 @@
 #include "utility/jsonAnalysis.h"
 #include "curl/curl.h"
 #include "utility/threadpool.h"
+#include "utility/SQLiteModule.h"
 INITIALIZE_EASYLOGGINGPP
 //using namespace cv;
 using namespace std;
@@ -33,6 +34,7 @@ using namespace std;
 #pragma comment(lib, "../lib/dlib_d.lib") 
 #pragma comment(lib, "../lib/freetype.lib") 
 #pragma comment(lib, "../lib/libcurl.lib") 
+#pragma comment(lib, "../lib/sqlite3.lib") 
 #else
 #pragma comment(lib, "../lib/opencv_core2413.lib") 
 #pragma comment(lib, "../lib/opencv_highgui2413.lib") 
@@ -42,6 +44,7 @@ using namespace std;
 #pragma comment(lib, "../lib/dlib.lib") 
 #pragma comment(lib, "../lib/freetype.lib") 
 #pragma comment(lib, "../lib/libcurl.lib") 
+#pragma comment(lib, "../lib/sqlite3.lib") 
 #endif
 
 /****************************************
@@ -82,6 +85,8 @@ int main()
 {
 	try
 	{
+		SQLite sqlite;
+		sqlite.initSQLite();
 		// global init curl
 		curl_global_init(CURL_GLOBAL_ALL);
 

FaceDetection/FaceDetection.vcxproj

@@ -154,6 +154,7 @@
     <ClInclude Include="targetver.h" />
     <ClInclude Include="utility\iniOperation.h" />
     <ClInclude Include="utility\jsonAnalysis.h" />
+    <ClInclude Include="utility\SQLiteModule.h" />
     <ClInclude Include="utility\threadpool.h" />
     <ClInclude Include="utility\utilCommonAPI.h" />
   </ItemGroup>
@@ -178,6 +179,10 @@
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
     </ClCompile>
+    <ClCompile Include="utility\SQLiteModule.cpp">
+      <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
+      <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
+    </ClCompile>
     <ClCompile Include="utility\utilCommonAPI.cpp">
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>

FaceDetection/FaceDetection.vcxproj.filters

@@ -48,6 +48,9 @@
     <ClInclude Include="utility\threadpool.h">
       <Filter>头文件</Filter>
     </ClInclude>
+    <ClInclude Include="utility\SQLiteModule.h">
+      <Filter>utility</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="stdafx.cpp">
@@ -71,5 +74,8 @@
     <ClCompile Include="utility\jsonAnalysis.cpp">
       <Filter>utility</Filter>
     </ClCompile>
+    <ClCompile Include="utility\SQLiteModule.cpp">
+      <Filter>utility</Filter>
+    </ClCompile>
   </ItemGroup>
 </Project>
\ No newline at end of file

FaceDetection/utility/SQLiteModule.cpp

+#include "SQLiteModule.h"
+#include "utilCommonAPI.h"
+#include "easylogging++.h"
+
+#define BUFFER_SIZE             1024 //缓冲区大小
+
+SQLite::SQLite()
+{
+	pDB = NULL;
+	errMsg = NULL;
+}
+
+SQLite::~SQLite()
+{
+
+}
+
+bool SQLite::initSQLite()
+{
+	std::string strDbFileName;      //*.db 名称
+	std::string strPath = GetProcDir();
+	strPath.append("fmdata.db");
+	strDbFileName = strPath;
+
+	//打开一个数据库，如果改数据库不存在，则创建一个名字为databaseName的数据库文件
+	int res = sqlite3_open(strDbFileName.c_str(), &pDB);
+	LOG(INFO) << "DbFileName:" << strDbFileName.c_str();
+	if (res != SQLITE_OK)
+	{
+		LOG(ERROR) << "打开数据库失败:" << strDbFileName.data();
+		return false;
+	}
+	try
+	{
+		_createTable();
+	}
+	catch (std::exception &ex)
+	{
+		LOG(ERROR) << "创建表异常：" << ex.what();
+		return false;
+	}
+	return true;
+}
+
+bool SQLite::_createTable()
+{
+	//插入一个表，返回值为SQLITE_OK为成功，否则输出出错信息
+	//函数参数：第一个为操作数据库的指针，第二句为SQL命令字符串
+	//第三个参数为callback函数，这里没有用，第四个参数为callback函数
+	//中的第一个参数，第五个为出错信息
+	bool rlt = true;
+	std::string strTableName = "fmTest";
+	if (!isTableExist(pDB, strTableName)) {
+		char szCreate[1024] = { 0 };
+		sprintf_s(szCreate, "create table %s(fm_id varchar(50) NOT NULL,\
+                                    statusCode int, \
+                                    msg varchar(200),\
+                                    prompt bool,\
+                                    fm_open_id varchar(50),\
+                                    total_amount int,\
+                                    paid_total_amount int,\
+                                    invoice_amount int,\
+                                    incentives_amount int,\
+                                    logtime TIMESTAMP default (datetime('now', 'localtime')),\
+                                    PRIMARY KEY (fm_id))", strTableName.c_str());
+		if (!_execSql(szCreate)) {
+			//执行失败，退出
+			exit(0);
+		}
+	}
+	else {
+		LOG(INFO) << "已存在表:" << strTableName.data() ;
+	}
+	return rlt;
+
+}
+
+void SQLite::closeSQLite()
+{
+	if (pDB)
+		sqlite3_close(pDB);
+}
+
+//查看表是否存在
+bool SQLite::isTableExist(sqlite3 *sqDb, std::string strTableName)
+{
+	char szQuery[1024] = { 0 };
+	sprintf_s(szQuery, "SELECT count(*) FROM sqlite_master WHERE type=\"table\" AND name=\"%s\"", strTableName.c_str());
+	sqlite3_stmt *pstmt;
+	sqlite3_prepare(sqDb, szQuery, strlen(szQuery), &pstmt, NULL);
+	sqlite3_step(pstmt);
+	int count = sqlite3_column_int(pstmt, 0);
+	sqlite3_finalize(pstmt);
+	if (count > 0)
+		return true;
+	return false;
+}
+
+int SQLite::isRecordExist(std::string strTable, std::string strKey)
+{
+	char* lpSql = new char[BUFFER_SIZE];
+	sprintf_s(lpSql, BUFFER_SIZE, "select * from %s where trans_id = '%s'", strTable.c_str(), strKey.c_str());
+	char** pResult;
+	int nRow = 0;
+	int nCol = 0;
+	int nResult = sqlite3_get_table(pDB, lpSql, &pResult, &nRow, &nCol, &errMsg);
+	delete[] lpSql;
+	if (nResult != SQLITE_OK)
+	{
+		LOG(ERROR) << "sqlite3_get_table error:" << errMsg;
+		return -1;
+	}
+	return nRow;
+}
+
+bool SQLite::_execSql(const char* sql)
+{
+	bool rlt = true;
+	int res = sqlite3_exec(pDB, sql, 0, 0, &errMsg);
+	if (res != SQLITE_OK)
+	{
+		LOG(ERROR) << "执行sql失败:" << sql << "  失败信息:" << errMsg;
+		rlt = false;
+	}
+	return rlt;
+}
+
+bool SQLite::insert(const char* sql)
+{
+	return _execSql(sql);
+}
+
+bool SQLite::update(const char* sql)
+{
+	return _execSql(sql);
+}
+
+bool SQLite::remove(const char* sql)
+{
+	return _execSql(sql);
+}
+
+int callbackQuery(void* data, int n_columns, char** column_values, char** column_names)
+{
+	try
+	{
+		std::string col0 = column_values[0];
+		std::string col1 = column_values[1];
+		std::string col2 = column_values[2];
+		LOG(INFO) << "column0:" << col0 << " column1:" << col1 << " column2:" << col2;
+	}
+	catch (std::exception &ex)
+	{
+		LOG(ERROR) << "读取数据失败:" << ex.what();
+		return -1;
+	}
+
+	return 0;
+}
+
+void SQLite::query(const char* sql)
+{
+	int rlt = sqlite3_exec(pDB, sql, callbackQuery, 0, &errMsg);
+	if (rlt != SQLITE_OK)
+	{
+		LOG(ERROR) << "查询数据失败:" << errMsg;
+	}
+}
\ No newline at end of file

FaceDetection/utility/SQLiteModule.h

+#ifndef SQLITE_MODULE_H
+#define SQLITE_MODULE_H
+#include <iostream>
+#include "sqlite3.h"
+
+class SQLite
+{
+public:
+    SQLite();
+    ~SQLite();
+
+    bool initSQLite();
+    void closeSQLite();
+
+    int isRecordExist(std::string strTable,std::string strKey);
+    bool isTableExist(sqlite3 *sqDb, std::string strTableName);
+    bool insert(const char* sql);
+    bool update(const char* sql);
+    bool remove(const char* sql);
+    void query(const char* sql);
+private:
+    bool _execSql(const char* sql);
+    bool _createTable();
+private:
+    
+    sqlite3 *pDB;
+    char * errMsg;
+};
+
+#endif
+