sourcemod/extensions/sqlite/driver/SqQuery.cpp

/**
 * vim: set ts=4 :
 * ===============================================================
 * SourceMod SQLite Driver 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
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * 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_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've got results, throw them away */
	if (m_pResults)
	{
		m_pResults->ResetResultCount();
	}

	while ((rc = sqlite3_step(m_pStmt)) == SQLITE_ROW)
	{
		/* Delay creation as long as possible... */
		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;
}