--- sql/pgsql/parser3pgsql.C 2012/04/18 09:39:15 1.38 +++ sql/pgsql/parser3pgsql.C 2021/11/03 14:51:51 1.47 @@ -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.38 2012/04/18 09:39:15 moko Exp $" IDENT_PA_SQL_DRIVER_H; +volatile const char * IDENT_PARSER3PGSQL_C="$Id: parser3pgsql.C,v 1.47 2021/11/03 14:51:51 moko Exp $" IDENT_PA_SQL_DRIVER_H; // from catalog/pg_type.h #define BOOLOID 16 @@ -99,7 +99,7 @@ struct Connection { PGconn *conn; const char* client_charset; bool autocommit; - bool without_default_transactions; + bool standard_conforming_strings; }; /** @@ -133,8 +133,8 @@ 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, @@ -153,11 +153,12 @@ public: char* datestyle=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, @@ -181,17 +182,11 @@ 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 @@ -201,20 +196,21 @@ public: } if(charset){ - char statement[MAX_STRING]="SET CLIENT_ENCODING="; + char statement[MAX_STRING+1]="SET CLIENT_ENCODING="; strncat(statement, charset, MAX_STRING); _execute_cmd(connection, statement); } if(datestyle){ - char statement[MAX_STRING]="SET DATESTYLE="; + char statement[MAX_STRING+1]="SET DATESTYLE="; strncat(statement, datestyle, MAX_STRING); _execute_cmd(connection, statement); } - _transaction_begin(connection); + if(!connection.autocommit) + _execute_cmd(connection, "set AUTOCOMMIT off"); } void disconnect(void *aconnection){ @@ -225,14 +221,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) { @@ -244,36 +240,52 @@ public: // thus we can't use the sql server quoting support const char* quote(void *aconnection, const char *str, unsigned int length) { + Connection& connection=*static_cast(aconnection); + const char* from; const char* from_end=str+length; size_t quoted=0; - for(from=str; from(aconnection); char *result=(char*)connection.services->malloc_atomic(length + quoted + 1); char *to = result; - for(from=str; from "''" - *to++='\''; - break; - case '\\': // "\" -> "\\" - *to++='\\'; - break; + if(connection.standard_conforming_strings){ + for(from=str; from "''" + *to++=*from; + } + } else { + for(from=str; from "''" + *to++= '\''; + break; + case '\\': // "\" -> "\\" + *to++='\\'; + break; + } + *to++=*from; } - *to++=*from; } *to=0; @@ -332,11 +344,9 @@ public: 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; @@ -460,9 +470,6 @@ cleanup: PQclear(res); if(failed) services._throw(sql_error); - - if(connection.autocommit) - commit(aconnection); } private: @@ -498,24 +505,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)) @@ -546,9 +535,9 @@ private: char *cur=result; memcpy(cur, astatement, statement_size); cur+=statement_size; if(limit!=SQL_NO_LIMIT) - cur+=snprintf(cur, 7+MAX_NUMBER, " limit %u", limit); + cur+=snprintf(cur, 7+MAX_NUMBER, " limit %lu", limit); if(offset) - cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset); + cur+=snprintf(cur, 8+MAX_NUMBER, " offset %lu", offset); o=result; } else o=astatement; @@ -689,11 +678,19 @@ private: // conn client library funcs private: // conn client library funcs linking const char *dlink(const char *dlopen_file_spec) { - if(lt_dlinit()) - return lt_dlerror(); + if(lt_dlinit()){ + if(const char* result=lt_dlerror()) + return result; + return "can not prepare to dynamic loading"; + } + lt_dlhandle handle=lt_dlopen(dlopen_file_spec); - if(!handle) + + if(!handle){ + if(const char* result=lt_dlerror()) + return result; return "can not open the dynamic link module"; + } #define DSLINK(name, action) \ name=(t_##name)lt_dlsym(handle, #name); \