IBQuery.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <stdlib.h>     // malloc, free
 
#include <ibase.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 <dcl/OutputStream.h>
#include <dcl/Charset.h>
 
#include "IBConnection.h"
#include "IBQuery.h"
#include "IBField.h"
#include "IBParam.h"
 
#ifndef SQLDA_CURRENT_VERSION
    #define SQLDA_CURRENT_VERSION SQLDA_VERSION1
#endif
 
#undef __THIS_FILE__
static const wchar_t __THIS_FILE__[] = __T("dcl/sql/IBQuery.cpp");
 
__DCL_BEGIN_NAMESPACE
 
IMPLEMENT_CLASSINFO(IBQuery, SQL::Query)
 
IBQuery::IBQuery(IBConnection* _connHandle)
    : Query(_connHandle)
{
    __stmtHandle = NULL;
    __inSQLDA = NULL;
    __outSQLDA = NULL;
    __outBuffer = NULL;
 
    __stmtType = _IB_STMT_TYPE_UNKNOWN;    // unknown
    __fields = NULL;
    __params = NULL;
}
 
IBQuery::~IBQuery()
{
#ifdef __DCL_DEBUG
    if (!reset()) {
        ByteString s;
        size_t n = 512;
        ByteBuffer* buf = ByteBuffer::create(n);
        bool b = conn()->__getErrorMessage(buf->data(), &n);
        if (b) {
            buf->__dataLength = n;
            s = buf;
        }
        buf->release();
 
        if (b) {
            __DCL_TRACE1(__T("Warning! %s\n"), s.data());
        }
        else {
            __DCL_TRACE0(__T("Warning! Query reset error\n"));
        }
    }
#else
    (void)reset();
#endif
}
 
void IBQuery::__destroy()
{
    delete this;
}
 
bool IBQuery::reset()
{
    Query::__eof = true;
    Query::__affectedRows = -1;
    __stmtType = _IB_STMT_TYPE_UNKNOWN;
 
    // 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;
    }
 
    if (__outBuffer) {
        free(__outBuffer);
        __outBuffer = NULL;
    }
 
    if (__outSQLDA) {
        free(__outSQLDA);
        __outSQLDA = NULL;
    }
 
    if (__inSQLDA) {
        free(__inSQLDA);
        __inSQLDA = NULL;
    }
 
    ISC_STATUS* statusVector = conn()->statusVector();
    bool r = true;
    if (__stmtHandle) {
        if (isc_dsql_free_statement(
                statusVector, 
                &__stmtHandle, 
                DSQL_drop)
                ) {
//          __DCL_ASSERT(STATUS_FAILED(statusVector));
            __SET_ERROR(SQL::eServerError);
            r = false;
        }
        __stmtHandle = NULL;
    }
    return r;
}
 
static size_t __TYPE_ALIGN(size_t _offset, int _sqltype)
{
    size_t size = 0;
    switch(_sqltype & ~1) {
    case SQL_SHORT :
        size = sizeof(short);
        break;
    case SQL_LONG :
        size = sizeof(long);
        break;
    case SQL_INT64 :
        size = sizeof(ISC_INT64);
        break;
    case SQL_FLOAT :
        size = sizeof(float);
        break;
    case SQL_DOUBLE :
        size = sizeof(double);
        break;
    case SQL_TYPE_DATE :
        size = sizeof(ISC_DATE);
        break;
    case SQL_TYPE_TIME :
        size = sizeof(ISC_TIME);
        break;
    case SQL_TIMESTAMP :
        size = sizeof(ISC_TIMESTAMP);
        break;
    case SQL_TEXT :
        size = sizeof(char);
        break;
    case SQL_VARYING :
        size = sizeof(short);
        break;
    case SQL_BLOB :
        size = sizeof(ISC_QUAD);
        break;
    default :
        __DCL_TRACE3(__T("_offset[%d] _sqltype[%d] [%d]\n"), _offset, _sqltype, _sqltype & ~1);
        __DCL_ASSERT(false);
    }
 
    return (_offset + size - 1) & ~(size - 1);
}
 
bool IBQuery::initFields()
{
    __DCL_ASSERT((__fields == NULL) 
            && (__outBuffer == NULL)
            && (Query::__fieldCount == 0)
            );
 
    ISC_STATUS* statusVector = conn()->statusVector();
 
    __outSQLDA = (XSQLDA*)malloc(XSQLDA_LENGTH(1));
    __outSQLDA->version = SQLDA_CURRENT_VERSION;
    __outSQLDA->sqln = 1;
 
    if(isc_dsql_describe(
            statusVector, 
            &__stmtHandle, 
            SQLDA_CURRENT_VERSION,
            __outSQLDA)
            ) {
//      __DCL_ASSERT(STATUS_FAILED(statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    if (__outSQLDA->sqld == 0) {
        // 필드가 없다.
        return true;
    }
 
    if (__outSQLDA->sqln < __outSQLDA->sqld) {
        __outSQLDA = (XSQLDA*)realloc(__outSQLDA, XSQLDA_LENGTH(__outSQLDA->sqld));
        __outSQLDA->version = SQLDA_CURRENT_VERSION;
        __outSQLDA->sqln = __outSQLDA->sqld;
    
        if (isc_dsql_describe(
                statusVector, 
                &__stmtHandle, 
                SQLDA_CURRENT_VERSION,
                __outSQLDA)
                ) {
//          __DCL_ASSERT(STATUS_FAILED(statusVector));
            __SET_ERROR(SQL::eServerError);
            return false;
        }
    }
 
    Query::__fieldCount = __outSQLDA->sqld;
    __fields = new IBField[Query::__fieldCount];
    if (__fields == NULL) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    size_t offset = 0;
    XSQLVAR* sqlvar = __outSQLDA->sqlvar;
    for(size_t i = 0; i < Query::__fieldCount; i++, sqlvar++) {
        offset = __TYPE_ALIGN(offset, sqlvar->sqltype);
        switch(sqlvar->sqltype & ~1) {
            case SQL_TEXT :
                offset += sqlvar->sqllen + 1;
                break;
            case SQL_VARYING :
                offset += sqlvar->sqllen + sizeof(short) + 1;
                break;
            default :
                offset += sqlvar->sqllen;
        }
    }
 
    if (offset > 0) {
        __outBuffer = (char*)malloc(offset);
        __DCL_ASSERT((size_t)__outBuffer % sizeof(void*) == 0);
        if (!__outBuffer) {
            __SET_ERROR(SQL::eOutOfMemory);
            return false;
        }
    }
 
    sqlvar = __outSQLDA->sqlvar;
    offset = 0;
    for(size_t i = 0; i < Query::__fieldCount; i++, sqlvar++) {
        offset = __TYPE_ALIGN(offset, sqlvar->sqltype);
        sqlvar->sqldata = (char*)__outBuffer + offset;
        switch(sqlvar->sqltype & ~1) {
            case SQL_TEXT :
                offset += sqlvar->sqllen + 1;
                break;
            case SQL_VARYING :
                offset += sqlvar->sqllen + sizeof(short) + 1;
                break;
            default :
                offset += sqlvar->sqllen;
        }
 
        if (!__fields[i].init(this, sqlvar))
            return false;
    }
 
    return true;
}
 
bool IBQuery::initParams(size_t _paramCount)
{
    ISC_STATUS* statusVector = conn()->statusVector();
 
    __inSQLDA = (XSQLDA*)malloc(XSQLDA_LENGTH(1));
    __inSQLDA->version = SQLDA_CURRENT_VERSION;
    __inSQLDA->sqln = 1;
 
    if (isc_dsql_describe_bind(
            statusVector, 
            &__stmtHandle, 
            SQLDA_CURRENT_VERSION, 
            __inSQLDA)
            ) {
//      __DCL_ASSERT(STATUS_FAILED(statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    if (__inSQLDA->sqld == 0) {
        // 파라미터가 없다. bind 할 필요가 없다.
        return true;
    }
 
    if (__inSQLDA->sqln < __inSQLDA->sqld) {
        __inSQLDA = (XSQLDA*)realloc(__inSQLDA, XSQLDA_LENGTH(__inSQLDA->sqld));
        __inSQLDA->version = SQLDA_CURRENT_VERSION;
        __inSQLDA->sqln = __inSQLDA->sqld;
    
        if (isc_dsql_describe_bind(
                statusVector, 
                &__stmtHandle, 
                SQLDA_CURRENT_VERSION, 
                __inSQLDA)
                ) {
//          __DCL_ASSERT(STATUS_FAILED(m_statusVector));
            __SET_ERROR(SQL::eServerError);
            return false;
        }
    }
    __DCL_ASSERT(__inSQLDA->sqld == _paramCount);
 
    Query::__paramCount = _paramCount;
    __params = new IBParam[Query::__paramCount];
    if (__params == NULL) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    XSQLVAR* sqlvar = __inSQLDA->sqlvar;
    for(size_t i = 0; i < Query::__paramCount; i++, sqlvar++) {
        if (!__params[i].init(this, sqlvar)) {
            return false;
        }
    }
 
    return true;
}
 
static char stmt_info_item[] = { isc_info_sql_stmt_type };
static char count_info_item[] = { isc_info_sql_records };
 
bool IBQuery::__prepare(const char* _sql, size_t _sqllen,
        size_t _paramCount)
{
    if(!reset())
        return false;
 
    IBConnection* connHandle = conn();
    ISC_STATUS* statusVector = connHandle->statusVector();
 
    if (isc_dsql_allocate_statement(
            statusVector, 
            connHandle->dbHandlePtr(), 
            &__stmtHandle)
            ) {
//      __DCL_ASSERT(STATUS_FAILED(m_statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    ISC_STATUS rs = isc_dsql_prepare(
            statusVector, 
            connHandle->trHandlePtr(), 
            &__stmtHandle, 
            0, 
            (_CONST char*)_sql,
            connHandle->dialect(), 
            NULL
            );
    if (rs) {
//      __DCL_ASSERT(STATUS_FAILED(m_statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
 
    int nLength;
    char res_buffer[8]; 
 
    if (isc_dsql_sql_info(
            statusVector,
            &__stmtHandle,
            sizeof(stmt_info_item),
            stmt_info_item,
            sizeof(res_buffer),
            res_buffer)
            ) {
//      __DCL_ASSERT(STATUS_FAILED(m_statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    if (res_buffer[0] != isc_info_sql_stmt_type) {
        __SET_ERROR(SQL::eServerError);
        return false;
    }
    nLength = isc_vax_integer(&res_buffer[1], 2);
    __stmtType = isc_vax_integer(&res_buffer[3], nLength);
 
    // SELECT, EXECUTE PROCEDURE 는 Row Sets을 반환한다.
    if ((__stmtType == isc_info_sql_stmt_select)
            || (__stmtType == isc_info_sql_stmt_exec_procedure)
            || (__stmtType == isc_info_sql_stmt_select_for_upd)) {
        if (!initFields())
            return false;
    }
 
    if (_paramCount > 0) {
        if (!initParams(_paramCount))
            return false;
    }
 
    return true;
//  rs = isc_dsql_set_cursor_name(statusVector, &__stmtHandle, "CURSOR1", 0);
}
 
bool IBQuery::__execute()
{
    IBConnection* connHandle = conn();
    ISC_STATUS* statusVector = connHandle->statusVector();
 
    if (__stmtType == isc_info_sql_stmt_exec_procedure) {
        if (isc_dsql_execute2(
                statusVector,
                connHandle->trHandlePtr(),
                &__stmtHandle,
                connHandle->dialect(),
                __inSQLDA,
                __outSQLDA)) {
            __SET_ERROR(SQL::eServerError);
            return false;
        }
        return true;
    }
 
    if (isc_dsql_execute(
            statusVector, 
            connHandle->trHandlePtr(),
            &__stmtHandle, 
            connHandle->dialect(),
            __inSQLDA)) {
//      __DCL_ASSERT(STATUS_FAILED(statusVector));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    for(size_t i = 0; i < Query::__paramCount; i++)
        __params[i].onAfterExecute();
 
    // INSERT, DELETE, UPDATE statement의 실행 후
    // affected records를 구한다.
 
    unsigned char count_type = 0;
    switch(__stmtType) {
        case isc_info_sql_stmt_update: {
            count_type = isc_info_req_update_count;
            break;
        }
        case isc_info_sql_stmt_delete: {
            count_type = isc_info_req_delete_count;
            break;
        }
        case isc_info_sql_stmt_insert: {
            count_type = isc_info_req_insert_count;
            break;
        }
        case isc_info_sql_stmt_select:
        case isc_info_sql_stmt_select_for_upd: {
            Query::__eof = false;
            break;
        }
        case isc_info_sql_stmt_exec_procedure: {
            for (size_t i = 0; i < Query::__fieldCount; i++) {
                if (!__fields[i].onAfterFetch())
                    return false;
            }
            break;
        }
    }
    
    if (count_type) {
        unsigned char count_is = 0;
        char count_buffer[33];
        unsigned short length;
 
        if (isc_dsql_sql_info(
                statusVector, 
                &__stmtHandle, 
                sizeof(count_info_item),
                count_info_item,
                sizeof(count_buffer),
                count_buffer)) {
//          __DCL_ASSERT(STATUS_FAILED(statusVector));
            __SET_ERROR(SQL::eServerError);
            return false;
        }
 
        for(char* pch = count_buffer + 3; *pch != isc_info_end; ) {
            count_is = *pch++;
            length = (unsigned short)isc_vax_integer(pch, 2);
            pch += 2;
            Query::__affectedRows = isc_vax_integer(pch, length);
            pch += length;
            if (count_is == count_type)
                break;
        }
    }
 
    return true;
}
 
bool IBQuery::__fetch()
{
    __DCL_ASSERT(!eof());
    __DCL_ASSERT(__stmtType == isc_info_sql_stmt_select);
 
    ISC_STATUS rs = isc_dsql_fetch(conn()->statusVector(), 
                &__stmtHandle, 
                SQLDA_CURRENT_VERSION, 
                __outSQLDA);
    if (rs == 0) {
        for(size_t i = 0; i < Query::__fieldCount; i++) {
            if (!__fields[i].onAfterFetch())
                return false;
        }
        return true;
    }
    else if (rs == 100) {
        Query::__eof = true;
        return true;
    }
 
    __DCL_TRACE1(L"[%zd]\n", (int64_t)rs);
    __SET_ERROR(SQL::eServerError);
    return false;
}
  
bool IBQuery::__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 IBQuery::__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