mysqlconnection_p.cpp

/* This file is part of the KDE project
   Copyright (C) 2004 Jaroslaw Staniek <js@iidea.pl>
   Copyright (C) 2004 Martin Ellis <martin.ellis@kdemail.net>
 
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.
 
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Library General Public License for more details.
 
You should have received a copy of the GNU Library General Public License
along with this program; see the file COPYING.  If not, write to
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
*/
 
#include <tqcstring.h>
#include <tqstring.h>
#include <tqstringlist.h>
#include <tqfile.h>
 
#include <kdebug.h>
 
#include "mysqlconnection_p.h"
 
#include <kexidb/connectiondata.h>
 
#ifdef MYSQLMIGRATE_H
#define NAMESPACE KexiMigration
#else
#define NAMESPACE KexiDB
#endif
 
using namespace NAMESPACE;
 
/* ************************************************************************** */
MySqlConnectionInternal::MySqlConnectionInternal(KexiDB::Connection* connection)
    : ConnectionInternal(connection)
    , mysql(0)
    , mysql_owned(true)
    , res(0)
{
}
 
MySqlConnectionInternal::~MySqlConnectionInternal()
{
    if (mysql_owned && mysql) {
        mysql_close(mysql);
        mysql = 0;
    }
}
 
void MySqlConnectionInternal::storeResult()
{
    res = mysql_errno(mysql);
    errmsg = mysql_error(mysql);
}
 
/* ************************************************************************** */
//bool MySqlConnectionInternal::db_connect(TQCString host, TQCString user,
//  TQCString password, unsigned short int port, TQString socket)
bool MySqlConnectionInternal::db_connect(const KexiDB::ConnectionData& data)
{
    if (!(mysql = mysql_init(mysql)))
        return false;
 
    KexiDBDrvDbg << "MySqlConnectionInternal::connect()" << endl;
    TQCString localSocket;
    TQString hostName = data.hostName;
    if (hostName.isEmpty() || hostName.lower()=="localhost") {
        if (data.useLocalSocketFile) {
            if (data.localSocketFileName.isEmpty()) {
                TQStringList sockets;
    #ifndef TQ_WS_WIN
                sockets.append("/var/lib/mysql/mysql.sock");
                sockets.append("/var/run/mysqld/mysqld.sock");
                sockets.append("/tmp/mysql.sock");
        
                for(TQStringList::ConstIterator it = sockets.constBegin(); it != sockets.constEnd(); it++)
                {
                    if(TQFile(*it).exists()) {
                        localSocket = ((TQString)(*it)).local8Bit();
                        break;
                    }
                }
    #endif
            }
            else
                localSocket = TQFile::encodeName(data.localSocketFileName);
        }
        else {
            //we're not using local socket
            hostName = "127.0.0.1"; //this will force mysql to connect to localhost
        }
    }

    const char *pwd = data.password.isNull() ? 0 : data.password.latin1();
    mysql_real_connect(mysql, hostName.latin1(), data.userName.latin1(), 
        pwd, 0, data.port, localSocket, 0);
    if(mysql_errno(mysql) == 0)
        return true;
 
    storeResult(); //store error msg, if any - can be destroyed after disconnect()
    db_disconnect();
//  setError(ERR_DB_SPECIFIC,err);
    return false;
}

bool MySqlConnectionInternal::db_disconnect()
{
    mysql_close(mysql);
    mysql = 0;
    KexiDBDrvDbg << "MySqlConnection::disconnect()" << endl;
    return true;
}
 
/* ************************************************************************** */
bool MySqlConnectionInternal::useDatabase(const TQString &dbName) {
//TODO is here escaping needed?
    return executeSQL("USE " + dbName);
}

bool MySqlConnectionInternal::executeSQL(const TQString& statement) {
//  KexiDBDrvDbg << "MySqlConnectionInternal::executeSQL: "
//               << statement << endl;
    TQCString queryStr=statement.utf8();
    const char *query=queryStr;
    if(mysql_real_query(mysql, query, strlen(query)) == 0)
    {
        return true;
    }
 
    storeResult();
//  setError(ERR_DB_SPECIFIC,mysql_error(m_mysql));
    return false;
}
 
TQString MySqlConnectionInternal::escapeIdentifier(const TQString& str) const {
    return TQString(str).replace('`', "'");
}
 
//--------------------------------------
 
MySqlCursorData::MySqlCursorData(KexiDB::Connection* connection)
: MySqlConnectionInternal(connection)
, mysqlres(0)
, mysqlrow(0)
, lengths(0)
, numRows(0)
{
    mysql_owned = false;
}
 
MySqlCursorData::~MySqlCursorData()
{
}