IFXConnection.cpp

Go to the documentation of this file.

#line 1 "ifxconnection.ec"
#include <dcl/Config.h>
 
#include <sqlhdr.h>
#include <sqliapi.h>
 
#include <sqlstype.h>
 
#include <ctype.h>      // isspace
#include <string.h>     // strlen, strncpy
 
#include <dcl/Object.h>
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#include <dcl/Numeric.h>
#include <dcl/Regex.h>
#include <dcl/SQLCore.h>
 
#include "IFXConnection.h"
#include "IFXQuery.h"
#include "IFXField.h" // for _getServerInfo
 
#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 char_t __THIS_FILE__[] = __T("dcl/sql/IFXConnection.ec");
 
__DCL_BEGIN_NAMESPACE
 
#define __SET_ERROR(_error) \
    setErrorHandle(_error, 0L, __THIS_FILE__, __LINE__)
#define __SET_ERROR_HANDLE(_SQLCODE) \
    setErrorHandle(SQL::eServerError, _SQLCODE, __THIS_FILE__, __LINE__)
#define __SET_ERROR_MSG(_message) \
    setErrorMessage(_message, __THIS_FILE__, __LINE__)
 
IMPLEMENT_CLASSINFO(IFXConnection, SQL::Connection)
 
IFXConnection::IFXConnection(const wchar_t* _serverTitle)
    : Connection(_serverTitle)
{
}
 
IFXConnection::~IFXConnection()
{
    if (!__connectionID.isEmpty()) {
        __DCL_TRACE0_N(L"Warning!! The connection was not closed\n");
        close();
    }
}
 
void IFXConnection::destroy()
{
    delete this;
}
 
/*
_conns에 가능한 프로퍼티
    USER,
    PASSWORD,
    SERVER, -- host name
    DATABASE
*/
 
void IFXConnection::reset()
{
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 79 "ifxconnection.ec"
#line 80 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 81 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL SET CONNECTION :connID;
 */
#line 83 "ifxconnection.ec"
  {
#line 83 "ifxconnection.ec"
  sqli_connect_set(0, connID, 0);
#line 83 "ifxconnection.ec"
  }
 
/*
 *  EXEC SQL BEGIN WORK;
 */
#line 85 "ifxconnection.ec"
  {
#line 85 "ifxconnection.ec"
  sqli_trans_begin2((mint)1);
#line 85 "ifxconnection.ec"
  }
    if (SQLCODE == 0) {
        Connection::__canTransact = true;
/*
 *      EXEC SQL ROLLBACK WORK;
 */
#line 88 "ifxconnection.ec"
  {
#line 88 "ifxconnection.ec"
  sqli_trans_rollback();
#line 88 "ifxconnection.ec"
  }
    }
    else {
        Connection::__canTransact = false;
    }
 
    __UNSET_STATE(stInTransaction);
}
 
bool IFXConnection::__open(const char* _conns, size_t _connlen)
{
    ListedByteStringToByteStringMap map;
    Connection::splitConnStr(_conns, _connlen, map);
 
    ByteString _SERVER = map["SERVER"];
    ByteString _USER = map["USER"];
    ByteString _PASSWORD = map["PASSWORD"];
    ByteString _DATABASE = map["DATABASE"];
    ByteString _TARGET;
 
    {
        ByteStringBuilder sb = _DATABASE;
        if (!_SERVER.isEmpty()) {
            sb.append("@").append(_SERVER);
        }
        _TARGET = sb.toByteString();
        __DCL_TRACE1_N(L"TARGET [%hs]\n", _TARGET.data());
    }
 
    ByteString connectionID = ByteString::format("conn_%zx", (size_t)this);
 
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 119 "ifxconnection.ec"
#line 120 "ifxconnection.ec"
  char *user = (_CONST char*) _USER.data();
  char *pass = (_CONST char*) _PASSWORD.data();
  char *target = (_CONST char*) _TARGET.data();
  char *connID = (_CONST char*) connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 124 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL CONNECT TO :target AS :connID
 *      USER :user USING :pass
 *      WITH CONCURRENT TRANSACTION;
 */
#line 126 "ifxconnection.ec"
  {
#line 128 "ifxconnection.ec"
  ifx_conn_t      *_sqiconn;
  _sqiconn = (ifx_conn_t *)ifx_alloc_conn_user(user, pass);
  sqli_connect_open(ESQLINTVERSION, 0, target, connID, _sqiconn, 1);
  ifx_free_conn_user(&_sqiconn);
#line 128 "ifxconnection.ec"
  }
 
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
    __connectionID = connectionID;
 
    if (!_DATABASE.isEmpty()) {
        reset();
    }
    return true;
}
 
bool IFXConnection::__close()
{
    if (__connectionID.isEmpty()) {
        __SET_ERROR(SQL::eNotConnected);
        return false;
    }
 
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 149 "ifxconnection.ec"
#line 150 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 151 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL DISCONNECT :connID;
 */
#line 153 "ifxconnection.ec"
  {
#line 153 "ifxconnection.ec"
  sqli_connect_close(0, connID, 0, 0);
#line 153 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
    __connectionID.clear();
    return true;
}
 
enum StmtType {
    StmtOther,
    StmtTransBegin,
    StmtTransEnd,
    StmtDatabase
};
 
typedef struct {
    StmtType type;
    const char* pattern;
} STMT_PATTERN;
 
static STMT_PATTERN sp[] = {
    { StmtTransBegin, "BEGIN"             },
    { StmtTransEnd,     "COMMIT|ROLLBACK"   },
    { StmtDatabase,     "DATABASE"          },
    { StmtOther,       NULL                }
};
 
static StmtType __GetStmtType(const char* _sql)
{
    for(size_t i = 0; sp[i].type != StmtOther; i++) {
        try {
            if (Regex::test(sp[i].pattern, _sql, true))
                return sp[i].type;
        }
        catch(Exception* e) {
            e->destroy();
        }
    }
    return StmtOther;
}
 
bool IFXConnection::__execute(const char* _sql, size_t _sqllen)
{
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 198 "ifxconnection.ec"
#line 199 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
  char *pszSqlStatement = (_CONST char*) _sql;
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 201 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL SET CONNECTION :connID;
 */
#line 203 "ifxconnection.ec"
  {
#line 203 "ifxconnection.ec"
  sqli_connect_set(0, connID, 0);
#line 203 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
/*
 *  EXEC SQL EXECUTE IMMEDIATE :pszSqlStatement;
 */
#line 209 "ifxconnection.ec"
  {
#line 209 "ifxconnection.ec"
  sqli_exec_immed(pszSqlStatement);
#line 209 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
    switch(__GetStmtType(_sql)) {
        case StmtTransBegin :
            __SET_STATE(stInTransaction);
            break;
        case StmtTransEnd :
            __UNSET_STATE(stInTransaction);
            break;
        case StmtDatabase :
            reset();
            break;
        case StmtOther :
        default :
            ;
    }
 
    return true;
}
 
bool IFXConnection::__startTrans()
{
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 235 "ifxconnection.ec"
#line 236 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 237 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL SET CONNECTION :connID;
 */
#line 239 "ifxconnection.ec"
  {
#line 239 "ifxconnection.ec"
  sqli_connect_set(0, connID, 0);
#line 239 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
/*
 *  EXEC SQL BEGIN WORK;
 */
#line 245 "ifxconnection.ec"
  {
#line 245 "ifxconnection.ec"
  sqli_trans_begin2((mint)1);
#line 245 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
    __SET_STATE(stInTransaction);
    return true;
}
 
bool IFXConnection::__commitTrans()
{
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 257 "ifxconnection.ec"
#line 258 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 259 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL SET CONNECTION :connID;
 */
#line 261 "ifxconnection.ec"
  {
#line 261 "ifxconnection.ec"
  sqli_connect_set(0, connID, 0);
#line 261 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
/*
 *  EXEC SQL COMMIT WORK;
 */
#line 267 "ifxconnection.ec"
  {
#line 267 "ifxconnection.ec"
  sqli_trans_commit();
#line 267 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
    __UNSET_STATE(stInTransaction);
    return true;
}
 
bool IFXConnection::__rollbackTrans()
{   
/*
 *  EXEC SQL BEGIN DECLARE SECTION;
 */
#line 279 "ifxconnection.ec"
#line 280 "ifxconnection.ec"
  char *connID = (_CONST char*) __connectionID.data();
/*
 *  EXEC SQL END DECLARE SECTION;
 */
#line 281 "ifxconnection.ec"
 
 
/*
 *  EXEC SQL SET CONNECTION :connID;
 */
#line 283 "ifxconnection.ec"
  {
#line 283 "ifxconnection.ec"
  sqli_connect_set(0, connID, 0);
#line 283 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
/*
 *  EXEC SQL ROLLBACK WORK;
 */
#line 289 "ifxconnection.ec"
  {
#line 289 "ifxconnection.ec"
  sqli_trans_rollback();
#line 289 "ifxconnection.ec"
  }
    if (SQLCODE < 0) {
        __SET_ERROR_HANDLE(SQLCODE);
        return false;
    }
 
    __UNSET_STATE(stInTransaction);
    return true;
}
 
bool IFXConnection::__createQueryInstance(SQL::Query** _queryHandleOut)
{
    __DCL_ASSERT(_queryHandleOut != NULL);
 
    SQL::Query* pNewQuery = new IFXQuery(this);
    if (!pNewQuery) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    *_queryHandleOut = pNewQuery;
    return true;
}
 
void IFXConnection::setErrorHandle(
    SQL::Error _error, long _SQLCODE,
    const wchar_t* _filename, int _line
)
{
    Connection::setErrorStatus(_error, _filename, _line);
    if (_SQLCODE == 0) {
        __lastErrorMessage.clear();
        return;
    }
 
    __DCL_ASSERT(_SQLCODE == sqlca.sqlcode);
 
    ByteStringBuilder sb;
    sb.format("SQLCODE(%d) ", _SQLCODE);
 
    char buf[1024];
    mint actual = 0;
    mint r = rgetlmsg(_SQLCODE, buf, sizeof(buf), &actual);
    if (!r) {
        while (actual > 0) {
            if (!isspace(buf[actual - 1]))
                break;
            actual--;
        }
        buf[actual] = '\0';
        sb.format(buf, sqlca.sqlerrm);
    }
    else {
        sb.format("(%d)", r);
        switch (r) {
        case -1227:
            sb.append("Message file not found");
            break;
        case -1228:
            sb.append("Message number not found in message file");
            break;
        case -1231:
            sb.append("Cannot seek within messge file");
            break;
        case -1232:
            sb.append("Message buffer too small");
            break;
        default:
            sb.append("Unkndown");
        }
    }
    __lastErrorMessage = sb.toByteString();
}
 
bool IFXConnection::__getErrorMessage(char* _buf, size_t* _buflen)
{
    __DCL_ASSERT(Connection::__errorCode == SQL::eServerError);
    if (__lastErrorMessage.length() < *_buflen) {
        *_buflen = __lastErrorMessage.length();
        *(_buf + *_buflen) = '\0';
    }
    strncpy(_buf, __lastErrorMessage.data(), *_buflen);
    return true;
}
 
bool IFXConnection::__getServerInfo(char* _buf, size_t* _buflen)
{
    IFXQuery* query = new IFXQuery(this);
    if (query == NULL) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    bool localTrans = false;
    if (!inState(SQL::Connection::stInTransaction)) {
        __startTrans();
        localTrans = true;
    }
 
    const char* _sql =
        "SELECT DBINFO('version', 'full')"
        "\n FROM systables WHERE tabid = 1";
    for ( ; ; ) {
        if (!query->prepare(_sql, ByteString::length(_sql), 0))
            break;
 
        if (!query->execute())
            break;
 
        if (!query->fetch())
            break;
 
        if (query->eof()) {
            __SET_ERROR(SQL::eNotAvailable);
            break;
        }
 
        IFXField* field = NULL;
        if (!query->__getField(0, (SQL::Field**)&field))
            break;
 
        size_t size = (size_t)-1;
        if (!field->__getDataSize(&size, false) || size == (size_t)-1) {
            __SET_ERROR(SQL::eNotAvailable);
            break;
        }
 
        if (!field->__getData(_buf, _buflen, SQL::typeText))
            break;
 
        // *_buflen에 '\0'이 포함되어 있다.
        *_buflen = ByteString::length(_buf, *_buflen);
 
        query->__destroy();
        query = NULL;
 
        if (localTrans) {
            __commitTrans();
        }
        return true;
    }
 
    if (query) {
        query->__destroy();
    }
 
    if (localTrans) {
        __rollbackTrans();
    }
    return false;
}       
 
__DCL_END_NAMESPACE
 
#line 441 "ifxconnection.ec"