/src/shared/Database/DatabaseMysql.cpp
C++ | 409 lines | 306 code | 73 blank | 30 comment | 70 complexity | 3ac316703cff2f3a5a3dd8adc9a36d9e MD5 | raw file
Possible License(s): CC-BY-SA-3.0, GPL-2.0
- /*
- * Copyright (C) 2005-2010 MaNGOS <http://getmangos.com/>
- * Copyright (C) 2010 NeoCore
- */
- #ifndef DO_POSTGRESQL
- #include "Util.h"
- #include "Policies/SingletonImp.h"
- #include "Platform/Define.h"
- #include "Threading.h"
- #include "DatabaseEnv.h"
- #include "Database/MySQLDelayThread.h"
- #include "Database/SqlOperations.h"
- #include "Timer.h"
- void DatabaseMysql::ThreadStart()
- {
- mysql_thread_init();
- }
- void DatabaseMysql::ThreadEnd()
- {
- mysql_thread_end();
- }
- size_t DatabaseMysql::db_count = 0;
- DatabaseMysql::DatabaseMysql() : Database(), mMysql(0)
- {
- // before first connection
- if( db_count++ == 0 )
- {
- // Mysql Library Init
- mysql_library_init(-1, NULL, NULL);
- if (!mysql_thread_safe())
- {
- sLog.outError("FATAL ERROR: Used MySQL library isn't thread-safe.");
- exit(1);
- }
- }
- }
- DatabaseMysql::~DatabaseMysql()
- {
- if (m_delayThread)
- HaltDelayThread();
- if (mMysql)
- mysql_close(mMysql);
- //Free Mysql library pointers for last ~DB
- if(--db_count == 0)
- mysql_library_end();
- }
- bool DatabaseMysql::Initialize(const char *infoString)
- {
- if(!Database::Initialize(infoString))
- return false;
- tranThread = NULL;
- MYSQL *mysqlInit;
- mysqlInit = mysql_init(NULL);
- if (!mysqlInit)
- {
- sLog.outError( "Could not initialize Mysql connection" );
- return false;
- }
- InitDelayThread();
- Tokens tokens = StrSplit(infoString, ";");
- Tokens::iterator iter;
- std::string host, port_or_socket, user, password, database;
- int port;
- char const* unix_socket;
- iter = tokens.begin();
- if(iter != tokens.end())
- host = *iter++;
- if(iter != tokens.end())
- port_or_socket = *iter++;
- if(iter != tokens.end())
- user = *iter++;
- if(iter != tokens.end())
- password = *iter++;
- if(iter != tokens.end())
- database = *iter++;
- mysql_options(mysqlInit,MYSQL_SET_CHARSET_NAME,"utf8");
- if(host==".") // socket use option (Unix/Linux)
- {
- unsigned int opt = MYSQL_PROTOCOL_SOCKET;
- mysql_options(mysqlInit,MYSQL_OPT_PROTOCOL,(char const*)&opt);
- host = "localhost";
- port = 0;
- unix_socket = port_or_socket.c_str();
- }
- else // generic case
- {
- port = atoi(port_or_socket.c_str());
- unix_socket = 0;
- }
- mMysql = mysql_real_connect(mysqlInit, host.c_str(), user.c_str(),
- password.c_str(), database.c_str(), port, unix_socket, 0);
- if (mMysql)
- {
- DETAIL_LOG( "Connected to MySQL database at %s",
- host.c_str());
- sLog.outString( "MySQL client library: %s", mysql_get_client_info());
- sLog.outString( "MySQL server ver: %s ", mysql_get_server_info( mMysql));
- /*----------SET AUTOCOMMIT ON---------*/
- // It seems mysql 5.0.x have enabled this feature
- // by default. In crash case you can lose data!!!
- // So better to turn this off
- // ---
- // This is wrong since NeoCore use transactions,
- // autocommit is turned of during it.
- // Setting it to on makes atomic updates work
- if (!mysql_autocommit(mMysql, 1))
- DETAIL_LOG("AUTOCOMMIT SUCCESSFULLY SET TO 1");
- else
- DETAIL_LOG("AUTOCOMMIT NOT SET TO 1");
- /*-------------------------------------*/
- // set connection properties to UTF8 to properly handle locales for different
- // server configs - core sends data in UTF8, so MySQL must expect UTF8 too
- PExecute("SET NAMES `utf8`");
- PExecute("SET CHARACTER SET `utf8`");
- return true;
- }
- else
- {
- sLog.outError( "Could not connect to MySQL database at %s: %s\n",
- host.c_str(),mysql_error(mysqlInit));
- mysql_close(mysqlInit);
- return false;
- }
- }
- bool DatabaseMysql::_Query(const char *sql, MYSQL_RES **pResult, MYSQL_FIELD **pFields, uint64* pRowCount, uint32* pFieldCount)
- {
- if (!mMysql)
- return 0;
- {
- // guarded block for thread-safe mySQL request
- ACE_Guard<ACE_Thread_Mutex> query_connection_guard(mMutex);
- uint32 _s = getMSTime();
- if(mysql_query(mMysql, sql))
- {
- sLog.outErrorDb( "SQL: %s", sql );
- sLog.outErrorDb("query ERROR: %s", mysql_error(mMysql));
- return false;
- }
- else
- {
- DEBUG_FILTER_LOG(LOG_FILTER_SQL_TEXT, "[%u ms] SQL: %s", getMSTimeDiff(_s,getMSTime()), sql );
- }
- *pResult = mysql_store_result(mMysql);
- *pRowCount = mysql_affected_rows(mMysql);
- *pFieldCount = mysql_field_count(mMysql);
- // end guarded block
- }
- if (!*pResult )
- return false;
- if (!*pRowCount)
- {
- mysql_free_result(*pResult);
- return false;
- }
- *pFields = mysql_fetch_fields(*pResult);
- return true;
- }
- QueryResult* DatabaseMysql::Query(const char *sql)
- {
- MYSQL_RES *result = NULL;
- MYSQL_FIELD *fields = NULL;
- uint64 rowCount = 0;
- uint32 fieldCount = 0;
- if(!_Query(sql,&result,&fields,&rowCount,&fieldCount))
- return NULL;
- QueryResultMysql *queryResult = new QueryResultMysql(result, fields, rowCount, fieldCount);
- queryResult->NextRow();
- return queryResult;
- }
- QueryNamedResult* DatabaseMysql::QueryNamed(const char *sql)
- {
- MYSQL_RES *result = NULL;
- MYSQL_FIELD *fields = NULL;
- uint64 rowCount = 0;
- uint32 fieldCount = 0;
- if(!_Query(sql,&result,&fields,&rowCount,&fieldCount))
- return NULL;
- QueryFieldNames names(fieldCount);
- for (uint32 i = 0; i < fieldCount; i++)
- names[i] = fields[i].name;
- QueryResultMysql *queryResult = new QueryResultMysql(result, fields, rowCount, fieldCount);
- queryResult->NextRow();
- return new QueryNamedResult(queryResult,names);
- }
- bool DatabaseMysql::Execute(const char *sql)
- {
- if (!mMysql)
- return false;
- // don't use queued execution if it has not been initialized
- if (!m_threadBody) return DirectExecute(sql);
- tranThread = ACE_Based::Thread::current(); // owner of this transaction
- TransactionQueues::iterator i = m_tranQueues.find(tranThread);
- if (i != m_tranQueues.end() && i->second != NULL)
- { // Statement for transaction
- i->second->DelayExecute(sql);
- }
- else
- {
- // Simple sql statement
- m_threadBody->Delay(new SqlStatement(sql));
- }
- return true;
- }
- bool DatabaseMysql::DirectExecute(const char* sql)
- {
- if (!mMysql)
- return false;
- {
- // guarded block for thread-safe mySQL request
- ACE_Guard<ACE_Thread_Mutex> query_connection_guard(mMutex);
- uint32 _s = getMSTime();
- if(mysql_query(mMysql, sql))
- {
- sLog.outErrorDb("SQL: %s", sql);
- sLog.outErrorDb("SQL ERROR: %s", mysql_error(mMysql));
- return false;
- }
- else
- {
- DEBUG_FILTER_LOG(LOG_FILTER_SQL_TEXT, "[%u ms] SQL: %s", getMSTimeDiff(_s,getMSTime()), sql );
- }
- // end guarded block
- }
- return true;
- }
- bool DatabaseMysql::_TransactionCmd(const char *sql)
- {
- if (mysql_query(mMysql, sql))
- {
- sLog.outError("SQL: %s", sql);
- sLog.outError("SQL ERROR: %s", mysql_error(mMysql));
- return false;
- }
- else
- {
- DEBUG_FILTER_LOG(LOG_FILTER_SQL_TEXT, "SQL: %s", sql);
- }
- return true;
- }
- bool DatabaseMysql::BeginTransaction()
- {
- if (!mMysql)
- return false;
- // don't use queued execution if it has not been initialized
- if (!m_threadBody)
- {
- if (tranThread == ACE_Based::Thread::current())
- return false; // huh? this thread already started transaction
- mMutex.acquire();
- if (!_TransactionCmd("START TRANSACTION"))
- {
- mMutex.release(); // can't start transaction
- return false;
- }
- return true; // transaction started
- }
- tranThread = ACE_Based::Thread::current(); // owner of this transaction
- TransactionQueues::iterator i = m_tranQueues.find(tranThread);
- if (i != m_tranQueues.end() && i->second != NULL)
- // If for thread exists queue and also contains transaction
- // delete that transaction (not allow trans in trans)
- delete i->second;
- m_tranQueues[tranThread] = new SqlTransaction();
- return true;
- }
- bool DatabaseMysql::CommitTransaction()
- {
- if (!mMysql)
- return false;
- // don't use queued execution if it has not been initialized
- if (!m_threadBody)
- {
- if (tranThread != ACE_Based::Thread::current())
- return false;
- bool _res = _TransactionCmd("COMMIT");
- tranThread = NULL;
- mMutex.release();
- return _res;
- }
- tranThread = ACE_Based::Thread::current();
- TransactionQueues::iterator i = m_tranQueues.find(tranThread);
- if (i != m_tranQueues.end() && i->second != NULL)
- {
- m_threadBody->Delay(i->second);
- i->second = NULL;
- return true;
- }
- else
- return false;
- }
- bool DatabaseMysql::RollbackTransaction()
- {
- if (!mMysql)
- return false;
- // don't use queued execution if it has not been initialized
- if (!m_threadBody)
- {
- if (tranThread != ACE_Based::Thread::current())
- return false;
- bool _res = _TransactionCmd("ROLLBACK");
- tranThread = NULL;
- mMutex.release();
- return _res;
- }
- tranThread = ACE_Based::Thread::current();
- TransactionQueues::iterator i = m_tranQueues.find(tranThread);
- if (i != m_tranQueues.end() && i->second != NULL)
- {
- delete i->second;
- i->second = NULL;
- }
- return true;
- }
- unsigned long DatabaseMysql::escape_string(char *to, const char *from, unsigned long length)
- {
- if (!mMysql || !to || !from || !length)
- return 0;
- return(mysql_real_escape_string(mMysql, to, from, length));
- }
- void DatabaseMysql::InitDelayThread()
- {
- assert(!m_delayThread);
- //New delay thread for delay execute
- m_threadBody = new MySQLDelayThread(this); // will deleted at m_delayThread delete
- m_delayThread = new ACE_Based::Thread(m_threadBody);
- }
- void DatabaseMysql::HaltDelayThread()
- {
- if (!m_threadBody || !m_delayThread) return;
- m_threadBody->Stop(); //Stop event
- m_delayThread->wait(); //Wait for flush to DB
- delete m_delayThread; //This also deletes m_threadBody
- m_delayThread = NULL;
- m_threadBody = NULL;
- }
- #endif