--- sql/odbc/parser3odbc.C 2004/01/30 07:30:40 1.19 +++ sql/odbc/parser3odbc.C 2008/07/01 08:56:06 1.31 @@ -5,7 +5,7 @@ Author: Alexandr Petrosian (http://paf.design.ru) */ -static const char *RCSId="$Id: parser3odbc.C,v 1.19 2004/01/30 07:30:40 paf Exp $"; +static const char *RCSId="$Id: parser3odbc.C,v 1.31 2008/07/01 08:56:06 misha Exp $"; #ifndef _MSC_VER # error compile ISAPI module with MSVC [no urge for now to make it autoconf-ed (PAF)] @@ -44,22 +44,39 @@ static const char *RCSId="$Id: parser3od #ifndef strncasecmp # define strncasecmp _strnicmp #endif +#ifndef strcasecmp +# define strcasecmp _stricmp +#endif -static char *lsplit(char *string, char delim) { - if(string) { - char *v=strchr(string, delim); - if(v) { +static char *lsplit(char *string, char delim){ + if(string){ + if(char* v=strchr(string, delim)){ *v=0; return v+1; } - } - return 0; + } + return 0; +} + +static char *lsplit(char **string_ref, char delim){ + char *result=*string_ref; + char *next=lsplit(*string_ref, delim); + *string_ref=next; + return result; +} + +static void toupper_str(char *out, const char *in, size_t size){ + while(size--) + *out++=(char)toupper(*in++); } struct Connection { SQL_Driver_services* services; CDatabase* db; + const char* client_charset; + bool autocommit; + bool fast_offset_search; }; /** @@ -73,32 +90,67 @@ public: /// get api version int api_version() { return SQL_DRIVER_API_VERSION; } + const char *initialize(char *dlopen_file_spec) { return 0; } /** connect - @param used_only_in_connect_url - format: @b DSN=dsn;UID=user;PWD=password (ODBC connect string) + @param url + format: @b DSN=dsn;UID=user;PWD=password? (ODBC connect string) + ClientCharset=charset& // transcode with parser + autocommit=1& // 0 -- disable auto commit + FastOffsetSearch=0 WARNING: must be used only to connect, for buffer doesn't live long */ + void connect( - char *used_only_in_connect_url, - SQL_Driver_services& services, - void **connection_ref ///< output: Connection* - ) { - Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); + char *url, + SQL_Driver_services& services, + void **connection_ref ///< output: Connection* + ){ + Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); *connection_ref=&connection; connection.services=&services; + connection.client_charset=0; + connection.fast_offset_search=false; + connection.autocommit=true; + + size_t url_length=strlen(url); + char *options=lsplit(url, '?'); + + while(options){ + if(char *key=lsplit(&options, '&')){ + if(*key){ + if(char *value=lsplit(key, '=')){ + if(strcmp(key, "ClientCharset")==0){ + toupper_str(value, value, strlen(value)); + connection.client_charset=value; + } else if(strcasecmp(key, "autocommit")==0){ + if(atoi(value)==0) + connection.autocommit=false; + } else if(strcmp(key, "FastOffsetSearch")==0){ + if(atoi(value)==1) + connection.fast_offset_search=true; + } else + services._throw("unknown connect option" /*key*/); + } else + services._throw("connect option without =value" /*key*/); + } + } + } TRY { connection.db=new CDatabase(); - connection.db->OpenEx(used_only_in_connect_url, CDatabase::noOdbcDialog); - connection.db->BeginTrans(); + connection.db->OpenEx(url, CDatabase::noOdbcDialog); + + if(!connection.autocommit) + connection.db->BeginTrans(); } CATCH_ALL (e) { _throw(services, e); } END_CATCH_ALL } - void disconnect(void *aconnection) { + + void disconnect(void *aconnection){ Connection& connection=*static_cast(aconnection); TRY delete connection.db; @@ -108,36 +160,42 @@ public: } END_CATCH_ALL } - void commit(void *aconnection) { + + void commit(void *aconnection){ Connection& connection=*static_cast(aconnection); - TRY - connection.db->CommitTrans(); - connection.db->BeginTrans(); - CATCH_ALL (e) { - _throw(*connection.services, e); + if(!connection.autocommit){ + TRY + connection.db->CommitTrans(); + connection.db->BeginTrans(); + CATCH_ALL (e) { + _throw(*connection.services, e); + } + END_CATCH_ALL } - END_CATCH_ALL } - void rollback(void *aconnection) { + + void rollback(void *aconnection){ Connection& connection=*static_cast(aconnection); - TRY - connection.db->Rollback(); - connection.db->BeginTrans(); - CATCH_ALL (e) { - _throw(*connection.services, e); + if(!connection.autocommit){ + TRY + connection.db->Rollback(); + connection.db->BeginTrans(); + CATCH_ALL (e) { + _throw(*connection.services, e); + } + END_CATCH_ALL } - END_CATCH_ALL } - bool ping(void *connection) { + bool ping(void *connection){ return true; } - const char* quote(void *aconnection, const char *from, unsigned int length) { + const char* quote(void *aconnection, const char *from, unsigned int length){ Connection& connection=*static_cast(aconnection); char *result=(char*)connection.services->malloc_atomic(length*2+1); char *to=result; - while(length--) { + while(length--){ if(*from=='\'') { // ' -> '' *to++='\''; } @@ -146,19 +204,41 @@ public: *to=0; return result; } - void query(void *aconnection, - const char *statement, unsigned long offset, unsigned long limit, - SQL_Driver_query_event_handlers& handlers) { + void query(void *aconnection, + const char *statement, + size_t placeholders_count, + Placeholder* placeholders, + unsigned long offset, + unsigned long limit, + SQL_Driver_query_event_handlers& handlers + ){ Connection& connection=*static_cast(aconnection); CDatabase *db=connection.db; SQL_Driver_services& services=*connection.services; - while(isspace(*statement)) + if(placeholders_count>0) + services._throw("bind variables not supported (yet)"); + + const char* client_charset=connection.client_charset; + const char* request_charset=services.request_charset(); + bool transcode_needed=(client_charset && strcmp(client_charset, request_charset)!=0); + + // transcode query from $request:charset to ?ClientCharset + if(transcode_needed){ + size_t length=strlen(statement); + services.transcode(statement, length, + statement, length, + request_charset, + client_charset); + } + + while(isspace((unsigned char)*statement)) statement++; - + TRY { // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\adosql.chm::/adoprg02_4g33.htm + // or http://msdn.microsoft.com/en-us/library/aa905899(SQL.80).aspx // Server cursors are created only for statements that begin with: // SELECT // EXEC[ute] procedure_name @@ -166,17 +246,30 @@ public: // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\odbcsql.chm::/od_6_035_5dnp.htm // The ODBC CALL escape sequence for calling a procedure is: // {[?=]call procedure_name[([parameter][,[parameter]]...)]} - if(strncasecmp(statement, "select", 6)==0 + if(strncasecmp(statement, "SELECT", 6)==0 || strncasecmp(statement, "EXEC", 4)==0 || strncasecmp(statement, "call", 4)==0 - || strncasecmp(statement, "{", 1)==0) { - CRecordset rs(db); + || strncasecmp(statement, "{", 1)==0 + ){ + CRecordset rs(db); + DWORD options=CRecordset::executeDirect|CRecordset::readOnly; + //CRecordset::skipDeletedRecords + //CRecordset::useMultiRowFetch + //CRecordset::userAllocMultiRowBuffers + //CRecordset::useExtendedFetch +/* + if(connection.fast_offset_search){ + options+=CRecordset::useMultiRowFetch+CRecordset::userAllocMultiRowBuffers; + } else { + options+=CRecordset::skipDeletedRecords; + } +*/ TRY { rs.Open( - CRecordset::forwardOnly, + (connection.fast_offset_search)?CRecordset::dynamic:CRecordset::forwardOnly, statement, - CRecordset::executeDirect - ); + options + ); } CATCH_ALL (e) { // could not fetch a table TRY { @@ -194,10 +287,12 @@ public: if(!column_count) services._throw("result contains no columns"); - SWORD column_types[MAX_COLS]; if(column_count>MAX_COLS) column_count=MAX_COLS; + SWORD column_types[MAX_COLS]; + bool transcode_column[MAX_COLS]; + SQL_Error sql_error; #define CHECK(afailed) if(afailed) services._throw(sql_error) @@ -206,47 +301,94 @@ public: CODBCFieldInfo fieldinfo; rs.GetODBCFieldInfo(i, fieldinfo); column_types[i]=fieldinfo.m_nSQLType; - size_t size=fieldinfo.m_strName.GetLength(); + switch(fieldinfo.m_nSQLType){ + case SQL_NUMERIC: + case SQL_DECIMAL: + case SQL_INTEGER: + case SQL_SMALLINT: + case SQL_FLOAT: + case SQL_REAL: + case SQL_DOUBLE: + case SQL_DATETIME: + case SQL_SMALLDATETIME: + case SQL_BIGINT: + case SQL_TINYINT: + transcode_column[i]=false; + break; + default: + transcode_column[i]=transcode_needed; + + } + size_t length=fieldinfo.m_strName.GetLength(); char *str=0; - if(size) { - str=(char*)services.malloc_atomic(size+1); - memcpy(str, (char *)LPCTSTR(fieldinfo.m_strName), size+1); + if(length){ + str=(char*)services.malloc_atomic(length+1); + memcpy(str, (char*)LPCTSTR(fieldinfo.m_strName), length+1); + + // transcode column name from ?ClientCharset to $request:charset + if(transcode_needed){ + services.transcode(str, length, + str, length, + client_charset, + request_charset); + } } - CHECK(handlers.add_column(sql_error, str, size)); + CHECK(handlers.add_column(sql_error, str, length)); } CHECK(handlers.before_rows(sql_error)); + + // skip offset rows + if(offset){ + if(connection.fast_offset_search){ + rs.Move(offset); + } else { + unsigned long row=offset; + while(!rs.IsEOF() && row>0){ + rs.MoveNext(); + row--; + } + } + } unsigned long row=0; CDBVariant v; CString s; - while(!rs.IsEOF() && (!limit||(row=offset) { - CHECK(handlers.add_row(sql_error)); - for(int i=0; ihour, v.m_pdate->minute, v.m_pdate->second, - v.m_pdate->fraction); + v.m_pdate->fraction/1000000); // lexical parser of INCOMING literal choked on times like hh:mm:ss.123000000 str=(char*)services.malloc_atomic(length+1); memcpy(str, local_buf, length+1); break; @@ -323,8 +470,8 @@ public: } } - void getFromString(SQL_Driver_services& services, CString& s, char*& astr, size_t& length) { - if(s.IsEmpty()) { + void getFromString(SQL_Driver_services& services, CString& s, char*& astr, size_t& length){ + if(s.IsEmpty()){ astr=0; length=0; } else { @@ -335,8 +482,9 @@ public: } } - void _throw(SQL_Driver_services& services, CException *e) { - char szCause[MAX_STRING]; szCause[0]=0; + void _throw(SQL_Driver_services& services, CException *e){ + char szCause[MAX_STRING]; + szCause[0]=0; e->GetErrorMessage(szCause, MAX_STRING); char msg[MAX_STRING]; snprintf(msg, MAX_STRING, "%s: %s", @@ -349,4 +497,4 @@ public: extern "C" SQL_Driver *SQL_DRIVER_CREATE() { return new ODBC_Driver(); -} \ No newline at end of file +}