diff options
author | StormBytePP <stormbyte@gmail.com> | 2015-08-19 19:02:10 +0200 |
---|---|---|
committer | StormBytePP <stormbyte@gmail.com> | 2015-08-21 17:52:42 +0200 |
commit | 1d2aafd39bcb79a67357d198ce9b2345642fdd39 (patch) | |
tree | c32cf1c3717625c60da59c82ba5a4fca2530119a /src/server/database/Database/Transaction.h | |
parent | 172293acee1607727ebd8070ab3e1390590d02a8 (diff) |
Core/Build: Merge common library and move database out of shared
Diffstat (limited to 'src/server/database/Database/Transaction.h')
-rw-r--r-- | src/server/database/Database/Transaction.h | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/src/server/database/Database/Transaction.h b/src/server/database/Database/Transaction.h new file mode 100644 index 00000000000..4fbbe1ed45b --- /dev/null +++ b/src/server/database/Database/Transaction.h @@ -0,0 +1,77 @@ +/* + * Copyright (C) 2008-2015 TrinityCore <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, see <http://www.gnu.org/licenses/>. + */ + +#ifndef _TRANSACTION_H +#define _TRANSACTION_H + +#include "SQLOperation.h" +#include "StringFormat.h" + +//- Forward declare (don't include header to prevent circular includes) +class PreparedStatement; + +/*! Transactions, high level class. */ +class Transaction +{ + friend class TransactionTask; + friend class MySQLConnection; + + template <typename T> + friend class DatabaseWorkerPool; + + public: + Transaction() : _cleanedUp(false) { } + ~Transaction() { Cleanup(); } + + void Append(PreparedStatement* statement); + void Append(const char* sql); + template<typename Format, typename... Args> + void PAppend(Format&& sql, Args&&... args) + { + Append(Trinity::StringFormat(std::forward<Format>(sql), std::forward<Args>(args)...).c_str()); + } + + size_t GetSize() const { return m_queries.size(); } + + protected: + void Cleanup(); + std::list<SQLElementData> m_queries; + + private: + bool _cleanedUp; + +}; +typedef std::shared_ptr<Transaction> SQLTransaction; + +/*! Low level class*/ +class TransactionTask : public SQLOperation +{ + template <class T> friend class DatabaseWorkerPool; + friend class DatabaseWorker; + + public: + TransactionTask(SQLTransaction trans) : m_trans(trans) { } + ~TransactionTask() { } + + protected: + bool Execute() override; + + SQLTransaction m_trans; + static std::mutex _deadlockLock; +}; + +#endif |