mysql的c++封装类

阳光穿透心脏的1/2处 2022-09-29 15:59 201阅读 0赞

// CppMysql - A C++ wrapper around the mysql database library.  
//  
// Copyright (c) 2009 Rob Groves. All Rights Reserved.  [lizp.net@gmail.com][lizp.net_gmail.com]  
//   
// Permission to use, copy, modify, and distribute this software and its  
// documentation for any purpose, without fee, and without a written  
// agreement, is hereby granted, provided that the above copyright notice,   
// this paragraph and the following two paragraphs appear in all copies,   
// modifications, and distributions.  
//  
// IN NO EVENT SHALL THE AUTHOR BE LIABLE TO ANY PARTY FOR DIRECT,  
// INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST  
// PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION,  
// EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.  
//  
// THE AUTHOR SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT  
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A  
// PARTICULAR PURPOSE. THE SOFTWARE AND ACCOMPANYING DOCUMENTATION, IF  
// ANY, PROVIDED HEREUNDER IS PROVIDED "AS IS". THE AUTHOR HAS NO OBLIGATION  
// TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.  
//  
// u can use it for anything, but u must show the source  
// frome  [http://blog.csdn.net/bat603][http_blog.csdn.net_bat603]  
// by ben  
// if u find some questions, please tell me with email  
//  
// V1.0  18/09/2009 -Initial Version for cppmysql  
  
\#ifndef \_CPP\_MYSQL\_1\_H\_  
\#define \_CPP\_MYSQL\_1\_H\_

\#include "mysql.h"

typedef unsigned int u\_int;  
typedef unsigned long u\_long;  
typedef MYSQL\*  DB\_HANDLE;

class CppMySQL3DB;

class CppMySQLQuery  
\{  
 friend class CppMySQL3DB;  
public:

CppMySQLQuery();

//当执行拷贝构造函数后，括号里的类已经无效，不能再使用  
    CppMySQLQuery(CppMySQLQuery& rQuery);

//当执行赋值构造函数后，=右边的类已经无效，不能再使用  
    CppMySQLQuery& operator=(CppMySQLQuery& rQuery);

virtual ~CppMySQLQuery();

u\_long numRow();//多少行  
    int numFields();//多少列

int fieldIndex(const char\* szField);  
 //0...n-1列  
    const char\* fieldName(int nCol);

//   const char\* fieldDeclType(int nCol);  
 //   int fieldDataType(int nCol);

u\_long seekRow(u\_long offerset);

int getIntField(int nField, int nNullValue=0);  
    int getIntField(const char\* szField, int nNullValue=0);

double getFloatField(int nField, double fNullValue=0.0);  
    double getFloatField(const char\* szField, double fNullValue=0.0);

//0...n-1列  
    const char\* getStringField(int nField, const char\* szNullValue="");  
    const char\* getStringField(const char\* szField, const char\* szNullValue="");

const unsigned char\* getBlobField(int nField, int& nLen);  
    const unsigned char\* getBlobField(const char\* szField, int& nLen);

bool fieldIsNull(int nField);  
    bool fieldIsNull(const char\* szField);

bool eof();

void nextRow();

void finalize();

private:  
 void freeRes();  
    void checkVM();  
private:  
 MYSQL\_RES\*  \_mysql\_res;  
 MYSQL\_FIELD\* \_field;  
 MYSQL\_ROW  \_row;  
 u\_long   \_row\_count;  
 u\_int   \_field\_count;

\};

class CppMySQL3DB  
\{  
public:

CppMySQL3DB();

virtual ~CppMySQL3DB();

int open(const char\* host, const char\* user, const char\* passwd, const char\* db,  
  unsigned int port = 0, unsigned long client\_flag = 0);

void close();

/\* 返回句柄 \*/  
 MYSQL\* getMysql();

/\* 处理返回多行的查询，返回影响的行数 \*/  
 //返回引用是因为在CppMySQLQuery的赋值构造函数中要把成员变量\_mysql\_res置为空  
 CppMySQLQuery& querySQL(const char \*sql);  
 /\* 执行非返回结果查询 \*/  
 int execSQL(const char\* sql);  
 /\* 测试mysql服务器是否存活 \*/  
 int ping();  
 /\* 关闭mysql 服务器 \*/  
 int shutDown();  
 /\* 主要功能:重新启动mysql 服务器 \*/  
 int reboot();  
 /\*  
 \* 说明:事务支持InnoDB or BDB表类型  
    \*/  
 /\* 主要功能:开始事务 \*/  
 int startTransaction();  
 /\* 主要功能:提交事务 \*/  
 int commit();  
 /\* 主要功能:回滚事务 \*/  
 int rollback();  
 /\* 得到客户信息 \*/  
 const char \* getClientInfo();  
 /\* 主要功能:得到客户版本信息 \*/  
 const unsigned long  getClientVersion();  
 /\* 主要功能:得到主机信息 \*/  
 const char \* getHostInfo();  
 /\* 主要功能:得到服务器信息 \*/  
 const char \* getServerInfo();  
 /\*主要功能:得到服务器版本信息\*/  
 const unsigned long  getServerVersion();  
 /\*主要功能:得到 当前连接的默认字符集\*/  
 const char \*  getCharacterSetName();  
 /\* 得到系统时间 \*/  
 const char \* getSysTime();  
 /\* 建立新数据库 \*/  
 int createDB(const char\* name);  
 /\* 删除制定的数据库\*/  
 int dropDB(const char\* name);

bool tableExists(const char\* table);

u\_int lastRowId();

void setBusyTimeout(int nMillisecs)\{\};  
private:

CppMySQL3DB(const CppMySQL3DB& db);  
    CppMySQL3DB& operator=(const CppMySQL3DB& db);

void checkDB();

private:  
 /\* msyql 连接句柄 \*/  
 MYSQL\* \_db\_ptr;  
 CppMySQLQuery \_db\_query;  
\};

\#endif

//.cpp

///

// CppMysql - A C++ wrapper around the mysql database library.  
//  
// Copyright (c) 2009 Rob Groves. All Rights Reserved.  [lizp.net@gmail.com][lizp.net_gmail.com]  
//   
// Permission to use, copy, modify, and distribute this software and its  
// documentation for any purpose, without fee, and without a written  
// agreement, is hereby granted, provided that the above copyright notice,   
// this paragraph and the following two paragraphs appear in all copies,   
// modifications, and distributions.  
//  
// IN NO EVENT SHALL THE AUTHOR BE LIABLE TO ANY PARTY FOR DIRECT,  
// INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST  
// PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION,  
// EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.  
//  
// THE AUTHOR SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT  
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A  
// PARTICULAR PURPOSE. THE SOFTWARE AND ACCOMPANYING DOCUMENTATION, IF  
// ANY, PROVIDED HEREUNDER IS PROVIDED "AS IS". THE AUTHOR HAS NO OBLIGATION  
// TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.  
//  
// u can use it for anything, but u must show the source  
// frome  [http://rainfish.cublog.cn][http_rainfish.cublog.cn]  
// by ben  
//  
// V1.0  18/09/2009 -Initial Version for cppmysql

\#include "cppmysql.h"  
\#include <string.h>  
\#include <stdlib.h>  
\#include <stdio.h>

CppMySQLQuery::CppMySQLQuery()  
\{  
 \_mysql\_res = NULL;  
 \_field = NULL;  
 \_row = NULL;  
 \_row\_count = 0;  
 \_field\_count = 0;

CppMySQLQuery::CppMySQLQuery(CppMySQLQuery& rQuery)  
\{  
 \*this = rQuery;  
\}

CppMySQLQuery& CppMySQLQuery::operator=(CppMySQLQuery& rQuery)  
\{  
 if ( this == &rQuery )  
  return \*this;

\_mysql\_res = rQuery.\_mysql\_res;  
 \_row = NULL;  
 \_row\_count = 0;  
 \_field\_count = 0;  
 \_field = NULL;

if ( \_mysql\_res != NULL )  
 \{  
  //定位游标位置到第一个位置  
  mysql\_data\_seek(\_mysql\_res, 0);  
  \_row =  mysql\_fetch\_row( \_mysql\_res );  
  \_row\_count = mysql\_num\_rows( \_mysql\_res );   
   //得到字段数量  
  \_field\_count = mysql\_num\_fields( \_mysql\_res );  
 \}

rQuery.\_mysql\_res = NULL;  
 rQuery.\_field = NULL;  
 rQuery.\_row = NULL;  
 rQuery.\_row\_count = 0;  
 rQuery.\_field\_count = 0;

return \*this;

CppMySQLQuery::~CppMySQLQuery()  
\{  
 freeRes();  
\}

void CppMySQLQuery::freeRes()  
\{  
 if ( \_mysql\_res != NULL )  
 \{  
  mysql\_free\_result(\_mysql\_res);  
  \_mysql\_res = NULL;  
 \}  
\}

u\_long CppMySQLQuery::numRow()  
\{  
 return \_row\_count;  
\}

int CppMySQLQuery::numFields()  
\{  
 return \_field\_count;  
\}

u\_long CppMySQLQuery::seekRow(u\_long offerset)  
\{  
 if ( offerset < 0 )  
  offerset = 0;  
 if ( offerset >= \_row\_count )  
  offerset = \_row\_count -1;

mysql\_data\_seek(\_mysql\_res, offerset);  
   
 \_row = mysql\_fetch\_row(\_mysql\_res);  
 return offerset;  
\}

int CppMySQLQuery::fieldIndex(const char\* szField)  
\{  
 if ( NULL == \_mysql\_res )  
  return -1;  
 if ( NULL == szField )  
  return -1;

mysql\_field\_seek(\_mysql\_res, 0);//定位到第0列  
 u\_int i = 0;  
 while ( i < \_field\_count )  
 \{  
  \_field = mysql\_fetch\_field( \_mysql\_res );  
  if ( \_field != NULL && strcmp(\_field->name, szField) == 0 )//找到  
   return i;

i++;  
 \}

return -1;  
\}

const char\* CppMySQLQuery::fieldName(int nCol)  
\{  
 if ( \_mysql\_res == NULL )  
  return NULL;

mysql\_field\_seek(\_mysql\_res, nCol);  
 \_field = mysql\_fetch\_field(\_mysql\_res);

if ( \_field != NULL )  
  return \_field->name;  
 else  
  return  NULL;  
\}

int CppMySQLQuery::getIntField(int nField, int nNullValue/\*=0\*/)  
\{  
 if ( NULL == \_mysql\_res )  
  return nNullValue;  
   
 if ( nField + 1 > (int)\_field\_count )  
  return nNullValue;  
   
 if ( NULL == \_row )  
  return nNullValue;  
   
 return atoi(\_row\[nField\]);  
\}

int CppMySQLQuery::getIntField(const char\* szField, int nNullValue/\*=0\*/)  
\{  
 if ( NULL == \_mysql\_res || NULL == szField )  
  return nNullValue;  
   
 if ( NULL == \_row )  
  return nNullValue;

const char\* filed = getStringField(szField);

if ( NULL == filed )  
  return nNullValue;

return atoi(filed);  
\}

const char\* CppMySQLQuery::getStringField(int nField, const char\* szNullValue/\*=""\*/)  
\{  
 if ( NULL == \_mysql\_res )  
  return szNullValue;

if ( nField + 1 > (int)\_field\_count )  
  return szNullValue;

if ( NULL == \_row )  
  return szNullValue;

return \_row\[nField\];  
\}  
      
const char\* CppMySQLQuery::getStringField(const char\* szField, const char\* szNullValue/\*=""\*/)  
\{  
 if ( NULL == \_mysql\_res )  
  return szNullValue;

int nField = fieldIndex(szField);  
 if ( nField == -1 )  
  return szNullValue;  
   
 return getStringField(nField);  
\}

double CppMySQLQuery::getFloatField(int nField, double fNullValue/\*=0.0\*/)  
\{  
 const char\* field = getStringField(nField);

if ( NULL == field )  
  return fNullValue;

return atol(field);  
\}  
   
double CppMySQLQuery::getFloatField(const char\* szField, double fNullValue/\*=0.0\*/)  
\{  
 const char\* field = getStringField(szField);  
 if ( NULL == field )  
  return fNullValue;

return atol(field);  
\}

void CppMySQLQuery::nextRow()  
\{  
 if ( NULL == \_mysql\_res )  
  return;

\_row = mysql\_fetch\_row(\_mysql\_res);  
\}

bool CppMySQLQuery::eof()  
\{  
 if ( \_row == NULL )  
  return true;

return false;  
\}

CppMySQL3DB::CppMySQL3DB()  
\{  
 \_db\_ptr = NULL;  
\}

CppMySQL3DB::~CppMySQL3DB()  
\{  
 if ( \_db\_ptr != NULL )  
 \{  
  close();  
 \}  
\}

int CppMySQL3DB::open(const char\* host, const char\* user, const char\* passwd, const char\* db,  
  unsigned int port /\*= 0\*/, unsigned long client\_flag /\*= 0\*/)  
\{  
 int ret = -1;

\_db\_ptr = mysql\_init(NULL);  
 if( NULL == \_db\_ptr )   
  goto EXT;  
   
 //如果连接失败，返回NULL。对于成功的连接，返回值与第1个参数的值相同。  
 if ( NULL == mysql\_real\_connect( \_db\_ptr, host, user, passwd, db,port, NULL, client\_flag) )  
  goto EXT;

//选择制定的数据库失败  
 //0表示成功，非0值表示出现错误。  
 if ( mysql\_select\_db( \_db\_ptr, db ) != 0 )   
 \{  
  mysql\_close(\_db\_ptr);  
  \_db\_ptr = NULL;  
  goto EXT;  
 \}

ret = 0;  
EXT:  
 //初始化mysql结构失败  
 if ( ret == -1 && \_db\_ptr != NULL )  
 \{  
  mysql\_close( \_db\_ptr );  
  \_db\_ptr = NULL;  
 \}

return ret;  
\}

void CppMySQL3DB::close()  
\{  
 if ( \_db\_ptr != NULL )  
 \{  
  mysql\_close( \_db\_ptr );  
  \_db\_ptr = NULL;  
 \}  
\}

MYSQL\* CppMySQL3DB::getMysql()  
\{  
 return \_db\_ptr;  
\}

/\* 处理返回多行的查询，返回影响的行数 \*/  
CppMySQLQuery& CppMySQL3DB::querySQL(const char \*sql)  
\{  
 if ( !mysql\_real\_query( \_db\_ptr, sql, strlen(sql) ) )  
 \{  
  \_db\_query.\_mysql\_res = mysql\_store\_result( \_db\_ptr );  
//   \_db\_query.\_row =  mysql\_fetch\_row( \_db\_query.\_mysql\_res );  
//   \_db\_query.\_row\_count = mysql\_num\_rows( \_db\_query.\_mysql\_res );   
//   //得到字段数量  
//   \_db\_query.\_field\_count = mysql\_num\_fields( \_db\_query.\_mysql\_res );  
 \}

return \_db\_query;  
\}

/\* 执行非返回结果查询 \*/  
int CppMySQL3DB::execSQL(const char\* sql)  
\{  
 if( !mysql\_real\_query( \_db\_ptr, sql, strlen(sql) ) )  
 \{  
  //得到受影响的行数  
  return (int)mysql\_affected\_rows(\_db\_ptr) ;  
 \}  
 else  
 \{  
  //执行查询失败  
  return -1;  
 \}  
\}

/\* 测试mysql服务器是否存活 \*/  
int CppMySQL3DB::ping()  
\{  
 if( mysql\_ping(\_db\_ptr) == 0 )  
  return 0;  
 else   
  return -1;   
\}

/\* 关闭mysql 服务器 \*/  
int CppMySQL3DB::shutDown()  
\{  
 if( mysql\_shutdown(\_db\_ptr,SHUTDOWN\_DEFAULT) == 0 )  
  return 0;  
 else   
  return -1;  
\}

/\* 主要功能:重新启动mysql 服务器 \*/  
int CppMySQL3DB::reboot()  
\{  
 if(!mysql\_reload(\_db\_ptr))  
  return 0;  
 else  
  return -1;  
\}

/\*  
\* 说明:事务支持InnoDB or BDB表类型  
\*/  
/\* 主要功能:开始事务 \*/  
int CppMySQL3DB::startTransaction()  
\{  
 if(!mysql\_real\_query(\_db\_ptr, "START TRANSACTION" ,  
  (unsigned long)strlen("START TRANSACTION") ))  
 \{  
  return 0;  
 \}  
 else  
  //执行查询失败  
  return -1;  
\}

/\* 主要功能:提交事务 \*/  
int CppMySQL3DB::commit()  
\{  
 if(!mysql\_real\_query( \_db\_ptr, "COMMIT",  
  (unsigned long)strlen("COMMIT") ) )  
 \{  
  return 0;  
 \}  
 else        
  //执行查询失败  
  return -1;  
\}

/\* 主要功能:回滚事务 \*/  
int CppMySQL3DB::rollback()  
\{  
 if(!mysql\_real\_query(\_db\_ptr, "ROLLBACK",  
  (unsigned long)strlen("ROLLBACK") ) )  
  return 0;  
 else    
  //执行查询失败  
  return -1;   
\}

/\* 得到客户信息 \*/  
const char \* CppMySQL3DB::getClientInfo()  
\{  
 return mysql\_get\_client\_info();  
\}

/\* 主要功能:得到客户版本信息 \*/  
const unsigned long  CppMySQL3DB::getClientVersion()  
\{  
 return mysql\_get\_client\_version();  
\}

/\* 主要功能:得到主机信息 \*/  
const char \* CppMySQL3DB::getHostInfo()  
\{  
 return mysql\_get\_host\_info(\_db\_ptr);  
\}

/\* 主要功能:得到服务器信息 \*/  
const char \* CppMySQL3DB::getServerInfo()  
\{  
 return mysql\_get\_server\_info( \_db\_ptr );

\}  
/\*主要功能:得到服务器版本信息\*/  
const unsigned long  CppMySQL3DB::getServerVersion()  
\{  
 return mysql\_get\_server\_version(\_db\_ptr);

/\*主要功能:得到 当前连接的默认字符集\*/  
const char \*  CppMySQL3DB::getCharacterSetName()  
\{  
 return mysql\_character\_set\_name(\_db\_ptr);

/\* 得到系统时间 \*/  
const char \* CppMySQL3DB::getSysTime()  
\{  
 //return ExecQueryGetSingValue("select now()");  
 return NULL;

/\* 建立新数据库 \*/  
int CppMySQL3DB::createDB(const char\* name)  
\{  
 char temp\[1024\];

sprintf(temp, "CREATE DATABASE %s", name);

if(!mysql\_real\_query( \_db\_ptr, temp, strlen(temp)) )  
  return 0;  
   
 else  
  //执行查询失败  
  return -1;  
\}

/\* 删除制定的数据库\*/  
int CppMySQL3DB::dropDB(const char\*  name)  
\{  
 char temp\[1024\];  
   
 sprintf(temp, "DROP DATABASE %s", name);

if(!mysql\_real\_query( \_db\_ptr, temp, strlen(temp)) )  
  return 0;  
 else  
  //执行查询失败  
  return -1;  
\}

bool CppMySQL3DB::tableExists(const char\* table)  
\{  
 return false;  
\}

u\_int CppMySQL3DB::lastRowId()  
\{  
 return 0;  
\}

[lizp.net_gmail.com]: mailto:lizp.net@gmail.com
[http_blog.csdn.net_bat603]: http://blog.csdn.net/bat603
[http_rainfish.cublog.cn]: http://rainfish.cublog.cn/