126 lines
2.4 KiB
C++
126 lines
2.4 KiB
C++
// vim: set ts=4 sw=4 tw=99 noet:
|
|
//
|
|
// AMX Mod X, based on AMX Mod by Aleksander Naszko ("OLO").
|
|
// Copyright (C) The AMX Mod X Development Team.
|
|
//
|
|
// This software is licensed under the GNU General Public License, version 3 or higher.
|
|
// Additional exceptions apply. For full license details, see LICENSE.txt or visit:
|
|
// https://alliedmods.net/amxmodx-license
|
|
|
|
//
|
|
// SQLite Module
|
|
//
|
|
|
|
#include "amxxmodule.h"
|
|
#include "SqliteQuery.h"
|
|
#include "SqliteDatabase.h"
|
|
#include "SqliteResultSet.h"
|
|
#include <amtl/am-string.h>
|
|
|
|
using namespace SourceMod;
|
|
|
|
SqliteQuery::SqliteQuery(SqliteDatabase *db, const char *query) :
|
|
m_pDatabase(db), m_LastRes(NULL)
|
|
{
|
|
m_QueryString = new char[strlen(query)+1];
|
|
strcpy(m_QueryString, query);
|
|
}
|
|
|
|
SqliteQuery::~SqliteQuery()
|
|
{
|
|
if (m_LastRes)
|
|
{
|
|
m_LastRes->FreeHandle();
|
|
m_LastRes = NULL;
|
|
}
|
|
|
|
delete [] m_QueryString;
|
|
}
|
|
|
|
void SqliteQuery::FreeHandle()
|
|
{
|
|
delete this;
|
|
}
|
|
|
|
bool SqliteQuery::Execute(QueryInfo *info, char *error, size_t maxlength)
|
|
{
|
|
bool res = ExecuteR(info, error, maxlength);
|
|
|
|
if (m_LastRes)
|
|
{
|
|
m_LastRes->FreeHandle();
|
|
}
|
|
|
|
m_LastRes = (SqliteResultSet *)info->rs;
|
|
|
|
return res;
|
|
}
|
|
|
|
bool SqliteQuery::Execute2(QueryInfo *info, char *error, size_t maxlength)
|
|
{
|
|
bool res = ExecuteR(info, error, maxlength);
|
|
|
|
if (m_LastRes)
|
|
m_LastRes->FreeHandle();
|
|
|
|
m_LastRes = (SqliteResultSet *)info->rs;
|
|
|
|
if (info->success)
|
|
{
|
|
info->insert_id = sqlite3_last_insert_rowid(m_pDatabase->m_pSql);
|
|
} else {
|
|
info->insert_id = 0;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
const char *SqliteQuery::GetQueryString()
|
|
{
|
|
return m_QueryString;
|
|
}
|
|
|
|
bool SqliteQuery::ExecuteR(QueryInfo *info, char *error, size_t maxlength)
|
|
{
|
|
int err;
|
|
char *errmsg;
|
|
char **results;
|
|
int rows, cols;
|
|
|
|
err = sqlite3_get_table(m_pDatabase->m_pSql, m_QueryString, &results, &rows, &cols, &errmsg);
|
|
|
|
if (err != SQLITE_OK)
|
|
{
|
|
if (error && maxlength && errmsg)
|
|
{
|
|
ke::SafeSprintf(error, maxlength, "%s", errmsg);
|
|
}
|
|
info->affected_rows = 0;
|
|
info->errorcode = err;
|
|
info->rs = NULL;
|
|
info->success = false;
|
|
} else {
|
|
info->affected_rows = sqlite3_changes(m_pDatabase->m_pSql);
|
|
info->errorcode = 0;
|
|
info->success = true;
|
|
if (cols)
|
|
{
|
|
SqliteResults data;
|
|
data.cols = cols;
|
|
data.rows = rows;
|
|
data.results = results;
|
|
|
|
SqliteResultSet *pRes = new SqliteResultSet(data);
|
|
info->rs = static_cast<IResultSet *>(pRes);
|
|
} else {
|
|
info->rs = NULL;
|
|
if (results)
|
|
{
|
|
sqlite3_free_table(results);
|
|
}
|
|
}
|
|
}
|
|
|
|
return info->success;
|
|
}
|