From 1d2aafd39bcb79a67357d198ce9b2345642fdd39 Mon Sep 17 00:00:00 2001 From: StormBytePP Date: Wed, 19 Aug 2015 19:02:10 +0200 Subject: Core/Build: Merge common library and move database out of shared --- src/server/database/Database/Transaction.h | 77 ++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) create mode 100644 src/server/database/Database/Transaction.h (limited to 'src/server/database/Database/Transaction.h') 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 + * + * 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 . + */ + +#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 + friend class DatabaseWorkerPool; + + public: + Transaction() : _cleanedUp(false) { } + ~Transaction() { Cleanup(); } + + void Append(PreparedStatement* statement); + void Append(const char* sql); + template + void PAppend(Format&& sql, Args&&... args) + { + Append(Trinity::StringFormat(std::forward(sql), std::forward(args)...).c_str()); + } + + size_t GetSize() const { return m_queries.size(); } + + protected: + void Cleanup(); + std::list m_queries; + + private: + bool _cleanedUp; + +}; +typedef std::shared_ptr SQLTransaction; + +/*! Low level class*/ +class TransactionTask : public SQLOperation +{ + template 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 -- cgit v1.2.3