extensions/sqlite/driver/SqQuery.h
author David Anderson <dvander@alliedmods.net>
Fri Aug 23 00:18:13 2013 -0700 (2013-08-23)
changeset 4040 b6924c1a1d71
parent 2289 44045ead61cd
permissions -rw-r--r--
Add atomic reference counting and port DBI (bug 5876 part 3, r=ds).
dvander@1203
     1
/**
dvander@4040
     2
 * vim: set ts=4 sw=4 tw=99 noet :
ds@1330
     3
 * =============================================================================
dvander@1226
     4
 * SourceMod SQLite Extension
ds@1947
     5
 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1330
     6
 * =============================================================================
dvander@1203
     7
 *
ds@1330
     8
 * This program is free software; you can redistribute it and/or modify it under
ds@1330
     9
 * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330
    10
 * Free Software Foundation.
dvander@1226
    11
 * 
ds@1330
    12
 * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330
    13
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330
    14
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330
    15
 * details.
dvander@1226
    16
 *
ds@1330
    17
 * You should have received a copy of the GNU General Public License along with
ds@1330
    18
 * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1226
    19
 *
ds@1330
    20
 * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330
    21
 * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330
    22
 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330
    23
 * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330
    24
 * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330
    25
 * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330
    26
 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330
    27
 * or <http://www.sourcemod.net/license.php>.
dvander@1203
    28
 *
dvander@1203
    29
 * Version: $Id$
dvander@1203
    30
 */
dvander@1203
    31
dvander@1203
    32
#ifndef _INCLUDE_SQLITE_SOURCEMOD_QUERY_H_
dvander@1203
    33
#define _INCLUDE_SQLITE_SOURCEMOD_QUERY_H_
dvander@1203
    34
dvander@4040
    35
#include <am-refcounting.h>
dvander@1203
    36
#include "SqDatabase.h"
dvander@1203
    37
#include "SqResults.h"
dvander@1203
    38
dvander@1203
    39
class SqQuery : 
dvander@1203
    40
	public IPreparedQuery
dvander@1203
    41
{
dvander@1203
    42
public:
dvander@1203
    43
	SqQuery(SqDatabase *parent, sqlite3_stmt *stmt);
dvander@1203
    44
	~SqQuery();
dvander@1203
    45
public: //IQuery
dvander@1203
    46
	IResultSet *GetResultSet();
dvander@1203
    47
	bool FetchMoreResults();
dvander@1203
    48
	void Destroy();
dvander@1203
    49
public: //IPreparedQuery
dvander@1203
    50
	bool BindParamInt(unsigned int param, int num, bool signd=true);
dvander@1203
    51
	bool BindParamFloat(unsigned int param, float f);
dvander@1203
    52
	bool BindParamNull(unsigned int param);
dvander@1203
    53
	bool BindParamString(unsigned int param, const char *text, bool copy);
dvander@1203
    54
	bool BindParamBlob(unsigned int param, const void *data, size_t length, bool copy);
dvander@1203
    55
	bool Execute();
dvander@1203
    56
	const char *GetError(int *errCode=NULL);
dvander@1203
    57
	unsigned int GetAffectedRows();
dvander@1203
    58
	unsigned int GetInsertID();
dvander@1203
    59
public: //IResultSet
dvander@1203
    60
	unsigned int GetRowCount();
dvander@1203
    61
	unsigned int GetFieldCount();
dvander@1203
    62
	const char *FieldNumToName(unsigned int columnId);
dvander@1203
    63
	bool FieldNameToNum(const char *name, unsigned int *columnId);
dvander@1203
    64
	bool MoreRows();
dvander@1203
    65
	IResultRow *FetchRow();
dvander@1203
    66
	IResultRow *CurrentRow();
dvander@1203
    67
	bool Rewind();
dvander@1203
    68
	DBType GetFieldType(unsigned int field);
dvander@1203
    69
	DBType GetFieldDataType(unsigned int field);
dvander@1203
    70
public: //IResultRow
dvander@1203
    71
	DBResult GetString(unsigned int columnId, const char **pString, size_t *length);
dvander@1203
    72
	DBResult CopyString(unsigned int columnId, 
dvander@1203
    73
		char *buffer, 
dvander@1203
    74
		size_t maxlength, 
dvander@1203
    75
		size_t *written);
dvander@1203
    76
	DBResult GetFloat(unsigned int columnId, float *pFloat);
dvander@1203
    77
	DBResult GetInt(unsigned int columnId, int *pInt);
dvander@1203
    78
	bool IsNull(unsigned int columnId);
dvander@1203
    79
	size_t GetDataSize(unsigned int columnId);
dvander@1203
    80
	DBResult GetBlob(unsigned int columnId, const void **pData, size_t *length);
dvander@1203
    81
	DBResult CopyBlob(unsigned int columnId, void *buffer, size_t maxlength, size_t *written);
dvander@1203
    82
public:
dvander@1203
    83
	sqlite3_stmt *GetStmt();
dvander@1203
    84
private:
dvander@4040
    85
	ke::Ref<SqDatabase> m_pParent;
dvander@1203
    86
	sqlite3_stmt *m_pStmt;
dvander@1203
    87
	SqResults *m_pResults;
dvander@1203
    88
	unsigned int m_ParamCount;
dvander@1203
    89
	String m_LastError;
dvander@1203
    90
	int m_LastErrorCode;
dvander@1203
    91
	unsigned int m_AffectedRows;
dvander@1203
    92
	unsigned int m_InsertID;
dvander@1210
    93
	unsigned int m_ColCount;
dvander@1203
    94
};
dvander@1203
    95
dvander@1203
    96
#endif //_INCLUDE_SQLITE_SOURCEMOD_QUERY_H_