From 1f66d719f2cbbcb144b5080c89dd73fcae261798 Mon Sep 17 00:00:00 2001
From: StormBytePP <stormbyte@gmail.com>
Date: Sat, 15 Aug 2015 02:19:10 +0200
Subject: Core/BuildSystem: Merge collision, debugging, threading, utilities
 and configuration into "common" which does not depend on shared anymore and
 moved database out of shared library These changes enables to build tools
 only without even having MySQL installed

---
 src/server/database/Database/MySQLConnection.h | 139 +++++++++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 src/server/database/Database/MySQLConnection.h

(limited to 'src/server/database/Database/MySQLConnection.h')

diff --git a/src/server/database/Database/MySQLConnection.h b/src/server/database/Database/MySQLConnection.h
new file mode 100644
index 00000000000..78d8d2fb5dd
--- /dev/null
+++ b/src/server/database/Database/MySQLConnection.h
@@ -0,0 +1,139 @@
+/*
+ * 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/>.
+ */
+
+#include "DatabaseWorkerPool.h"
+#include "Transaction.h"
+#include "Util.h"
+#include "ProducerConsumerQueue.h"
+
+#ifndef _MYSQLCONNECTION_H
+#define _MYSQLCONNECTION_H
+
+class DatabaseWorker;
+class PreparedStatement;
+class MySQLPreparedStatement;
+class PingOperation;
+
+enum ConnectionFlags
+{
+    CONNECTION_ASYNC = 0x1,
+    CONNECTION_SYNCH = 0x2,
+    CONNECTION_BOTH = CONNECTION_ASYNC | CONNECTION_SYNCH
+};
+
+struct MySQLConnectionInfo
+{
+    explicit MySQLConnectionInfo(std::string const& infoString)
+    {
+        Tokenizer tokens(infoString, ';');
+
+        if (tokens.size() != 5)
+            return;
+
+        uint8 i = 0;
+
+        host.assign(tokens[i++]);
+        port_or_socket.assign(tokens[i++]);
+        user.assign(tokens[i++]);
+        password.assign(tokens[i++]);
+        database.assign(tokens[i++]);
+    }
+
+    std::string user;
+    std::string password;
+    std::string database;
+    std::string host;
+    std::string port_or_socket;
+};
+
+typedef std::map<uint32 /*index*/, std::pair<std::string /*query*/, ConnectionFlags /*sync/async*/> > PreparedStatementMap;
+
+class MySQLConnection
+{
+    template <class T> friend class DatabaseWorkerPool;
+    friend class PingOperation;
+
+    public:
+        MySQLConnection(MySQLConnectionInfo& connInfo);                               //! Constructor for synchronous connections.
+        MySQLConnection(ProducerConsumerQueue<SQLOperation*>* queue, MySQLConnectionInfo& connInfo);  //! Constructor for asynchronous connections.
+        virtual ~MySQLConnection();
+
+        virtual uint32 Open();
+        void Close();
+
+        bool PrepareStatements();
+
+    public:
+        bool Execute(const char* sql);
+        bool Execute(PreparedStatement* stmt);
+        ResultSet* Query(const char* sql);
+        PreparedResultSet* Query(PreparedStatement* stmt);
+        bool _Query(const char *sql, MYSQL_RES **pResult, MYSQL_FIELD **pFields, uint64* pRowCount, uint32* pFieldCount);
+        bool _Query(PreparedStatement* stmt, MYSQL_RES **pResult, uint64* pRowCount, uint32* pFieldCount);
+
+        void BeginTransaction();
+        void RollbackTransaction();
+        void CommitTransaction();
+        int ExecuteTransaction(SQLTransaction& transaction);
+
+        operator bool () const { return m_Mysql != NULL; }
+        void Ping() { mysql_ping(m_Mysql); }
+
+        uint32 GetLastError() { return mysql_errno(m_Mysql); }
+
+    protected:
+        bool LockIfReady()
+        {
+            /// Tries to acquire lock. If lock is acquired by another thread
+            /// the calling parent will just try another connection
+            return m_Mutex.try_lock();
+        }
+
+        void Unlock()
+        {
+            /// Called by parent databasepool. Will let other threads access this connection
+            m_Mutex.unlock();
+        }
+
+        MYSQL* GetHandle()  { return m_Mysql; }
+        MySQLPreparedStatement* GetPreparedStatement(uint32 index);
+        void PrepareStatement(uint32 index, const char* sql, ConnectionFlags flags);
+
+        virtual void DoPrepareStatements() = 0;
+
+    protected:
+        std::vector<MySQLPreparedStatement*> m_stmts;         //! PreparedStatements storage
+        PreparedStatementMap                 m_queries;       //! Query storage
+        bool                                 m_reconnecting;  //! Are we reconnecting?
+        bool                                 m_prepareError;  //! Was there any error while preparing statements?
+
+    private:
+        bool _HandleMySQLErrno(uint32 errNo);
+
+    private:
+        ProducerConsumerQueue<SQLOperation*>* m_queue;      //! Queue shared with other asynchronous connections.
+        DatabaseWorker*       m_worker;                     //! Core worker task.
+        MYSQL *               m_Mysql;                      //! MySQL Handle.
+        MySQLConnectionInfo&  m_connectionInfo;             //! Connection info (used for logging)
+        ConnectionFlags       m_connectionFlags;            //! Connection flags (for preparing relevant statements)
+        std::mutex            m_Mutex;
+
+        MySQLConnection(MySQLConnection const& right) = delete;
+        MySQLConnection& operator=(MySQLConnection const& right) = delete;
+};
+
+#endif
-- 
cgit v1.2.3