#include "DBuntil.h"
#include <mysql.h>
#include <cstdio>
#include <iostream>


MYSQL myCont;
MYSQL_RES *result;
MYSQL_ROW sql_row;

DBuntil::DBuntil() {}

DBuntil::DBuntil(my_db mydb)
{
	if(mysql_init(&myCont)!=NULL)
	{
		std::cout<<"init succeed"<<std::endl;
	}
	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)
	{
		std::cout<<"mysql_real_connect succeed"<<std::endl;
	}
	else
		std::cout<<"mysql_real_connect failed"<<std::endl;
}

DBuntil::~DBuntil()
{
	if (result != NULL)
		mysql_free_result(result);

	mysql_close(&myCont);
}

bool DBuntil::db_search(person* per)
{
	std::cout<<"db_search start"<<std::endl;
	if(db_select(per))
	{

		return true;
	}
	return false;
}

bool DBuntil::db_select(person* per)
{
	std::cout<<"db_select start"<<std::endl;
	sprintf( sql, "select a.p_id,b.`name`,b.img from face_person a,user_info b where a.face_id = %d AND a.p_id = b.pid " ,per->f_id );

	mysql_query(&myCont, "SET NAMES utf8"); //设置编码格式
	res = mysql_query(&myCont,sql);//查询
	if (!res)
	{
		result = mysql_store_result(&myCont);
		if (result)
		{
			while (sql_row = mysql_fetch_row(result))
			{
				//获取具体的数据
				per->p_id = atoi( sql_row[0]);
				per->name = sql_row[1];
			}
		}
	}
	else
	{
		std::cout<<"query sql failed!"<<std::endl;
		return false;
	}
	return true;
}

bool DBuntil::db_add(person *per)
{
	//
	sprintf( sql, "INSERT INTO user_info(NAME, img) VALUES('%s', NULL)" , per->name );
	res = mysql_query(&myCont,sql);
	if(!res)
	{
		sprintf( sql, "INSERT INTO face_person(p_id, face_id) VALUES (LAST_INSERT_ID(), %d)" , per->f_id );
		res = mysql_query(&myCont,sql);
		if(!res)
		{
			return true;
		}
	}
	return false;
}

bool DBuntil::db_register(person *per)
{
	if(db_add(per))
	{
		return true;
	}
	return false;
}