用于操作mysql数据库的c++类

//DBMysql.h文件

#ifndef DB_MYSQL_H
#define DB_MYSQL_H
#include <winsock2.h>
#include <mysql.h>
#include <string>
#include <map>
#include <vector>
#include <iostream>
using namespace std;
typedef map<string,string > strMap;

/*mysql操作类，封装了c语言相关的api，可实现基本的查询、插入、修改和删除动作*/
class DBMysql
{
protected:
 MYSQL *mysql; //代表一个到数据库的连接
private:
 string host; //连接的服务器
 string user; //用户名
 string password; //连接密码
 unsigned int port; //连接端口
 string db; //操作的数据库的名称
 MYSQL_RES *result; //操作的结果
 string query; //sql语句
 unsigned long num; //返回查询得到的结果数
 string error; //错误提示信息
 unsigned int debug; //是否显示调试信息
 strMap info ; //查询语句返回一条结果
 vector<strMap> arrInfo; //查询语句可能会返回多条结果
 vector<string> fields; //返回查询结果的列
 void disPlayError();
public:
 DBMysql(string host,string user,string password,unsigned int port);// 构造函数
 DBMysql(); //构造函数
 void SetConnect(string host,string user,string password,unsigned int port);//确定连接参数
 unsigned int DBConnect();//连接数据库
 unsigned int DBSelect(string db); //连接一个数据库
 void SetQuery(string q); //设定查询语句
 unsigned int DBQuery(); //查询数据库
 strMap GetInfo(); //返回查询得到的一条结果
 vector<strMap> GetArray(); //返回查询得到的结果
 string GetError(); //返回错误信息
 vector<string> GetFields();//返回查询后的列值
 unsigned int InsertData(string table,strMap *data); //向数据库中插入一条数据
 unsigned long GetLastID(); //返回最后一个自动增量的值
 unsigned long GetNum(); //返回一条sql语句影响的行数
 unsigned int UpdateData(string table,strMap *data,string condition); //根据条件修改一条数据
 unsigned int DeleteData(string table,string condition); //根据条件删除数据
 ~DBMysql();//析构函数
};
#endif

 
 //DBMysql.cpp文件

#include 'DBMysql.h'
#include <assert.h>
/*构造函数，设定连接的服务器，用户名，密码和端口*/
DBMysql::DBMysql(string host,string user,string password,unsigned int port=3306)
{
 mysql = mysql_init(NULL);
 num = 0;
 error='';
 query='';
 result = NULL;
 SetConnect(host,user,password,port);
}
DBMysql::DBMysql()
{
}
/*设定连接的服务器，用户名，密码和端口*/
void DBMysql::SetConnect(string host,string user,string password,unsigned int port)
{
 DBMysql::host = host;
 DBMysql::user = user;
 DBMysql::password = password;
 DBMysql::port = port;
}
/*连接数据库*/
unsigned int DBMysql::DBConnect()
{
 MYSQL *con;
 if(mysql == NULL)
 {
  error = '初始化mysql错误';
  return 1;
 }
 con = mysql_real_connect(mysql,host.c_str(),user.c_str(),password.c_str(),NULL,port,NULL,0);
 if(con == NULL)
 {
  error=mysql_error(mysql);
  return mysql_errno(mysql);
 }
 return 0;
}
/*选择一个数据库*/
unsigned int DBMysql::DBSelect(string database)
{
 unsigned int re;
 if( mysql == NULL) return 1;
 db = database;
 re = mysql_select_db(mysql,db.c_str());
 if(re != 0)
 {
  error+=mysql_error(mysql);
 }
 return re;
}
/*设定sql语句*/
void DBMysql::SetQuery(string q)
{

 assert(!q.empty());
 if(result != NULL )
 {
  mysql_free_result(result);
 }
 query = q;
}
/*执行sql语句*/
unsigned int DBMysql::DBQuery()
{
 unsigned int re;
 if( mysql == NULL) return 1;
 assert(!query.empty());
 re = mysql_query(mysql,query.c_str());
 if(re == 0)
 {
  result = mysql_store_result(mysql);
  num = mysql_affected_rows(mysql);
  info.clear();
  arrInfo.clear();
  fields.clear();
 }
 else
 {
  re = mysql_errno(mysql);
  error = mysql_error(mysql);
  cout<<error<<endl;
 }
 return re;
}

/*获取查询得到的一条结果*/
strMap DBMysql::GetInfo()
{
 MYSQL_ROW row;
 unsigned int i;
 assert(mysql != NULL);
 if(info.size() > 0) return info;
 if(result != NULL)
 {
  GetFields();
  row = mysql_fetch_row(result);
  if(row != NULL)
  {
   for(i=0;i<fields.size();i++)
   {
    info[fields[i]] = (char*)row[i];
   }
  }
  //mysql_free_result(result);
 }
 return info;
}
/*获取查询得到的所有结果*/
vector<strMap> DBMysql::GetArray()
{
 MYSQL_ROW row;
 unsigned int i;
 strMap tmp;
 assert(mysql != NULL);
 if(arrInfo.size() > 0) return arrInfo;
 if(result != NULL)
 {
  GetFields();
  while(row = mysql_fetch_row(result))
  {
   if(row != NULL)
   {
    for(i=0;i<fields.size();i++)
    {
     tmp[fields[i]] = (char *)row[i];
    }
    arrInfo.push_back(tmp);
   }
  }
 }
 return arrInfo;
}
/*获取sql语句执行影响的行数*/
unsigned long DBMysql::GetNum()
{
 return num;
}
/*获取插入后的id号*/
unsigned long DBMysql::GetLastID()
{
 return mysql_insert_id(mysql);
}
/*向数据库插入数据*/
unsigned int DBMysql::InsertData(string table,strMap *data)
{
 strMap::const_iterator iter;
 string q1;
 int flag=0;
 assert(mysql != NULL);
 assert(!table.empty());
 assert(data != NULL);
 for(iter = data->begin();iter!= data->end();iter++)
 {
  if(flag == 0)
  {
   q1 = 'insert into ';
   q1+=table;
   q1+=' set ';
   q1+=iter->first;
   q1+='='';
   q1+=iter->second;
   q1+=''';
   flag++;
  }
  else
  {
   q1+=',';
   q1+=iter->first;
   q1+='='';
   q1+=iter->second;
   q1+=''';
  }
 }
 SetQuery(q1);
 return DBQuery();
}
/*根据条件修改数据*/
unsigned int DBMysql::UpdateData(string table,strMap *data,string condition)
{
 strMap::const_iterator iter;
 string q1;
 int flag=0;
 assert(mysql != NULL);
 assert(!table.empty());
 assert(data != NULL);
 for(iter = data->begin();iter!= data->end();iter++)
 {
  if(flag == 0)
  {
   q1 = 'update ';
   q1+=table;
   q1+=' set ';
   q1+=iter->first;
   q1+='='';
   q1+=iter->second;
   q1+=''';
   flag++;
  }
  else
  {
   q1+=',';
   q1+=iter->first;
   q1+='='';
   q1+=iter->second;
   q1+=''';
  }
 }
 if(!condition.empty())
 {
  q1+=' where ';
  q1+=condition;
 }
 SetQuery(q1);
 return DBQuery();
}
/*根据条件删除数据*/
unsigned int DBMysql::DeleteData(string table,string condition)
{
 string q;
 assert(mysql != NULL);
 assert(!table.empty());
 q='delete from ';
 q+=table;
 if(!condition.empty())
 {
  q+=' where ';
  q+=condition;
 }
 SetQuery(q);
 return DBQuery();
}

/*获取返回的错误信息*/
string DBMysql::GetError()
{
 return error;
}

/*返回查询后的列值*/
vector<string> DBMysql::GetFields()
{
 /*
  field = mysql_fetch_fields(result);
  然后通过field[i].name访问在此有错误，不知道为什么，可能是mysql的bug
 */
 MYSQL_FIELD *field;
 assert(mysql != NULL);
 if(fields.size()>0) return fields;
 while(field = mysql_fetch_field(result))
 {
  fields.push_back(field->name);
 }
 return fields;
}

/*析构函数*/
DBMysql::~DBMysql()
{
 if(result != NULL)
  mysql_free_result(result);
 fields.clear();
 error='';
 info.clear();
 db='';
 arrInfo.clear();
 mysql_close(mysql);
}

//例子test.php

#include 'DBMysql.h'
#include <string>
#include <vector>
using namespace std;
int main()
{
 vector<strMap> info;
 DBMysql db('localhost','root','');
 db.DBConnect();
 db.DBSelect('mysql');
 string query = 'select user,password from user';
 db.SetQuery(query);
 db.DBQuery();
 info = db.GetArray();
 for(int i=0;i<info.size();i++)
 {
  cout<<info[i]['user']<<':'<<info[i]['password']<<endl;
 }
 return 1;
}

//呵呵，希望大家能够喜欢，有bug或改进意见希望告诉我。

 

【来源】