db_mysql.cpp

Go to the documentation of this file.

#include "db_mysql.h"
#include <unistd.h>
db_mysql::db_mysql()
    :m_Mysql(NULL)
    ,m_Res(NULL)
    ,m_bConnect(false)
    ,m_bSelectDB(false)
    ,m_bExecute(false)
    ,m_bGetRow(false)
    ,m_bInit(false)
{
    ;   
}
db_mysql::db_mysql(string host,string user,string passwd,string db)
    :m_Mysql(NULL)
    ,m_Res(NULL)
    ,m_bConnect(false)
    ,m_bSelectDB(false)
    ,m_bExecute(false)
    ,m_bGetRow(false)
{
    Init(host,user,passwd,db);
}   
 
void db_mysql::Init(string host,string user,string passwd,string db)
{
    m_strHost = host;
    m_strUser = user;
    m_strPasswd = passwd;
    m_strDB = db;
    m_bInit = true;
    if(!db.empty())
    {
        m_bSelectDB = true;
    }
}
void db_mysql::SetQuery(string query)
{
    m_strQuery=query;
}
bool db_mysql::Connect(void)
{
    if(!m_bConnect)
    {
        my_bool b = 0;
        if(!(m_Mysql=mysql_init(NULL)))
            {
            cout<<"Init mysql error!!"<<endl;
            return false; 
            }
//      mysql_options(m_Mysql,MYSQL_REPORT_DATA_TRUNCATION,&b);
            if(!mysql_real_connect(m_Mysql,m_strHost.c_str(),m_strUser.c_str(),m_strPasswd.c_str(),m_strDB.c_str(),0,NULL,0))
        {
            cout<<"Connect error!!"<<endl;
            for(int i=0;i<10;i++)
            {
                cout<<"Reconnect ..."<<endl;
                sleep(1);
                if(mysql_real_connect(m_Mysql,m_strHost.c_str(),m_strUser.c_str(),m_strPasswd.c_str(),m_strDB.c_str(),0,NULL,0))
                {
                    cout<<"Reconnect success"<<endl;
                    m_bConnect = true;
                    return true;
                }
                cout<<"Reconnect fail"<<endl;
            }
            mysql_close(m_Mysql);
            return false;
        }
    }
    else
    {
        cout<<"You are reconnecting!"<<endl;
        return false;   
    }
    m_bConnect = true;
    return true;
}
bool db_mysql::SelectDB(string db)
{
    if(!m_bConnect)
    {
        cout<<"You must connect before select db!";
        return false;
    }
    if(mysql_select_db(m_Mysql,db.c_str()))
    {
        cout<<"Select database error!"<<endl;
        return false;
    }
    m_strDB = db;
    m_bSelectDB = true;
    return true;
}
    
bool db_mysql::Execute(string query)
{
    if((!m_bConnect)||(!m_bSelectDB))
    {
        cout<<"You executing query before connection or select db!"<<endl;
        return false;
    }
    if(!query.empty())
    {
        m_strQuery=query;
    }
    if(m_strQuery.empty())
    {
        cout<<"Your haven't set query string!"<<endl;
        return false;   
    }
    else
    {
        Free_Result();
        if(mysql_real_query(m_Mysql,m_strQuery.c_str(),m_strQuery.size()))
        {
            cout<<"Execute query error!"<<endl;
            return false;   
        }   
    }
//zhaohs
        Field_num=mysql_field_count(m_Mysql);
        if(Field_num==0){
    m_result=false;
    }
    else {m_result=true;}
//zhaohs
    m_Res = mysql_store_result(m_Mysql);
    m_bExecute = true;
    return true;
}
    
bool db_mysql::GetRow(my_ulonglong row)
{
    if(!m_bExecute)
    {
        cout<<"You must execute query before get row!"<<endl;
        return false;
    }
    if(row!=NEXT_ROW)
    {
        if(row>=mysql_num_rows(m_Res)||row<0)
        {
//          cout<<"Select row error! Row number must be set in range!"<<endl;
            return false;
        }
        mysql_data_seek(m_Res,row); 
    }
    m_Row=mysql_fetch_row(m_Res);
    if(m_Row==NULL)
    {
        cout<<"fetch_row error!"<<endl; 
        return false;
    }
    
    m_bGetRow = true;
    return true;
}
 
string db_mysql::GetField(unsigned int n)
{
        
    if(!m_bGetRow)
    {
        cout<<"You must get a row before getfield value!"<<endl;
        return "";
    }
    if(m_Row[n]!=NULL){ return (string)m_Row[n];}
    else return "";
}
//// gaizao
/*string db_mysql::GetField(string name)
{
 
         if(!m_bGetRow)
             {cout<<"You must get a row before getfield value!"<<endl;
          return "";
         }
        if(m_Row[name]!=NULL){ return (string)m_Row[name];}
        else return "";
}*/
/////  myself---zhaohs
string db_mysql::GetField_name(unsigned int n)
{
        if(!m_result)
            {
         cout<<"error:no select!"<<endl;
        return "";
        }
        m_field = mysql_fetch_field_direct(m_Res, n);
    return string(m_field->name);
 
 
//  num_fields = mysql_num_fields(result);
//  m_field = mysql_fetch_fields(m_Res);
  //      std::cout<<m_field[n].name<<std::endl;
//  return m_field[n].name;
 
       
}
 
////
unsigned long db_mysql::Num_Rows(void)
{
    if(!m_bExecute)
    {
        cout<<"You are get number of rows before execute!"<<endl;
        return 0;
    }
    return mysql_num_rows(m_Res);
}
 
void db_mysql::Free_Result(void)
{
    if(m_bExecute)
    {
        mysql_free_result(m_Res);
        m_bExecute = false;
        m_bGetRow = false;  
    }
}
 
void db_mysql::Close(void)
{
    if(m_bConnect)
    {
        Free_Result();
        mysql_close(m_Mysql);
        m_bConnect = false;
    }
}
 
db_mysql::~db_mysql()
{
    Free_Result();
    Close();
}