mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-20 01:15:35 +01: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
97 lines
2.6 KiB
C++
97 lines
2.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 _SQLOPERATION_H
|
|
#define _SQLOPERATION_H
|
|
|
|
#include <ace/Method_Request.h>
|
|
#include <ace/Activation_Queue.h>
|
|
|
|
#include "Common.h"
|
|
#include "Callback.h"
|
|
|
|
class MySQLConnection;
|
|
|
|
class SQLOperation : public ACE_Method_Request
|
|
{
|
|
public:
|
|
SQLOperation(){};
|
|
virtual int call()
|
|
{
|
|
Execute();
|
|
return 0;
|
|
}
|
|
virtual bool Execute() = 0;
|
|
virtual void SetConnection(MySQLConnection* con) { m_conn = con; }
|
|
|
|
MySQLConnection* m_conn;
|
|
};
|
|
|
|
typedef ACE_Future<QueryResult_AutoPtr> QueryResultFuture;
|
|
/*! Raw, ad-hoc query. */
|
|
class BasicStatementTask : public SQLOperation
|
|
{
|
|
public:
|
|
BasicStatementTask(const char* sql);
|
|
BasicStatementTask(const char* sql, QueryResultFuture result);
|
|
~BasicStatementTask();
|
|
|
|
bool Execute();
|
|
|
|
private:
|
|
const char* m_sql; //- Raw query to be executed
|
|
bool m_has_result;
|
|
QueryResultFuture m_result;
|
|
};
|
|
|
|
|
|
|
|
class SQLQueryHolder
|
|
{
|
|
friend class SQLQueryHolderTask;
|
|
private:
|
|
typedef std::pair<const char*, QueryResult_AutoPtr> SQLResultPair;
|
|
std::vector<SQLResultPair> m_queries;
|
|
public:
|
|
SQLQueryHolder() {}
|
|
~SQLQueryHolder();
|
|
bool SetQuery(size_t index, const char *sql);
|
|
bool SetPQuery(size_t index, const char *format, ...) ATTR_PRINTF(3,4);
|
|
void SetSize(size_t size);
|
|
QueryResult_AutoPtr GetResult(size_t index);
|
|
void SetResult(size_t index, QueryResult_AutoPtr result);
|
|
};
|
|
|
|
typedef ACE_Future<SQLQueryHolder*> QueryResultHolderFuture;
|
|
|
|
class SQLQueryHolderTask : public SQLOperation
|
|
{
|
|
private:
|
|
SQLQueryHolder * m_holder;
|
|
QueryResultHolderFuture m_result;
|
|
|
|
public:
|
|
SQLQueryHolderTask(SQLQueryHolder *holder, QueryResultHolderFuture res)
|
|
: m_holder(holder), m_result(res){};
|
|
bool Execute();
|
|
|
|
};
|
|
|
|
|
|
#endif
|