--- parser3/src/main/pa_sql_driver_manager.C 2001/09/14 15:49:06 1.35 +++ parser3/src/main/pa_sql_driver_manager.C 2001/11/12 10:00:32 1.55 @@ -2,10 +2,10 @@ Parser: sql driver manager implementation. Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexander Petrosyan (http://paf.design.ru) - Author: Alexander Petrosyan (http://design.ru/paf) + $Id: pa_sql_driver_manager.C,v 1.55 2001/11/12 10:00:32 paf Exp $ */ -static const char *RCSId="$Id: pa_sql_driver_manager.C,v 1.35 2001/09/14 15:49:06 parser Exp $"; #include "pa_sql_driver_manager.h" #include "ltdl.h" @@ -13,6 +13,9 @@ static const char *RCSId="$Id: pa_sql_dr #include "pa_exception.h" #include "pa_common.h" #include "pa_threads.h" +#include "pa_stack.h" +#include "pa_vhash.h" +#include "pa_vtable.h" // globals @@ -23,6 +26,18 @@ SQL_Driver_manager *SQL_driver_manager; const int EXPIRE_UNUSED_CONNECTION_SECONDS=60; const int CHECK_EXPIRED_CONNECTIONS_SECONDS=EXPIRE_UNUSED_CONNECTION_SECONDS*2; +// helpers + +const String& url_without_login(Pool& pool, const String& url) { + String& result=*new(pool) String(pool); + result << url.mid(0, url.pos(":")) << "://****"; + + int at_pos=url.pos("@"); + if(at_pos>0) + result << url.mid(at_pos, url.size()); + + return result; +} /// SQL_Driver_services Pooled implementation class SQL_Driver_services_impl : public SQL_Driver_services, public Pooled { @@ -31,39 +46,82 @@ public: furl(aurl) { } - /// allocates some bytes on pool - void *malloc(size_t size) { return Pooled::malloc(size); } - /// allocates some bytes clearing them with zeros - void *calloc(size_t size) { return Pooled::calloc(size); } - /// throw exception - void _throw(const char *comment) { - THROW(0, 0, - &furl, + virtual void *malloc(size_t size) { return Pooled::malloc(size, 8); } + virtual void *calloc(size_t size) { return Pooled::calloc(size); } + + /** + the idea is to #1 jump to C++ some function to main body, where + every function stack frame has exception unwind information + and from there... #2 propagate_exception() + */ + virtual void _throw(const char *comment) { + // hiding passwords and addresses from accidental show [imagine user forgot @exception] + e=Exception(0, 0, + &url_without_login(pool(), furl), comment); + + longjmp(mark, 1); + } + virtual void propagate_exception() { + throw e; } private: const String& furl; + Exception e; }; +// helpers + +static void expire_connection(Array::Item *value, void *info) { + SQL_Connection& connection=*static_cast(value); + time_t older_dies=reinterpret_cast(info); + + if(connection.connected() && connection.expired(older_dies)) + connection.disconnect(); +} +static void expire_connections(const Hash::Key& key, Hash::Val *value, void *info) { + Stack& stack=*static_cast(value); + for(int i=0; i<=stack.top_index(); i++) + expire_connection(stack.get(i), info); +} + // SQL_Driver_manager +SQL_Driver_manager::SQL_Driver_manager(Pool& apool) : Cache_manager(apool), + driver_cache(apool), + connection_cache(apool), + prev_expiration_pass_time(0) { +} + +SQL_Driver_manager::~SQL_Driver_manager() { + connection_cache.for_each(expire_connections, + reinterpret_cast((time_t)0/*=in past=expire all*/)); +} + /// @param request_url protocol://[driver-dependent] -SQL_Connection& SQL_Driver_manager::get_connection(const String& request_url, +SQL_Connection& SQL_Driver_manager::get_connection(const String& request_url, + const String& request_origin, Table *protocol2driver_and_client) { - Pool& pool=request_url.pool(); // request pool + Pool& pool=request_origin.pool(); // request pool // we have table for locating protocol's library if(!protocol2driver_and_client) - PTHROW(0, 0, + throw Exception(0, 0, &request_url, "$"MAIN_SQL_NAME":"MAIN_SQL_DRIVERS_NAME" table must be defined"); + // construct services[request] (deassociates at close) + SQL_Driver_services *services=new(pool) SQL_Driver_services_impl(pool, request_url); + // first trying to get cached connection SQL_Connection *result=get_connection_from_cache(request_url); - if(result && !result->ping()) { // we have some cached connection, is it pingable? - result->disconnect(); // kill unpingabe=dead connection - result=0; + if(result) { + result->set_services(services); + if(!result->ping()) { // we have some cached connection, is it pingable? + result->disconnect(); // kill unpingabe=dead connection + result=0; + } } char *request_url_cstr; @@ -72,12 +130,12 @@ SQL_Connection& SQL_Driver_manager::get_ else { // no cached connection or it were unpingabe: connect/reconnect int pos=request_url.pos("://", 3); if(pos<0) - PTHROW(0, 0, - &request_url, - "no protocol specified"); // NOTE: not THROW, but PTHROW + throw Exception(0, 0, + request_url.size()?&request_url:&request_origin, + "connection string must start with protocol://"); // NOTE: not THROW, but PTHROW // make global_url C-string on global pool - request_url_cstr=request_url.cstr(String::UL_AS_IS); + request_url_cstr=request_url.cstr(); char *global_url_cstr=(char *)malloc(strlen(request_url_cstr)+1); strcpy(global_url_cstr, request_url_cstr); // make global_url string on global pool @@ -95,40 +153,40 @@ SQL_Connection& SQL_Driver_manager::get_ global_protocol_cstr); SQL_Driver *driver; - const String *dlopen_file_spec=0; // first trying to get cached driver if(!(driver=get_driver_from_cache(global_protocol))) { // no cached const String *library=0; + const String *dlopen_file_spec=0; if(protocol2driver_and_client->locate(0, global_protocol)) { if(!(library=protocol2driver_and_client->item(1)) || library->size()==0) - PTHROW(0, 0, + throw Exception(0, 0, protocol2driver_and_client->origin_string(), "driver library column for protocol '%s' is empty", request_protocol_cstr); dlopen_file_spec=protocol2driver_and_client->item(2); } else - PTHROW(0, 0, + throw Exception(0, 0, &request_url, "undefined protocol '%s'", request_protocol_cstr); if(lt_dlinit()) - PTHROW(0, 0, + throw Exception(0, 0, library, "prepare to dynamic loading failed, %s", lt_dlerror()); const char *filename=library->cstr(String::UL_FILE_SPEC); lt_dlhandle handle=lt_dlopen(filename); if (!handle) - PTHROW(0, 0, + throw Exception(0, 0, library, "can not open the module, %s", lt_dlerror()); SQL_Driver_create_func create=(SQL_Driver_create_func)lt_dlsym(handle, SQL_DRIVER_CREATE_NAME); if(!create) - PTHROW(0, 0, + throw Exception(0, 0, library, "function '"SQL_DRIVER_CREATE_NAME"' was not found"); @@ -138,22 +196,21 @@ SQL_Connection& SQL_Driver_manager::get_ // validate driver api version int driver_api_version=driver->api_version(); if(driver_api_version!=SQL_DRIVER_API_VERSION) - PTHROW(0, 0, + throw Exception(0, 0, library, "driver implements API version 0x%04X not equal to 0x%04X", driver_api_version, SQL_DRIVER_API_VERSION); // initialise by connecting to sql client dynamic link library - bool specified_dlopen_file_spec=dlopen_file_spec && dlopen_file_spec->size(); - const char *dlopen_file_spec_cstr= - specified_dlopen_file_spec? - dlopen_file_spec->cstr(String::UL_FILE_SPEC):0; + char *dlopen_file_spec_cstr= + dlopen_file_spec && dlopen_file_spec->size()? + dlopen_file_spec->cstr(String::UL_AS_IS):0; if(const char *error=driver->initialize( dlopen_file_spec_cstr)) - PTHROW(0, 0, + throw Exception(0, 0, library, "driver failed to initialize client library '%s', %s", - specified_dlopen_file_spec?dlopen_file_spec_cstr:"unspecifed", + dlopen_file_spec_cstr?dlopen_file_spec_cstr:"unspecifed", error); // cache it @@ -164,10 +221,10 @@ SQL_Connection& SQL_Driver_manager::get_ // associate with services[request] // NOTE: never freed up! result=new(this->pool()) SQL_Connection(this->pool(), global_url, *driver); + // associate with services[request] (deassociates at close) + result->set_services(services); } - // associate with services[request] (deassociates at close) - result->set_services(new(pool) SQL_Driver_services_impl(pool, request_url)); // if not connected yet, do that now, when result has services if(!result->connected()) result->connect(request_url_cstr); @@ -203,8 +260,6 @@ void SQL_Driver_manager::put_driver_to_c SQL_Connection *SQL_Driver_manager::get_connection_from_cache(const String& url) { SYNCHRONIZED; - maybe_expire_connection_cache(); - if(Stack *connections=static_cast(connection_cache.get(url))) while(connections->top_index()>=0) { // there are cached connections to that 'url' SQL_Connection *result=static_cast(connections->pop()); @@ -227,19 +282,7 @@ void SQL_Driver_manager::put_connection_ connections->push(&connection); } -static void expire_connection(Array::Item *value, void *info) { - SQL_Connection& connection=*static_cast(value); - time_t older_dies=reinterpret_cast(info); - - if(connection.connected() && connection.expired(older_dies)) - connection.disconnect(); -} -static void expire_connections(const Hash::Key& key, Hash::Val *value, void *info) { - Stack& stack=*static_cast(value); - for(int i=0; i<=stack.top_index(); i++) - expire_connection(stack.get(i), info); -} -void SQL_Driver_manager::maybe_expire_connection_cache() { +void SQL_Driver_manager::maybe_expire_cache() { time_t now=time(0); if(prev_expiration_pass_time(value); + Table& table=*static_cast(info); + + if(connection.connected()) { + Pool& pool=table.pool(); + Array& row=*new(pool) Array(pool); + + // url + row+=&url_without_login(pool, connection.get_url()); + // time + time_t time_stamp=connection.get_time_stamp(); + const char *unsafe_time_cstr=ctime(&time_stamp); + int time_buf_size=strlen(unsafe_time_cstr); + char *safe_time_buf=(char *)pool.malloc(time_buf_size); + memcpy(safe_time_buf, unsafe_time_cstr, time_buf_size); + row+=new(pool) String(pool, safe_time_buf, time_buf_size); + + table+=&row; + } +} +static void add_connections_to_status_cache_table(const Hash::Key& key, Hash::Val *value, void *info) { + Stack& stack=*static_cast(value); + Array_iter iter(stack); + for(int countdown=stack.top_index(); countdown-->=0; ) + add_connection_to_status_cache_table(iter.next(), info); +} +Value& SQL_Driver_manager::get_status(Pool& pool, const String *source) { + VHash& result=*new(pool) VHash(pool); + + // cache + { + Array& columns=*new(pool) Array(pool); + columns+=new(pool) String(pool, "url"); + columns+=new(pool) String(pool, "time"); + Table& table=*new(pool) Table(pool, 0, &columns, connection_cache.size()); + + connection_cache.for_each(add_connections_to_status_cache_table, &table); + + result.hash(source).put(*new(pool) String(pool, "cache"), new(pool) VTable(pool, &table)); + } + + return result; +} \ No newline at end of file