ODBCConnection Class Reference

#include <ODBCConnection.h>

Inheritance diagram for ODBCConnection:

Public Member Functions
SQLHDBC	connHandle () const
bool	IS_MS_SS () const
void	setErrorHandle (SQLRETURN _rc, SQLSMALLINT _htype, SQLHANDLE _handle, const wchar_t *_filename, int _line)
void	setErrorMessage (const ByteString &_message, const wchar_t *_filename, int _line)
	ODBCConnection (const wchar_t *_serverTitle)
virtual void	destroy ()
virtual	~ODBCConnection ()
virtual bool	__open (const char *_cons, size_t _conslen)
virtual bool	__close ()
virtual bool	__execute (const char *_sql, size_t _sqllen)
virtual bool	__startTrans ()
virtual bool	__commitTrans ()
virtual bool	__rollbackTrans ()
virtual bool	__createQueryInstance (SQL::Query **_queryHandleOut)
virtual bool	__getErrorMessage (char *_buf, size_t *_buflen)
virtual bool	__getServerInfo (char *_buf, size_t *_buflen)
Public Member Functions inherited from SQL::Connection
long	refCount () const
long	addRef ()
long	release ()
bool	open (const char *_conn, size_t _connlen)
bool	close ()
bool	execute (const char *_sql, size_t _sqllen)
bool	startTrans ()
bool	commitTrans ()
bool	rollbackTrans ()
bool	getErrorMessage (char *_buf, size_t *_buflen)
bool	getServerInfo (char *_buf, size_t *_buflen)
bool	createQueryInstance (Query **_queryHandle)
bool	destroyQueryInstance (Query *_queryHandle)
Error	errorCode () const
bool	canTransact () const
const wchar_t *	serverTitle () const
bool	inState (unsigned int _state) const
Public Member Functions inherited from Object
virtual String	toString () const
String	className () const
bool	isInstanceOf (const std::type_info &typeinfo) const
virtual const std::type_info &	typeInfo () const

Additional Inherited Members
Public Types inherited from SQL::Connection
enum	State { stClosed = 0x0001 , stOpenned = 0x0002 , stInTransaction = 0x0004 }
Static Public Member Functions inherited from SQL::Connection
static size_t	splitConnStr (const char *_connstr, size_t _strlen, ListedByteStringToByteStringMap &_results)
Public Attributes inherited from SQL::Connection
_PROTECTED	: void setErrorStatus( Error _errorCode
_PROTECTED const wchar_t *	_filename
_PROTECTED const wchar_t int	_line
Protected Member Functions inherited from SQL::Connection
	Connection (const wchar_t *_serverTitle)
virtual	~Connection ()
Protected Member Functions inherited from Object
virtual	~Object ()
	Object ()
Protected Attributes inherited from SQL::Connection
bool	__canTransact
const wchar_t *	__serverTitle
Error	__errorCode
const wchar_t *	__errorFileName
int	__errorLine
unsigned int	__states

Detailed Description

Definition at line 8 of file ODBCConnection.h.

Constructor & Destructor Documentation

ODBCConnection()

ODBCConnection::ODBCConnection

(

const wchar_t *

_serverTitle

)

~ODBCConnection()

ODBCConnection::~ODBCConnection ( )

virtual

Definition at line 67 of file ODBCConnection.cpp.

{
    if (__hdbc) {
        __DCL_TRACE0_N(L"Warning!! The connection was not closed\n");
        close();
    }
 
    if (__henv) {
        SQLRETURN rc_ = SQLFreeHandle(SQL_HANDLE_ENV, __henv);
        if (rc_ != SQL_SUCCESS) {
            __DCL_TRACE1_N(L"SQLFreeHandle(ENV) Fail! [%d]\n", rc_);
        }
        __henv = NULL;
    }
}

Member Function Documentation

__close()

bool ODBCConnection::__close ( )

virtual

Implements SQL::Connection.

Definition at line 174 of file ODBCConnection.cpp.

{
    if (!__hdbc) {
        __SET_ERROR(SQL::eNotConnected);
        return false;
    }
 
    SQLRETURN rc = SQL_SUCCESS;
    rc = SQLDisconnect(__hdbc);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
    }
 
    SQLRETURN rc_ = SQLFreeHandle(SQL_HANDLE_DBC, __hdbc);
    if (rc_ != SQL_SUCCESS) {
        __DCL_TRACE1_N(L"SQLFreeHandle(DBC) Fail! [%d]\n", rc_);
    }
    __hdbc = NULL;
 
    return rc == SQL_SUCCESS;
}

__commitTrans()

bool ODBCConnection::__commitTrans ( )

virtual

Implements SQL::Connection.

Definition at line 231 of file ODBCConnection.cpp.

{
    __UNSET_STATE(Connection::stInTransaction);
    SQLRETURN rc = SQLEndTran(SQL_HANDLE_DBC, __hdbc, SQL_COMMIT);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
 
    return true;
}

__createQueryInstance()

bool ODBCConnection::__createQueryInstance ( SQL::Query ** _queryHandleOut )

virtual

Implements SQL::Connection.

Definition at line 255 of file ODBCConnection.cpp.

{
    __DCL_ASSERT(_ppQuery != NULL);
 
    ODBCQuery* pNewQuery = new ODBCQuery(this);
    if (!pNewQuery) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    *_ppQuery = pNewQuery;
    return true;
}

__execute()

bool ODBCConnection::__execute ( const char * _sql, size_t _sqllen )

virtual

Implements SQL::Connection.

Definition at line 196 of file ODBCConnection.cpp.

{
    SQLHSTMT hstmt = NULL;
    SQLRETURN rc = SQLAllocHandle(SQL_HANDLE_STMT, __hdbc, &hstmt);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
 
    rc = SQLExecDirectA(hstmt, (SQLCHAR*) _sql, (SQLINTEGER)_sqllen);
    if (!SQL_SUCCEEDED(rc)) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_STMT, hstmt);
    }
    __SET_INFO_HANDLE(rc, SQL_HANDLE_STMT, hstmt);
 
    SQLRETURN rc_ = SQLFreeStmt(hstmt, SQL_CLOSE);
    if (rc_ != SQL_SUCCESS) {
        __DCL_TRACE1_N(L"SQLFreeStmt(STMT) Fail! [%d]\n", rc_);
    }
 
    rc_ = SQLFreeHandle(SQL_HANDLE_STMT, hstmt);
    if (rc_ != SQL_SUCCESS) {
        __DCL_TRACE1_N(L"SQLFreeHandle(STMT) Fail! [%d]\n", rc_);
    }
    hstmt = NULL;
 
    return rc == SQL_SUCCESS;
}

__getErrorMessage()

bool ODBCConnection::__getErrorMessage ( char * _buf, size_t * _buflen )

virtual

Implements SQL::Connection.

Definition at line 269 of file ODBCConnection.cpp.

{
    if (__lastErrorMessage.length() < *_buflen) {
        *_buflen = __lastErrorMessage.length();
        *(_buf + *_buflen) = '\0';
    }
    strncpy(_buf, __lastErrorMessage.data(), *_buflen);
    __lastErrorMessage.clear();
    return true;
}

__getServerInfo()

bool ODBCConnection::__getServerInfo ( char * _buf, size_t * _buflen )

virtual

Implements SQL::Connection.

Definition at line 280 of file ODBCConnection.cpp.

{
#if 0 && defined(__WINNT__)
    StringBuilder sb;
    WCHAR buf[1024];
    SQLSMALLINT length;
 
    SQLRETURN rc = SQLGetInfoW(__hdbc, SQL_DBMS_NAME, buf, __countof(buf, WCHAR) , &length);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
    sb.append(buf, length / sizeof(WCHAR));
 
    rc = SQLGetInfoW(__hdbc, SQL_DBMS_VER, buf, __countof(buf, WCHAR) , &length);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
    sb.append(L' ');
    sb.append(buf, length / sizeof(WCHAR));
 
    ByteString s = UTF8Encoder::encode(sb.toString());
    if (s.length() < *_buflen) {
        *_buflen = s.length();
    }
    strncpy(_buf, s.data(), *_buflen);
    return true;
#else
    SQLSMALLINT size = (SQLSMALLINT) __MIN(*_buflen, SHRT_MAX);
    SQLSMALLINT offset = 0;
    SQLSMALLINT length;
    SQLRETURN rc = SQL_SUCCESS;
 
    rc = SQLGetInfoA(__hdbc, SQL_DBMS_NAME,
        _buf + offset, size - offset, &length);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
 
    if ((offset + length) >= size) {
        *_buflen = size;
        return true;
    }
 
    offset += length;
    *(_buf + offset) = ' ';
    offset++;
 
    rc = SQLGetInfoA(__hdbc, SQL_DBMS_VER,
        _buf + offset, size - offset, &length);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
 
    if ((offset + length) >= size) {
        *_buflen = size;
        return true;
    }
 
    *_buflen = offset + length;
    return true;
#endif
}

__open()

bool ODBCConnection::__open ( const char * _cons, size_t _conslen )

virtual

Implements SQL::Connection.

Definition at line 88 of file ODBCConnection.cpp.

{
    SQLRETURN rc = SQL_SUCCESS;
    if (!__henv) {
        rc = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &__henv);
        if (rc != SQL_SUCCESS) {
            __SET_ERROR_MSG(
                ByteString::format("Unable to allocate an environment handle [%d]",
                rc));
            return false;
        }
 
        rc = SQLSetEnvAttr(__henv, SQL_ATTR_ODBC_VERSION, (SQLPOINTER)SQL_OV_ODBC3, 0);
        if (rc != SQL_SUCCESS) {
            __SET_ERROR_HANDLE(rc, SQL_HANDLE_ENV, __henv);
            return false;
        }
    }
 
    __DCL_ASSERT_HANDLE(__hdbc == NULL);
    rc = SQLAllocHandle(SQL_HANDLE_DBC, __henv, &__hdbc);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_ENV, __henv);
        return false;
    }
 
    for ( ; ; ) {
        SQLSMALLINT length2 = 0;
        rc = SQLDriverConnectA(__hdbc, NULL,
                (SQLCHAR*)_cons, (SQLSMALLINT)_conslen,
                NULL, 0, &length2, SQL_DRIVER_NOPROMPT
            );
        if (!SQL_SUCCEEDED(rc)) {
            __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
            break;
        }
        __SET_INFO_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
 
#if defined(__WINNT__) && !defined(strncasecmp)
#define strncasecmp(s1,s2,n) _strnicmp(s1,s2,n)
#endif
        char buf[128];
        SQLSMALLINT len = sizeof(buf);
        rc = SQLGetInfoA(__hdbc, SQL_DBMS_NAME, buf, len, &len);
        if (rc != SQL_SUCCESS) {
            __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
            return false;
        }
        __IS_MS_SS = strncasecmp(buf, "Microsoft SQL Server", __MIN(9, len)) == 0;
        __DCL_TRACE3_N(L"[%hs][%d] [%d]\n", buf, len, __IS_MS_SS);
 
        rc = SQLSetConnectAttrA(__hdbc, SQL_ATTR_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF, 0);
        if (rc != SQL_SUCCESS) {
            __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
            break;
        }
 
    #if __TRACE_THIS
        SQLUINTEGER GetDataSupport = 0;
        SQLSMALLINT length;
 
        rc = SQLGetInfoA(__hdbc, SQL_GETDATA_EXTENSIONS,
            &GetDataSupport, sizeof(GetDataSupport), &length);
        __DCL_TRACE3_N(L"[%04x] AC[%04x] AO[%04x]\n",
            GetDataSupport,
            GetDataSupport & SQL_GD_ANY_COLUMN,
            GetDataSupport & SQL_GD_ANY_ORDER
            );
        __DCL_TRACE3_N(L"BL[%04x] BO[%04x] OP[%04x]\n",
            GetDataSupport & SQL_GD_BLOCK,
            GetDataSupport & SQL_GD_BOUND,
            GetDataSupport & SQL_GD_OUTPUT_PARAMS
        );
    #endif
        return true;
    }
 
    // Connect Failed
    SQLRETURN rc_ = SQLFreeHandle(SQL_HANDLE_DBC, __hdbc);
    if (rc_ != SQL_SUCCESS) {
        __DCL_TRACE1_N(L"SQLFreeHandle(DBC) Fail! [%d]\n", rc_);
    }
    __hdbc = NULL;
    return false;
}

__rollbackTrans()

bool ODBCConnection::__rollbackTrans ( )

virtual

Implements SQL::Connection.

Definition at line 243 of file ODBCConnection.cpp.

{   
    __UNSET_STATE(Connection::stInTransaction);
    SQLRETURN rc = SQLEndTran(SQL_HANDLE_DBC, __hdbc, SQL_ROLLBACK);
    if (rc != SQL_SUCCESS) {
        __SET_ERROR_HANDLE(rc, SQL_HANDLE_DBC, __hdbc);
        return false;
    }
 
    return true;
}

__startTrans()

bool ODBCConnection::__startTrans ( )

virtual

Implements SQL::Connection.

Definition at line 225 of file ODBCConnection.cpp.

{
    __SET_STATE(Connection::stInTransaction);
    return true;
}

connHandle()

SQLHDBC ODBCConnection::connHandle ( ) const

inline

Definition at line 50 of file ODBCConnection.h.

{
    return __hdbc;
}

destroy()

void ODBCConnection::destroy ( )

virtual

파생클래스에서 new 연산자를 override했거나, 추가적인 행위가 필요하다면 이것도 override하라

Implements SQL::Connection.

Definition at line 83 of file ODBCConnection.cpp.

{
    delete this;
}

IS_MS_SS()

bool ODBCConnection::IS_MS_SS ( ) const

inline

Definition at line 55 of file ODBCConnection.h.

{
    return __IS_MS_SS;
}

setErrorHandle()

void ODBCConnection::setErrorHandle	(	SQLRETURN	_rc,
		SQLSMALLINT	_htype,
		SQLHANDLE	_handle,
		const wchar_t *	_filename,
		int	_line )

Definition at line 347 of file ODBCConnection.cpp.

{
    Connection::setErrorStatus(SQL::eServerError, _filename, _line);
 
    if (_rc == SQL_SUCCESS_WITH_INFO || _rc == SQL_ERROR) {
        __DCL_ASSERT(
            _htype == SQL_HANDLE_ENV
            || _htype == SQL_HANDLE_DBC
            || _htype == SQL_HANDLE_STMT
            || _htype == SQL_HANDLE_DESC
            // || _htype == SQL_HANDLE_DBC_INFO_TOKEN
        );
        __DCL_ASSERT(_handle != NULL);
 
#if 0 && defined(__WINNT__)
        /*
         * sizeof(WCHAR) == 2 인 Windows에서만 사용할 수 있다.
         * Linux에서 WCHAR은 기본적으로 signed short로 wchar_t가 아니다.
         * 이 코드는 LANG=C.UTF-4 환경변수가 적용되지 않는 Windows에서만 적용한다.
         */
        StringBuilder sb;
        SQLSMALLINT RecNumber = 0;
        SQLINTEGER  NativeError;
        WCHAR       SQLState[SQL_SQLSTATE_SIZE + 1];
        WCHAR       MessageText[1024];
        SQLSMALLINT MessageTextLength;
        SQLRETURN   rc;
        while ((rc = SQLGetDiagRecW(
            _htype, _handle, ++RecNumber, SQLState, &NativeError,
            MessageText, __countof(MessageText, WCHAR), &MessageTextLength
        )) == SQL_SUCCESS) {
            if (sb.length() > 0) {
                sb.append('\n');
            }
            sb.format(L"[%d][%ls]", RecNumber, SQLState);
            sb.append(MessageText, (size_t) MessageTextLength);
        }
        //__DCL_TRACE1_N(L"rc[%d]\n", rc);
        String s = sb.toString();
    #if __TRACE_THIS
        if (_rc == SQL_SUCCESS_WITH_INFO) {
            __DCL_TRACE1_N(L"SQL_SUCCESS_WITH_INFO:\n  %ls\n",
                s.replace(L"\n", L"\n  ").data());
        }
    #endif
        try {
            __lastErrorMessage = UTF8Encoder::encode(s);
        }
        catch (CharsetConvertException* _e) {
            __DCL_TRACE1(L"Fatal!! [%ls]\n", _e->toString().data());
            _e->destroy();
        }
#else
        ByteStringBuilder sb;
        SQLSMALLINT RecNumber = 0;
        SQLINTEGER  NativeError;
        SQLCHAR     SQLState[SQL_SQLSTATE_SIZE + 1];
        SQLCHAR     ErrorMsg[1024];
        SQLSMALLINT ErrorMsgLen;
        SQLRETURN   rc;
        while ((rc = SQLGetDiagRecA(
            _htype, _handle, ++RecNumber, SQLState, &NativeError,
            ErrorMsg, __countof(ErrorMsg, SQLCHAR), &ErrorMsgLen
        )) == SQL_SUCCESS) {
            if (sb.length() > 0) {
                sb.append('\n');
            }
            sb.format("[%d][%hs]", RecNumber, SQLState);
            sb.append((const char*)ErrorMsg, (size_t)ErrorMsgLen);
        }
        __DCL_ASSERT(rc == SQL_NO_DATA);
        ByteString s;
#ifdef __WINNT__
        /*
         * Windows에서 LANG=C.UTF-4 환경변수가 적용되지 않으므로,
         * wchar_t로 변환 후 UTF-8로 다시 변환한다.
         */
        try {
            String ws = LocaleDecoder::decode(sb.toByteString());
            s = UTF8Encoder::encode(ws);
    #if __TRACE_THIS
            if (_rc == SQL_SUCCESS_WITH_INFO) {
                __DCL_TRACE1_N(L"SQL_SUCCESS_WITH_INFO:\n  %ls\n",
                    ws.replace(L"\n", L"\n  ").data());
            }
    #endif
        }
        catch (CharsetConvertException* _e) {
            __DCL_TRACE1(L"Fatal!! [%ls]\n", _e->toString().data());
            _e->destroy();
        }
#else
        s = sb.toByteString();
    #if __TRACE_THIS
        if (_rc == SQL_SUCCESS_WITH_INFO) {
            __DCL_TRACE1_N(L"SQL_SUCCESS_WITH_INFO:\n  %hs\n",
                s.replace("\n", "\n  ").data());
        }
    #endif
#endif
        __lastErrorMessage = s;
#endif
    }
    else {
        #define CASE_STR(rc) case rc: __lastErrorMessage = #rc; break;
        switch (_rc) {
            CASE_STR(SQL_SUCCESS)
            CASE_STR(SQL_SUCCESS_WITH_INFO)
            CASE_STR(SQL_ERROR)
            CASE_STR(SQL_NO_DATA)
            CASE_STR(SQL_INVALID_HANDLE)
            default:
                __DCL_ASSERT(false);
                __lastErrorMessage = ByteString::format("SQLRETURN[%d] mabe bug!!", _rc);
        }
    }
}

setErrorMessage()

void ODBCConnection::setErrorMessage ( const ByteString & _message, const wchar_t * _filename, int _line )

inline

Definition at line 60 of file ODBCConnection.h.

{
    Connection::setErrorStatus(SQL::eServerError, _filename, _line);
    __lastErrorMessage = _message;
}

---

The documentation for this class was generated from the following files:

• ODBCConnection.h
• ODBCConnection.cpp