Annotation of parser3/src/sql/pgsql/parser3pgsql.C, revision 1.1
1.1 ! parser 1: /** @file
! 2: Parser PgSQL driver.
! 3:
! 4: Copyright(c) 2001 ArtLebedev Group(http://www.artlebedev.com)
! 5:
! 6: Author: Alexander Petrosyan <paf@design.ru>(http://design.ru/paf)
! 7:
! 8: 2001.07.30 using PgSQL 7.1.2
! 9: */
! 10: static const char *RCSId="$Id: parser3pgsql.C,v 1.26 2001/07/23 13:59:52 parser Exp $";
! 11:
! 12: #include "config_includes.h"
! 13:
! 14: #include "pa_sql_driver.h"
! 15:
! 16: #include <libpq-fe.h>
! 17: #include "ltdl.h"
! 18:
! 19: #define MAX_STRING 0x400
! 20: #define MAX_NUMBER 20
! 21:
! 22: #if _MSC_VER
! 23: # define snprintf _snprintf
! 24: #endif
! 25: #ifndef strcasecmp
! 26: # define strcasecmp _stricmp
! 27: #endif
! 28:
! 29: static char *lsplit(char *string, char delim) {
! 30: if(string) {
! 31: char *v=strchr(string, delim);
! 32: if(v) {
! 33: *v=0;
! 34: return v+1;
! 35: }
! 36: }
! 37: return 0;
! 38: }
! 39:
! 40: /**
! 41: PgSQL server driver
! 42: */
! 43: class PgSQL_Driver : public SQL_Driver {
! 44: public:
! 45:
! 46: PgSQL_Driver() : SQL_Driver() {
! 47: }
! 48:
! 49: /// get api version
! 50: int api_version() { return SQL_DRIVER_API_VERSION; }
! 51: /// initialize driver by loading sql dynamic link library
! 52: const char *initialize(const char *dlopen_file_spec) {
! 53: return dlopen_file_spec?
! 54: dlink(dlopen_file_spec):"client library column is empty";
! 55: }
! 56: /** connect
! 57: @param used_only_in_connect_url
! 58: format: @b user:pass@host[:port]|[local]/database
! 59: 3.23.22b
! 60: Currently the only option for @b character_set_name is cp1251_koi8.
! 61: WARNING: must be used only to connect, for buffer doesn't live long
! 62: */
! 63: void connect(
! 64: char *used_only_in_connect_url,
! 65: SQL_Driver_services& services,
! 66: void **connection ///< output: PGconn *
! 67: ) {
! 68: char *user=used_only_in_connect_url;
! 69: char *host=lsplit(user, '@');
! 70: char *db=lsplit(host, '/');
! 71: char *pwd=lsplit(user, ':');
! 72: char *port=lsplit(host, ':');
! 73:
! 74: // _asm int 3;
! 75: PGconn *pgsql=PQsetdbLogin(
! 76: strcasecmp(host, "local")==0?NULL/* local Unix domain socket */:host, port,
! 77: NULL, NULL, db, user, pwd);
! 78: if(!pgsql)
! 79: services._throw("PQsetdbLogin failed");
! 80: if(PQstatus(pgsql)!=CONNECTION_OK)
! 81: services._throw(PQerrorMessage(pgsql));
! 82:
! 83: *(PGconn **)connection=pgsql;
! 84: }
! 85: void disconnect(SQL_Driver_services&, void *connection) {
! 86: PQfinish((PGconn *)connection);
! 87: }
! 88: void commit(SQL_Driver_services&, void *) {}
! 89: void rollback(SQL_Driver_services&, void *) {}
! 90:
! 91: bool ping(SQL_Driver_services&, void *connection) {
! 92: return PQstatus((PGconn *)connection)==CONNECTION_OK;
! 93: }
! 94:
! 95: unsigned int quote(
! 96: SQL_Driver_services&, void *connection,
! 97: char *to, const char *from, unsigned int length) {
! 98: /*
! 99: 3.23.22b
! 100: You must allocate the to buffer to be at least length*2+1 bytes long.
! 101: (In the worse case, each character may need to be encoded as using two bytes,
! 102: and you need room for the terminating null byte.)
! 103:
! 104: it's already UNTAINT_TIMES_BIGGER
! 105: */
! 106: // ' -> ''
! 107: unsigned int result=length;
! 108: while(length--) {
! 109: if(*from=='\'')
! 110: *to++='\'';
! 111: *to++=*from++;
! 112: }
! 113: return result;
! 114: }
! 115: void query(
! 116: SQL_Driver_services& services, void *connection,
! 117: const char *astatement, unsigned long offset, unsigned long limit,
! 118: SQL_Driver_query_event_handlers& handlers) {
! 119:
! 120: PGconn *pgsql=(PGconn *)connection;
! 121:
! 122: const char *statement;
! 123: if(offset || limit) {
! 124: size_t statement_size=strlen(astatement);
! 125: char *statement_limited=(char *)services.malloc(
! 126: statement_size+MAX_NUMBER*2+15/* limit # offset #*/+1);
! 127: char *cur=statement_limited;
! 128: memcpy(cur, astatement, statement_size); cur+=statement_size;
! 129: if(limit)
! 130: cur+=snprintf(cur, 7+MAX_NUMBER, " limit %u", limit);
! 131: if(offset)
! 132: cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset);
! 133: statement=statement_limited;
! 134: } else
! 135: statement=astatement;
! 136:
! 137: PGresult *res=PQexec(pgsql, statement);
! 138: if(!res)
! 139: services._throw(PQerrorMessage(pgsql));
! 140:
! 141: switch(PQresultStatus(res)) {
! 142: case PGRES_EMPTY_QUERY:
! 143: PQclear(res);
! 144: services._throw("no query");
! 145: break;
! 146: case PGRES_COMMAND_OK:
! 147: // empty result: insert|delete|update|...
! 148: return;
! 149: case PGRES_TUPLES_OK:
! 150: break;
! 151: default:
! 152: PQclear(res);
! 153: services._throw("unknown PQexec error");
! 154: break;
! 155: }
! 156:
! 157: int column_count=PQnfields(res);
! 158: if(!column_count) {
! 159: PQclear(res);
! 160: services._throw("result contains no columns");
! 161: }
! 162:
! 163: for(int i=0; i<column_count; i++){
! 164: char *name=PQfname(res, i);
! 165: size_t size=strlen(name);
! 166: void *ptr=services.malloc(size);
! 167: memcpy(ptr, name, size);
! 168: handlers.add_column(ptr, size);
! 169: }
! 170:
! 171: handlers.before_rows();
! 172:
! 173: if(unsigned long row_count=(unsigned long)PQntuples(res))
! 174: for(unsigned long r=0; r<row_count; r++) {
! 175: handlers.add_row();
! 176: for(int i=0; i<column_count; i++){
! 177: size_t size=(size_t)PQgetlength(res, r, i);
! 178: void *ptr;
! 179: if(size) {
! 180: ptr=services.malloc(size);
! 181: memcpy(ptr, PQgetvalue(res, r, i), size);
! 182: } else
! 183: ptr=0;
! 184: handlers.add_row_cell(ptr, size);
! 185: }
! 186: }
! 187:
! 188: PQclear(res);
! 189: }
! 190:
! 191: private: // pgsql client library funcs
! 192:
! 193: typedef PGconn* (*t_PQsetdbLogin)(
! 194: const char *pghost,
! 195: const char *pgport,
! 196: const char *pgoptions,
! 197: const char *pgtty,
! 198: const char *dbName,
! 199: const char *login,
! 200: const char *pwd); t_PQsetdbLogin PQsetdbLogin;
! 201: typedef void (*t_PQfinish)(PGconn *conn); t_PQfinish PQfinish;
! 202: typedef char *(*t_PQerrorMessage)(const PGconn* conn); t_PQerrorMessage PQerrorMessage;
! 203: typedef ConnStatusType (*t_PQstatus)(const PGconn *conn); t_PQstatus PQstatus;
! 204: typedef PGresult *(*t_PQexec)(PGconn *conn,
! 205: const char *query); t_PQexec PQexec;
! 206: typedef ExecStatusType (*t_PQresultStatus)(const PGresult *res); t_PQresultStatus PQresultStatus;
! 207: typedef int (*t_PQgetlength)(const PGresult *res,
! 208: int tup_num,
! 209: int field_num); t_PQgetlength PQgetlength;
! 210: typedef char* (*t_PQgetvalue)(const PGresult *res,
! 211: int tup_num,
! 212: int field_num); t_PQgetvalue PQgetvalue;
! 213: typedef int (*t_PQntuples)(const PGresult *res); t_PQntuples PQntuples;
! 214: typedef char *(*t_PQfname)(const PGresult *res,
! 215: int field_index); t_PQfname PQfname;
! 216: typedef int (*t_PQnfields)(const PGresult *res); t_PQnfields PQnfields;
! 217: typedef void (*t_PQclear)(PGresult *res); t_PQclear PQclear;
! 218:
! 219:
! 220: private: // pgsql client library funcs linking
! 221:
! 222: const char *dlink(const char *dlopen_file_spec) {
! 223: lt_dlhandle handle=lt_dlopen(dlopen_file_spec);
! 224: if(!handle)
! 225: return "can not open the dynamic link module";
! 226:
! 227: #define DSLINK(name, action) \
! 228: name=(t_##name)lt_dlsym(handle, #name); \
! 229: if(!name) \
! 230: action;
! 231:
! 232: #define DLINK(name) DSLINK(name, return "function " #name " was not found")
! 233:
! 234: DLINK(PQsetdbLogin);
! 235: DLINK(PQerrorMessage);
! 236: DLINK(PQstatus);
! 237: DLINK(PQfinish);
! 238: DLINK(PQgetvalue);
! 239: DLINK(PQgetlength);
! 240: DLINK(PQntuples);
! 241: DLINK(PQfname);
! 242: DLINK(PQnfields);
! 243: DLINK(PQclear);
! 244: DLINK(PQresultStatus);
! 245: DLINK(PQexec);
! 246:
! 247: return 0;
! 248: }
! 249:
! 250: };
! 251:
! 252: extern "C" SQL_Driver *create() {
! 253: return new PgSQL_Driver();
! 254: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to parser3pgsql.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / sql / pgsql