First Commitv0.10.1

For Azeroth!

1 files changed, 137 insertions, 0 deletions

diff --git a/src/server/shared/Database/MySQLConnection.h b/src/server/shared/Database/MySQLConnection.h
new file mode 100644
index 0000000000..256898ccff
--- /dev/null
+++ b/src/server/shared/Database/MySQLConnection.h
@@ -0,0 +1,137 @@
+/*
+ * Copyright (C) 
+ *
+ * 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 <ace/Activation_Queue.h>
+
+#include "DatabaseWorkerPool.h"
+#include "Transaction.h"
+#include "Util.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
+{
+    MySQLConnectionInfo() { }
+    MySQLConnectionInfo(const std::string& 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(ACE_Activation_Queue* queue, MySQLConnectionInfo& connInfo);  //! Constructor for asynchronous connections.
+        virtual ~MySQLConnection();
+
+        virtual bool Open();
+        void Close();
+
+    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();
+        bool 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.tryacquire() != -1;
+        }
+
+        void Unlock()
+        {
+            /// Called by parent databasepool. Will let other threads access this connection
+            m_Mutex.release();
+        }
+
+        MYSQL* GetHandle()  { return m_Mysql; }
+        MySQLPreparedStatement* GetPreparedStatement(uint32 index);
+        void PrepareStatement(uint32 index, const char* sql, ConnectionFlags flags);
+
+        bool PrepareStatements();
+        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:
+        ACE_Activation_Queue* 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)
+        ACE_Thread_Mutex      m_Mutex;
+};
+
+#endif