|
|
| version 1.6, 2001/07/31 14:22:00 | version 1.12, 2001/08/24 07:04:25 |
|---|---|
| Line 14 static const char *RCSId="$Id$"; | Line 14 static const char *RCSId="$Id$"; |
| #include "pa_sql_driver.h" | #include "pa_sql_driver.h" |
| #include <libpq-fe.h> | #include <libpq-fe.h> |
| #include <libpq-fs.h> | #include <libpq/libpq-fs.h> |
| // #include <catalog/pg_type.h> | // OIDOID from catalog/pg_type.h |
| #define OIDOID 26 | #define OIDOID 26 |
| #define LO_BUFSIZE 8192 | // LO_BUFSIZE from interfaces\libpq\fe-lobj.c = 8192 (0x2000) |
| // actually writing chunks of that size failed, reduced it twice | |
| #define LO_BUFSIZE 0x1000 | |
| // from postgres_ext.h | |
| #define InvalidOid ((Oid) 0) | |
| #include "ltdl.h" | #include "ltdl.h" |
| Line 63 public: | Line 67 public: |
| 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"; |
| } | } |
| #define throwPQerror services._throw(PQerrorMessage(conn)) | |
| /** connect | /** connect |
| @param used_only_in_connect_url | @param used_only_in_connect_url |
| format: @b user:pass@host[:port]|[local]/database | format: @b user:pass@host[:port]|[local]/database |
| 3.23.22b | |
| Currently the only option for @b character_set_name is cp1251_koi8. | |
| WARNING: must be used only to connect, for buffer doesn't live long | |
| */ | */ |
| void connect( | void connect( |
| char *used_only_in_connect_url, | char *used_only_in_connect_url, |
| Line 81 public: | Line 85 public: |
| char *pwd=lsplit(user, ':'); | char *pwd=lsplit(user, ':'); |
| char *port=lsplit(host, ':'); | char *port=lsplit(host, ':'); |
| // _asm int 3; | |
| PGconn *conn=PQsetdbLogin( | PGconn *conn=PQsetdbLogin( |
| strcasecmp(host, "local")==0?NULL/* local Unix domain socket */:host, port, | strcasecmp(host, "local")==0?NULL/* local Unix domain socket */:host, port, |
| NULL, NULL, db, user, pwd); | NULL, NULL, db, user, pwd); |
| if(!conn) | if(!conn) |
| services._throw("PQsetdbLogin failed"); | services._throw("PQsetdbLogin failed"); |
| if(PQstatus(conn)!=CONNECTION_OK) | if(PQstatus(conn)!=CONNECTION_OK) |
| services._throw(PQerrorMessage(conn)); | throwPQerror; |
| *(PGconn **)connection=conn; | *(PGconn **)connection=conn; |
| begin_transaction(services, conn); | begin_transaction(services, conn); |
| } | } |
| void disconnect(SQL_Driver_services&, void *connection) { | void disconnect(void *connection) { |
| // _asm int 3; | |
| PQfinish((PGconn *)connection); | PQfinish((PGconn *)connection); |
| } | } |
| void commit(SQL_Driver_services& services, void *connection) { | void commit(SQL_Driver_services& services, void *connection) { |
| // _asm int 3; | |
| PGconn *conn=(PGconn *)connection; | PGconn *conn=(PGconn *)connection; |
| if(PGresult *res=PQexec(conn, "COMMIT")) | if(PGresult *res=PQexec(conn, "COMMIT")) |
| PQclear(res); | PQclear(res); |
| else | else |
| services._throw(PQerrorMessage(conn)); | throwPQerror; |
| begin_transaction(services, conn); | begin_transaction(services, conn); |
| } | } |
| void rollback(SQL_Driver_services& services, void *connection) { | void rollback(SQL_Driver_services& services, void *connection) { |
| // _asm int 3; | |
| PGconn *conn=(PGconn *)connection; | PGconn *conn=(PGconn *)connection; |
| if(PGresult *res=PQexec(conn, "ROLLBACK")) | if(PGresult *res=PQexec(conn, "ROLLBACK")) |
| PQclear(res); | PQclear(res); |
| else | else |
| services._throw(PQerrorMessage(conn)); | throwPQerror; |
| begin_transaction(services, conn); | begin_transaction(services, conn); |
| } | } |
| Line 158 public: | Line 158 public: |
| PGresult *res=PQexec(conn, statement); | PGresult *res=PQexec(conn, statement); |
| if(!res) | if(!res) |
| services._throw(PQerrorMessage(conn)); | throwPQerror; |
| switch(PQresultStatus(res)) { | switch(PQresultStatus(res)) { |
| case PGRES_EMPTY_QUERY: | case PGRES_EMPTY_QUERY: |
| Line 198 public: | Line 198 public: |
| void *ptr; | void *ptr; |
| if(PQftype(res, i)==OIDOID) { | if(PQftype(res, i)==OIDOID) { |
| // ObjectID column, read object bytes | // ObjectID column, read object bytes |
| // _asm int 3; | |
| char *error_pos=0; | char *error_pos=0; |
| Oid oid=cell?atoi(cell):0; | Oid oid=cell?atoi(cell):0; |
| Line 218 public: | Line 217 public: |
| if(size) { | if(size) { |
| // read | // read |
| ptr=services.malloc(size); | ptr=services.malloc(size); |
| char *buf=(char *)ptr; | if(!lo_read_ex(conn, fd, (const char *)ptr, size_tell)) |
| int countdown=size_tell; | |
| int size_read; | |
| while(countdown && (size_read=lo_read(conn, fd, buf, min(LO_BUFSIZE, countdown)))>0) { | |
| buf+=size_read; | |
| countdown-=size_read; | |
| } | |
| if(countdown) | |
| PQclear_throw("lo_read can not read all bytes of object"); | PQclear_throw("lo_read can not read all bytes of object"); |
| } else | } else |
| ptr=0; | ptr=0; |
| Line 234 public: | Line 226 public: |
| } else | } else |
| PQclear_throwPQerror; | PQclear_throwPQerror; |
| } else { | } else { |
| // normal column, read it as ASCII string | // normal column, read it normally |
| size=(size_t)PQgetlength(res, r, i); | size=(size_t)PQgetlength(res, r, i); |
| if(size) { | if(size) { |
| ptr=services.malloc(size); | ptr=services.malloc(size); |
| Line 255 private: // private funcs | Line 247 private: // private funcs |
| if(PGresult *res=PQexec(conn, "BEGIN")) | if(PGresult *res=PQexec(conn, "BEGIN")) |
| PQclear(res); | PQclear(res); |
| else | else |
| services._throw(PQerrorMessage(conn)); | throwPQerror; |
| } | } |
| const char *preprocess_statement(SQL_Driver_services& services, PGconn *conn, | const char *preprocess_statement(SQL_Driver_services& services, PGconn *conn, |
| const char *astatement, unsigned long offset, unsigned long limit) { | const char *astatement, unsigned long offset, unsigned long limit) { |
| size_t statement_size=strlen(astatement); | size_t statement_size=strlen(astatement); |
| //_asm int 3; | |
| char *result=(char *)services.malloc(statement_size | char *result=(char *)services.malloc(statement_size |
| +MAX_NUMBER*2+15 // limit # offset # | +MAX_NUMBER*2+15 // limit # offset # |
| Line 280 private: // private funcs | Line 271 private: // private funcs |
| } else | } else |
| o=astatement; | o=astatement; |
| // /*:zzzz*/'literal' -> oid | // /**xxx**/'literal' -> oid |
| char *n=result; | char *n=result; |
| while(*o) { | while(*o) { |
| if( | if( |
| o[0]=='/' && | o[0]=='/' && |
| o[1]=='*' && | o[1]=='*' && |
| o[2]==':') { // name start | o[2]=='*') { // name start |
| o+=3; | o+=3; |
| while(*o) | while(*o) |
| if( | if( |
| o[0]=='*' && | o[0]=='*' && |
| o[1]=='/' && | o[1]=='*' && |
| o[2]=='\'') { // name end | o[2]=='/' && |
| o+=3; | o[3]=='\'') { // name end |
| o+=4; | |
| Oid oid=lo_creat(conn, INV_READ|INV_WRITE); | Oid oid=lo_creat(conn, INV_READ|INV_WRITE); |
| if(oid==InvalidOid) | |
| throwPQerror; | |
| int fd=lo_open(conn, oid, INV_WRITE); | int fd=lo_open(conn, oid, INV_WRITE); |
| const char *start=o; | if(fd>=0) { |
| bool escaped=false; | const char *start=o; |
| while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) { | bool escaped=false; |
| escaped=*o=='\\' || (o[0]=='\'' && o[1]=='\''); | while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) { |
| if(escaped) { | escaped=*o=='\\' || (o[0]=='\'' && o[1]=='\''); |
| // write pending, skip "\" or "'" | if(escaped) { |
| if(o!=start) | // write pending, skip "\" or "'" |
| lo_write(conn, fd, start, o-start); | if(!lo_write_ex(conn, fd, start, o-start)) |
| start=++o; | services._throw("lo_write could not write all bytes of object (1)"); |
| } else | start=++o; |
| o++; | } else |
| } | o++; |
| if(o!=start) | } |
| lo_write(conn, fd, start, o-start); | if(!lo_write_ex(conn, fd, start, o-start)) |
| lo_close(conn, fd); | services._throw("lo_write can not write all bytes of object (2)"); |
| if(lo_close(conn, fd)<0) | |
| throwPQerror; | |
| } else | |
| throwPQerror; | |
| if(*o) | if(*o) |
| o++; // skip "'" | o++; // skip "'" |
| n+=snprintf(n, MAX_NUMBER, "%u", oid); | n+=snprintf(n, MAX_NUMBER, "%u", oid); |
| break; | break; |
| } else | } else |
| o++; // /*:skip*/ | o++; // /**skip**/'xxx' |
| } else | } else |
| *n++=*o++; | *n++=*o++; |
| } | } |
| Line 326 private: // private funcs | Line 324 private: // private funcs |
| return result; | return result; |
| } | } |
| private: // lo_read/write exchancements | |
| bool lo_read_ex(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len) { | |
| int size_read; | |
| while(len && (size_read=lo_read(conn, fd, buf, min(LO_BUFSIZE, len)))>0) { | |
| buf+=size_read; | |
| len-=size_read; | |
| } | |
| return len==0; | |
| } | |
| bool lo_write_ex(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len) { | |
| int size_written; | |
| while(len && (size_written=lo_write(conn, fd, buf, min(LO_BUFSIZE, len)))>0) { | |
| buf+=size_written; | |
| len-=size_written; | |
| } | |
| return len==0; | |
| } | |
| private: // conn client library funcs | private: // conn client library funcs |
| typedef PGconn* (*t_PQsetdbLogin)( | typedef PGconn* (*t_PQsetdbLogin)( |