94 lines
1.7 KiB
C++
94 lines
1.7 KiB
C++
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include "SqliteQuery.h"
|
||
|
|
#include "SqliteDatabase.h"
|
||
|
|
#include "SqliteResultSet.h"
|
||
|
|
|
||
|
|
#if defined WIN32
|
||
|
|
#define snprintf _snprintf
|
||
|
|
#endif
|
||
|
|
|
||
|
|
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::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)
|
||
|
|
{
|
||
|
|
snprintf(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);
|
||
|
|
m_LastRes = pRes;
|
||
|
|
info->rs = static_cast<IResultSet *>(pRes);
|
||
|
|
} else {
|
||
|
|
info->rs = NULL;
|
||
|
|
if (results)
|
||
|
|
{
|
||
|
|
sqlite3_free_table(results);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return info->success;
|
||
|
|
}
|