//mysqlhelper.h

#ifndef __MYSQL_HELPER_H__
#define __MYSQL_HELPER_H__

#include <stdlib.h>

#include <map>
#include <vector>
#include <string>
using namespace std;

#include <mysql.h>

namespace mysqlhelper
{

/*********************
*@brief 數(shù)據(jù)庫(kù)異常類
**********************/
struct MysqlHelper_Exception //: public TC_Exception
{
 MysqlHelper_Exception(const string &sBuffer):errorInfo(sBuffer){}; //: TC_Exception(sBuffer){};
 ~MysqlHelper_Exception() throw(){}; 

 string errorInfo;
};

/***********************
* @brief 數(shù)據(jù)庫(kù)配置接口
***********************/
struct DBConf
{

 string _host;//主機(jī)地址
 string _user; //用戶名 
 string _password;//密碼
 string _database; //數(shù)據(jù)庫(kù)
 string _charset; //字符集
 int _port;//端口
 int _flag; //客戶端標(biāo)識(shí)

 /*****************
 * @brief 構(gòu)造函數(shù)
 *****************/
 DBConf():_port(0), _flag(0){}

 /**********************************
 * @brief 讀取數(shù)據(jù)庫(kù)配置. 
 * @param mpParam 存放數(shù)據(jù)庫(kù)配置的map 
 * dbhost: 主機(jī)地址
 * dbuser:用戶名
 * dbpass:密碼
 * dbname:數(shù)據(jù)庫(kù)名稱
 * dbport:端口
 **********************************/
 void loadFromMap(const map<string, string> &mpParam)
 {
 map<string, string> mpTmp = mpParam;

 _host = mpTmp["dbhost"];
 _user = mpTmp["dbuser"];
 _password = mpTmp["dbpass"];
 _database = mpTmp["dbname"];
 _charset = mpTmp["charset"];
 _port = atoi(mpTmp["dbport"].c_str());
 _flag = 0;

 if(mpTmp["dbport"] == "")
 {
 _port = 3306;
 }
 }
};

/**************************************************************
* @brief:MySQL數(shù)據(jù)庫(kù)操作類 
* @feature:非線程安全，通常一個(gè)線程一個(gè)MysqlHelper對(duì)象；
* 對(duì)于insert/update可以有更好的函數(shù)封裝，保證SQL注入；
* MysqlHelper::DB_INT表示組裝sql語(yǔ)句時(shí)，不加””和轉(zhuǎn)義；
* MysqlHelper::DB_STR表示組裝sql語(yǔ)句時(shí)，加””并轉(zhuǎn)義；
**************************************************************/
class MysqlHelper{

public:
 /**
 * @brief 構(gòu)造函數(shù)
 */
 MysqlHelper();

 /**
 * @brief 構(gòu)造函數(shù). 
 * @param: sHost:主機(jī)IP
 * @param sUser 用戶
 * @param sPasswd 密碼
 * @param sDatebase 數(shù)據(jù)庫(kù)
 * @param port 端口
 * @param iUnixSocket socket
 * @param iFlag 客戶端標(biāo)識(shí)
 */
 MysqlHelper(const string& sHost, const string& sUser = "", const string& sPasswd = "", const string& sDatabase = "", const string &sCharSet = "", int port = 0, int iFlag = 0);

 /**
 * @brief 構(gòu)造函數(shù). 
 * @param tcDBConf 數(shù)據(jù)庫(kù)配置
 */
 MysqlHelper(const DBConf& tcDBConf);

 /**
 * @brief 析構(gòu)函數(shù).
 */
 ~MysqlHelper();

 /**
 * @brief 初始化. 
 * 
 * @param sHost 主機(jī)IP
 * @param sUser 用戶
 * @param sPasswd 密碼
 * @param sDatebase 數(shù)據(jù)庫(kù)
 * @param port 端口
 * @param iUnixSocket socket
 * @param iFlag 客戶端標(biāo)識(shí)
 * @return 無(wú)
 */
 void init(const string& sHost, const string& sUser = "", const string& sPasswd = "", const string& sDatabase = "", const string &sCharSet = "", int port = 0, int iFlag = 0);

 /**
 * @brief 初始化. 
 * 
 * @param tcDBConf 數(shù)據(jù)庫(kù)配置
 */
 void init(const DBConf& tcDBConf);

 /**
 * @brief 連接數(shù)據(jù)庫(kù). 
 * 
 * @throws MysqlHelper_Exception
 * @return 無(wú)
 */
 void connect();

 /**
 * @brief 斷開數(shù)據(jù)庫(kù)連接. 
 * @return 無(wú)
 */
 void disconnect();

 /**
 * @brief 獲取數(shù)據(jù)庫(kù)變量. 
 * @return 數(shù)據(jù)庫(kù)變量
 */
 string getVariables(const string &sName);

 /**
 * @brief 直接獲取數(shù)據(jù)庫(kù)指針. 
 * 
 * @return MYSQL* 數(shù)據(jù)庫(kù)指針
 */
 MYSQL *getMysql();

 /**
 * @brief 字符轉(zhuǎn)義. 
 * 
 * @param sFrom 源字符串
 * @param sTo 輸出字符串
 * @return 輸出字符串
 */
 string escapeString(const string& sFrom);

 /**
 * @brief 更新或者插入數(shù)據(jù). 
 * 
 * @param sSql sql語(yǔ)句
 * @throws MysqlHelper_Exception
 * @return
 */
 void execute(const string& sSql);

 /**
 * @brief mysql的一條記錄
 */
 class MysqlRecord
 {
 public:
 /**
 * @brief 構(gòu)造函數(shù).
 * 
 * @param record
 */
 MysqlRecord(const map<string, string> &record);

 /**
 * @brief 獲取數(shù)據(jù)，s一般是指數(shù)據(jù)表的某個(gè)字段名 
 * @param s 要獲取的字段
 * @return 符合查詢條件的記錄的s字段名
 */
 const string& operator[](const string &s);
 protected:
 const map<string, string> &_record;
 };

 /**
 * @brief 查詢出來(lái)的mysql數(shù)據(jù)
 */
 class MysqlData
 {
 public:
 /**
 * @brief 所有數(shù)據(jù).
 * 
 * @return vector<map<string,string>>&
 */
 vector<map<string, string> >& data();

 /**
 * 數(shù)據(jù)的記錄條數(shù)
 * 
 * @return size_t
 */
 size_t size();

 /**
 * @brief 獲取某一條記錄. 
 * 
 * @param i 要獲取第幾條記錄 
 * @return MysqlRecord類型的數(shù)據(jù)，可以根據(jù)字段獲取相關(guān)信息，
 */
 MysqlRecord operator[](size_t i);

 protected:
 vector<map<string, string> > _data;
 };

 /**
 * @brief Query Record. 
 * 
 * @param sSql sql語(yǔ)句
 * @throws MysqlHelper_Exception
 * @return MysqlData類型的數(shù)據(jù)，可以根據(jù)字段獲取相關(guān)信息
 */
 MysqlData queryRecord(const string& sSql);

 /**
 * @brief 定義字段類型， 
 * DB_INT:數(shù)字類型 
 * DB_STR:字符串類型
 */
 enum FT
 {
 DB_INT, 
 DB_STR, 
 };

 /**
 * 數(shù)據(jù)記錄
 */
 typedef map<string, pair<FT, string> > RECORD_DATA;

 /**
 * @brief 更新記錄. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @param sCondition where子語(yǔ)句,例如:where A = B
 * @throws MysqlHelper_Exception
 * @return size_t 影響的行數(shù)
 */
 size_t updateRecord(const string &sTableName, const map<string, pair<FT, string> > &mpColumns, const string &sCondition);

 /**
 * @brief 插入記錄. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @throws MysqlHelper_Exception
 * @return size_t 影響的行數(shù)
 */
 size_t insertRecord(const string &sTableName, const map<string, pair<FT, string> > &mpColumns);

 /**
 * @brief 替換記錄. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @throws MysqlHelper_Exception
 * @return size_t 影響的行數(shù)
 */
 size_t replaceRecord(const string &sTableName, const map<string, pair<FT, string> > &mpColumns);

 /**
 * @brief 刪除記錄. 
 * 
 * @param sTableName 表名
 * @param sCondition where子語(yǔ)句,例如:where A = B
 * @throws MysqlHelper_Exception
 * @return size_t 影響的行數(shù)
 */
 size_t deleteRecord(const string &sTableName, const string &sCondition = "");

 /**
 * @brief 獲取Table查詢結(jié)果的數(shù)目. 
 * 
 * @param sTableName 用于查詢的表名
 * @param sCondition where子語(yǔ)句,例如:where A = B
 * @throws MysqlHelper_Exception
 * @return size_t 查詢的記錄數(shù)目
 */
 size_t getRecordCount(const string& sTableName, const string &sCondition = "");

 /**
 * @brief 獲取Sql返回結(jié)果集的個(gè)數(shù). 
 * 
 * @param sCondition where子語(yǔ)句,例如:where A = B
 * @throws MysqlHelper_Exception
 * @return 查詢的記錄數(shù)目
 */
 size_t getSqlCount(const string &sCondition = "");

 /**
 * @brief 存在記錄. 
 * 
 * @param sql sql語(yǔ)句
 * @throws MysqlHelper_Exception
 * @return 操作是否成功
 */
 bool existRecord(const string& sql);

 /**
 * @brief 獲取字段最大值. 
 * 
 * @param sTableName 用于查詢的表名
 * @param sFieldName 用于查詢的字段
 * @param sCondition where子語(yǔ)句,例如:where A = B
 * @throws MysqlHelper_Exception
 * @return 查詢的記錄數(shù)目
 */
 int getMaxValue(const string& sTableName, const string& sFieldName, const string &sCondition = "");

 /**
 * @brief 獲取auto_increment最后插入得ID. 
 * 
 * @return ID值
 */
 long lastInsertID();

 /**
 * @brief 構(gòu)造Insert-SQL語(yǔ)句. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @return string insert-SQL語(yǔ)句
 */
 string buildInsertSQL(const string &sTableName, const map<string, pair<FT, string> > &mpColumns);

 /**
 * @brief 構(gòu)造Replace-SQL語(yǔ)句. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @return string insert-SQL語(yǔ)句
 */
 string buildReplaceSQL(const string &sTableName, const map<string, pair<FT, string> > &mpColumns);

 /**
 * @brief 構(gòu)造Update-SQL語(yǔ)句. 
 * 
 * @param sTableName 表名
 * @param mpColumns 列名/值對(duì)
 * @param sCondition where子語(yǔ)句
 * @return string Update-SQL語(yǔ)句
 */
 string buildUpdateSQL(const string &sTableName,const map<string, pair<FT, string> > &mpColumns, const string &sCondition);

 /**
 * @brief 獲取最后執(zhí)行的SQL語(yǔ)句.
 * 
 * @return SQL語(yǔ)句
 */
 string getLastSQL() { return _sLastSql; }

 /**
 * @brief 獲取查詢影響數(shù)
 * @return int
 */
 size_t getAffectedRows();
protected:
 /**
 * @brief copy contructor，只申明,不定義,保證不被使用 
 */
 MysqlHelper(const MysqlHelper &tcMysql);

 /**
 * 
 * @brief 只申明,不定義,保證不被使用
 */
 MysqlHelper &operator=(const MysqlHelper &tcMysql);


private:

 /**
 * 數(shù)據(jù)庫(kù)指針
 */
 MYSQL *_pstMql;

 /**
 * 數(shù)據(jù)庫(kù)配置
 */
 DBConf _dbConf;

 /**
 * 是否已經(jīng)連接
 */
 bool _bConnected;

 /**
 * 最后執(zhí)行的sql
 */
 string _sLastSql;

};
}
#endif //__MYSQL_HELPER_H__

//mysqlhelper.cpp

#include "mysqlHelper.h"

#include <string.h>
#include <sstream>

using namespace std;

namespace mysqlhelper{

MysqlHelper::MysqlHelper():_bConnected(false)
{
 _pstMql = mysql_init(NULL);
}

MysqlHelper::MysqlHelper(const string& sHost, const string& sUser, const string& sPasswd, const string& sDatabase, const string &sCharSet, int port, int iFlag)
:_bConnected(false)
{
 init(sHost, sUser, sPasswd, sDatabase, sCharSet, port, iFlag);

 _pstMql = mysql_init(NULL);
}

MysqlHelper::MysqlHelper(const DBConf& tcDBConf)
:_bConnected(false)
{
 _dbConf = tcDBConf;

 _pstMql = mysql_init(NULL); 
}

MysqlHelper::~MysqlHelper()
{
 if (_pstMql != NULL)
 {
 mysql_close(_pstMql);
 _pstMql = NULL;
 }
}

void MysqlHelper::init(const string& sHost, const string& sUser, const string& sPasswd, const string& sDatabase, const string &sCharSet, int port, int iFlag)
{
 _dbConf._host = sHost;
 _dbConf._user = sUser;
 _dbConf._password = sPasswd;
 _dbConf._database = sDatabase;
 _dbConf._charset = sCharSet;
 _dbConf._port = port;
 _dbConf._flag = iFlag;
}

void MysqlHelper::init(const DBConf& tcDBConf)
{
 _dbConf = tcDBConf;
}

void MysqlHelper::connect()
{
 disconnect();

 if( _pstMql == NULL)
 {
 _pstMql = mysql_init(NULL);
 }

 //建立連接后, 自動(dòng)調(diào)用設(shè)置字符集語(yǔ)句
 if(!_dbConf._charset.empty()) {
 if (mysql_options(_pstMql, MYSQL_SET_CHARSET_NAME, _dbConf._charset.c_str())) {
 throw MysqlHelper_Exception(string("MysqlHelper::connect: mysql_options MYSQL_SET_CHARSET_NAME ") + _dbConf._charset + ":" + string(mysql_error(_pstMql)));
 }
 }

 if (mysql_real_connect(_pstMql, _dbConf._host.c_str(), _dbConf._user.c_str(), _dbConf._password.c_str(), _dbConf._database.c_str(), _dbConf._port, NULL, _dbConf._flag) == NULL) 
 {
 throw MysqlHelper_Exception("[MysqlHelper::connect]: mysql_real_connect: " + string(mysql_error(_pstMql)));
 }

 _bConnected = true;
}

void MysqlHelper::disconnect()
{
 if (_pstMql != NULL)
 {
 mysql_close(_pstMql);
 _pstMql = mysql_init(NULL);
 }

 _bConnected = false; 
}

string MysqlHelper::escapeString(const string& sFrom)
{
 if(!_bConnected)
 {
 connect();
 }

 string sTo;
 string::size_type iLen = sFrom.length() * 2 + 1;
 char *pTo = (char *)malloc(iLen);

 memset(pTo, 0x00, iLen);

 mysql_real_escape_string(_pstMql, pTo, sFrom.c_str(), sFrom.length());

 sTo = pTo;

 free(pTo);

 return sTo;
}

MYSQL *MysqlHelper::getMysql(void)
{
 return _pstMql;
}

string MysqlHelper::buildInsertSQL(const string &sTableName, const RECORD_DATA &mpColumns)
{
 ostringstream sColumnNames;
 ostringstream sColumnValues;

 map<string, pair<FT, string> >::const_iterator itEnd = mpColumns.end();

 for(map<string, pair<FT, string> >::const_iterator it = mpColumns.begin(); it != itEnd; ++it)
 {
 if (it == mpColumns.begin())
 {
 sColumnNames << "`" << it->first << "`";
 if(it->second.first == DB_INT)
 {
 sColumnValues << it->second.second;
 }
 else
 {
 sColumnValues << "'" << escapeString(it->second.second) << "'";
 }
 }
 else
 {
 sColumnNames << ",`" << it->first << "`";
 if(it->second.first == DB_INT)
 {
 sColumnValues << "," + it->second.second;
 }
 else
 {
 sColumnValues << ",'" + escapeString(it->second.second) << "'";
 }
 }
 }

 ostringstream os;
 os << "insert into " << sTableName << " (" << sColumnNames.str() << ") values (" << sColumnValues.str() << ")";
 return os.str();
}

string MysqlHelper::buildReplaceSQL(const string &sTableName, const RECORD_DATA &mpColumns)
{
 ostringstream sColumnNames;
 ostringstream sColumnValues;

 map<string, pair<FT, string> >::const_iterator itEnd = mpColumns.end();
 for(map<string, pair<FT, string> >::const_iterator it = mpColumns.begin(); it != itEnd; ++it)
 {
 if (it == mpColumns.begin())
 {
 sColumnNames << "`" << it->first << "`";
 if(it->second.first == DB_INT)
 {
 sColumnValues << it->second.second;
 }
 else
 {
 sColumnValues << "'" << escapeString(it->second.second) << "'";
 }
 }
 else
 {
 sColumnNames << ",`" << it->first << "`";
 if(it->second.first == DB_INT)
 {
 sColumnValues << "," + it->second.second;
 }
 else
 {
 sColumnValues << ",'" << escapeString(it->second.second) << "'";
 }
 }
 }

 ostringstream os;
 os << "replace into " << sTableName << " (" << sColumnNames.str() << ") values (" << sColumnValues.str() << ")";
 return os.str();
}

string MysqlHelper::buildUpdateSQL(const string &sTableName,const RECORD_DATA &mpColumns, const string &sWhereFilter)
{
 ostringstream sColumnNameValueSet;

 map<string, pair<FT, string> >::const_iterator itEnd = mpColumns.end();

 for(map<string, pair<FT, string> >::const_iterator it = mpColumns.begin(); it != itEnd; ++it)
 {
 if (it == mpColumns.begin())
 {
 sColumnNameValueSet << "`" << it->first << "`";
 }
 else
 {
 sColumnNameValueSet << ",`" << it->first << "`";
 }

 if(it->second.first == DB_INT)
 {
 sColumnNameValueSet << "= " << it->second.second;
 }
 else
 {
 sColumnNameValueSet << "= '" << escapeString(it->second.second) << "'";
 }
 }

 ostringstream os;
 os << "update " << sTableName << " set " << sColumnNameValueSet.str() << " " << sWhereFilter;

 return os.str();
}

string MysqlHelper::getVariables(const string &sName)
{
 string sql = "SHOW VARIABLES LIKE '" + sName + "'";

 MysqlData data = queryRecord(sql);
 if(data.size() == 0)
 {
 return "";
 }

 if(sName == data[0]["Variable_name"])
 {
 return data[0]["Value"];
 }

 return "";
}

void MysqlHelper::execute(const string& sSql)
{
 /**
 沒(méi)有連上, 連接數(shù)據(jù)庫(kù)
 */
 if(!_bConnected)
 {
 connect();
 }

 _sLastSql = sSql;

 int iRet = mysql_real_query(_pstMql, sSql.c_str(), sSql.length());
 if(iRet != 0)
 {
 /**
 自動(dòng)重新連接
 */
 int iErrno = mysql_errno(_pstMql);
 if (iErrno == 2013 || iErrno == 2006)
 {
 connect();
 iRet = mysql_real_query(_pstMql, sSql.c_str(), sSql.length());
 }
 }

 if (iRet != 0)
 {
 throw MysqlHelper_Exception("[MysqlHelper::execute]: mysql_query: [ " + sSql+" ] :" + string(mysql_error(_pstMql))); 
 }
}

MysqlHelper::MysqlData MysqlHelper::queryRecord(const string& sSql)
{
 MysqlData data;

 /**
 沒(méi)有連上, 連接數(shù)據(jù)庫(kù)
 */
 if(!_bConnected)
 {
 connect();
 }

 _sLastSql = sSql;

 int iRet = mysql_real_query(_pstMql, sSql.c_str(), sSql.length());
 if(iRet != 0)
 {
 /**
 自動(dòng)重新連接
 */
 int iErrno = mysql_errno(_pstMql);
 if (iErrno == 2013 || iErrno == 2006)
 {
 connect();
 iRet = mysql_real_query(_pstMql, sSql.c_str(), sSql.length());
 }
 }

 if (iRet != 0)
 {
 throw MysqlHelper_Exception("[MysqlHelper::execute]: mysql_query: [ " + sSql+" ] :" + string(mysql_error(_pstMql))); 
 }

 MYSQL_RES *pstRes = mysql_store_result(_pstMql);

 if(pstRes == NULL)
 {
 throw MysqlHelper_Exception("[MysqlHelper::queryRecord]: mysql_store_result: " + sSql + " : " + string(mysql_error(_pstMql)));
 }

 vector<string> vtFields;
 MYSQL_FIELD *field;
 while((field = mysql_fetch_field(pstRes)))
 {
 vtFields.push_back(field->name);
 }

 map<string, string> mpRow;
 MYSQL_ROW stRow;

 while((stRow = mysql_fetch_row(pstRes)) != (MYSQL_ROW)NULL)
 {
 mpRow.clear();
 unsigned long * lengths = mysql_fetch_lengths(pstRes);
 for(size_t i = 0; i < vtFields.size(); i++)
 {
 if(stRow[i])
 {
 mpRow[vtFields[i]] = string(stRow[i], lengths[i]);
 }
 else
 {
 mpRow[vtFields[i]] = "";
 }
 }

 data.data().push_back(mpRow);
 }

 mysql_free_result(pstRes);

 return data;
}

size_t MysqlHelper::updateRecord(const string &sTableName, const RECORD_DATA &mpColumns, const string &sCondition)
{
 string sSql = buildUpdateSQL(sTableName, mpColumns, sCondition);
 execute(sSql);

 return mysql_affected_rows(_pstMql);
}

size_t MysqlHelper::insertRecord(const string &sTableName, const RECORD_DATA &mpColumns)
{
 string sSql = buildInsertSQL(sTableName, mpColumns);
 execute(sSql);

 return mysql_affected_rows(_pstMql);
}

size_t MysqlHelper::replaceRecord(const string &sTableName, const RECORD_DATA &mpColumns)
{
 string sSql = buildReplaceSQL(sTableName, mpColumns);
 execute(sSql);

 return mysql_affected_rows(_pstMql);
}

size_t MysqlHelper::deleteRecord(const string &sTableName, const string &sCondition)
{
 ostringstream sSql;
 sSql << "delete from " << sTableName << " " << sCondition;

 execute(sSql.str());

 return mysql_affected_rows(_pstMql);
}

size_t MysqlHelper::getRecordCount(const string& sTableName, const string &sCondition)
{
 ostringstream sSql;
 sSql << "select count(*) as num from " << sTableName << " " << sCondition;

 MysqlData data = queryRecord(sSql.str());

 long n = atol(data[0]["num"].c_str());

 return n;

}

size_t MysqlHelper::getSqlCount(const string &sCondition)
{
 ostringstream sSql;
 sSql << "select count(*) as num " << sCondition;

 MysqlData data = queryRecord(sSql.str());

 long n = atol(data[0]["num"].c_str());

 return n;
}

int MysqlHelper::getMaxValue(const string& sTableName, const string& sFieldName,const string &sCondition)
{
 ostringstream sSql;
 sSql << "select " << sFieldName << " as f from " << sTableName << " " << sCondition << " order by f desc limit 1";

 MysqlData data = queryRecord(sSql.str());

 int n = 0;

 if(data.size() == 0)
 {
 n = 0;
 }
 else
 {
 n = atol(data[0]["f"].c_str());
 }

 return n;
}

bool MysqlHelper::existRecord(const string& sql)
{
 return queryRecord(sql).size() > 0;
}

long MysqlHelper::lastInsertID()
{
 return mysql_insert_id(_pstMql);
}

size_t MysqlHelper::getAffectedRows()
{
 return mysql_affected_rows(_pstMql);
}

//////////////////////////////////////////////////////////////////////////////////////////////////////////////
MysqlHelper::MysqlRecord::MysqlRecord(const map<string, string> &record):_record(record){}

const string& MysqlHelper::MysqlRecord::operator[](const string &s)
{
 map<string, string>::const_iterator it = _record.find(s);
 if(it == _record.end())
 {
 throw MysqlHelper_Exception("field '" + s + "' not exists.");
 }
 return it->second;
}

//////////////////////////////////////////////////////////////////////////////////////////////////////////////

vector<map<string, string> >& MysqlHelper::MysqlData::data()
{
 return _data;
}

size_t MysqlHelper::MysqlData::size()
{
 return _data.size();
}

MysqlHelper::MysqlRecord MysqlHelper::MysqlData::operator[](size_t i)
{
 return MysqlRecord(_data[i]);
}

}//end namespace 

/****************************************************
*@brief:mysqlhelper demo
*@autor:lvlv
*@date:2016.06.12
*@MySQL version:MySQL Community Server 5.6.30 (GPL)
****************************************************/

#include <string.h>
#include <iostream>
#include <string>
using namespace std;

#include "mysqlHelper.h"
using namespace mysqlhelper;


int main(int argc,char* argv[]){

 //初始化mysql對(duì)象并建立連接
 MysqlHelper mysqlHelper;
 mysqlHelper.init("119.29.184.114","root","123456","StudentCourse");
 try{
 mysqlHelper.connect();
 }catch(MysqlHelper_Exception& excep){
 cout<<excep.errorInfo;
 return -1;
 }

 //增加一條學(xué)生記錄
 //示例插入語(yǔ)句
 //string sql="insert into student values("201421031060","呂呂","華南理工大學(xué)","2014","軟件工程",1)";

 MysqlHelper::RECORD_DATA record;
 record.insert(make_pair("studentNo",make_pair(MysqlHelper::DB_STR,"201421031060")));
 record.insert(make_pair("name",make_pair(MysqlHelper::DB_STR,"呂呂")));
 record.insert(make_pair("school",make_pair(MysqlHelper::DB_STR,"廣州中醫(yī)藥大學(xué)")));
 record.insert(make_pair("grade",make_pair(MysqlHelper::DB_STR,"2014")));
 record.insert(make_pair("major",make_pair(MysqlHelper::DB_STR,"計(jì)算機(jī)科學(xué)與技術(shù)")));
 record.insert(make_pair("gender",make_pair(MysqlHelper::DB_INT,"1")));
 int res=0;
 try{
 res=mysqlHelper.insertRecord("student",record);
 }catch(MysqlHelper_Exception& excep){
 cout<<excep.errorInfo;
 return -1;
 }
 cout<<"res:"<<res<<" insert successfully "<<endl;

 //刪除一條學(xué)生記錄，學(xué)號(hào)為201421031059
 try{
 res=mysqlHelper.deleteRecord("student","where studentNo=\"201421031059\"");
 }catch(MysqlHelper_Exception& excep){
 cout<<excep.errorInfo;
 return -1;
 }
 cout<<"res:"<<res<<" delete successfully "<<endl;

 //查找學(xué)號(hào)為201421031059的學(xué)生選擇的所有課程名稱
 MysqlHelper::MysqlData dataSet;
 string querySQL="select courseName from course co where co.courseNo in (select courseNo from courseSelection where studentNo=\"201421031060\")";
 try{
 dataSet=mysqlHelper.queryRecord(querySQL);
 }catch(MysqlHelper_Exception& excep){
 cout<<excep.errorInfo;
 return -1;
 }
 cout<<"query successfully"<<endl;
 for(size_t i=0;i<dataSet.size();++i){
 cout<<dataSet[i]["courseName"]<<endl;
 }

 //修改學(xué)號(hào)為201421031060的學(xué)生專業(yè)
 MysqlHelper::RECORD_DATA recordChange;
 recordChange.insert(make_pair("major",make_pair(MysqlHelper::DB_STR,"軟件工程")));
 try{
 res=mysqlHelper.updateRecord("student",recordChange,"where studentNo=\"201421031060\"");
 }catch(MysqlHelper_Exception& excep){
 cout<<excep.errorInfo;
 return -1;
 }
 cout<<"res:"<<res<<" update successfully"<<endl;

 return 0;
}

##################################
# @brief:make scripts
# @date:2016.05.28
# @author:lvlv
##################################

#environment var
VPATH+=.

CC:=g++
FLAGS=-g -Wall -std=c++11
INC+=-I/usr/local/mysql/include
LIBDIR+=-L/usr/local/mysql/lib

CPPDIRS=.
CPPS=$(shell for dir in ${CPPDIRS};do echo $${dir}/*.cpp;done)

OBJDIR=obj
OBJS=$(patsubst %.cpp,${OBJDIR}/%.o,$(notdir ${CPPS}))

TARGET:=mysqlDemo.out

${TARGET}:${OBJS}
 ${CC} ${FLAGS} ${OBJS} -o $@ ${LIBDIR} -lmysqlclient

${OBJDIR}/%.o:./%.cpp
 ${CC} ${FLAGS} ${INC} -o $@ -c $<

.PHONY:clean
clean:
 rm -f ${TARGET} ${OBJDIR}/*