initial import of sqlitex
This commit is contained in:
93
dlls/sqlite/sqlitepp/SqliteQuery.cpp
Normal file
93
dlls/sqlite/sqlitepp/SqliteQuery.cpp
Normal file
@ -0,0 +1,93 @@
|
||||
#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;
|
||||
}
|
Reference in New Issue
Block a user