Mysql数据库学习（四）：常用Mysql C API 介绍和使用、封装一个访问Mysql数据库的类MysqlDB

#ifndef _MYSQL_DB_H_
#define _MYSQL_DB_H_

//#define WIN32_LEAN_AND_MEAN
#include <winsock2.h>
#include <mysql.h>

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

namespace DAL
{

class MysqlDB;
class MysqlRecordset
{
    friend class MysqlDB;
public:
    const string& GetItem(unsigned int nRow, unsigned int nCol) const
    {
        return rows_[nRow][nCol];
    }

    const string& GetItem(unsigned int nRow, const string& name) const
    {
        unsigned int index = GetFieldIndex(name);
        return rows_[nRow][index];
    }

    unsigned int GetRows() const
    {
        return rows_.size();
    }

    unsigned int GetCols() const
    {
        return fields_.size();
    }

    unsigned int GetFieldIndex(const std::string &name) const
    {
        unsigned int index = -1;
        for(unsigned int i = 0; i < fields_.size(); ++i)
        {
            if (fields_[i].name == name)
                index = fields_[i].index;
        }
        return index;
    }

    void Clear()
    {
        rows_.clear();
        fields_.clear();
    }

    typedef struct Field
    {
        string name; //列的字段名
        unsigned int index; //字段名对应的下标
    } FIELD;

    typedef vector<FIELD> FIELDS; //所有列的字段结构体集合
    typedef vector<string> ROW;  //每一行存储值

private:
    vector<ROW> rows_; // 总共存储多个行
    FIELDS fields_;
};

class MysqlDB
{
public:
    MysqlDB();
    ~MysqlDB();
    void Open(const char* host,
        const char* user,
        const char* passwd,
        const char* db,
        unsigned int port);
    void Close();

    unsigned long long ExecSQL(const char* sql);
    MysqlRecordset QuerySQL(const char* sql);

    unsigned long long GetInsertId() const;
    void StartTransaction();
    void Commit();
    void Rollback();


private:
    MYSQL* mysql_;
};

}

#endif // _MYSQL_DB_H_

#include <exception>
#include "MysqlDB.h"

using namespace std;

MysqlDB::MysqlDB() : mysql_(NULL)
{
}

MysqlDB::~MysqlDB()
{
    if (mysql_)
    {
        Close();
    }
}
void MysqlDB::Open(const char* host,
                   const char* user,
                   const char* passwd,
                   const char* db,
                   unsigned int port)
{

    mysql_ = mysql_init(NULL);
    if (mysql_ == NULL)
    {
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }

    my_bool reconnect = true;
    mysql_options(mysql_, MYSQL_OPT_RECONNECT, &reconnect);
    mysql_options(mysql_, MYSQL_SET_CHARSET_NAME, "gbk");

    if (!mysql_real_connect(mysql_, host, user,
        passwd, db, 0, NULL, 0))
    {
        string errmsg = mysql_error(mysql_);
        Close();
        throw Exception("DB ERROR:"+errmsg);
    }
}

void MysqlDB::Close()
{
    if (NULL != mysql_)
    {
        mysql_close(mysql_);
        mysql_ = NULL;
    }
}

MysqlRecordset MysqlDB::QuerySQL(const char* sql)
{
    if (mysql_query(mysql_, sql) != 0)
    {
        //int errno = mysql_errno(mysql_);
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }

    MYSQL_RES*  mysql_res;
    mysql_res = mysql_store_result(mysql_);

    //得到查询返回的行数
    //unsigned long n = mysql_affected_rows(mysql_);    

    //指向  mysql 的查询字段集
    MYSQL_FIELD* mysql_field = NULL;
    
    MysqlRecordset rs;
    unsigned int i = 0;
    unsigned int nCols = mysql_num_fields(mysql_res);
    while ((mysql_field = mysql_fetch_field(mysql_res)) != NULL)
    {
        MysqlRecordset::FIELD field;
        field.name = mysql_field->name;
        field.index = i;
        ++i;
        rs.fields_.push_back(field); //压入某个列字段的结构体
    }
 
    MYSQL_ROW mysql_row;
    while ((mysql_row = mysql_fetch_row(mysql_res)))
    {
        MysqlRecordset::ROW row(nCols);
        for (unsigned int i = 0; i< nCols; ++i)
        {
            row[i] = mysql_row[i] ? mysql_row[i] : "";
        }
        rs.rows_.push_back(row); //压入某一行的存储值
        
    }
    

    mysql_free_result(mysql_res);

    return rs;

}

unsigned long long MysqlDB::ExecSQL(const char* sql)
{
    if (mysql_query(mysql_, sql) != 0)
    {
        //int errno = mysql_errno(mysql_);
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }

    return mysql_affected_rows(mysql_);

}

void MysqlDB::StartTransaction()
{
    if (mysql_query(mysql_, "START TRANSACTION") != 0)
    {
        //int errno = mysql_errno(mysql_);
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }
}

void MysqlDB::Commit()
{
    if (mysql_query( mysql_, "COMMIT") != 0)
    {
        //int errno = mysql_errno(mysql_);
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }
}

void MysqlDB::Rollback()
{
    if (mysql_query(mysql_, "ROLLBACK") == 0)
    {
        //int errno = mysql_errno(mysql_);
        string errmsg = mysql_error(mysql_);
        throw Exception("DB ERROR:"+errmsg);
    }
}


unsigned long long MysqlDB::GetInsertId() const
{
    return mysql_insert_id(mysql_);  //auto_increment字段
}