D:/Zythum/DinoKod/DataBase/MySQL.cpp

//---------------------------------------------------------------------------------------------
//      This file is a part of "DinoKod".
//      Copyright © 2003 Dino Productions. All Rights Reserved.
//      
//      File                    : MySQL.cpp
//      Author                  : Sebastien LEIX        sebastien.leix@wanadoo.fr
//      Date                    : 06/05/2002
//      Modification    :
//
//---------------------------------------------------------------------------------------------
#include <winsock2.h>
#include "Common/Error.h"
#include "Common/Console.h"

#include "DataBase/MySQL.h"

//---------------------------------------------------------------------------------------------------------------------
KMySQL::KMySQL()
{
        m_pMySQL        = NULL;
        m_nFields       = 0;
        m_nRows         = 0;
        m_pResult       = NULL;
}

//---------------------------------------------------------------------------------------------------------------------
KMySQL::~KMySQL()
{
        FlushResult();
}

//---------------------------------------------------------------------------------------------------------------------
bool KMySQL::Init()
{
        if( m_pMySQL )
        {
                KError::Error( NULL, "KMySQL::Init() : MySQL already initialized" );
                return false;
        }

        m_pMySQL = mysql_init( NULL );
        if( !m_pMySQL )
        {
                KError::Error( NULL, "KMySQL::Init() : Cannot init mysql" );
                return false;
        }

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KMySQL::End()
{
        if( m_pMySQL )
                if( !Disconnect() )
                        return false;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KMySQL::Connect( char* pHost, char* pLogin, char* pPassword, char* pDataBase )
{
        if( !mysql_real_connect( m_pMySQL, pHost, pLogin, pPassword, pDataBase, 0, NULL, 0 ) )
                return false;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KMySQL::Disconnect()
{
        if( !m_pMySQL )
        {
                KError::Error( NULL, "KMySQL::Disconnect() : MySQL not initialized" );
                return false;
        }

        mysql_close( m_pMySQL );
        m_pMySQL = NULL;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
bool KMySQL::Query( char* pQuery, s32* pnFields, s32* pnRows, char**** pResult )
{
        *pnFields       = 0;
        *pnRows         = 0;
        *pResult        = NULL;

        if( !m_pMySQL )
        {
                KError::Error( NULL, "KMySQL::Query() : MySQL not initialized" );
                return false;
        }

        if( mysql_query( m_pMySQL, pQuery ) )
                return false;

        MYSQL_RES*      pMySQLResult;
        
        pMySQLResult = mysql_store_result( m_pMySQL );
        if( !pMySQLResult )
                return false;

        // Creation de la table de resultat
        MYSQL_ROW       Row;
        s32                     r, f;

        *pnFields       = mysql_num_fields( pMySQLResult );
        *pnRows         = (s32)mysql_num_rows( pMySQLResult );

        FlushResult();
        
        if( !(*pnFields) || !(*pnRows) )
                return false;

        AllocResult( *pnFields, *pnRows );

        r = 0;
        while( Row = mysql_fetch_row( pMySQLResult ) )
        {
                for( f = 0; f < *pnFields; f ++ )
                {
                        m_pResult[f][r] = strdup( Row[f] ? Row[f] : "" );
                }
                r ++;
        }

        *pResult = m_pResult;

        return true;
}

//---------------------------------------------------------------------------------------------------------------------
void KMySQL::AllocResult( s32 nFields, s32 nRows )
{
        KASSERT( !m_pResult );

        m_nFields       = nFields;
        m_nRows         = nRows;

        m_pResult = new char**[nFields];
        for( s32 f = 0; f < nFields; f ++ )
        {
                m_pResult[f] = new char*[nRows];
                for( s32 r = 0; r < m_nRows; r ++ )
                        m_pResult[f][r] = NULL;
        }
}

//---------------------------------------------------------------------------------------------------------------------
void KMySQL::FlushResult()
{
        if( !m_pResult )
                return;

        for( s32 f = 0; f < m_nFields; f ++ )
        {
                for( s32 r = 0; r < m_nRows; r ++ )
                        if( m_pResult[f][r] )
                                Freep( m_pResult[f][r] );

                Deletev( m_pResult[f] );
        }
        Deletev( m_pResult );

        m_nFields       = 0;
        m_nRows         = 0;
}

---