SqQuery.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <stdlib.h>     // malloc
#include <string.h>     // memset
 
#include <sqlite3.h>
 
#include <dcl/Object.h>
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#include <dcl/SQLCore.h>
 
#include "SqConnection.h"
#include "SqQuery.h"
#include "SqField.h"
#include "SqParam.h"
#include "SqTypes.h"
 
#define __TRACE_THIS    0
#if __TRACE_THIS
#define __DCL_TRACE0_N  __DCL_TRACE0
#define __DCL_TRACE1_N  __DCL_TRACE1
#define __DCL_TRACE2_N  __DCL_TRACE2
#define __DCL_TRACE3_N  __DCL_TRACE3
#define __DCL_TRACE4_N  __DCL_TRACE4
#define __DCL_TRACE5_N  __DCL_TRACE5
#define __DCL_TRACE6_N  __DCL_TRACE6
#else
#define __DCL_TRACE0_N(fmt)
#define __DCL_TRACE1_N(fmt, arg)
#define __DCL_TRACE2_N(fmt, arg1, arg2)
#define __DCL_TRACE3_N(fmt, arg1, arg2, arg3)
#define __DCL_TRACE4_N(fmt, arg1, arg2, arg3, arg4)
#define __DCL_TRACE5_N(fmt, arg1, arg2, arg3, arg4, arg5)
#define __DCL_TRACE6_N(fmt, arg1, arg2, arg3, arg4, arg5, arg6)
#endif
 
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/sql/SqQuery.cpp");
 
__DCL_BEGIN_NAMESPACE
 
#define __SET_ERROR(_error) \
    connection()->setErrorStatus(_error, __THIS_FILE__, __LINE__)
#define __SET_ERROR_MSG(_message) \
    conn()->setErrorMessage(_message, __THIS_FILE__, __LINE__)
 
IMPLEMENT_CLASSINFO(SqQuery, SQL::Query)
 
SqQuery::SqQuery(SqConnection* pConnection)
    : Query(pConnection)
{
    Query::__placeholder = L'$';
    __stmt = NULL;
    __first = true;
 
    __fields = NULL;
    __params = NULL;
}
 
SqQuery::~SqQuery()
{
#ifdef __DCL_DEBUG
    if (!reset()) {
        char buf[256];
        size_t buflen = sizeof(buf) - 1;
        bool b = conn()->__getErrorMessage(buf, &buflen);
        buf[b ? buflen : 0] = '\0';
        __DCL_TRACE1(L"Warning! Query reset error! %hs\n", buf);
    }
#else
    (void)reset();
#endif
}
 
void SqQuery::__destroy()
{
    delete this;
}
 
bool SqQuery::reset()
{
    Query::__eof = true;
    Query::__affectedRows = -1;
 
    // clear fields
    if (__fields) {
        __DCL_ASSERT(Query::__fieldCount > 0);
        delete[] __fields;
        __fields = NULL;
        Query::__fieldCount = 0;
    }
 
    // clear binds
    if (__params) {
        __DCL_ASSERT(Query::__paramCount > 0);
        delete[] __params;
        __params = NULL;
        Query::__paramCount = 0;
    }
 
    bool r = true;
    if (__stmt) {
        int rc = sqlite3_finalize(__stmt);
        if (rc != SQLITE_OK) {
            __SET_ERROR_MSG(sqlite3_errmsg(conn()->connHandle()));
            r = false;
        }
    }
 
    return r;
}
 
bool SqQuery::__prepare(
    const char* _sql, size_t _sqllen, size_t _paramCount
)
{
    if(!reset())
        return false;
 
    sqlite3_stmt* stmt = NULL;
    const char* pzTail = NULL;
    int rc = sqlite3_prepare_v2(conn()->connHandle(),
        _sql, (int)_sqllen, &stmt, &pzTail);
    if (rc != SQLITE_OK) {
        __SET_ERROR_MSG(sqlite3_errmsg(conn()->connHandle()));
        return false;
    }
 
#if defined(__DCL_DEBUG) && __TRACE_THIS
    __DCL_TRACE1_N(L"bind_parameter_count [%d]\n",
        sqlite3_bind_parameter_count(stmt)
    );
    __DCL_TRACE1_N(L"column_count [%d]\n",
        sqlite3_column_count(stmt)
    );
#endif
 
    __DCL_ASSERT((__params == NULL)
        && (Query::__paramCount == 0)
    );
 
    Query::__paramCount = sqlite3_bind_parameter_count(stmt);
    if (Query::__paramCount) {
        __params = new SqParam[Query::__paramCount];
        if (__params == NULL) {
            __SET_ERROR(SQL::eOutOfMemory);
            return false;
        }
 
        for (size_t i = 0; i < Query::__paramCount; i++) {
            int number = (int) i + 1;
            const char* name = sqlite3_bind_parameter_name(stmt, number);
            if (!__params[i].init(this, number, name)) {
                return false;
            }
        }
    }
 
    __stmt = stmt;
    return true;
}
 
bool SqQuery::__execute()
{
    __DCL_ASSERT(__stmt != NULL);
 
    for (size_t i = 0; i < Query::__paramCount; i++) {
        if (!(__params[i].onBeforeExecute()))
            return false;
    }
 
    int rc = sqlite3_step(__stmt);
    switch (rc) {
        case SQLITE_ROW: {
            __DCL_ASSERT(
                (__fields == NULL)
                && (Query::__fieldCount == 0)
            );
            Query::__fieldCount = sqlite3_column_count(__stmt);
            __fields = new SqField[Query::__fieldCount];
            if (__fields == NULL) {
                __SET_ERROR(SQL::eOutOfMemory);
                return false;
            }
 
            for (size_t i = 0; i < Query::__fieldCount; i++) {
                if (!__fields[i].init(this, (int) i))
                    return false;
            }
            Query::__eof = false;
            __first = true;
            break;
        }
        case SQLITE_OK:
        case SQLITE_DONE: {
#if SQLITE_VERSION_NUMBER >= 3037000
            Query::__affectedRows = sqlite3_changes64(conn()->connHandle());
#else
            Query::__affectedRows = sqlite3_changes(conn()->connHandle());
#endif
            __DCL_TRACE2_N(L"__execute sqlite3_step[%d][%lld]\n",
                rc, Query::__affectedRows
            );
            break;
        }
        default: {
            __SET_ERROR_MSG(sqlite3_errmsg(conn()->connHandle()));
            return false;
        }
    }
 
    if (Query::__paramCount) {
        for (size_t i = 0; i < Query::__paramCount; i++) {
            if (!(__params[i].onAfterExecute()))
                return false;
        }
        rc = sqlite3_reset(__stmt);
        if (rc != SQLITE_OK) {
            __SET_ERROR_MSG(sqlite3_errmsg(conn()->connHandle()));
            return false;
        }
    }
 
    return true;
}
 
bool SqQuery::__fetch()
{
    if (__first) {
        __first = false;
    }
    else {
        int rc = sqlite3_step(__stmt);
        __DCL_TRACE1_N(L"sqlite3_step [%d]\n", rc);
        switch (rc) {
            case SQLITE_ROW:
                break;
            case SQLITE_DONE:
                Query::__eof = true;
                break;
            default: {
                __SET_ERROR_MSG(sqlite3_errmsg(conn()->connHandle()));
                return false;
            }
        }
    }
    return true;
}
 
bool SqQuery::__getField(size_t _index, SQL::Field** _fieldHandleOut)
{
    __DCL_ASSERT(Query::__fieldCount > 0);
    __DCL_ASSERT((0 <= _index) && (_index < Query::__fieldCount));
    *_fieldHandleOut = &__fields[_index];
    return true;
}
 
bool SqQuery::__getParam(size_t _index, SQL::Param** _paramHandleOut)
{
    __DCL_ASSERT(Query::__paramCount > 0);
    __DCL_ASSERT((0 <= _index) && (_index < Query::__paramCount));
    *_paramHandleOut = &__params[_index];
    return true;
}
 
__DCL_END_NAMESPACE