|
|
| version 1.29, 2008/06/24 17:43:48 | version 1.46, 2019/11/30 22:11:11 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser PgSQL driver. | Parser PgSQL driver. |
| Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2019 Art. Lebedev Studio (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) |
| 2007.10.25 using PgSQL 8.1.5 | 2007.10.25 using PgSQL 8.1.5 |
| */ | */ |
| static const char *RCSId="$Id$"; | |
| #include "config_includes.h" | #include "config_includes.h" |
| Line 16 static const char *RCSId="$Id$"; | Line 15 static const char *RCSId="$Id$"; |
| #include <libpq-fe.h> | #include <libpq-fe.h> |
| #include <libpq/libpq-fs.h> | #include <libpq/libpq-fs.h> |
| // OIDOID from catalog/pg_type.h | volatile const char * IDENT_PARSER3PGSQL_C="$Id$" IDENT_PA_SQL_DRIVER_H; |
| // from catalog/pg_type.h | |
| #define BOOLOID 16 | |
| #define INT8OID 20 | |
| #define INT2OID 21 | |
| #define INT4OID 23 | |
| #define OIDOID 26 | #define OIDOID 26 |
| #define FLOAT4OID 700 | |
| #define FLOAT8OID 701 | |
| #define DATEOID 1082 | |
| #define TIMEOID 1083 | |
| #define TIMESTAMPOID 1114 | |
| #define TIMESTAMPTZOID 1184 | |
| #define TIMETZOID 1266 | |
| #define NUMERICOID 1700 | |
| // LO_BUFSIZE from interfaces\libpq\fe-lobj.c = 8192 (0x2000) | // LO_BUFSIZE from interfaces\libpq\fe-lobj.c = 8192 (0x2000) |
| // actually writing chunks of that size failed, reduced it twice | // actually writing chunks of that size failed, reduced it twice |
| #define LO_BUFSIZE 0x1000 | #define LO_BUFSIZE 0x1000 |
| // from postgres_ext.h | |
| //#define InvalidOid ((Oid) 0) | |
| #include "ltdl.h" | #include "ltdl.h" |
| #define MAX_COLS 500 | |
| #define MAX_STRING 0x400 | #define MAX_STRING 0x400 |
| #define MAX_NUMBER 20 | #define MAX_NUMBER 20 |
| Line 36 static const char *RCSId="$Id$"; | Line 50 static const char *RCSId="$Id$"; |
| #endif | #endif |
| #ifndef max | #ifndef max |
| inline int max(int a,int b) { return a>b?a:b; } | inline int max(int a,int b){ return a>b?a:b; } |
| inline int min(int a,int b){ return a<b?a:b; } | inline int min(int a,int b){ return a<b?a:b; } |
| #endif | #endif |
| static char *lsplit(char *string, char delim) { | static char *lsplit(char *string, char delim){ |
| if(string) { | if(string){ |
| char *v=strchr(string, delim); | if(char *v=strchr(string, delim)){ |
| if(v) { | |
| *v=0; | *v=0; |
| return v+1; | return v+1; |
| } | } |
| Line 51 static char *lsplit(char *string, char d | Line 64 static char *lsplit(char *string, char d |
| return 0; | return 0; |
| } | } |
| static char *lsplit(char **string_ref, char delim) { | static char *lsplit(char **string_ref, char delim){ |
| char *result=*string_ref; | char *result=*string_ref; |
| char *next=lsplit(*string_ref, delim); | char *next=lsplit(*string_ref, delim); |
| *string_ref=next; | *string_ref=next; |
| return result; | return result; |
| } | } |
| static char* rsplit(char* string, char delim) { | static char* rsplit(char* string, char delim){ |
| if(string) { | if(string){ |
| char* v=strrchr(string, delim); | if(char* v=strrchr(string, delim)){ |
| if(v) { | |
| *v=0; | *v=0; |
| return v+1; | return v+1; |
| } | } |
| Line 69 static char* rsplit(char* string, char d | Line 81 static char* rsplit(char* string, char d |
| return NULL; | return NULL; |
| } | } |
| static void toupper_str(char *out, const char *in, size_t size) { | static void toupper_str(char *out, const char *in, size_t size){ |
| while(size--) | while(size--) |
| *out++=(char)toupper(*in++); | *out++=(char)toupper(*in++); |
| } | } |
| inline static const char* strdup(SQL_Driver_services& services, char* str, size_t length) { | |
| char *strm=(char*)services.malloc_atomic(length+1); | |
| memcpy(strm, str, length); | |
| strm[length]=0; | |
| return (const char*)strm; | |
| } | |
| struct Connection { | struct Connection { |
| SQL_Driver_services* services; | SQL_Driver_services* services; |
| PGconn *conn; | PGconn *conn; |
| const char* cstrClientCharset; | const char* client_charset; |
| bool autocommit; | bool autocommit; |
| bool with_default_transactions; | |
| bool standard_conforming_strings; | |
| }; | }; |
| /** | /** |
| Line 92 public: | Line 113 public: |
| } | } |
| /// get api version | /// get api version |
| int api_version() { return SQL_DRIVER_API_VERSION; } | int api_version(){ return SQL_DRIVER_API_VERSION; } |
| /// initialize driver by loading sql dynamic link library | /// initialize driver by loading sql dynamic link library |
| const char *initialize(char *dlopen_file_spec) { | const char *initialize(char *dlopen_file_spec){ |
| return dlopen_file_spec? | return dlopen_file_spec? |
| dlink(dlopen_file_spec):"client library column is empty"; | dlink(dlopen_file_spec):"client library column is empty"; |
| } | } |
| Line 110 public: | Line 131 public: |
| /** connect | /** connect |
| @param url | @param url |
| format: @b user:pass@host[:port]|[local]/database? | format: @b user:pass@host[:port]|[local]/database? |
| ClientCharset=xyz& // transcode by parser | ClientCharset=charset& // transcode by parser |
| charset=xyz& // transcode by server with 'set CLIENT_ENCODING=xyz' | charset=value& // transcode by server with 'SET CLIENT_ENCODING=value' |
| datestyle=xyz& // set DATESTYLE=xyz | datestyle=value& // 'SET DATESTYLE=value' available values are: ISO|SQL|Postgres|European|US|German [default=ISO] |
| autocommit=1& | autocommit=0& // 1 -- each statement is commited automatically, only when with_default_transaction enabled |
| WithoutDefaultTransaction=1 // == autocommit=0 | standard_conforming_strings=1& // 0 -- escape \ char that could be needed for old servers |
| with_default_transaction=0 // 1 -- wrap connection into BEGIN TRAN/COMMIT/ROLLBACK | |
| */ | */ |
| void connect( | void connect( |
| char *url, | char* url, |
| SQL_Driver_services& services, | SQL_Driver_services& services, |
| void **connection_ref ///< output: Connection* | void** connection_ref ///< output: Connection* |
| ) { | ){ |
| char *user=url; | char* user=url; |
| char *host=rsplit(user, '@'); | char* host=rsplit(user, '@'); |
| char *db=lsplit(host, '/'); | char* db=lsplit(host, '/'); |
| char *pwd=lsplit(user, ':'); | char* pwd=lsplit(user, ':'); |
| char *port=lsplit(host, ':'); | char* port=lsplit(host, ':'); |
| char *options=lsplit(db, '?'); | char *options=lsplit(db, '?'); |
| char *cstrBackwardCompAskServerToTranscode=0; | char* charset=0; |
| char* datestyle=0; | char* datestyle=0; |
| Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); | Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); |
| *connection_ref=&connection; | *connection_ref=&connection; |
| connection.services=&services; | connection.services=&services; |
| connection.cstrClientCharset=0; | connection.client_charset=0; |
| connection.autocommit=true; | connection.autocommit=false; |
| connection.with_default_transactions=false; | |
| connection.standard_conforming_strings=true; | |
| connection.conn=PQsetdbLogin( | connection.conn=PQsetdbLogin( |
| (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port, | (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port, |
| NULL, NULL, db, user, pwd); | NULL, NULL, db, user, pwd); |
| Line 151 public: | Line 177 public: |
| if(char *key=lsplit(&options, '&')){ | if(char *key=lsplit(&options, '&')){ |
| if(*key){ | if(*key){ |
| if(char *value=lsplit(key, '=')){ | if(char *value=lsplit(key, '=')){ |
| if(strcmp(key, "ClientCharset")==0) { // transcoding with parser | if(strcmp(key, "ClientCharset")==0){ |
| toupper_str(value, value, strlen(value)); | toupper_str(value, value, strlen(value)); |
| connection.cstrClientCharset=value; | connection.client_charset=value; |
| } else if(strcasecmp(key, "charset")==0){ // transcoding with server | } else if(strcasecmp(key, "charset")==0){ |
| cstrBackwardCompAskServerToTranscode=value; | charset=value; |
| } else if(strcasecmp(key, "datestyle")==0) { | } else if(strcasecmp(key, "datestyle")==0){ |
| datestyle=value; | datestyle=value; |
| } else if(strcasecmp(key, "autocommit")==0){ | } else if(strcasecmp(key, "autocommit")==0){ |
| if(atoi(value)==0) | if(atoi(value)==1){ |
| connection.autocommit=false; | if(!connection.with_default_transactions) |
| } else if(strcmp(key, "WithoutDefaultTransaction")==0){ // backward, use autocommit=0 | services._throw("autocommit can be used only with_default_transaction enabled"); |
| connection.autocommit=true; | |
| } | |
| } else if(strcmp(key, "with_default_transaction")==0){ | |
| if(atoi(value)==1) | if(atoi(value)==1) |
| connection.autocommit=false; | connection.with_default_transactions=true; |
| } else if(strcmp(key, "WithoutDefaultTransaction")==0){ | |
| if(atoi(value)==0) | |
| connection.with_default_transactions=true; | |
| } else if(strcasecmp(key, "standard_conforming_strings")==0){ | |
| if(atoi(value)==0) | |
| connection.standard_conforming_strings=false; | |
| } else | } else |
| services._throw("unknown connect option" /*key*/); | services._throw("unknown connect option" /*key*/); |
| } else | } else |
| Line 172 public: | Line 207 public: |
| } | } |
| } | } |
| if(cstrBackwardCompAskServerToTranscode){ | if(charset){ |
| char statement[MAX_STRING]="set CLIENT_ENCODING="; | char statement[MAX_STRING+1]="SET CLIENT_ENCODING="; |
| strncat(statement, cstrBackwardCompAskServerToTranscode, MAX_STRING); | strncat(statement, charset, MAX_STRING); |
| execute_resultless(connection, statement); | _execute_cmd(connection, statement); |
| } | } |
| if(datestyle){ | if(datestyle){ |
| char statement[MAX_STRING]="set DATESTYLE="; // ISO,SQL,Postgres,European,NonEuropean=US,German,DEFAULT=ISO | char statement[MAX_STRING+1]="SET DATESTYLE="; |
| strncat(statement, datestyle, MAX_STRING); | strncat(statement, datestyle, MAX_STRING); |
| execute_resultless(connection, statement); | _execute_cmd(connection, statement); |
| } | } |
| begin_transaction(connection); | _transaction_begin(connection); |
| } | } |
| void disconnect(void *aconnection) { | void disconnect(void *aconnection){ |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| PQfinish(connection.conn); | PQfinish(connection.conn); |
| connection.conn=0; | connection.conn=0; |
| } | } |
| void commit(void *aconnection) { | void commit(void *aconnection){ |
| execute_transaction_cmd(aconnection, "COMMIT"); | Connection& connection=*static_cast<Connection*>(aconnection); |
| _transaction_commit(connection); | |
| _transaction_begin(connection); | |
| } | } |
| void rollback(void *aconnection) { | void rollback(void *aconnection){ |
| execute_transaction_cmd(aconnection, "ROLLBACK"); | Connection& connection=*static_cast<Connection*>(aconnection); |
| _transaction_rollback(connection); | |
| _transaction_begin(connection); | |
| } | } |
| bool ping(void *aconnection) { | bool ping(void *aconnection) { |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| return PQstatus(connection.conn)==CONNECTION_OK; | return PQstatus(connection.conn)==CONNECTION_OK; |
| } | } |
| const char* quote( | // charset here is services.request_charset(), not connection.client_charset |
| void *aconnection, | // thus we can't use the sql server quoting support |
| const char *from, unsigned int length) { | const char* quote(void *aconnection, const char *str, unsigned int length) |
| { | |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| char *result=(char*)connection.services->malloc_atomic(length*2+1); | const char* from; |
| int err=0; | const char* from_end=str+length; |
| PQescapeStringConn (connection.conn, | |
| result, from, length, | size_t quoted=0; |
| &err); | |
| if(connection.standard_conforming_strings){ | |
| for(from=str; from<from_end; from++){ | |
| if(*from=='\'') | |
| quoted++; | |
| } | |
| } else { | |
| for(from=str; from<from_end; from++){ | |
| switch (*from) { | |
| case '\'': | |
| case '\\': | |
| quoted++; | |
| } | |
| } | |
| } | |
| if(!quoted) | |
| return str; | |
| char *result=(char*)connection.services->malloc_atomic(length + quoted + 1); | |
| char *to = result; | |
| if(connection.standard_conforming_strings){ | |
| for(from=str; from<from_end; from++){ | |
| if(*from=='\'') | |
| *to++= '\''; // "'" -> "''" | |
| *to++=*from; | |
| } | |
| } else { | |
| for(from=str; from<from_end; from++){ | |
| switch (*from) { | |
| case '\'': // "'" -> "''" | |
| *to++= '\''; | |
| break; | |
| case '\\': // "\" -> "\\" | |
| *to++='\\'; | |
| break; | |
| } | |
| *to++=*from; | |
| } | |
| } | |
| *to=0; | |
| return result; | return result; |
| } | } |
| void query(void *aconnection, | void query(void *aconnection, |
| const char *astatement, | const char *astatement, |
| size_t placeholders_count, Placeholder* placeholders, | size_t placeholders_count, Placeholder* placeholders, |
| unsigned long offset, unsigned long limit, | unsigned long offset, unsigned long limit, |
| SQL_Driver_query_event_handlers& handlers) { | SQL_Driver_query_event_handlers& handlers |
| // _asm int 3; | ){ |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| const char* cstrClientCharset=connection.cstrClientCharset; | |
| SQL_Driver_services& services=*connection.services; | SQL_Driver_services& services=*connection.services; |
| PGconn *conn=connection.conn; | PGconn *conn=connection.conn; |
| 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; | |
| const char** paramValues; | const char** paramValues; |
| if(placeholders_count>0){ | if(placeholders_count>0){ |
| //services._throw("bind variables not supported (yet)"); | int binds_size=sizeof(char)*placeholders_count; |
| int binds_size=sizeof(char) * placeholders_count; | |
| paramValues = static_cast<const char**>(services.malloc_atomic(binds_size)); | paramValues = static_cast<const char**>(services.malloc_atomic(binds_size)); |
| bind_parameters(placeholders_count, placeholders, paramValues, connection); | _bind_parameters(placeholders_count, placeholders, paramValues, connection, transcode_needed); |
| } | } |
| // transcode from $request:charset to ?ClientCharset | size_t statement_size=0; |
| if(cstrClientCharset) { | if(transcode_needed){ |
| size_t transcoded_statement_size; | // transcode query from $request:charset to ?ClientCharset |
| services.transcode(astatement, strlen(astatement), | statement_size=strlen(astatement); |
| astatement, transcoded_statement_size, | services.transcode(astatement, statement_size, |
| services.request_charset(), | astatement, statement_size, |
| cstrClientCharset); | request_charset, |
| client_charset); | |
| } | } |
| const char *statement=preprocess_statement(connection, | const char *statement=_preprocess_statement(connection, astatement, statement_size, offset, limit); |
| astatement, offset, limit); | // error after prepare? |
| PGresult *res; | PGresult *res; |
| if(placeholders_count>0){ | if(placeholders_count>0){ |
| Line 263 public: | Line 346 public: |
| if(!res) | if(!res) |
| throwPQerror; | throwPQerror; |
| bool failed=false; | |
| SQL_Error sql_error; | |
| switch(PQresultStatus(res)) { | switch(PQresultStatus(res)) { |
| case PGRES_EMPTY_QUERY: | case PGRES_EMPTY_QUERY: |
| PQclear_throw("no query"); | PQclear_throw("no query"); |
| break; | break; |
| case PGRES_COMMAND_OK: | case PGRES_COMMAND_OK: // empty result: insert|delete|update|... |
| // empty result: insert|delete|update|... | PQclear(res); |
| PQclear(res); | if(connection.autocommit) |
| return; | commit(aconnection); |
| case PGRES_TUPLES_OK: | return; |
| break; | case PGRES_TUPLES_OK: |
| default: | break; |
| PQclear_throwPQerror; | default: |
| break; | PQclear_throwPQerror; |
| break; | |
| } | } |
| int column_count=PQnfields(res); | |
| if(!column_count) | |
| PQclear_throw("result contains no columns"); | |
| bool failed=false; | |
| SQL_Error sql_error; | |
| #define CHECK(afailed) \ | #define CHECK(afailed) \ |
| if(afailed) { \ | if(afailed) { \ |
| failed=true; \ | failed=true; \ |
| goto cleanup; \ | goto cleanup; \ |
| } | } |
| for(int i=0; i<column_count; i++){ | size_t column_count=PQnfields(res); |
| if(!column_count) | |
| PQclear_throw("result contains no columns"); | |
| if(column_count>MAX_COLS) | |
| column_count=MAX_COLS; | |
| unsigned int column_types[MAX_COLS]; | |
| for(size_t i=0; i<column_count; i++){ | |
| column_types[i]=PQftype(res, i); | |
| char *name=PQfname(res, i); | char *name=PQfname(res, i); |
| size_t length=strlen(name); | size_t length=strlen(name); |
| char* strm=(char*)services.malloc(length+1); | const char* str=strdup(services, name, length); |
| memcpy(strm, name, length+1); | |
| const char* str=strm; | |
| // transcode from ?ClientCharset to $request:charset | if(transcode_needed) |
| if(cstrClientCharset) | // transcode column name from ?ClientCharset to $request:charset |
| services.transcode(str, length, | services.transcode(str, length, |
| str, length, | str, length, |
| cstrClientCharset, | client_charset, |
| services.request_charset()); | request_charset); |
| CHECK(handlers.add_column(sql_error, str, length)); | CHECK(handlers.add_column(sql_error, str, length)); |
| } | } |
| Line 312 public: | Line 402 public: |
| if(unsigned long row_count=(unsigned long)PQntuples(res)) | if(unsigned long row_count=(unsigned long)PQntuples(res)) |
| for(unsigned long r=0; r<row_count; r++) { | for(unsigned long r=0; r<row_count; r++) { |
| CHECK(handlers.add_row(sql_error)); | CHECK(handlers.add_row(sql_error)); |
| for(int i=0; i<column_count; i++){ | for(size_t i=0; i<column_count; i++){ |
| const char *cell=PQgetvalue(res, r, i); | char *cell=PQgetvalue(res, r, i); |
| size_t length; | |
| const char* str; | |
| if(PQftype(res, i)==OIDOID) { | |
| // ObjectID column, read object bytes | |
| char *error_pos=0; | size_t length=0; |
| Oid oid=cell?atoi(cell):0; | const char* str; |
| int fd=lo_open(conn, oid, INV_READ); | |
| if(fd>=0) { | |
| // seek to end | |
| if(lo_lseek(conn, fd, 0, SEEK_END)<0) | |
| PQclear_throwPQerror; | |
| // get length | |
| int size_tell=lo_tell(conn, fd); | |
| if(size_tell<0) | |
| PQclear_throwPQerror; | |
| // seek to begin | |
| if(lo_lseek(conn, fd, 0, SEEK_SET)<0) | |
| PQclear_throwPQerror; | |
| length=(size_t)size_tell; | |
| if(length) { | |
| // read | |
| char* strm=(char*)services.malloc(length+1); | |
| if(!lo_read_ex(conn, fd, strm, size_tell)) | |
| PQclear_throw("lo_read can not read all bytes of object"); | |
| strm[length]=0; | |
| str=strm; | |
| } else | |
| str=0; | |
| if(lo_close(conn, fd)<0) | |
| PQclear_throwPQerror; | |
| } else | |
| PQclear_throwPQerror; | |
| } else { | |
| // normal column, read it normally | |
| length=(size_t)PQgetlength(res, r, i); | |
| if(length) { | |
| char* strm=(char*)services.malloc(length+1); | |
| memcpy(strm, cell, length+1); | |
| str=strm; | |
| } else | |
| str=0; | |
| } | |
| if(str && length) { | switch(column_types[i]){ |
| // transcode from ?ClientCharset to $request:charset | case BOOLOID: |
| if(cstrClientCharset) | case INT8OID: |
| services.transcode(str, length, | case INT2OID: |
| str, length, | case INT4OID: |
| cstrClientCharset, | case FLOAT4OID: |
| services.request_charset()); | case FLOAT8OID: |
| case DATEOID: | |
| case TIMEOID: | |
| case TIMESTAMPOID: | |
| case TIMESTAMPTZOID: | |
| case TIMETZOID: | |
| case NUMERICOID: | |
| length=(size_t)PQgetlength(res, r, i); | |
| str=length ? strdup(services, cell, length) : 0; | |
| // transcode is never required for these types | |
| break; | |
| case OIDOID: | |
| { | |
| Oid oid=cell?atoi(cell):0; | |
| int fd=lo_open(conn, oid, INV_READ); | |
| if(fd>=0){ | |
| // seek to end | |
| if(lo_lseek(conn, fd, 0, SEEK_END)<0) | |
| PQclear_throwPQerror; | |
| // get length | |
| int size_tell=lo_tell(conn, fd); | |
| if(size_tell<0) | |
| PQclear_throwPQerror; | |
| // seek to begin | |
| if(lo_lseek(conn, fd, 0, SEEK_SET)<0) | |
| PQclear_throwPQerror; | |
| length=(size_t)size_tell; | |
| if(length){ | |
| // read | |
| char* strm=(char*)services.malloc(length+1); | |
| if(!lo_read_ex(conn, fd, strm, size_tell)) | |
| PQclear_throw("lo_read can not read all bytes of object"); | |
| strm[length]=0; | |
| str=strm; | |
| if(transcode_needed) { | |
| // transcode cell value from ?ClientCharset to $request:charset | |
| services.transcode(str, length, | |
| str, length, | |
| client_charset, | |
| request_charset); | |
| } | |
| } else | |
| str=0; | |
| if(lo_close(conn, fd)<0) | |
| PQclear_throwPQerror; | |
| } else | |
| PQclear_throwPQerror; | |
| break; | |
| } | |
| default: | |
| // normal column, read it normally | |
| length=(size_t)PQgetlength(res, r, i); | |
| str=length ? strdup(services, cell, length) : 0; | |
| if(transcode_needed) { | |
| // transcode cell value from ?ClientCharset to $request:charset | |
| services.transcode(str, length, | |
| str, length, | |
| client_charset, | |
| request_charset); | |
| } | |
| break; | |
| } | } |
| CHECK(handlers.add_row_cell(sql_error, str, length)); | CHECK(handlers.add_row_cell(sql_error, str, length)); |
| } | } |
| } | } |
| Line 374 cleanup: | Line 483 cleanup: |
| PQclear(res); | PQclear(res); |
| if(failed) | if(failed) |
| services._throw(sql_error); | services._throw(sql_error); |
| } | |
| private: // private funcs | if(connection.autocommit) |
| commit(aconnection); | |
| } | |
| void bind_parameters( | private: |
| size_t placeholders_count, | void _bind_parameters( |
| Placeholder* placeholders, | size_t placeholders_count, |
| const char** paramValues, | Placeholder* placeholders, |
| Connection& connection | const char** paramValues, |
| ) { | Connection& connection, |
| for(size_t i=0; i<placeholders_count; i++) { | bool transcode_needed |
| ){ | |
| for(size_t i=0; i<placeholders_count; i++){ | |
| Placeholder& ph=placeholders[i]; | Placeholder& ph=placeholders[i]; |
| size_t value_length; | if(transcode_needed){ |
| if(connection.cstrClientCharset) { | |
| size_t name_length; | size_t name_length; |
| connection.services->transcode(ph.name, strlen(ph.name), | connection.services->transcode(ph.name, strlen(ph.name), |
| ph.name, name_length, | ph.name, name_length, |
| connection.services->request_charset(), | connection.services->request_charset(), |
| connection.cstrClientCharset); | connection.client_charset); |
| if(ph.value) { | if(ph.value) { |
| size_t value_length; | |
| connection.services->transcode(ph.value, strlen(ph.value), | connection.services->transcode(ph.value, strlen(ph.value), |
| ph.value, value_length, | ph.value, value_length, |
| connection.services->request_charset(), | connection.services->request_charset(), |
| connection.cstrClientCharset); | connection.client_charset); |
| } | } |
| } | } |
| if( atoi(ph.name) <= 0 || atoi(ph.name) > placeholders_count) { | int name_number=atoi(ph.name); |
| if(name_number <= 0 || (size_t)name_number > placeholders_count) | |
| connection.services->_throw("bad bind parameter key"); | connection.services->_throw("bad bind parameter key"); |
| } | |
| paramValues[atoi(ph.name)-1] = ph.value; | paramValues[name_number-1]=ph.value; |
| } | } |
| } | } |
| void execute_transaction_cmd(void *aconnection, const char *query) { | void _transaction_begin(Connection& connection){ |
| Connection& connection=*static_cast<Connection*>(aconnection); | _execute_transactions_cmd(connection, "BEGIN"); |
| } | |
| if(connection.autocommit){ | void _transaction_commit(Connection& connection){ |
| Connection& connection=*static_cast<Connection*>(aconnection); | _execute_transactions_cmd(connection, "COMMIT"); |
| execute_resultless(connection, query); | } |
| } | |
| begin_transaction(connection); | void _transaction_rollback(Connection& connection){ |
| _execute_transactions_cmd(connection, "ROLLBACK"); | |
| } | } |
| /** | void _execute_transactions_cmd(const Connection& connection, const char *query){ |
| Executes a query and throws the result. | if(connection.with_default_transactions) // without ?with_default_transaction=1 user must execute BEGIN/COMMIT/ROLLBACK by himself |
| */ | _execute_cmd(connection, query); |
| void execute_resultless(const Connection& connection, const char *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)) | if(PGresult *res=PQexec(connection.conn, query)) |
| PQclear(res); // throw out the result [don't need but must call] | PQclear(res); // throw away the result [don't need but must call] |
| else | else |
| throwPQerror; | throwPQerror; |
| } | } |
| void begin_transaction(Connection& connection) { | const char *_preprocess_statement( |
| if(connection.autocommit){ | Connection& connection, |
| execute_resultless(connection, "BEGIN"); | const char *astatement, |
| } | size_t statement_size, |
| } | unsigned long offset, |
| unsigned long limit | |
| const char *preprocess_statement(Connection& connection, | ){ |
| const char *astatement, unsigned long offset, unsigned long limit) { | |
| PGconn *conn=connection.conn; | PGconn *conn=connection.conn; |
| size_t statement_size=strlen(astatement); | if(!statement_size) |
| statement_size=strlen(astatement); | |
| char *result=(char *)connection.services->malloc(statement_size | char *result=(char *)connection.services->malloc(statement_size |
| +MAX_NUMBER*2+15 // limit # offset # | +MAX_NUMBER*2+15 // " limit # offset #" |
| +MAX_STRING // in case of short 'strings' | +MAX_STRING // in case of short 'strings' |
| +1); | +1); |
| // offset & limit -> suffixes | // offset & limit -> suffixes |
| const char *o; | const char *o; |
| if(offset || limit) { | if(offset || limit!=SQL_NO_LIMIT){ |
| char *cur=result; | char *cur=result; |
| memcpy(cur, astatement, statement_size); cur+=statement_size; | memcpy(cur, astatement, statement_size); cur+=statement_size; |
| if(limit) | 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) | if(offset) |
| cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset); | cur+=snprintf(cur, 8+MAX_NUMBER, " offset %lu", offset); |
| o=result; | o=result; |
| } else | } else |
| o=astatement; | o=astatement; |
| Line 464 private: // private funcs | Line 581 private: // private funcs |
| while(*o) { | while(*o) { |
| if( | if( |
| o[0]=='/' && | o[0]=='/' && |
| o[1]=='*' && | o[1]=='*' && |
| o[2]=='*') { // name start | o[2]=='*') { // name start |
| const char* saved_o=o; | const char* saved_o=o; |
| o+=3; | o+=3; |
| Line 569 private: // conn client library funcs | Line 686 private: // conn client library funcs |
| typedef char* (*t_PQgetvalue)(const PGresult *res, | typedef char* (*t_PQgetvalue)(const PGresult *res, |
| int tup_num, | int tup_num, |
| int field_num); t_PQgetvalue PQgetvalue; | int field_num); t_PQgetvalue PQgetvalue; |
| typedef int (*t_PQntuples)(const PGresult *res); t_PQntuples PQntuples; | typedef int (*t_PQntuples)(const PGresult *res); t_PQntuples PQntuples; |
| typedef char *(*t_PQfname)(const PGresult *res, | typedef char *(*t_PQfname)(const PGresult *res, |
| int field_index); t_PQfname PQfname; | int field_index); t_PQfname PQfname; |
| typedef int (*t_PQnfields)(const PGresult *res); t_PQnfields PQnfields; | typedef int (*t_PQnfields)(const PGresult *res); t_PQnfields PQnfields; |
| Line 595 private: // conn client library funcs | Line 712 private: // conn client library funcs |
| private: // conn client library funcs linking | private: // conn client library funcs linking |
| const char *dlink(const char *dlopen_file_spec) { | const char *dlink(const char *dlopen_file_spec) { |
| if(lt_dlinit()) | if(lt_dlinit()){ |
| return lt_dlerror(); | if(const char* result=lt_dlerror()) |
| return result; | |
| return "can not prepare to dynamic loading"; | |
| } | |
| lt_dlhandle handle=lt_dlopen(dlopen_file_spec); | 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"; | return "can not open the dynamic link module"; |
| } | |
| #define DSLINK(name, action) \ | #define DSLINK(name, action) \ |
| name=(t_##name)lt_dlsym(handle, #name); \ | name=(t_##name)lt_dlsym(handle, #name); \ |