mirror of
https://github.com/araxiaonline/TrinityCore2.git
synced 2026-06-16 04:49:38 -04:00
994186f267
- Make SQL Transactions actual objects used in code. (Thanks to Derex for the idea) * Uncommitted transactions will be automatically rolled back and cleaned up using ACE_Refcounted_Auto_Ptr, so no need to call Rollback() in the code. * Prevents recursive transactions and makes developers aware of transactions going on. * Gets rid of unneccesary overhead iterating over a concurrent map. - Some cleanups in affected code, including better usage of transaction control in AH / mail related code to prevent data loss. *** Experimental, use at own risk, recommended to backup your DBs. *** --HG-- branch : trunk
123 lines
4.6 KiB
C++
123 lines
4.6 KiB
C++
/*
|
|
* Copyright (C) 2008-2010 Trinity <http://www.trinitycore.org/>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#ifndef _DATABASEWORKERPOOL_H
|
|
#define _DATABASEWORKERPOOL_H
|
|
|
|
#include "Common.h"
|
|
|
|
#include <ace/Atomic_Op_T.h>
|
|
#include <ace/Thread_Mutex.h>
|
|
|
|
#include "SQLOperation.h"
|
|
#include "QueryResult.h"
|
|
#include "Callback.h"
|
|
#include "MySQLConnection.h"
|
|
#include "Transaction.h"
|
|
|
|
enum MySQLThreadBundle
|
|
{
|
|
MYSQL_BUNDLE_NONE = 0x00, //- Each task will run their own MySQL connection
|
|
MYSQL_BUNDLE_CLI = 0x01, //- Commandline interface thread
|
|
MYSQL_BUNDLE_RA = 0x02, //- Remote admin thread
|
|
MYSQL_BUNDLE_RAR = 0x04, //- Reactor runnable thread
|
|
MYSQL_BUNDLE_WORLD = 0x08, //- WorldRunnable
|
|
MYSQL_BUNDLE_ALL = MYSQL_BUNDLE_CLI | MYSQL_BUNDLE_RA | MYSQL_BUNDLE_RAR | MYSQL_BUNDLE_WORLD,
|
|
};
|
|
|
|
class DatabaseWorkerPoolEnd : public SQLOperation
|
|
{
|
|
public:
|
|
DatabaseWorkerPoolEnd(ACE_Condition_Thread_Mutex &mtx) : shutdown_Mtx(mtx) {}
|
|
~DatabaseWorkerPoolEnd() { shutdown_Mtx.broadcast(); } //! Tells the Worker Pool to enqueue the next DatabaseWorkerPoolEnd operation.
|
|
|
|
int call() { return -1; }
|
|
bool Execute() { return false;} //! Not called - fool the compiler
|
|
|
|
ACE_Condition_Thread_Mutex &shutdown_Mtx;
|
|
};
|
|
|
|
class DatabaseWorkerPool
|
|
{
|
|
public:
|
|
DatabaseWorkerPool();
|
|
~DatabaseWorkerPool();
|
|
|
|
bool Open(const std::string& infoString, uint8 num_threads);
|
|
void Close();
|
|
|
|
void Init_MySQL_Connection();
|
|
void End_MySQL_Connection();
|
|
|
|
void Execute(const char* sql);
|
|
void PExecute(const char* sql, ...);
|
|
void DirectExecute(const char* sql);
|
|
void DirectPExecute(const char* sql, ...);
|
|
QueryResult_AutoPtr Query(const char* sql);
|
|
QueryResult_AutoPtr PQuery(const char* sql, ...);
|
|
ACE_Future<QueryResult_AutoPtr> AsyncQuery(const char* sql);
|
|
ACE_Future<QueryResult_AutoPtr> AsyncPQuery(const char* sql, ...);
|
|
QueryResultHolderFuture DelayQueryHolder(SQLQueryHolder* holder);
|
|
|
|
SQLTransaction BeginTransaction();
|
|
void CommitTransaction(SQLTransaction transaction);
|
|
|
|
void escape_string(std::string& str)
|
|
{
|
|
if (str.empty())
|
|
return;
|
|
|
|
char* buf = new char[str.size()*2+1];
|
|
escape_string(buf,str.c_str(),str.size());
|
|
str = buf;
|
|
delete[] buf;
|
|
}
|
|
|
|
unsigned long escape_string(char *to, const char *from, unsigned long length)
|
|
{
|
|
if (!to || !from || !length)
|
|
return 0;
|
|
return (mysql_real_escape_string(GetConnection()->GetHandle(), to, from, length));
|
|
}
|
|
|
|
private:
|
|
void Enqueue(SQLOperation* op)
|
|
{
|
|
m_queue->enqueue(op);
|
|
}
|
|
|
|
MySQLConnection* GetConnection();
|
|
|
|
private:
|
|
typedef UNORDERED_MAP<ACE_Based::Thread*, MySQLConnection*> ConnectionMap;
|
|
typedef UNORDERED_MAP<ACE_Based::Thread*, TransactionTask*> TransactionQueues;
|
|
typedef ACE_Atomic_Op<ACE_SYNCH_MUTEX, uint32> AtomicUInt;
|
|
|
|
private:
|
|
ACE_Activation_Queue* m_queue; //! Queue shared by async worker threads.
|
|
ACE_Thread_Mutex m_queue_mtx; //! For thread safe enqueues of delayed statements.
|
|
std::vector<MySQLConnection*> m_async_connections;
|
|
ConnectionMap m_sync_connections; //! Holds a mysql connection+thread per mapUpdate thread and unbundled runnnables.
|
|
ACE_Thread_Mutex m_connectionMap_mtx; //! For thread safe access to the synchroneous connection map
|
|
MySQLConnection* m_bundle_conn; //! Bundled connection (see Database.ThreadBundleMask config)
|
|
AtomicUInt m_connections; //! Counter of MySQL connections;
|
|
std::string m_infoString; //! Infostring that is passed on to child connections.
|
|
};
|
|
|
|
#endif
|