aboutsummaryrefslogtreecommitdiff
path: root/src/server/shared/Database/PreparedStatement.cpp
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/shared/Database/PreparedStatement.cpp
parent172293acee1607727ebd8070ab3e1390590d02a8 (diff)
Core/Build: Merge common library and move database out of shared
Diffstat (limited to 'src/server/shared/Database/PreparedStatement.cpp')
-rw-r--r--src/server/shared/Database/PreparedStatement.cpp500
1 files changed, 0 insertions, 500 deletions
diff --git a/src/server/shared/Database/PreparedStatement.cpp b/src/server/shared/Database/PreparedStatement.cpp
deleted file mode 100644
index 2c9ce136e6e..00000000000
--- a/src/server/shared/Database/PreparedStatement.cpp
+++ /dev/null
@@ -1,500 +0,0 @@
-/*
- * 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 "PreparedStatement.h"
-#include "MySQLConnection.h"
-#include "Log.h"
-
-PreparedStatement::PreparedStatement(uint32 index) :
-m_stmt(NULL),
-m_index(index) { }
-
-PreparedStatement::~PreparedStatement() { }
-
-void PreparedStatement::BindParameters()
-{
- ASSERT (m_stmt);
-
- uint8 i = 0;
- for (; i < statement_data.size(); i++)
- {
- switch (statement_data[i].type)
- {
- case TYPE_BOOL:
- m_stmt->setBool(i, statement_data[i].data.boolean);
- break;
- case TYPE_UI8:
- m_stmt->setUInt8(i, statement_data[i].data.ui8);
- break;
- case TYPE_UI16:
- m_stmt->setUInt16(i, statement_data[i].data.ui16);
- break;
- case TYPE_UI32:
- m_stmt->setUInt32(i, statement_data[i].data.ui32);
- break;
- case TYPE_I8:
- m_stmt->setInt8(i, statement_data[i].data.i8);
- break;
- case TYPE_I16:
- m_stmt->setInt16(i, statement_data[i].data.i16);
- break;
- case TYPE_I32:
- m_stmt->setInt32(i, statement_data[i].data.i32);
- break;
- case TYPE_UI64:
- m_stmt->setUInt64(i, statement_data[i].data.ui64);
- break;
- case TYPE_I64:
- m_stmt->setInt64(i, statement_data[i].data.i64);
- break;
- case TYPE_FLOAT:
- m_stmt->setFloat(i, statement_data[i].data.f);
- break;
- case TYPE_DOUBLE:
- m_stmt->setDouble(i, statement_data[i].data.d);
- break;
- case TYPE_STRING:
- m_stmt->setBinary(i, statement_data[i].binary, true);
- break;
- case TYPE_BINARY:
- m_stmt->setBinary(i, statement_data[i].binary, false);
- break;
- case TYPE_NULL:
- m_stmt->setNull(i);
- break;
- }
- }
- #ifdef _DEBUG
- if (i < m_stmt->m_paramCount)
- TC_LOG_WARN("sql.sql", "[WARNING]: BindParameters() for statement %u did not bind all allocated parameters", m_index);
- #endif
-}
-
-//- Bind to buffer
-void PreparedStatement::setBool(const uint8 index, const bool value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.boolean = value;
- statement_data[index].type = TYPE_BOOL;
-}
-
-void PreparedStatement::setUInt8(const uint8 index, const uint8 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.ui8 = value;
- statement_data[index].type = TYPE_UI8;
-}
-
-void PreparedStatement::setUInt16(const uint8 index, const uint16 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.ui16 = value;
- statement_data[index].type = TYPE_UI16;
-}
-
-void PreparedStatement::setUInt32(const uint8 index, const uint32 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.ui32 = value;
- statement_data[index].type = TYPE_UI32;
-}
-
-void PreparedStatement::setUInt64(const uint8 index, const uint64 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.ui64 = value;
- statement_data[index].type = TYPE_UI64;
-}
-
-void PreparedStatement::setInt8(const uint8 index, const int8 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.i8 = value;
- statement_data[index].type = TYPE_I8;
-}
-
-void PreparedStatement::setInt16(const uint8 index, const int16 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.i16 = value;
- statement_data[index].type = TYPE_I16;
-}
-
-void PreparedStatement::setInt32(const uint8 index, const int32 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.i32 = value;
- statement_data[index].type = TYPE_I32;
-}
-
-void PreparedStatement::setInt64(const uint8 index, const int64 value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.i64 = value;
- statement_data[index].type = TYPE_I64;
-}
-
-void PreparedStatement::setFloat(const uint8 index, const float value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.f = value;
- statement_data[index].type = TYPE_FLOAT;
-}
-
-void PreparedStatement::setDouble(const uint8 index, const double value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].data.d = value;
- statement_data[index].type = TYPE_DOUBLE;
-}
-
-void PreparedStatement::setString(const uint8 index, const std::string& value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].binary.resize(value.length() + 1);
- memcpy(statement_data[index].binary.data(), value.c_str(), value.length() + 1);
- statement_data[index].type = TYPE_STRING;
-}
-
-void PreparedStatement::setBinary(const uint8 index, const std::vector<uint8>& value)
-{
- if (index >= statement_data.size())
- statement_data.resize(index + 1);
-
- statement_data[index].binary = value;
- statement_data[index].type = TYPE_BINARY;
-}
-
-void PreparedStatement::setNull(const uint8 index)
-{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
-
- statement_data[index].type = TYPE_NULL;
-}
-
-MySQLPreparedStatement::MySQLPreparedStatement(MYSQL_STMT* stmt) :
-m_stmt(NULL),
-m_Mstmt(stmt),
-m_bind(NULL)
-{
- /// Initialize variable parameters
- m_paramCount = mysql_stmt_param_count(stmt);
- m_paramsSet.assign(m_paramCount, false);
- m_bind = new MYSQL_BIND[m_paramCount];
- memset(m_bind, 0, sizeof(MYSQL_BIND)*m_paramCount);
-
- /// "If set to 1, causes mysql_stmt_store_result() to update the metadata MYSQL_FIELD->max_length value."
- my_bool bool_tmp = 1;
- mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, &bool_tmp);
-}
-
-MySQLPreparedStatement::~MySQLPreparedStatement()
-{
- ClearParameters();
- if (m_Mstmt->bind_result_done)
- {
- delete[] m_Mstmt->bind->length;
- delete[] m_Mstmt->bind->is_null;
- }
- mysql_stmt_close(m_Mstmt);
- delete[] m_bind;
-}
-
-void MySQLPreparedStatement::ClearParameters()
-{
- for (uint32 i=0; i < m_paramCount; ++i)
- {
- delete m_bind[i].length;
- m_bind[i].length = NULL;
- delete[] (char*) m_bind[i].buffer;
- m_bind[i].buffer = NULL;
- m_paramsSet[i] = false;
- }
-}
-
-static bool ParamenterIndexAssertFail(uint32 stmtIndex, uint8 index, uint32 paramCount)
-{
- TC_LOG_ERROR("sql.driver", "Attempted to bind parameter %u%s on a PreparedStatement %u (statement has only %u parameters)", uint32(index) + 1, (index == 1 ? "st" : (index == 2 ? "nd" : (index == 3 ? "rd" : "nd"))), stmtIndex, paramCount);
- return false;
-}
-
-//- Bind on mysql level
-bool MySQLPreparedStatement::CheckValidIndex(uint8 index)
-{
- ASSERT(index < m_paramCount || ParamenterIndexAssertFail(m_stmt->m_index, index, m_paramCount));
-
- if (m_paramsSet[index])
- TC_LOG_WARN("sql.sql", "[WARNING] Prepared Statement (id: %u) trying to bind value on already bound index (%u).", m_stmt->m_index, index);
- return true;
-}
-
-void MySQLPreparedStatement::setBool(const uint8 index, const bool value)
-{
- setUInt8(index, value ? 1 : 0);
-}
-
-void MySQLPreparedStatement::setUInt8(const uint8 index, const uint8 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_TINY, &value, sizeof(uint8), true);
-}
-
-void MySQLPreparedStatement::setUInt16(const uint8 index, const uint16 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_SHORT, &value, sizeof(uint16), true);
-}
-
-void MySQLPreparedStatement::setUInt32(const uint8 index, const uint32 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_LONG, &value, sizeof(uint32), true);
-}
-
-void MySQLPreparedStatement::setUInt64(const uint8 index, const uint64 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_LONGLONG, &value, sizeof(uint64), true);
-}
-
-void MySQLPreparedStatement::setInt8(const uint8 index, const int8 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_TINY, &value, sizeof(int8), false);
-}
-
-void MySQLPreparedStatement::setInt16(const uint8 index, const int16 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_SHORT, &value, sizeof(int16), false);
-}
-
-void MySQLPreparedStatement::setInt32(const uint8 index, const int32 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_LONG, &value, sizeof(int32), false);
-}
-
-void MySQLPreparedStatement::setInt64(const uint8 index, const int64 value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_LONGLONG, &value, sizeof(int64), false);
-}
-
-void MySQLPreparedStatement::setFloat(const uint8 index, const float value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_FLOAT, &value, sizeof(float), (value > 0.0f));
-}
-
-void MySQLPreparedStatement::setDouble(const uint8 index, const double value)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- setValue(param, MYSQL_TYPE_DOUBLE, &value, sizeof(double), (value > 0.0f));
-}
-
-void MySQLPreparedStatement::setBinary(const uint8 index, const std::vector<uint8>& value, bool isString)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- size_t len = value.size();
- param->buffer_type = MYSQL_TYPE_BLOB;
- delete [] static_cast<char *>(param->buffer);
- param->buffer = new char[len];
- param->buffer_length = len;
- param->is_null_value = 0;
- delete param->length;
- param->length = new unsigned long(len);
- if (isString)
- {
- *param->length -= 1;
- param->buffer_type = MYSQL_TYPE_VAR_STRING;
- }
-
- memcpy(param->buffer, value.data(), len);
-}
-
-void MySQLPreparedStatement::setNull(const uint8 index)
-{
- CheckValidIndex(index);
- m_paramsSet[index] = true;
- MYSQL_BIND* param = &m_bind[index];
- param->buffer_type = MYSQL_TYPE_NULL;
- delete [] static_cast<char *>(param->buffer);
- param->buffer = NULL;
- param->buffer_length = 0;
- param->is_null_value = 1;
- delete param->length;
- param->length = NULL;
-}
-
-void MySQLPreparedStatement::setValue(MYSQL_BIND* param, enum_field_types type, const void* value, uint32 len, bool isUnsigned)
-{
- param->buffer_type = type;
- delete [] static_cast<char *>(param->buffer);
- param->buffer = new char[len];
- param->buffer_length = 0;
- param->is_null_value = 0;
- param->length = NULL; // Only != NULL for strings
- param->is_unsigned = isUnsigned;
-
- memcpy(param->buffer, value, len);
-}
-
-std::string MySQLPreparedStatement::getQueryString(std::string const& sqlPattern) const
-{
- std::string queryString = sqlPattern;
-
- size_t pos = 0;
- for (uint32 i = 0; i < m_stmt->statement_data.size(); i++)
- {
- pos = queryString.find('?', pos);
- std::stringstream ss;
-
- switch (m_stmt->statement_data[i].type)
- {
- case TYPE_BOOL:
- ss << uint16(m_stmt->statement_data[i].data.boolean);
- break;
- case TYPE_UI8:
- ss << uint16(m_stmt->statement_data[i].data.ui8); // stringstream will append a character with that code instead of numeric representation
- break;
- case TYPE_UI16:
- ss << m_stmt->statement_data[i].data.ui16;
- break;
- case TYPE_UI32:
- ss << m_stmt->statement_data[i].data.ui32;
- break;
- case TYPE_I8:
- ss << int16(m_stmt->statement_data[i].data.i8); // stringstream will append a character with that code instead of numeric representation
- break;
- case TYPE_I16:
- ss << m_stmt->statement_data[i].data.i16;
- break;
- case TYPE_I32:
- ss << m_stmt->statement_data[i].data.i32;
- break;
- case TYPE_UI64:
- ss << m_stmt->statement_data[i].data.ui64;
- break;
- case TYPE_I64:
- ss << m_stmt->statement_data[i].data.i64;
- break;
- case TYPE_FLOAT:
- ss << m_stmt->statement_data[i].data.f;
- break;
- case TYPE_DOUBLE:
- ss << m_stmt->statement_data[i].data.d;
- break;
- case TYPE_STRING:
- ss << '\'' << (char const*)m_stmt->statement_data[i].binary.data() << '\'';
- break;
- case TYPE_BINARY:
- ss << "BINARY";
- break;
- case TYPE_NULL:
- ss << "NULL";
- break;
- }
-
- std::string replaceStr = ss.str();
- queryString.replace(pos, 1, replaceStr);
- pos += replaceStr.length();
- }
-
- return queryString;
-}
-
-//- Execution
-PreparedStatementTask::PreparedStatementTask(PreparedStatement* stmt, bool async) :
-m_stmt(stmt), m_result(nullptr)
-{
- m_has_result = async; // If it's async, then there's a result
- if (async)
- m_result = new PreparedQueryResultPromise();
-}
-
-PreparedStatementTask::~PreparedStatementTask()
-{
- delete m_stmt;
- if (m_has_result && m_result != nullptr)
- delete m_result;
-}
-
-bool PreparedStatementTask::Execute()
-{
- if (m_has_result)
- {
- PreparedResultSet* result = m_conn->Query(m_stmt);
- if (!result || !result->GetRowCount())
- {
- delete result;
- m_result->set_value(PreparedQueryResult(NULL));
- return false;
- }
- m_result->set_value(PreparedQueryResult(result));
- return true;
- }
-
- return m_conn->Execute(m_stmt);
-}