--- sql/pgsql/parser3pgsql.C 2012/10/19 04:13:28 1.42 +++ sql/pgsql/parser3pgsql.C 2021/11/07 22:16:54 1.50 @@ -1,7 +1,7 @@ /** @file Parser PgSQL driver. - Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com) + Copyright (c) 2001-2019 Art. Lebedev Studio (http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) @@ -15,7 +15,7 @@ #include #include -volatile const char * IDENT_PARSER3PGSQL_C="$Id: parser3pgsql.C,v 1.42 2012/10/19 04:13:28 misha Exp $" IDENT_PA_SQL_DRIVER_H; +volatile const char * IDENT_PARSER3PGSQL_C="$Id: parser3pgsql.C,v 1.50 2021/11/07 22:16:54 moko Exp $" IDENT_PA_SQL_DRIVER_H; // from catalog/pg_type.h #define BOOLOID 16 @@ -78,7 +78,7 @@ static char* rsplit(char* string, char d return v+1; } } - return NULL; + return NULL; } static void toupper_str(char *out, const char *in, size_t size){ @@ -99,7 +99,6 @@ struct Connection { PGconn *conn; const char* client_charset; bool autocommit; - bool without_default_transactions; bool standard_conforming_strings; }; @@ -117,15 +116,14 @@ public: /// initialize driver by loading sql dynamic link library const char *initialize(char *dlopen_file_spec){ - return dlopen_file_spec? - dlink(dlopen_file_spec):"client library column is empty"; + return dlopen_file_spec ? dlink(dlopen_file_spec) : "client library column is empty"; } #define throwPQerror connection.services->_throw(PQerrorMessage(connection.conn)) #define PQclear_throw(msg) { \ PQclear(res); \ connection.services->_throw(msg); \ - } + } #define PQclear_throwPQerror PQclear_throw(PQerrorMessage(connection.conn)) /** connect @@ -134,43 +132,31 @@ public: ClientCharset=charset& // transcode by parser charset=value& // transcode by server with 'SET CLIENT_ENCODING=value' datestyle=value& // 'SET DATESTYLE=value' available values are: ISO|SQL|Postgres|European|US|German [default=ISO] - autocommit=1& // each transaction is commited automatically (default) - WithoutDefaultTransaction=0 // 1 -- disable any BEGIN TRAN/COMMIT/ROLLBACK [can NOT be used with autocommit option] + autocommit=0& // 1 -- each statement is commited automatically, only when with_default_transaction enabled + standard_conforming_strings=1& // 0 -- escape \ char that could be needed for old servers */ - void connect( - char* url, - SQL_Driver_services& services, - void** connection_ref ///< output: Connection* - ){ + void connect(char* url, SQL_Driver_services& services, void** connection_ref /* < output: Connection* */){ char* user=url; char* host=rsplit(user, '@'); char* db=lsplit(host, '/'); char* pwd=lsplit(user, ':'); - char* port=lsplit(host, ':'); char *options=lsplit(db, '?'); char* charset=0; char* datestyle=0; + char* pq_options=0; + size_t pq_options_len=options ? strlen(options) : 0; + Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); + *connection_ref=&connection; connection.services=&services; - connection.client_charset=0; + connection.client_charset=0; connection.autocommit=true; - connection.without_default_transactions=false; connection.standard_conforming_strings=true; - connection.conn=PQsetdbLogin( - (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port, - NULL, NULL, db, user, pwd); - - if(!connection.conn) - services._throw("PQsetdbLogin failed"); - - if(PQstatus(connection.conn)!=CONNECTION_OK) - throwPQerror; - while(options){ if(char *key=lsplit(&options, '&')){ if(*key){ @@ -183,28 +169,43 @@ public: } else if(strcasecmp(key, "datestyle")==0){ datestyle=value; } else if(strcasecmp(key, "autocommit")==0){ - if(connection.without_default_transactions) - services._throw("options WithoutDefaultTransaction and autocommit can't be used together"); if(atoi(value)==0) connection.autocommit=false; - } else if(strcmp(key, "WithoutDefaultTransaction")==0){ - if(!connection.autocommit) - services._throw("options WithoutDefaultTransaction and autocommit can't be used together"); - if(atoi(value)==1){ - connection.without_default_transactions=true; - connection.autocommit=false; - } } else if(strcasecmp(key, "standard_conforming_strings")==0){ if(atoi(value)==0) connection.standard_conforming_strings=false; - } else - services._throw("unknown connect option" /*key*/); + } else { + if(!pq_options) { + pq_options=(char*)services.malloc_atomic(pq_options_len+2); + strcpy(pq_options, "?"); + } else { + strcat(pq_options, "&"); + } + strcat(pq_options, key); + strcat(pq_options, "="); + strcat(pq_options, value); + } } else services._throw("connect option without =value" /*key*/); } } } + if(host && (strchr(host, ',') || pq_options)){ // pq_options can exist only if host and db are not null + char pq_url[MAX_STRING+1]; + snprintf(pq_url, MAX_STRING, "postgresql://%s/%s%s", host, db ? db : "", pq_options ? pq_options : ""); + connection.conn=PQsetdbLogin(NULL, NULL, NULL, NULL, pq_url, user, pwd); + } else { + char* port=lsplit(host, ':'); + connection.conn=PQsetdbLogin( (host && strcasecmp(host, "local") == 0) ? NULL /* local Unix domain socket */ : host, port, NULL, NULL, db, user, pwd); + } + + if(!connection.conn) + services._throw("PQsetdbLogin failed"); + + if(PQstatus(connection.conn)!=CONNECTION_OK) + throwPQerror; + if(charset){ char statement[MAX_STRING+1]="SET CLIENT_ENCODING="; strncat(statement, charset, MAX_STRING); @@ -219,7 +220,8 @@ public: _execute_cmd(connection, statement); } - _transaction_begin(connection); + if(!connection.autocommit) + _execute_cmd(connection, "BEGIN"); } void disconnect(void *aconnection){ @@ -230,14 +232,14 @@ public: void commit(void *aconnection){ Connection& connection=*static_cast(aconnection); - _transaction_commit(connection); - _transaction_begin(connection); + if(!connection.autocommit) + _execute_cmd(connection, "COMMIT"); } void rollback(void *aconnection){ Connection& connection=*static_cast(aconnection); - _transaction_rollback(connection); - _transaction_begin(connection); + if(!connection.autocommit) + _execute_cmd(connection, "ROLLBACK"); } bool ping(void *aconnection) { @@ -247,8 +249,7 @@ public: // charset here is services.request_charset(), not connection.client_charset // thus we can't use the sql server quoting support - const char* quote(void *aconnection, const char *str, unsigned int length) - { + const char* quote(void *aconnection, const char *str, unsigned int length){ Connection& connection=*static_cast(aconnection); const char* from; @@ -301,12 +302,7 @@ public: return result; } - void query(void *aconnection, - const char *astatement, - size_t placeholders_count, Placeholder* placeholders, - unsigned long offset, unsigned long limit, - SQL_Driver_query_event_handlers& handlers - ){ + void query(void *aconnection, const char *astatement, size_t placeholders_count, Placeholder* placeholders, unsigned long offset, unsigned long limit, SQL_Driver_query_event_handlers& handlers ){ Connection& connection=*static_cast(aconnection); SQL_Driver_services& services=*connection.services; PGconn *conn=connection.conn; @@ -326,10 +322,7 @@ public: if(transcode_needed){ // transcode query from $request:charset to ?ClientCharset statement_size=strlen(astatement); - services.transcode(astatement, statement_size, - astatement, statement_size, - request_charset, - client_charset); + services.transcode(astatement, statement_size, astatement, statement_size, request_charset, client_charset); } const char *statement=_preprocess_statement(connection, astatement, statement_size, offset, limit); @@ -348,16 +341,14 @@ public: SQL_Error sql_error; switch(PQresultStatus(res)) { - case PGRES_EMPTY_QUERY: + case PGRES_EMPTY_QUERY: PQclear_throw("no query"); break; case PGRES_COMMAND_OK: // empty result: insert|delete|update|... PQclear(res); - if(connection.autocommit) - commit(aconnection); return; case PGRES_TUPLES_OK: - break; + break; default: PQclear_throwPQerror; break; @@ -387,10 +378,7 @@ public: if(transcode_needed) // transcode column name from ?ClientCharset to $request:charset - services.transcode(str, length, - str, length, - client_charset, - request_charset); + services.transcode(str, length, str, length, client_charset, request_charset); CHECK(handlers.add_column(sql_error, str, length)); } @@ -481,34 +469,19 @@ cleanup: PQclear(res); if(failed) services._throw(sql_error); - - if(connection.autocommit) - commit(aconnection); } private: - void _bind_parameters( - size_t placeholders_count, - Placeholder* placeholders, - const char** paramValues, - Connection& connection, - bool transcode_needed - ){ + void _bind_parameters( size_t placeholders_count, Placeholder* placeholders, const char** paramValues, Connection& connection, bool transcode_needed){ for(size_t i=0; itranscode(ph.name, strlen(ph.name), - ph.name, name_length, - connection.services->request_charset(), - connection.client_charset); + connection.services->transcode(ph.name, strlen(ph.name), ph.name, name_length, connection.services->request_charset(), connection.client_charset); if(ph.value) { size_t value_length; - connection.services->transcode(ph.value, strlen(ph.value), - ph.value, value_length, - connection.services->request_charset(), - connection.client_charset); + connection.services->transcode(ph.value, strlen(ph.value), ph.value, value_length, connection.services->request_charset(), connection.client_charset); } } int name_number=atoi(ph.name); @@ -519,24 +492,6 @@ private: } } - - void _transaction_begin(Connection& connection){ - _execute_transactions_cmd(connection, "BEGIN"); - } - - void _transaction_commit(Connection& connection){ - _execute_transactions_cmd(connection, "COMMIT"); - } - - void _transaction_rollback(Connection& connection){ - _execute_transactions_cmd(connection, "ROLLBACK"); - } - - void _execute_transactions_cmd(const Connection& connection, const char *query){ - if(!connection.without_default_transactions) // with option ?WithoutDefaultTransaction=1 user must execute BEGIN/COMMIT/ROLLBACK by himself - _execute_cmd(connection, query); - } - // executes a query and throw away the result. void _execute_cmd(const Connection& connection, const char *query){ if(PGresult *res=PQexec(connection.conn, query)) @@ -545,13 +500,7 @@ private: throwPQerror; } - const char *_preprocess_statement( - Connection& connection, - const char *astatement, - size_t statement_size, - unsigned long offset, - unsigned long limit - ){ + const char *_preprocess_statement(Connection& connection, const char *astatement, size_t statement_size, unsigned long offset, unsigned long limit){ PGconn *conn=connection.conn; if(!statement_size) @@ -584,11 +533,7 @@ private: const char* saved_o=o; o+=3; while(*o) - if( - o[0]=='*' && - o[1]=='*' && - o[2]=='/' && - o[3]=='\'') { // name end + if(o[0]=='*' && o[1]=='*' && o[2]=='/' && o[3]=='\'') { // name end saved_o=0; // found, marking that o+=4; Oid oid=lo_creat(conn, INV_READ|INV_WRITE); @@ -746,11 +691,11 @@ private: // conn client library funcs li DLINK(PQexecParams); DLINK(PQftype); DLINK(PQescapeStringConn); - DLINK(lo_open); DLINK(lo_close); - DLINK(lo_read); DLINK(lo_write); - DLINK(lo_lseek); DLINK(lo_creat); - DLINK(lo_tell); DLINK(lo_unlink); - DLINK(lo_import); DLINK(lo_export); + DLINK(lo_open); DLINK(lo_close); + DLINK(lo_read); DLINK(lo_write); + DLINK(lo_lseek); DLINK(lo_creat); + DLINK(lo_tell); DLINK(lo_unlink); + DLINK(lo_import); DLINK(lo_export); return 0; }