diff -r c7b4237fff50 -r 03a67e41afbc extensions/sqlite/driver/SqQuery.cpp --- a/extensions/sqlite/driver/SqQuery.cpp Fri Mar 28 06:21:34 2008 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,205 +0,0 @@ -/** - * vim: set ts=4 : - * ============================================================================= - * SourceMod SQLite Extension - * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved. - * ============================================================================= - * - * This program is free software; you can redistribute it and/or modify it under - * the terms of the GNU General Public License, version 3.0, as published by the - * Free Software Foundation. - * - * 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 . - * - * As a special exception, AlliedModders LLC gives you permission to link the - * code of this program (as well as its derivative works) to "Half-Life 2," the - * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software - * by the Valve Corporation. You must obey the GNU General Public License in - * all respects for all other code used. Additionally, AlliedModders LLC grants - * this exception to all derivative works. AlliedModders LLC defines further - * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), - * or . - * - * Version: $Id$ - */ - -#include "SqQuery.h" - -SqQuery::SqQuery(SqDatabase *parent, sqlite3_stmt *stmt) : - m_pParent(parent), m_pStmt(stmt), m_pResults(NULL), m_AffectedRows(0), m_InsertID(0) -{ - m_ParamCount = sqlite3_bind_parameter_count(m_pStmt); - m_ColCount = sqlite3_column_count(m_pStmt); - m_pParent->IncReferenceCount(); -} - -SqQuery::~SqQuery() -{ - delete m_pResults; - sqlite3_finalize(m_pStmt); - m_pParent->Close(); -} - -IResultSet *SqQuery::GetResultSet() -{ - return m_pResults; -} - -bool SqQuery::FetchMoreResults() -{ - /* We never have multiple result sets */ - return false; -} - -void SqQuery::Destroy() -{ - delete this; -} - -bool SqQuery::BindParamFloat(unsigned int param, float f) -{ - /* SQLite is 1 indexed */ - param++; - - if (param > m_ParamCount) - { - return false; - } - - return (sqlite3_bind_double(m_pStmt, param, (double)f) == SQLITE_OK); -} - -bool SqQuery::BindParamNull(unsigned int param) -{ - /* SQLite is 1 indexed */ - param++; - - if (param > m_ParamCount) - { - return false; - } - - return (sqlite3_bind_null(m_pStmt, param) == SQLITE_OK); -} - -bool SqQuery::BindParamString(unsigned int param, const char *text, bool copy) -{ - /* SQLite is 1 indexed */ - param++; - - if (param > m_ParamCount) - { - return false; - } - - return (sqlite3_bind_text(m_pStmt, param, text, -1, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK); -} - -bool SqQuery::BindParamInt(unsigned int param, int num, bool signd/* =true */) -{ - /* SQLite is 1 indexed */ - param++; - - if (param > m_ParamCount) - { - return false; - } - - return (sqlite3_bind_int(m_pStmt, param, num) == SQLITE_OK); -} - -bool SqQuery::BindParamBlob(unsigned int param, const void *data, size_t length, bool copy) -{ - /* SQLite is 1 indexed */ - param++; - - if (param > m_ParamCount) - { - return false; - } - - return (sqlite3_bind_blob(m_pStmt, param, data, length, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK); -} - -sqlite3_stmt *SqQuery::GetStmt() -{ - return m_pStmt; -} - -bool SqQuery::Execute() -{ - int rc; - - /* If we don't have a result set and we have a column count, - * create a result set pre-emptively. This is in case there - * are no rows in the upcoming result set. - */ - if (!m_pResults && m_ColCount) - { - m_pResults = new SqResults(this); - } - - /* If we've got results, throw them away */ - if (m_pResults) - { - m_pResults->ResetResultCount(); - } - - /* Fetch each row, if any */ - while ((rc = sqlite3_step(m_pStmt)) == SQLITE_ROW) - { - /* This should NEVER happen but we're being safe. */ - if (!m_pResults) - { - m_pResults = new SqResults(this); - } - m_pResults->PushResult(); - } - - sqlite3 *db = m_pParent->GetDb(); - if (rc != SQLITE_OK && rc != SQLITE_DONE && rc == sqlite3_errcode(db)) - { - /* Something happened... */ - m_LastErrorCode = rc; - m_LastError.assign(sqlite3_errmsg(db)); - m_AffectedRows = 0; - m_InsertID = 0; - } else { - m_LastErrorCode = SQLITE_OK; - m_AffectedRows = (unsigned int)sqlite3_changes(db); - m_InsertID = (unsigned int)sqlite3_last_insert_rowid(db); - } - - /* Reset everything for the next execute */ - sqlite3_reset(m_pStmt); - sqlite3_clear_bindings(m_pStmt); - - return (m_LastErrorCode == SQLITE_OK); -} - -const char *SqQuery::GetError(int *errCode/* =NULL */) -{ - if (errCode) - { - *errCode = m_LastErrorCode; - } - return m_LastError.c_str(); -} - -unsigned int SqQuery::GetAffectedRows() -{ - return m_AffectedRows; -} - -unsigned int SqQuery::GetInsertID() -{ - return m_InsertID; -} - -