|
|
| version 1.27, 2002/08/15 10:38:18 | version 1.47, 2020/12/15 17:10:32 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: sql fconnection decl. | Parser: sql fconnection decl. |
| Copyright (c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2020 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) |
| */ | */ |
| #ifndef PA_SQL_CONNECTION_H | #ifndef PA_SQL_CONNECTION_H |
| #define PA_SQL_CONNECTION_H | #define PA_SQL_CONNECTION_H |
| static const char* IDENT_SQL_CONNECTION_H="$Date$"; | #define IDENT_PA_SQL_CONNECTION_H "$Id$" |
| #include "pa_pool.h" | |
| #include "pa_sql_driver.h" | #include "pa_sql_driver.h" |
| #include "pa_sql_driver_manager.h" | #include "pa_sql_driver_manager.h" |
| // defines | // defines |
| /// @see SQL_Driver_services_impl::_throw | /// @see SQL_Driver_services_impl::_throw |
| #define SQL_CONNECTION_SERVICED_FUNC_GUARDED(actions) \ | #ifdef PA_WITH_SJLJ_EXCEPTIONS |
| if(!fservices || !setjmp(fservices->mark)) { \ | #define SQL_CONNECTION_SERVICED_FUNC_GUARDED(actions) \ |
| actions; \ | use(); \ |
| } else \ | actions |
| fservices->propagate_exception(); | #else |
| #define SQL_CONNECTION_SERVICED_FUNC_GUARDED(actions) \ | |
| use(); \ | |
| if(!setjmp(fservices.mark)) { \ | |
| actions; \ | |
| } else \ | |
| fservices.propagate_exception(); | |
| #endif | |
| /// SQL connection. handy wrapper around low level SQL_Driver | /// SQL_Driver_services Pooled implementation |
| class SQL_Connection : public Pooled { | class SQL_Driver_services_impl: public SQL_Driver_services { |
| const String* furl; | |
| Exception fexception; | |
| const char* frequest_charset; | |
| const char* fdocument_root; | |
| public: | |
| SQL_Driver_services_impl(const char* arequest_charset, const char* adocument_root): furl(0), frequest_charset(arequest_charset), fdocument_root(adocument_root) {} | |
| void set_url(const String& aurl) { furl=&aurl;} | |
| const String& url_without_login() const; | |
| override void* malloc(size_t size) { return pa_malloc(size); } | |
| override void* malloc_atomic(size_t size) { return pa_malloc_atomic(size); } | |
| override void* realloc(void *ptr, size_t size) { return pa_realloc(ptr, size); } | |
| override const char* request_charset() { return frequest_charset; } | |
| override const char* request_document_root() { return fdocument_root; } | |
| override void transcode(const char* src, size_t src_length, | |
| const char*& dst, size_t& dst_length, | |
| const char* charset_from_name, | |
| const char* charset_to_name | |
| ); | |
| /** | |
| normally we can't 'throw' from dynamic library, so | |
| 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() | |
| but when parser configured --with-sjlj-exceptions | |
| one can simply 'throw' from dynamic library. | |
| [sad story: one can not longjump/throw due to some bug in gcc as of 3.2.1 version] | |
| */ | |
| override void _throw(const SQL_Error& aexception) { | |
| // converting SQL_exception to parser Exception | |
| // hiding passwords and addresses from accidental show [imagine user forgot @exception] | |
| #ifdef PA_WITH_SJLJ_EXCEPTIONS | |
| throw | |
| #else | |
| fexception= | |
| #endif | |
| Exception(aexception.type() ? aexception.type() : "sql.connect", &url_without_login(), aexception.comment()); | |
| #ifndef PA_WITH_SJLJ_EXCEPTIONS | |
| longjmp(mark, 1); | |
| #endif | |
| } | |
| virtual void propagate_exception() { | |
| #ifndef PA_WITH_SJLJ_EXCEPTIONS | |
| throw fexception; | |
| #endif | |
| } | |
| }; | |
| friend class SQL_Connection_ptr; | /// SQL connection. handy wrapper around low level SQL_Driver |
| class SQL_Connection: public PA_Object { | |
| const String& furl; | |
| SQL_Driver& fdriver; | |
| SQL_Driver_services_impl fservices; | |
| void *fconnection; | |
| time_t time_used; | |
| public: | public: |
| SQL_Connection(Pool& pool, const String& aurl, SQL_Driver& adriver) : Pooled(pool), | SQL_Connection(const String& aurl, SQL_Driver& adriver, const char* arequest_charset, const char* adocument_root): |
| furl(aurl), | furl(aurl), |
| fdriver(adriver), | fdriver(adriver), |
| fservices(arequest_charset, adocument_root), | |
| fconnection(0), | fconnection(0), |
| time_used(0), used(0), | time_used(0) { |
| marked_to_rollback(false) { | |
| } | } |
| SQL_Driver_services_impl& services() { return fservices; } | |
| const String& get_url() { return furl; } | const String& get_url() { return furl; } |
| void set_services(SQL_Driver_services *aservices) { | void set_url() { |
| fservices=aservices; | fservices.set_url(furl); |
| } | |
| void use() { | |
| time_used=time(0); // they started to use at this time | |
| } | } |
| bool expired(time_t older_dies) { | bool expired(time_t older_dies) { |
| return !used && time_used<older_dies; | return time_used<older_dies; |
| } | } |
| time_t get_time_used() { return time_used; } | time_t get_time_used() { return time_used; } |
| bool connected() { return fconnection!=0; } | bool connected() { return fconnection!=0; } |
| void connect(char *used_only_in_connect_url_cstr) { | void connect(char *used_only_in_connect_url_cstr) { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| fdriver.connect(used_only_in_connect_url_cstr, *fservices, &fconnection) | fdriver.connect(used_only_in_connect_url_cstr, fservices, &fconnection) |
| ); | ); |
| } | } |
| void disconnect() { | void disconnect() { |
| Line 59 public: | Line 130 public: |
| } | } |
| bool ping() { | bool ping() { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| return fdriver.ping(*fservices, fconnection) | return fdriver.ping(fconnection) |
| ); | ); |
| return 0; // never reached | return false; // never reached, warning war |
| } | } |
| uint quote(char *to, const char *from, unsigned int length) { | const char* quote(const char* str, unsigned int length) { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| return fdriver.quote(*fservices, fconnection, to, from, length) | return fdriver.quote(fconnection, str, length) |
| ); | ); |
| return 0; // never reached | return NULL; // never reached, warning war |
| } | } |
| void query( | void query( |
| const char *statement, unsigned long offset, unsigned long limit, | const char* statement, |
| size_t placeholders_count, SQL_Driver::Placeholder* placeholders, | |
| unsigned long offset, unsigned long limit, | |
| SQL_Driver_query_event_handlers& handlers, | SQL_Driver_query_event_handlers& handlers, |
| const String& source) { | const String& source) { |
| try { | try { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| fdriver.query(*fservices, fconnection, | fdriver.query(fconnection, statement, placeholders_count, placeholders, offset, limit, handlers) |
| statement, offset, limit, | ); |
| handlers) | |
| ); | |
| } catch(const Exception& e) { // query problem | } catch(const Exception& e) { // query problem |
| // give more specific source [were url] | if(strcmp(e.type(), "sql.connect")==0) { // if it is _throw exception, |
| throw Exception("sql.execute", | // show query instead of connect string |
| &source, | throw Exception("sql.execute", &source, "%s", e.comment()); |
| "%s", e.comment()); | } else |
| rethrow; | |
| } | } |
| } | } |
| void mark_to_rollback() { | |
| marked_to_rollback=true; | |
| } | |
| private: // closing process | |
| void commit() { | void commit() { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| fdriver.commit(*fservices, fconnection) | fdriver.commit(fconnection) |
| ); | ); |
| } | } |
| void rollback() { | void rollback() { |
| SQL_CONNECTION_SERVICED_FUNC_GUARDED( | SQL_CONNECTION_SERVICED_FUNC_GUARDED( |
| fdriver.rollback(*fservices, fconnection) | fdriver.rollback(fconnection) |
| ); | ); |
| } | } |
| /// return to cache | /// return to cache |
| void close() { | void close() { |
| if(marked_to_rollback) { | SQL_driver_manager->close_connection(furl, this); |
| rollback(); | |
| marked_to_rollback=false; | |
| } else | |
| commit(); | |
| SQL_driver_manager->close_connection(furl, *this); | |
| } | } |
| private: // connection usage methods | |
| void use() { | |
| time_used=time(0); // they started to use at this time | |
| used++; | |
| } | |
| void unuse() { | |
| used--; | |
| if(!used) | |
| close(); | |
| } | |
| private: // connection usage data | |
| int used; | |
| private: | |
| const String& furl; | |
| SQL_Driver& fdriver; | |
| SQL_Driver_services *fservices; | |
| void *fconnection; | |
| time_t time_used; | |
| bool marked_to_rollback; | |
| }; | |
| /// Auto-object used to track SQL_Connection usage | |
| class SQL_Connection_ptr { | |
| SQL_Connection *fconnection; | |
| public: | |
| explicit SQL_Connection_ptr(SQL_Connection *aconnection) : fconnection(aconnection) { | |
| fconnection->use(); | |
| } | |
| ~SQL_Connection_ptr() { | |
| fconnection->unuse(); | |
| } | |
| SQL_Connection* operator->() { | |
| return fconnection; | |
| } | |
| SQL_Connection* get() const { | |
| return fconnection; | |
| } | |
| // copying | |
| SQL_Connection_ptr(const SQL_Connection_ptr& src) : fconnection(src.fconnection) { | |
| fconnection->use(); | |
| } | |
| SQL_Connection_ptr& operator =(const SQL_Connection_ptr& src) { | |
| // may do without this=src check | |
| fconnection->unuse(); | |
| fconnection=src.fconnection; | |
| fconnection->use(); | |
| return *this; | |
| } | |
| }; | }; |
| #endif | #endif |