1.1 --- a/extensions/sqlite/driver/SqDatabase.cpp Fri Jul 11 06:51:21 2008 +0000
1.2 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000
1.3 @@ -1,247 +0,0 @@
1.4 -/**
1.5 - * vim: set ts=4 :
1.6 - * =============================================================================
1.7 - * SourceMod SQLite Extension
1.8 - * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
1.9 - * =============================================================================
1.10 - *
1.11 - * This program is free software; you can redistribute it and/or modify it under
1.12 - * the terms of the GNU General Public License, version 3.0, as published by the
1.13 - * Free Software Foundation.
1.14 - *
1.15 - * This program is distributed in the hope that it will be useful, but WITHOUT
1.16 - * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.17 - * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.18 - * details.
1.19 - *
1.20 - * You should have received a copy of the GNU General Public License along with
1.21 - * this program. If not, see <http://www.gnu.org/licenses/>.
1.22 - *
1.23 - * As a special exception, AlliedModders LLC gives you permission to link the
1.24 - * code of this program (as well as its derivative works) to "Half-Life 2," the
1.25 - * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.26 - * by the Valve Corporation. You must obey the GNU General Public License in
1.27 - * all respects for all other code used. Additionally, AlliedModders LLC grants
1.28 - * this exception to all derivative works. AlliedModders LLC defines further
1.29 - * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.30 - * or <http://www.sourcemod.net/license.php>.
1.31 - *
1.32 - * Version: $Id$
1.33 - */
1.34 -
1.35 -#include "extension.h"
1.36 -#include "SqDatabase.h"
1.37 -#include "SqQuery.h"
1.38 -
1.39 -SqDatabase::SqDatabase(sqlite3 *sq3, bool persistent) :
1.40 - m_sq3(sq3), m_refcount(1), m_pFullLock(NULL), m_Persistent(persistent)
1.41 -{
1.42 - m_pRefLock = threader->MakeMutex();
1.43 -}
1.44 -
1.45 -SqDatabase::~SqDatabase()
1.46 -{
1.47 - m_pRefLock->DestroyThis();
1.48 - if (m_pFullLock)
1.49 - {
1.50 - m_pFullLock->DestroyThis();
1.51 - }
1.52 - sqlite3_close(m_sq3);
1.53 -}
1.54 -
1.55 -void SqDatabase::IncReferenceCount()
1.56 -{
1.57 - m_pRefLock->Lock();
1.58 - m_refcount++;
1.59 - m_pRefLock->Unlock();
1.60 -}
1.61 -
1.62 -bool SqDatabase::Close()
1.63 -{
1.64 - m_pRefLock->Lock();
1.65 - if (m_refcount > 1)
1.66 - {
1.67 - m_refcount--;
1.68 - m_pRefLock->Unlock();
1.69 - return false;
1.70 - }
1.71 - m_pRefLock->Unlock();
1.72 -
1.73 - if (m_Persistent)
1.74 - {
1.75 - g_SqDriver.RemovePersistent(this);
1.76 - }
1.77 -
1.78 - delete this;
1.79 -
1.80 - return true;
1.81 -}
1.82 -
1.83 -const char *SqDatabase::GetError(int *errorCode/* =NULL */)
1.84 -{
1.85 - return sqlite3_errmsg(m_sq3);
1.86 -}
1.87 -
1.88 -bool SqDatabase::LockForFullAtomicOperation()
1.89 -{
1.90 - if (!m_pFullLock)
1.91 - {
1.92 - m_pFullLock = threader->MakeMutex();
1.93 - if (!m_pFullLock)
1.94 - {
1.95 - return false;
1.96 - }
1.97 - }
1.98 -
1.99 - m_pFullLock->Lock();
1.100 -
1.101 - return true;
1.102 -}
1.103 -
1.104 -void SqDatabase::UnlockFromFullAtomicOperation()
1.105 -{
1.106 - if (m_pFullLock)
1.107 - {
1.108 - m_pFullLock->Unlock();
1.109 - }
1.110 -}
1.111 -
1.112 -IDBDriver *SqDatabase::GetDriver()
1.113 -{
1.114 - return &g_SqDriver;
1.115 -}
1.116 -
1.117 -bool SqDatabase::QuoteString(const char *str, char buffer[], size_t maxlen, size_t *newSize)
1.118 -{
1.119 - char *res = sqlite3_snprintf(static_cast<int>(maxlen), buffer, "%q", str);
1.120 -
1.121 - if (res != NULL && newSize != NULL)
1.122 - {
1.123 - *newSize = strlen(buffer);
1.124 - }
1.125 -
1.126 - return (res != NULL);
1.127 -}
1.128 -
1.129 -unsigned int SqDatabase::GetInsertID()
1.130 -{
1.131 - return (unsigned int)sqlite3_last_insert_rowid(m_sq3);
1.132 -}
1.133 -
1.134 -unsigned int SqDatabase::GetAffectedRows()
1.135 -{
1.136 - return (unsigned int)sqlite3_changes(m_sq3);
1.137 -}
1.138 -
1.139 -bool SqDatabase::DoSimpleQuery(const char *query)
1.140 -{
1.141 - IQuery *pQuery = DoQuery(query);
1.142 - if (!pQuery)
1.143 - {
1.144 - return false;
1.145 - }
1.146 - pQuery->Destroy();
1.147 - return true;
1.148 -}
1.149 -
1.150 -/* this sounds like daiquiri.. i'm tired. */
1.151 -IQuery *SqDatabase::DoQuery(const char *query)
1.152 -{
1.153 - IPreparedQuery *pQuery = PrepareQuery(query, NULL, 0, NULL);
1.154 - if (!pQuery)
1.155 - {
1.156 - return NULL;
1.157 - }
1.158 - if (!pQuery->Execute())
1.159 - {
1.160 - pQuery->Destroy();
1.161 - return NULL;
1.162 - }
1.163 - return pQuery;
1.164 -}
1.165 -
1.166 -bool SqDatabase::DoSimpleQueryEx(const char *query, size_t len)
1.167 -{
1.168 - IQuery *pQuery = DoQueryEx(query, len);
1.169 - if (!pQuery)
1.170 - {
1.171 - return false;
1.172 - }
1.173 - pQuery->Destroy();
1.174 - return true;
1.175 -}
1.176 -
1.177 -IQuery *SqDatabase::DoQueryEx(const char *query, size_t len)
1.178 -{
1.179 - IPreparedQuery *pQuery = PrepareQueryEx(query, len, NULL, 0, NULL);
1.180 - if (!pQuery)
1.181 - {
1.182 - return NULL;
1.183 - }
1.184 - if (!pQuery->Execute())
1.185 - {
1.186 - pQuery->Destroy();
1.187 - return NULL;
1.188 - }
1.189 - return pQuery;
1.190 -}
1.191 -
1.192 -IPreparedQuery *SqDatabase::PrepareQuery(const char *query,
1.193 - char *error,
1.194 - size_t maxlength,
1.195 - int *errCode/* =NULL */)
1.196 -{
1.197 - sqlite3_stmt *stmt = NULL;
1.198 - if ((m_LastErrorCode = sqlite3_prepare_v2(m_sq3, query, -1, &stmt, NULL)) != SQLITE_OK
1.199 - || !stmt)
1.200 - {
1.201 - const char *msg;
1.202 - if (m_LastErrorCode != SQLITE_OK)
1.203 - {
1.204 - msg = sqlite3_errmsg(m_sq3);
1.205 - } else {
1.206 - msg = "Invalid query string";
1.207 - m_LastErrorCode = SQLITE_ERROR;
1.208 - }
1.209 - if (error)
1.210 - {
1.211 - strncopy(error, msg, maxlength);
1.212 - }
1.213 - m_LastError.assign(msg);
1.214 - return NULL;
1.215 - }
1.216 - return new SqQuery(this, stmt);
1.217 -}
1.218 -
1.219 -IPreparedQuery *SqDatabase::PrepareQueryEx(const char *query,
1.220 - size_t len,
1.221 - char *error,
1.222 - size_t maxlength,
1.223 - int *errCode/* =NULL */)
1.224 -{
1.225 - sqlite3_stmt *stmt = NULL;
1.226 - if ((m_LastErrorCode = sqlite3_prepare_v2(m_sq3, query, len, &stmt, NULL)) != SQLITE_OK
1.227 - || !stmt)
1.228 - {
1.229 - const char *msg;
1.230 - if (m_LastErrorCode != SQLITE_OK)
1.231 - {
1.232 - msg = sqlite3_errmsg(m_sq3);
1.233 - } else {
1.234 - msg = "Invalid query string";
1.235 - m_LastErrorCode = SQLITE_ERROR;
1.236 - }
1.237 - if (error)
1.238 - {
1.239 - strncopy(error, msg, maxlength);
1.240 - }
1.241 - m_LastError.assign(msg);
1.242 - return NULL;
1.243 - }
1.244 - return new SqQuery(this, stmt);
1.245 -}
1.246 -
1.247 -sqlite3 *SqDatabase::GetDb()
1.248 -{
1.249 - return m_sq3;
1.250 -}
