--- parser3/src/sql/mysql/Attic/parser3mysql.C 2001/04/04 12:13:20 1.3 +++ parser3/src/sql/mysql/Attic/parser3mysql.C 2001/04/05 11:01:59 1.7 @@ -1,44 +1,120 @@ /** @file Parser: MySQL driver. - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) + Copyright(c) 2001 ArtLebedev Group(http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) + Author: Alexander Petrosyan (http://design.ru/paf) - $Id: parser3mysql.C,v 1.3 2001/04/04 12:13:20 paf Exp $ + $Id: parser3mysql.C,v 1.7 2001/04/05 11:01:59 paf Exp $ */ +#include + #include "pa_sql_driver.h" +#include "mysql.h" +#include "pa_common.h" // +char *lsplit(char *string, char delim) { + if(string) { + char *v=strchr(string, delim); + if(v) { + *v=0; + return v+1; + } + } + return 0; +} -/// MySQL server driver +/** + MySQL server driver + + @todo + figure out about memory for errors: + - static=add multithread locks + - dynamic=who should free it up? +*/ class MySQL_Driver : public SQL_Driver { public: - MySQL_Driver() : SQL_Driver(), - info(0) { + MySQL_Driver() : SQL_Driver() { } /// get api version - int api_version() { return SQL_API_VERSION; } - /// connect @test move info inside - const char *connect(const char *url) { - /*return "mysql connect failed";*/ - return 0; - } - const char *disconnect() { - return "mysql disconnect failed"; - } - const char *commit() { - return 0;//"mysql commit failed"; + int api_version() { return SQL_DRIVER_API_VERSION; } + /// connect + void connect( + char *url, ///< @b user:pass@host[:port]/database + void **connection ///< output: MYSQL * + ) { + char *user=url; + char *host=lsplit(user, '@'); + char *db=lsplit(host, '/'); + char *pwd=lsplit(user, ':'); + char *error_pos=0; + char *port_cstr=lsplit(host, ':'); + int port=port_cstr?strtol(port_cstr, &error_pos, 0):0; + + MYSQL *mysql=mysql_init(NULL); + if(!mysql_real_connect(mysql, + host, user, pwd, db, port?port:MYSQL_PORT, NULL, 0)) + fservices->_throw(mysql_error(mysql)); + + *(MYSQL **)connection=mysql; } - const char *rollback() { - return "mysql rollback failed"; + void disconnect(void *connection) { + mysql_close((MYSQL *)connection); } - -private: + void commit(void *connection) {} + void rollback(void *connection) {} - void *info; + void query(void *connection, + const char *statement, + unsigned int *column_count, Cell **columns, + unsigned long *row_count, Cell ***rows) { + + MYSQL *mysql=(MYSQL *)connection; + MYSQL_RES *res=NULL; + + if(mysql_query(mysql, statement)) + fservices->_throw(mysql_error(mysql)); + if(!(res=mysql_store_result(mysql)) && mysql_field_count(mysql)) + fservices->_throw(mysql_error(mysql)); + if(!res) { + // empty result + *row_count=0; + *column_count=0; + return; + } + + *column_count=mysql_num_fields(res); + *columns=(Cell *)fservices->malloc(sizeof(Cell)*(*column_count)); + + *row_count=(unsigned long)mysql_num_rows(res); + *rows=(Cell **)fservices->malloc(sizeof(Cell *)*(*row_count)); + + for(unsigned int i=0; i<(*column_count); i++){ + MYSQL_FIELD *field=mysql_fetch_field(res); + size_t size=strlen(field->name); + (*columns)[i].size=size; + (*columns)[i].ptr=fservices->malloc(size); + memcpy((*columns)[i].ptr, field->name, size); + } + + for(unsigned long r=0; r<(*row_count); r++) + if(MYSQL_ROW mysql_row=mysql_fetch_row(res)) { // never false.. + unsigned long *lengths=mysql_fetch_lengths(res); + Cell *row=(Cell *)malloc(sizeof(Cell)*(*column_count)); + (*rows)[r]=row; + for(unsigned int i=0; i<(*column_count); i++){ + size_t size=(size_t)lengths[i]; + row[i].size=size; + row[i].ptr=fservices->malloc(size); + memcpy(row[i].ptr, mysql_row[i], size); + } + } + + mysql_free_result(res); + } }; extern "C" SQL_Driver *create() {