From dfa10a0616074fd82703516a2d3ef0f44cac64ab Mon Sep 17 00:00:00 2001
From: pans <pans@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期五, 30 十二月 2016 18:20:34 +0800
Subject: [PATCH] 

---
 RtspFace/demo/src/db/main.cpp    |    4 ++--
 RtspFace/demo/src/db/DBuntil.cpp |   42 +++++++++++++++++++++++++++++++-----------
 RtspFace/demo/src/db/DBuntil.h   |   12 ++++++------
 3 files changed, 39 insertions(+), 19 deletions(-)

diff --git a/RtspFace/demo/src/db/DBuntil.cpp b/RtspFace/demo/src/db/DBuntil.cpp
index 5eaeddc..ee9102f 100644
--- a/RtspFace/demo/src/db/DBuntil.cpp
+++ b/RtspFace/demo/src/db/DBuntil.cpp
@@ -8,18 +8,22 @@
 MYSQL_RES *result;
 MYSQL_ROW sql_row;
 
-DBuntil::() {}
+DBuntil::DBuntil(){}
 
 DBuntil::DBuntil(my_db mydb)
 {
-	if(mysql_init(&myCont)!=NULL) {
+	if(mysql_init(&myCont)!=NULL)
+	{
 		std::cout<<"init succeed"<<std::endl;
-	} else
+	}
+	else
 		std::cout<<"init failed"<<std::endl;
 
-	if(mysql_real_connect(&myCont, mydb->host, mydb->user, mydb->pswd, mydb->db, mydb->port, NULL, 0) != NULL) {
+	if(mysql_real_connect(&myCont, mydb.host, mydb.user, mydb.pswd, mydb.db, mydb.port, NULL, 0) != NULL)
+	{
 		std::cout<<"mysql_real_connect succeed"<<std::endl;
-	} else
+	}
+	else
 		std::cout<<"mysql_real_connect failed"<<std::endl;
 }
 
@@ -46,15 +50,21 @@
 
 	mysql_query(&myCont, "SET NAMES utf8"); //璁剧疆缂栫爜鏍煎紡
 	res = mysql_query(&myCont,sql);//鏌ヨ
-	if (!res) {
+	if (!res)
+	{
 		result = mysql_store_result(&myCont);
-		if (result) {
-			while (sql_row = mysql_fetch_row(result)) { //鑾峰彇鍏蜂綋鐨勬暟鎹�
+		if (result)
+		{
+			while (sql_row = mysql_fetch_row(result))
+			{
+				//鑾峰彇鍏蜂綋鐨勬暟鎹�
 				per->p_id = atoi( sql_row[0]);
 				per->name = sql_row[1];
 			}
 		}
-	} else {
+	}
+	else
+	{
 		std::cout<<"query sql failed!"<<std::endl;
 		return false;
 	}
@@ -66,12 +76,22 @@
 	//
 	sprintf( sql, "INSERT INTO user_info(NAME, img) VALUES('%s', NULL)" , per->name );
 	res = mysql_query(&myCont,sql);
-	if(!res) {
+	if(!res)
+	{
 		sprintf( sql, "INSERT INTO face_person(p_id, face_id) VALUES (LAST_INSERT_ID(), %d)" , f_id );
 		res = mysql_query(&myCont,sql);
-		if(!res) {
+		if(!res)
+		{
 			return true;
 		}
 	}
 	return false;
+}
+
+int DBuntil::db_register(int f_id,person *per)
+{
+	if(db_add(f_id,per)){
+		return true;
+	}
+	return false;
 }
\ No newline at end of file
diff --git a/RtspFace/demo/src/db/DBuntil.h b/RtspFace/demo/src/db/DBuntil.h
index 9a65d46..e642b02 100644
--- a/RtspFace/demo/src/db/DBuntil.h
+++ b/RtspFace/demo/src/db/DBuntil.h
@@ -9,12 +9,12 @@
 };
 
 struct my_db {
-	char user[];
-	char pswd[];
-	char host[];
-	char db[];
+	char user[25];
+	char pswd[25];
+	char host[25];
+	char db[25];
 	unsigned int port;
-}
+};
 
 class DBuntil
 {
@@ -25,7 +25,7 @@
 	~DBuntil();
 	person db_rearch(int f_id);
 
-	int db_register(int f_id,person p);
+	int db_register(int f_id,person *p);
 
 private:
 
diff --git a/RtspFace/demo/src/db/main.cpp b/RtspFace/demo/src/db/main.cpp
index 0443a47..a9304df 100644
--- a/RtspFace/demo/src/db/main.cpp
+++ b/RtspFace/demo/src/db/main.cpp
@@ -12,8 +12,8 @@
 	my_db mydb={"root","Basic@2017","localhost","demo",3306};
 	
 	DBuntil dbu = DBuntil(mydb);
-	person p={,"test"};
-	if(dbu.db_add(i,p)){
+	person p={0,"test"};
+	if(dbu.db_register(i,&p)){
 		cout<<"add succeed"<<endl;
 	}else
 		cout<<"add false"<<endl;

--
Gitblit v1.8.0