MyConnection.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <string.h>     // strlen, strncpy
#include <stdlib.h>     // strtol, strtoul, strtoll, strtoull
 
#include <dcl/Object.h>
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#include <dcl/Regex.h>
#include <dcl/ListedHashMap.h>
#include <dcl/SQLCore.h>
 
#include "MyConnection.h"
#include "MyQuery.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
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const wchar_t __THIS_FILE__[] = __T("dcl/sql/MyConnection.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
IMPLEMENT_CLASSINFO(MyConnection, SQL::Connection)
 
MyConnection::MyConnection(const wchar_t* _serverTitle)
    : Connection(_serverTitle)
{
    Connection::__canTransact = true;
    __mysql = NULL;
    __storeResult = false;
}
 
MyConnection::~MyConnection()
{
    if (__mysql) {
        __DCL_TRACE0_N(L"Warning!! The connection was not closed\n");
        close();
    }
}
 
void MyConnection::destroy()
{
    delete this;
}
 
/* 
pszConnectionByteString에 가능한 property
    for mysql_real_connect
        USER,
        PASSWORD,
        SERVER, -- host name
        DATABASE,
        PORT,       -- tcp port
        UNIX_SOCKET,
 
    for mysql_options
        APPLICATION
 
    driver options
        STORE_RESULT
 
*/
 
bool MyConnection::__open(const char* _pszConnString, size_t _n)
{
    if ((__mysql = mysql_init(NULL)) == NULL) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    ListedByteStringToByteStringMap map;
    Connection::splitConnectionString(_pszConnString, _n, map);
 
    ByteString strAppName = map["APPLICATION"];
 
    if (!strAppName.isEmpty()) {
        mysql_options(connHandle(), MYSQL_READ_DEFAULT_GROUP, strAppName.data());
    }
 
    ByteString strUser = map["USER"];
    ByteString strPassword = map["PASSWORD"];
    ByteString strHost = map["SERVER"];
    ByteString strDb = map["DATABASE"];
    ByteString strPort = map["PORT"];
    ByteString strUnixSocket = map["UNIX_SOCKET"];
    __storeResult = map.find("STORE_RESULT") != map.end();
 
    unsigned int port = 0;
    if (!strPort.isEmpty()) {
        char* endptr;
        unsigned long n = strtoul(strPort.data(), &endptr, 10);
        if (n == ULONG_MAX) {
            __SET_ERROR_MSG("연결 문자열이 잘못되었습니다. PORT=" + strPort);
            return false;
        }
        port = (unsigned int) n;
    }
 
    if (mysql_real_connect(
            connHandle(),
            strHost.isEmpty()? NULL : strHost.data(),
            strUser.isEmpty()? NULL : strUser.data(),
            strPassword.isEmpty()? NULL : strPassword.data(),
            strDb.isEmpty() ? NULL : strDb.data(),
            port,
            strUnixSocket.isEmpty() ? NULL : strUnixSocket.data(),
            0
            ) == NULL) {
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    if (mysql_set_character_set(connHandle(), "utf8")) {
        __SET_ERROR_MSG(ByteString::format("(%u) %hs",
            mysql_errno(connHandle()),
            mysql_error(connHandle())));
        return false;
    }
 
    if (mysql_autocommit(connHandle(), 0)) {
        __SET_ERROR_MSG(ByteString::format("(%u) %hs",
            mysql_errno(connHandle()),
            mysql_error(connHandle())));
        return false;
    }
 
    return true;
}
 
bool MyConnection::__close()
{
    if (!__mysql) {
        __SET_ERROR(SQL::eNotConnected);
        return false;
    }
 
    mysql_close(__mysql);
    __mysql = NULL;
 
    return true;
}
 
enum StmtType {
    StmtOther,
    StmtTransBegin,
    StmtTransEnd
};
 
typedef struct {
    StmtType type;
    const char* pattern;
} STMT_PATTERN;
 
static STMT_PATTERN sp[] = {
    {   StmtTransBegin,       "START[[:space:]]+TRANSACTION|BEGIN"    },
    {   StmtTransEnd,       "COMMIT|ROLLBACK"   },
    {   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* p) {
            p->destroy();
        }
    }
    return StmtOther;
}
 
bool MyConnection::__execute(const char* _sql, size_t _n)
{
    if (mysql_real_query(connHandle(), _sql, (unsigned long) _n)) {
        __SET_ERROR_MSG(ByteString::format("(%u) %hs",
            mysql_errno(connHandle()),
            mysql_error(connHandle())));
    }
 
    switch (__GetStmtType(_sql)) {
    case StmtTransBegin:
        __SET_STATE(stInTransaction);
        break;
    case StmtTransEnd:
        __UNSET_STATE(stInTransaction);
        break;
    case StmtOther:
    default:
        ;
    }
 
    return true;
}
 
bool MyConnection::__startTrans()
{
    return __execute("START TRANSACTION", 17);
}
 
bool MyConnection::__commitTrans()
{
    if (mysql_commit(connHandle())) {
        __SET_ERROR_MSG(ByteString::format("(%u) %hs",
            mysql_errno(connHandle()),
            mysql_error(connHandle())));
    }
 
    return true;
}
 
bool MyConnection::__rollbackTrans()
{   
    if (mysql_rollback(connHandle())) {
        __SET_ERROR_MSG(ByteString::format("(%u) %hs",
            mysql_errno(connHandle()),
            mysql_error(connHandle())));
    }
 
    return true;
}
 
bool MyConnection::__createQueryInstance(SQL::Query** _ppQuery)
{
    __DCL_ASSERT(_ppQuery != NULL);
 
    MyQuery* pNewQuery = new MyQuery(this);
    if (!pNewQuery) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
    
    *_ppQuery = pNewQuery;
    return true;
}
 
bool MyConnection::__getErrorMessage(char* _buffer, size_t* _size)
{
    if (__lastErrorMessage.length() < *_size)
        *_size = __lastErrorMessage.length();
 
    strncpy(_buffer, __lastErrorMessage.data(), *_size);
 
    return true;
}
 
bool MyConnection::__getServerInfo(char* _buffer, size_t* _size)
{
    const char* psz = mysql_get_server_info(connHandle());
    if (!psz)
        return false;
 
    size_t nLength = strlen(psz);
 
    if (nLength < *_size)
        *_size = nLength;
 
    strncpy(_buffer, psz, *_size);
 
    return true;
}       
 
__DCL_END_NAMESPACE