db_command.cpp

Go to the documentation of this file.

/*
**  ClanLib SDK
**  Copyright (c) 1997-2005 The ClanLib Team
**
**  This software is provided 'as-is', without any express or implied
**  warranty.  In no event will the authors be held liable for any damages
**  arising from the use of this software.
**
**  Permission is granted to anyone to use this software for any purpose,
**  including commercial applications, and to alter it and redistribute it
**  freely, subject to the following restrictions:
**
**  1. The origin of this software must not be misrepresented; you must not
**     claim that you wrote the original software. If you use this software
**     in a product, an acknowledgment in the product documentation would be
**     appreciated but is not required.
**  2. Altered source versions must be plainly marked as such, and must not be
**     misrepresented as being the original software.
**  3. This notice may not be removed or altered from any source distribution.
**
**  Note: Some of the libraries ClanLib link to may have additional
**  requirements or restrictions.
**
**  File Author(s):
**
**    Magnus Norddahl
*/

#include "precomp.h"
#include "db_command.h"
#include "db_connection.h"
#include "exception.h"
#include "bytearray.h"
#include "system.h"
#include "string_help.h"
#include "logger.h"
#include <sqlite3.h>
#include <stdarg.h>

// CL_DBCommand Construction:

CL_DBCommand::CL_DBCommand(CL_DBConnection *database)
: db(database), vm(0), last_insert_rowid(-1)
{
}

CL_DBCommand::CL_DBCommand(const CL_String &sql_statement, CL_DBConnection *database)
: db(database), vm(0), last_insert_rowid(-1)
{
        compile(sql_statement);
}

CL_DBCommand::~CL_DBCommand()
{
        reset();
}

// CL_DBCommand Attributes:

int CL_DBCommand::get_last_insert_rowid() const
{
        return last_insert_rowid;
}

int CL_DBCommand::get_columns() const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_count(vm);
}
        
const char *CL_DBCommand::get_name(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_name(vm, column);
}

int CL_DBCommand::get_index(const char *name) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        int count = get_columns();
        for (int index=0; index<count; index++)
        {
                const char *col_name = sqlite3_column_name(vm, index);
                if (col_name != 0 && strcmp(col_name, name) == 0)
                {
                        return index;
                }
        }

        CL_String name_text = CL_StringHelp::utf8_to_text(name);
        throw CL_Exception(TEXT("No such column name ") + name_text);
        return -1;
}

int CL_DBCommand::get_index(const CL_String &name) const
{
        return get_index(name.c_str());
}

const char *CL_DBCommand::get_decltype(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_decltype(vm, column);
}

const char *CL_DBCommand::get_decltype(const char *name) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_decltype(vm, get_index(name));
}

CL_ByteArray CL_DBCommand::get_blob(int column) const
{
        CL_ByteArray array;
        get_blob(column, array);
        return array;
}

CL_ByteArray CL_DBCommand::get_blob(const char *name) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return get_blob(get_index(name));
}

void CL_DBCommand::get_blob(int column, CL_ByteArray &array) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        const void *blob = sqlite3_column_blob(vm, column);
        if (blob == 0)
        {
                array.set_size(0);
                return;
        }
        
        array.set_size(sqlite3_column_bytes(vm, column));
        memcpy(array.get_data(), blob, array.get_size());
}

void CL_DBCommand::get_blob(const char *name, CL_ByteArray &array) const
{
        return get_blob(get_index(name), array);
}

const void *CL_DBCommand::get_blob_raw(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_blob(vm, column);
}

const void *CL_DBCommand::get_blob_raw(const char *name) const
{
        return get_blob_raw(get_index(name));
}

int CL_DBCommand::get_blob_size(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_bytes(vm, column);
}

int CL_DBCommand::get_blob_size(const char *name) const
{
        return get_blob_size(get_index(name));
}

double CL_DBCommand::get_double(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_double(vm, column);
}

double CL_DBCommand::get_double(const char *name) const
{
        return get_double(get_index(name));
}

int CL_DBCommand::get_int(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return sqlite3_column_int(vm, column);
}

int CL_DBCommand::get_int(const char *name) const
{
        return get_int(get_index(name));
}

const char *CL_DBCommand::get_text(int column) const
{
        CL_MutexSection mutex_lock(&db->mutex);
        return (char *) sqlite3_column_text(vm, column);
}

const char *CL_DBCommand::get_text(const char *name) const
{
        return get_text(get_index(name));
}

CL_DBReturnValue CL_DBCommand::operator[](int column) const
{
        return CL_DBReturnValue(this, column);
}
        
CL_DBReturnValue CL_DBCommand::operator[](const char *name) const
{
        return CL_DBReturnValue(this, get_index(name));
}
        
CL_DBReturnValue CL_DBCommand::operator[](const CL_String &name) const
{
        return CL_DBReturnValue(this, get_index(name));
}

int CL_DBCommand::get_param_column(const char *name) const
{
        return 0;
}

// CL_DBCommand Operations:

void CL_DBCommand::compile(const CL_String &sql_statement)
{
        compile(sql_statement.c_str());
}

void CL_DBCommand::compile(const char *sql_statement)
{
        cl_log_event("db", sql_statement);
        reset();
        
        CL_MutexSection mutex_lock(&db->mutex);
        const char *tail = 0;
        int result = sqlite3_prepare(db->db, sql_statement, strlen(sql_statement), &vm, &tail);
        if (result != SQLITE_OK)
        {
                CL_StringA error = sqlite3_errmsg(db->db);
                throw CL_Exception(error);
        }
}

void CL_DBCommand::compilef(const char *sql_statement, ...)
{
        CL_MutexSection mutex_lock(&db->mutex);
        va_list params;
        va_start(params, sql_statement);
        char *str = sqlite3_vmprintf(sql_statement, params);
        va_end(params);
        try
        {
                compile(str);
        }
        catch (CL_Exception e)
        {
                sqlite3_free(str);
                throw;
        }
        sqlite3_free(str);
}

void CL_DBCommand::set_param_double(int column, double value)
{
}

void CL_DBCommand::set_param_double(const char *column_name, double value)
{
        set_param_double(get_param_column(column_name), value);
}

void CL_DBCommand::set_param_int(int column, int value)
{
}

void CL_DBCommand::set_param_int(const char *column_name, int value)
{
        set_param_int(get_param_column(column_name), value);
}

void CL_DBCommand::set_param_blob(int column, const CL_ByteArray &blob)
{
}

void CL_DBCommand::set_param_blob(const char *column_name, const CL_ByteArray &blob)
{
        set_param_blob(get_param_column(column_name), blob);
}

void CL_DBCommand::set_param_text(int column, const char *text)
{
}

void CL_DBCommand::set_param_text(const char *column_name, const char *text)
{
        set_param_text(get_param_column(column_name), text);
}

bool CL_DBCommand::step()
{
        if (vm == 0)
                return false;

        for (int i=0; i<100; i++)
        {
                CL_MutexSection mutex_lock(&db->mutex);
                last_insert_rowid = -1;
                int result = sqlite3_step(vm);
                switch (result)
                {
                case SQLITE_BUSY:
                        CL_System::sleep(1);
                        break;
                case SQLITE_ROW:
                        last_insert_rowid = sqlite3_last_insert_rowid(db->db);
                        return true;
                case SQLITE_DONE:
                        last_insert_rowid = sqlite3_last_insert_rowid(db->db);
                        return false;
                case SQLITE_ERROR:
                        reset();
                        return false;
                case SQLITE_MISUSE:
                        throw CL_Exception(TEXT("Database Misuse!"));
                }
        }
        
        throw CL_Exception(TEXT("Database Busy!"));
}

void CL_DBCommand::reset()
{
        if (vm)
        {
                CL_MutexSection mutex_lock(&db->mutex);
                int result = sqlite3_finalize(vm);
                vm = 0;
                last_insert_rowid = -1;
                if (result != SQLITE_OK)
                {
                        const char *err = sqlite3_errmsg(db->db);
                        CL_StringA error = (err == 0) ? "Unknown error" : err;
                        throw CL_Exception(error);
                }
        }
}

// CL_DBCommand Implementation:

---