aboutsummaryrefslogtreecommitdiff
path: root/src/server/database/Database/PreparedStatement.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/server/database/Database/PreparedStatement.cpp')
-rw-r--r--src/server/database/Database/PreparedStatement.cpp120
1 files changed, 52 insertions, 68 deletions
diff --git a/src/server/database/Database/PreparedStatement.cpp b/src/server/database/Database/PreparedStatement.cpp
index 371543453da..e410df84464 100644
--- a/src/server/database/Database/PreparedStatement.cpp
+++ b/src/server/database/Database/PreparedStatement.cpp
@@ -26,15 +26,15 @@
#include <mysql.h>
#include <sstream>
-PreparedStatementBase::PreparedStatementBase(uint32 index) :
-m_stmt(NULL),
-m_index(index) { }
+PreparedStatementBase::PreparedStatementBase(uint32 index, uint8 capacity) :
+ m_stmt(nullptr), m_index(index), statement_data(capacity) { }
PreparedStatementBase::~PreparedStatementBase() { }
-void PreparedStatementBase::BindParameters()
+void PreparedStatementBase::BindParameters(MySQLPreparedStatement* stmt)
{
- ASSERT(m_stmt);
+ ASSERT(stmt);
+ m_stmt = stmt;
uint8 i = 0;
for (; i < statement_data.size(); i++)
@@ -42,51 +42,51 @@ void PreparedStatementBase::BindParameters()
switch (statement_data[i].type)
{
case TYPE_BOOL:
- m_stmt->setBool(i, statement_data[i].data.boolean);
+ stmt->setBool(i, statement_data[i].data.boolean);
break;
case TYPE_UI8:
- m_stmt->setUInt8(i, statement_data[i].data.ui8);
+ stmt->setUInt8(i, statement_data[i].data.ui8);
break;
case TYPE_UI16:
- m_stmt->setUInt16(i, statement_data[i].data.ui16);
+ stmt->setUInt16(i, statement_data[i].data.ui16);
break;
case TYPE_UI32:
- m_stmt->setUInt32(i, statement_data[i].data.ui32);
+ stmt->setUInt32(i, statement_data[i].data.ui32);
break;
case TYPE_I8:
- m_stmt->setInt8(i, statement_data[i].data.i8);
+ stmt->setInt8(i, statement_data[i].data.i8);
break;
case TYPE_I16:
- m_stmt->setInt16(i, statement_data[i].data.i16);
+ stmt->setInt16(i, statement_data[i].data.i16);
break;
case TYPE_I32:
- m_stmt->setInt32(i, statement_data[i].data.i32);
+ stmt->setInt32(i, statement_data[i].data.i32);
break;
case TYPE_UI64:
- m_stmt->setUInt64(i, statement_data[i].data.ui64);
+ stmt->setUInt64(i, statement_data[i].data.ui64);
break;
case TYPE_I64:
- m_stmt->setInt64(i, statement_data[i].data.i64);
+ stmt->setInt64(i, statement_data[i].data.i64);
break;
case TYPE_FLOAT:
- m_stmt->setFloat(i, statement_data[i].data.f);
+ stmt->setFloat(i, statement_data[i].data.f);
break;
case TYPE_DOUBLE:
- m_stmt->setDouble(i, statement_data[i].data.d);
+ stmt->setDouble(i, statement_data[i].data.d);
break;
case TYPE_STRING:
- m_stmt->setBinary(i, statement_data[i].binary, true);
+ stmt->setBinary(i, statement_data[i].binary, true);
break;
case TYPE_BINARY:
- m_stmt->setBinary(i, statement_data[i].binary, false);
+ stmt->setBinary(i, statement_data[i].binary, false);
break;
case TYPE_NULL:
- m_stmt->setNull(i);
+ stmt->setNull(i);
break;
}
}
#ifdef _DEBUG
- if (i < m_stmt->m_paramCount)
+ if (i < stmt->m_paramCount)
TC_LOG_WARN("sql.sql", "[WARNING]: BindParameters() for statement %u did not bind all allocated parameters", m_index);
#endif
}
@@ -94,8 +94,7 @@ void PreparedStatementBase::BindParameters()
//- Bind to buffer
void PreparedStatementBase::setBool(const uint8 index, const bool value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.boolean = value;
statement_data[index].type = TYPE_BOOL;
@@ -103,8 +102,7 @@ void PreparedStatementBase::setBool(const uint8 index, const bool value)
void PreparedStatementBase::setUInt8(const uint8 index, const uint8 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.ui8 = value;
statement_data[index].type = TYPE_UI8;
@@ -112,8 +110,7 @@ void PreparedStatementBase::setUInt8(const uint8 index, const uint8 value)
void PreparedStatementBase::setUInt16(const uint8 index, const uint16 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.ui16 = value;
statement_data[index].type = TYPE_UI16;
@@ -121,8 +118,7 @@ void PreparedStatementBase::setUInt16(const uint8 index, const uint16 value)
void PreparedStatementBase::setUInt32(const uint8 index, const uint32 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.ui32 = value;
statement_data[index].type = TYPE_UI32;
@@ -130,8 +126,7 @@ void PreparedStatementBase::setUInt32(const uint8 index, const uint32 value)
void PreparedStatementBase::setUInt64(const uint8 index, const uint64 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.ui64 = value;
statement_data[index].type = TYPE_UI64;
@@ -139,8 +134,7 @@ void PreparedStatementBase::setUInt64(const uint8 index, const uint64 value)
void PreparedStatementBase::setInt8(const uint8 index, const int8 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.i8 = value;
statement_data[index].type = TYPE_I8;
@@ -148,8 +142,7 @@ void PreparedStatementBase::setInt8(const uint8 index, const int8 value)
void PreparedStatementBase::setInt16(const uint8 index, const int16 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.i16 = value;
statement_data[index].type = TYPE_I16;
@@ -157,8 +150,7 @@ void PreparedStatementBase::setInt16(const uint8 index, const int16 value)
void PreparedStatementBase::setInt32(const uint8 index, const int32 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.i32 = value;
statement_data[index].type = TYPE_I32;
@@ -166,8 +158,7 @@ void PreparedStatementBase::setInt32(const uint8 index, const int32 value)
void PreparedStatementBase::setInt64(const uint8 index, const int64 value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.i64 = value;
statement_data[index].type = TYPE_I64;
@@ -175,8 +166,7 @@ void PreparedStatementBase::setInt64(const uint8 index, const int64 value)
void PreparedStatementBase::setFloat(const uint8 index, const float value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.f = value;
statement_data[index].type = TYPE_FLOAT;
@@ -184,8 +174,7 @@ void PreparedStatementBase::setFloat(const uint8 index, const float value)
void PreparedStatementBase::setDouble(const uint8 index, const double value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].data.d = value;
statement_data[index].type = TYPE_DOUBLE;
@@ -193,8 +182,7 @@ void PreparedStatementBase::setDouble(const uint8 index, const double value)
void PreparedStatementBase::setString(const uint8 index, const std::string& value)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].binary.resize(value.length() + 1);
memcpy(statement_data[index].binary.data(), value.c_str(), value.length() + 1);
@@ -203,8 +191,7 @@ void PreparedStatementBase::setString(const uint8 index, const std::string& valu
void PreparedStatementBase::setBinary(const uint8 index, const std::vector<uint8>& value)
{
- if (index >= statement_data.size())
- statement_data.resize(index + 1);
+ ASSERT(index < statement_data.size());
statement_data[index].binary = value;
statement_data[index].type = TYPE_BINARY;
@@ -212,16 +199,13 @@ void PreparedStatementBase::setBinary(const uint8 index, const std::vector<uint8
void PreparedStatementBase::setNull(const uint8 index)
{
- if (index >= statement_data.size())
- statement_data.resize(index+1);
+ ASSERT(index < statement_data.size());
statement_data[index].type = TYPE_NULL;
}
-MySQLPreparedStatement::MySQLPreparedStatement(MYSQL_STMT* stmt) :
-m_stmt(NULL),
-m_Mstmt(stmt),
-m_bind(NULL)
+MySQLPreparedStatement::MySQLPreparedStatement(MYSQL_STMT* stmt, std::string queryString) :
+ m_stmt(nullptr), m_Mstmt(stmt), m_bind(nullptr), m_queryString(std::move(queryString))
{
/// Initialize variable parameters
m_paramCount = mysql_stmt_param_count(stmt);
@@ -278,17 +262,17 @@ static void SetParameterValue(MYSQL_BIND* param, enum_field_types type, const vo
}
//- Bind on mysql level
-void MySQLPreparedStatement::CheckValidIndex(uint8 index)
+void MySQLPreparedStatement::AssertValidIndex(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);
+ TC_LOG_ERROR("sql.sql", "[ERROR] Prepared Statement (id: %u) trying to bind value on already bound index (%u).", m_stmt->m_index, index);
}
void MySQLPreparedStatement::setNull(const uint8 index)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
param->buffer_type = MYSQL_TYPE_NULL;
@@ -307,7 +291,7 @@ void MySQLPreparedStatement::setBool(const uint8 index, const bool value)
void MySQLPreparedStatement::setUInt8(const uint8 index, const uint8 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_TINY, &value, sizeof(uint8), true);
@@ -315,7 +299,7 @@ void MySQLPreparedStatement::setUInt8(const uint8 index, const uint8 value)
void MySQLPreparedStatement::setUInt16(const uint8 index, const uint16 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_SHORT, &value, sizeof(uint16), true);
@@ -323,7 +307,7 @@ void MySQLPreparedStatement::setUInt16(const uint8 index, const uint16 value)
void MySQLPreparedStatement::setUInt32(const uint8 index, const uint32 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_LONG, &value, sizeof(uint32), true);
@@ -331,7 +315,7 @@ void MySQLPreparedStatement::setUInt32(const uint8 index, const uint32 value)
void MySQLPreparedStatement::setUInt64(const uint8 index, const uint64 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_LONGLONG, &value, sizeof(uint64), true);
@@ -339,7 +323,7 @@ void MySQLPreparedStatement::setUInt64(const uint8 index, const uint64 value)
void MySQLPreparedStatement::setInt8(const uint8 index, const int8 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_TINY, &value, sizeof(int8), false);
@@ -347,7 +331,7 @@ void MySQLPreparedStatement::setInt8(const uint8 index, const int8 value)
void MySQLPreparedStatement::setInt16(const uint8 index, const int16 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_SHORT, &value, sizeof(int16), false);
@@ -355,7 +339,7 @@ void MySQLPreparedStatement::setInt16(const uint8 index, const int16 value)
void MySQLPreparedStatement::setInt32(const uint8 index, const int32 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_LONG, &value, sizeof(int32), false);
@@ -363,7 +347,7 @@ void MySQLPreparedStatement::setInt32(const uint8 index, const int32 value)
void MySQLPreparedStatement::setInt64(const uint8 index, const int64 value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_LONGLONG, &value, sizeof(int64), false);
@@ -371,7 +355,7 @@ void MySQLPreparedStatement::setInt64(const uint8 index, const int64 value)
void MySQLPreparedStatement::setFloat(const uint8 index, const float value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_FLOAT, &value, sizeof(float), (value > 0.0f));
@@ -379,7 +363,7 @@ void MySQLPreparedStatement::setFloat(const uint8 index, const float value)
void MySQLPreparedStatement::setDouble(const uint8 index, const double value)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
SetParameterValue(param, MYSQL_TYPE_DOUBLE, &value, sizeof(double), (value > 0.0f));
@@ -387,7 +371,7 @@ void MySQLPreparedStatement::setDouble(const uint8 index, const double value)
void MySQLPreparedStatement::setBinary(const uint8 index, const std::vector<uint8>& value, bool isString)
{
- CheckValidIndex(index);
+ AssertValidIndex(index);
m_paramsSet[index] = true;
MYSQL_BIND* param = &m_bind[index];
uint32 len = uint32(value.size());
@@ -407,9 +391,9 @@ void MySQLPreparedStatement::setBinary(const uint8 index, const std::vector<uint
memcpy(param->buffer, value.data(), len);
}
-std::string MySQLPreparedStatement::getQueryString(std::string const& sqlPattern) const
+std::string MySQLPreparedStatement::getQueryString() const
{
- std::string queryString = sqlPattern;
+ std::string queryString(m_queryString);
size_t pos = 0;
for (uint32 i = 0; i < m_stmt->statement_data.size(); i++)