mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-16 07:30:42 +01:00
95 lines
3.0 KiB
C++
95 lines
3.0 KiB
C++
/*
|
|
* This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
|
|
*
|
|
* 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 "QueryHolder.h"
|
|
#include "Errors.h"
|
|
#include "Log.h"
|
|
#include "MySQLConnection.h"
|
|
#include "PreparedStatement.h"
|
|
#include "QueryResult.h"
|
|
|
|
bool SQLQueryHolderBase::SetPreparedQueryImpl(size_t index, PreparedStatementBase* stmt)
|
|
{
|
|
if (m_queries.size() <= index)
|
|
{
|
|
TC_LOG_ERROR("sql.sql", "Query index ({}) out of range (size: {}) for prepared statement", uint32(index), (uint32)m_queries.size());
|
|
return false;
|
|
}
|
|
|
|
m_queries[index].first = stmt;
|
|
return true;
|
|
}
|
|
|
|
PreparedQueryResult SQLQueryHolderBase::GetPreparedResult(size_t index) const
|
|
{
|
|
// Don't call to this function if the index is of a prepared statement
|
|
ASSERT(index < m_queries.size(), "Query holder result index out of range, tried to access index " SZFMTD " but there are only " SZFMTD " results",
|
|
index, m_queries.size());
|
|
|
|
return m_queries[index].second;
|
|
}
|
|
|
|
void SQLQueryHolderBase::SetPreparedResult(size_t index, PreparedResultSet* result)
|
|
{
|
|
if (result && !result->GetRowCount())
|
|
{
|
|
delete result;
|
|
result = nullptr;
|
|
}
|
|
|
|
/// store the result in the holder
|
|
if (index < m_queries.size())
|
|
m_queries[index].second = PreparedQueryResult(result);
|
|
}
|
|
|
|
SQLQueryHolderBase::~SQLQueryHolderBase()
|
|
{
|
|
for (std::pair<PreparedStatementBase*, PreparedQueryResult>& query : m_queries)
|
|
{
|
|
/// if the result was never used, free the resources
|
|
/// results used already (getresult called) are expected to be deleted
|
|
delete query.first;
|
|
}
|
|
}
|
|
|
|
void SQLQueryHolderBase::SetSize(size_t size)
|
|
{
|
|
/// to optimize push_back, reserve the number of queries about to be executed
|
|
m_queries.resize(size);
|
|
}
|
|
|
|
bool SQLQueryHolderTask::Execute(MySQLConnection* conn, SQLQueryHolderBase* holder)
|
|
{
|
|
/// execute all queries in the holder and pass the results
|
|
for (size_t i = 0; i < holder->m_queries.size(); ++i)
|
|
if (PreparedStatementBase* stmt = holder->m_queries[i].first)
|
|
holder->SetPreparedResult(i, conn->Query(stmt));
|
|
|
|
return true;
|
|
}
|
|
|
|
bool SQLQueryHolderCallback::InvokeIfReady()
|
|
{
|
|
if (m_future.valid() && m_future.wait_for(std::chrono::seconds(0)) == std::future_status::ready)
|
|
{
|
|
m_callback(*m_holder);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|