aboutsummaryrefslogtreecommitdiff
path: root/src/server/database/Database/QueryResult.h
diff options
context:
space:
mode:
authorStormBytePP <stormbyte@gmail.com>2015-08-19 19:02:10 +0200
committerStormBytePP <stormbyte@gmail.com>2015-08-21 17:52:42 +0200
commit1d2aafd39bcb79a67357d198ce9b2345642fdd39 (patch)
treec32cf1c3717625c60da59c82ba5a4fca2530119a /src/server/database/Database/QueryResult.h
parent172293acee1607727ebd8070ab3e1390590d02a8 (diff)
Core/Build: Merge common library and move database out of shared
Diffstat (limited to 'src/server/database/Database/QueryResult.h')
-rw-r--r--src/server/database/Database/QueryResult.h111
1 files changed, 111 insertions, 0 deletions
diff --git a/src/server/database/Database/QueryResult.h b/src/server/database/Database/QueryResult.h
new file mode 100644
index 00000000000..a61fb6331c1
--- /dev/null
+++ b/src/server/database/Database/QueryResult.h
@@ -0,0 +1,111 @@
+/*
+ * Copyright (C) 2008-2015 TrinityCore <http://www.trinitycore.org/>
+ * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
+ *
+ * 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 QUERYRESULT_H
+#define QUERYRESULT_H
+
+#include <memory>
+#include "Field.h"
+
+#ifdef _WIN32
+ #include <winsock2.h>
+#endif
+#include <mysql.h>
+
+class ResultSet
+{
+ public:
+ ResultSet(MYSQL_RES* result, MYSQL_FIELD* fields, uint64 rowCount, uint32 fieldCount);
+ ~ResultSet();
+
+ bool NextRow();
+ uint64 GetRowCount() const { return _rowCount; }
+ uint32 GetFieldCount() const { return _fieldCount; }
+
+ Field* Fetch() const { return _currentRow; }
+ const Field & operator [] (uint32 index) const
+ {
+ ASSERT(index < _fieldCount);
+ return _currentRow[index];
+ }
+
+ protected:
+ uint64 _rowCount;
+ Field* _currentRow;
+ uint32 _fieldCount;
+
+ private:
+ void CleanUp();
+ MYSQL_RES* _result;
+ MYSQL_FIELD* _fields;
+
+ ResultSet(ResultSet const& right) = delete;
+ ResultSet& operator=(ResultSet const& right) = delete;
+};
+
+typedef std::shared_ptr<ResultSet> QueryResult;
+
+class PreparedResultSet
+{
+ public:
+ PreparedResultSet(MYSQL_STMT* stmt, MYSQL_RES* result, uint64 rowCount, uint32 fieldCount);
+ ~PreparedResultSet();
+
+ bool NextRow();
+ uint64 GetRowCount() const { return m_rowCount; }
+ uint32 GetFieldCount() const { return m_fieldCount; }
+
+ Field* Fetch() const
+ {
+ ASSERT(m_rowPosition < m_rowCount);
+ return m_rows[uint32(m_rowPosition)];
+ }
+
+ const Field & operator [] (uint32 index) const
+ {
+ ASSERT(m_rowPosition < m_rowCount);
+ ASSERT(index < m_fieldCount);
+ return m_rows[uint32(m_rowPosition)][index];
+ }
+
+ protected:
+ std::vector<Field*> m_rows;
+ uint64 m_rowCount;
+ uint64 m_rowPosition;
+ uint32 m_fieldCount;
+
+ private:
+ MYSQL_BIND* m_rBind;
+ MYSQL_STMT* m_stmt;
+ MYSQL_RES* m_res;
+
+ my_bool* m_isNull;
+ unsigned long* m_length;
+
+ void FreeBindBuffer();
+ void CleanUp();
+ bool _NextRow();
+
+ PreparedResultSet(PreparedResultSet const& right) = delete;
+ PreparedResultSet& operator=(PreparedResultSet const& right) = delete;
+};
+
+typedef std::shared_ptr<PreparedResultSet> PreparedQueryResult;
+
+#endif
+