PeModuleMain.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef _MSC_VER
    #pragma comment(lib, "libecpg.lib")
    #pragma comment(lib, "libpq.lib")
    #pragma comment(lib, "libpgtypes.lib")
#endif
 
#include <string.h>         // strchr
#include <locale.h>         // setlocale
#include <dcl/_stdlib.h>   // setenv
 
#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 "PeConnection.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 char_t __THIS_FILE__[] = __T("dcl/sql/PeModuleMain.cpp");
#endif
 
/*
#ifdef _WINDOWS
BOOL APIENTRY DllMain( HANDLE hModule, 
    DWORD  ul_reason_for_call, 
    LPVOID lpReserved
)
{
    switch (ul_reason_for_call) {
        case DLL_PROCESS_ATTACH:
        case DLL_PROCESS_DETACH:
            return TRUE;
        case DLL_THREAD_ATTACH:
        case DLL_THREAD_DETACH:
            break;
    }
    return FALSE;
}
#endif
*/
 
__DCL_BEGIN_NAMESPACE
 
static const wchar_t* _serverTitle = __T("PostgreSQL/ecpg");
 
SQL::Connection* CreateConnectionInstance()
{
    return new PeConnection(_serverTitle);
}
 
bool ModuleInitialize()
{
    /*
     * LANG에 UTF-8이 없는 경우 LANG을 변경한다.
     */
    ByteStringBuilder sb;
    char* lang = getenv("LANG");
    if (!lang) {
        // current locale
        lang = setlocale(LC_CTYPE, NULL);
        if (!(lang && strchr(lang, '.'))) {
            // set default locale
            lang = setlocale(LC_ALL, "");
        }
    }
    if (lang) {
        ByteString lang_ = ByteString(lang);
        ByteString lower = lang_.toLowerCase();
        if (!lower.contains("utf8")) {
            if (lower.contains("ko") || lower.contains("kr")) {
                sb.append("ko_KR");
            }
            else {
                size_t index = lang_.indexOf(L'.');
                sb.append(index == (size_t)-1 ?
                    lang_ : lang_.substring(0, index));
            }
            sb.append(".UTF-8");
        }
    }
    else {
        sb = "C.UTF-8";
    }
 
    if (sb.length() > 0) {
        ByteString langUTF8 = sb.toByteString();
        __DCL_TRACE1_N(L"setenv LANG=[%hs]\n", langUTF8.data());
        return setenv("LANG", langUTF8, 1) == 0;;
    }
    return true;
}
 
bool ModuleCleanup()
{
    return true;
}
 
extern "C"
{
DCL_DSO_EXPORT
SQL::DRIVER_MODULE DCL_DSO_ENTRY_POINT =
{
    // DCL common members
    sizeof(SQL::DRIVER_MODULE),       // structure size
    DCL_VERSION,
    __T(__BUILD_TIMESTAMP__),
    DCL_BUILD_FLAG,
    DCL_SQL_DRIVER_MODULE,         // module type
    __T("DCL SQL Adapter for PostgreSQL Database Server"),
 
    // private members
    DCL_SQL_VERSION,
    _serverTitle,
    __T(DCL_SQL_VERSION_STRING),
    ModuleInitialize,
    ModuleCleanup,
    CreateConnectionInstance
};
 
}
 
__DCL_END_NAMESPACE
 
#ifdef __WINNT_NEW_DELETE_OVERRIDE
#undef new
__WINNT_NEW_DELETE_OVERRIDE
#endif