OciQuery.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <oci.h>
 
#include <stdlib.h>     // abs
 
#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 "OciConnection.h"
#include "OciQuery.h"
#include "OciField.h"
#include "OciParam.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
#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)
#endif
 
#undef __THIS_FILE__
static const wchar_t __THIS_FILE__[] = __T("dcl/sql/OciQuery.cpp");
 
__DCL_BEGIN_NAMESPACE
 
#undef __SET_ERROR
#define __SET_ERROR(_error, status, pError) \
    conn()->setErrorStatus(_error, status, pError, \
            true, __THIS_FILE__, __LINE__)
 
#define __SET_ERROR_NORESET(_error, status, pError) \
    conn()->setErrorStatus(_error, status, pError, \
            false, __THIS_FILE__, __LINE__)
 
IMPLEMENT_CLASSINFO(OciQuery, SQL::Query)
 
OciQuery::OciQuery(OciConnection* _connHandle)
    : Query(_connHandle)
{
    __stmt = NULL;
    __stmtType = _OCI_STMT_TYPE_UNKNOWN;
 
    __fields = NULL;
    __params = NULL;
    Query::__placeholder = L':';
}
 
OciQuery::~OciQuery()
{
#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 OciQuery::__destroy()
{
    delete this;
}
 
bool OciQuery::reset()
{
    Query::__eof = true;
    Query::__affectedRows = -1;
    __stmtType = _OCI_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 (__stmt) {
        sword status = ::OCIStmtRelease(
                __stmt,
                conn()->errorHandle(),
                NULL, 0,
                OCI_DEFAULT
                );
        if (status != OCI_SUCCESS) {
            __SET_ERROR(SQL::eServerError, status, NULL);
            return false;
        }
        __stmt = NULL;
    }
 
    return true;
}
 
bool OciQuery::initParams(size_t _paramCount)
{
#if defined(__DCL_DEBUG) && 1
    ub4 size = 0;
    ub4 startloc = 1;
    sb4 found = 0;
    text* bvnp; ub1 bvnl;
    text* invp; ub1 inpl;
    ub1 dupl; OCIBind* hndl;
 
    sword status = ::OCIStmtGetBindInfo(
            __stmt,
            conn()->errorHandle(),
            size, startloc, &found,
            &bvnp, &bvnl, &invp, &inpl,
            &dupl, &hndl);
    __DCL_TRACE2_N(L"[%d][%d]\n", status, found);
    if (status != OCI_SUCCESS) {
        if (status == OCI_NO_DATA) {
            return true;
        }
        // 그렇다면 에러다.
        __SET_ERROR(SQL::eServerError, status, conn()->errorHandle());
        return false;
    }
    __DCL_ASSERT(abs(found) == _paramCount);
#endif
 
    Query::__paramCount = _paramCount;
    __params = new OciParam[Query::__paramCount];
    if (__params == NULL) {
        __SET_ERROR(SQL::eOutOfMemory, 0, NULL);
        return false;
    }
 
    for(size_t i = 0; i < Query::__paramCount; i++) {
        if (!__params[i].init(this, i + 1))
            return false;
    }
 
    return true;
}
 
bool OciQuery::__prepare(const char* _sql, size_t _sqllen,
        size_t _paramCount)
{
    OciConnection* pConnection = conn();
    OCIError* pError = pConnection->errorHandle();
 
    if (!reset())
        return false;
 
    sword status = OCI_SUCCESS;
 
    status = ::OCIStmtPrepare2(
            pConnection->svcctxHandle(),
            &__stmt,
            pError,
            (const text* )_sql,
            _sqllen,
            NULL, 0,
            OCI_NTV_SYNTAX,
            OCI_DEFAULT);
    if (status != OCI_SUCCESS) {
        __SET_ERROR(SQL::eServerError, status, pError);
        return false;
    }
 
    status = ::OCIAttrGet(__stmt, 
            OCI_HTYPE_STMT, 
            &__stmtType,
            NULL, 
            OCI_ATTR_STMT_TYPE, 
            pError);
    if (status != OCI_SUCCESS) {
        __SET_ERROR(SQL::eServerError, status, pError);
        return false;
    }
 
    if (_paramCount > 0) {
        if (!initParams(_paramCount))
            return false;
    }
 
    return true;
}
 
bool OciQuery::initFields()
{
    OciConnection* pConnection = conn();
    OCIError* pError = pConnection->errorHandle();
    sword status = OCI_SUCCESS;
 
    ub4 nFieldCount = 0;
    status = ::OCIAttrGet(__stmt, 
            OCI_HTYPE_STMT, 
            &nFieldCount, 
            NULL, 
            OCI_ATTR_PARAM_COUNT, 
            pError);
    if (status != OCI_SUCCESS) {
        __SET_ERROR(SQL::eServerError, status, pError);
        return false;
    }
 
    Query::__fieldCount = nFieldCount;
    __fields = new OciField[Query::__fieldCount];
    if (__fields == NULL) {
        __SET_ERROR(SQL::eOutOfMemory, 0, NULL);
        return false;
    }
 
    for(size_t i = 0; i < Query::__fieldCount; i++) {
        if (!__fields[i].init(this, i + 1))
            return false;
    }
 
    return true;
}
 
bool OciQuery::__execute()
{
    for(size_t i = 0; i < Query::__paramCount; i++) {
        if (!__params[i].doBind())
            return false;
    }
 
    OciConnection* pConnection = conn();
    OCIError*   pError = pConnection->errorHandle();
 
    sword status = OCI_SUCCESS;
    status = ::OCIStmtExecute(
            pConnection->svcctxHandle(),
            __stmt,
            pError,
            (__stmtType == OCI_STMT_SELECT) ? 0 : 1,
            0,
            NULL,
            NULL,
            OCI_DEFAULT);
    if (status != OCI_SUCCESS) {
        __SET_ERROR_NORESET(SQL::eServerError, status, pError);
        return false;
    }
 
    // OCI의 경우 stmt execute 후 fields를 결정할 수 있다.
    if (__stmtType == OCI_STMT_SELECT) {
        if (!__fields) {
            if (!initFields()) 
                return false;
        }
 
        Query::__eof = false;
    }
 
    for(size_t i = 0; i < Query::__paramCount; i++) {
        if (!__params[i].onAfterExecute())
            return false;
    }
 
/*
    for(size_t i = 0; i < Query::__paramCount; i++)
    {
        if (__params[i]._dataType() == SQLT_BLOB)
            if(!__params[i].writeBlob())
                return false;
    }
*/
    if (__stmtType == OCI_STMT_UPDATE
        || __stmtType == OCI_STMT_INSERT
        || __stmtType == OCI_STMT_DELETE) {
 
        ub8 nAffectedRows = 0;
        status = ::OCIAttrGet(__stmt,
                OCI_HTYPE_STMT, 
                &nAffectedRows,
                NULL, 
                OCI_ATTR_UB8_ROW_COUNT,
                pError);
        if (status != OCI_SUCCESS) {
            __SET_ERROR(SQL::eServerError, status, pError);
            return false;
        }
        Query::__affectedRows = nAffectedRows;
    }
    return true;
}
bool OciQuery::__fetch()
{
    __DCL_ASSERT(!eof());
    OCIError* pError = conn()->errorHandle();
    sword status = ::OCIStmtFetch2(
            __stmt,
            pError,
            1,
            OCI_FETCH_NEXT, 0,
            OCI_DEFAULT);
    if (status == OCI_SUCCESS) {
        for(size_t i = 0; i < Query::__fieldCount; i++) {
            if (!__fields[i].onAfterFetch())
                return false;
        }
        return true;
    }
    else if (status == OCI_NO_DATA) {
        Query::__eof = true;
        return true;
    }
 
    __SET_ERROR_NORESET(SQL::eServerError, status, pError);
    return false;
}
 
bool OciQuery::__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 OciQuery::__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