|
|
| version 1.28, 2001/03/11 08:16:32 | version 1.73, 2001/12/15 21:28:19 |
|---|---|
| Line 1 | Line 1 |
| /* | /** @file |
| Parser | Parser: pool class decl. |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | |
| Author: Alexander Petrosyan <paf@design.ru> (http://paf.design.ru) | |
| $Id$ | $Id$ |
| */ | */ |
| Line 9 | Line 11 |
| #ifndef PA_POOL_H | #ifndef PA_POOL_H |
| #define PA_POOL_H | #define PA_POOL_H |
| #include <stddef.h> | #include "pa_config_includes.h" |
| #ifdef XML | |
| #include <memory> | |
| #include <XalanDOM/XalanDOMString.hpp> | |
| #include <util/TransService.hpp> | |
| #endif | |
| // forwards | |
| class Exception; | class Exception; |
| class Temp_exception; | class String; |
| class Charset; | |
| /** | |
| Pool mechanizm allows users not to free up allocated memory, | |
| leaving that problem to 'pools'. | |
| @see Pooled | |
| */ | |
| class Pool { | class Pool { |
| friend Temp_exception; | |
| public: | public: |
| Pool() : fexception(0) {} | Pool(void *astorage); |
| ~Pool() {} | |
| ///{@ statistics | |
| size_t total_allocated() { return ftotal_allocated; } | |
| unsigned int total_times() { return ftotal_times; } | |
| ///}@ | |
| Exception& exception() const { return *fexception; } | void set_context(void *acontext) { fcontext=acontext; } |
| void *get_context() { return fcontext; } | |
| void *malloc(size_t size) { | /// allocates some bytes on pool |
| return check(real_malloc(size), size); | void *malloc(size_t size, int place=0) { |
| return check(real_malloc(size, place), size); | |
| } | } |
| /// allocates some bytes clearing them with zeros | |
| void *calloc(size_t size) { | void *calloc(size_t size) { |
| return check(real_calloc(size), size); | return check(real_calloc(size), size); |
| } | } |
| private: // implementation defined | /// registers a routine to clean up non-pooled allocations |
| void register_cleanup(void (*cleanup) (void *), void *data) { | |
| if(!real_register_cleanup(cleanup, data)) | |
| fail_register_cleanup(); | |
| } | |
| ///{@ source charset | |
| void set_source_charset(Charset& acharset); | |
| Charset& get_source_charset(); | |
| ///}@ | |
| ///{@ client charset | |
| void set_client_charset(Charset& charset); | |
| Charset& get_client_charset(); | |
| ///}@ | |
| #ifdef XML | |
| const char *transcode_cstr(const XalanDOMString& s); | |
| String& transcode(const XalanDOMString& s); | |
| /// @see Charset::transcode(const String& s) | |
| std::auto_ptr<XalanDOMString> transcode(const String& s); | |
| #endif | |
| private: | |
| void *real_malloc(size_t size); | void *fstorage; |
| void *fcontext; | |
| Charset *source_charset; | |
| Charset *client_charset; | |
| private: | |
| //{ | |
| /// @name implementation defined | |
| void *real_malloc(size_t size, int place); | |
| void *real_calloc(size_t size); | void *real_calloc(size_t size); |
| bool real_register_cleanup(void (*cleanup) (void *), void *data); | |
| //} | |
| private: | private: |
| // checks whether mem allocated OK. throws exception otherwise | /// checks whether mem allocated OK. throws exception otherwise |
| void *check(void *ptr, size_t size) { | void *check(void *ptr, size_t size) { |
| if(ptr) | if(ptr) { |
| ftotal_allocated+=size; | |
| ftotal_times++; | |
| return ptr; | return ptr; |
| } | |
| fail(size); | fail_alloc(size); |
| // never reached | // never reached |
| return 0; | return 0; |
| } | } |
| // throws proper exception | /// throws allocation exception |
| void fail(size_t size) const; | void fail_alloc(size_t size) const; |
| protected: // exception handling | |
| // exception replacement mechanism is 'protected' from direct usage | |
| // Temp_exception_change object enforces paired set/restore | |
| Exception *set_exception(Exception *e){ | |
| Exception *r=fexception; | |
| fexception=e; | |
| return r; | |
| } | |
| void restore_exception(Exception *e) { | |
| fexception=e; | |
| } | |
| private: | /// throws register cleanup exception |
| void fail_register_cleanup() const; | |
| // current request's exception object | private: // statistics |
| Exception *fexception; | |
| size_t ftotal_allocated; | |
| unsigned int ftotal_times; | |
| private: //disabled | private: //disabled |
| // Pool(const Pool&) {} | Pool(const Pool&); |
| Pool& operator = (const Pool&) { return *this; } | Pool& operator= (const Pool&); |
| }; | }; |
| /** | |
| Base for all classes that are allocated in 'pools'. | |
| Holds Pool object. Contains useful wrappers to it's methods. | |
| @see NEW | |
| */ | |
| class Pooled { | class Pooled { |
| // the pool i'm allocated on | // the pool i'm allocated on |
| Pool& fpool; | Pool *fpool; |
| public: | public: |
| /// the Pooled-sole: Pooled instances can be allocated in Pool rather then on heap | |
| static void *operator new(size_t size, Pool& apool) { | static void *operator new(size_t size, Pool& apool) { |
| return apool.malloc(size); | return apool.malloc(size, 1); |
| } | } |
| Pooled(Pool& apool) : fpool(apool) { | Pooled(Pool& apool) : fpool(&apool) {} |
| } | |
| Pool& pool() const { return fpool; } | /// my pool |
| Pool& pool() const { return *fpool; } | |
| void *malloc(size_t size) const { return fpool.malloc(size); } | /** used for moving objects from one pool to another. |
| void *calloc(size_t size) const { return fpool.calloc(size); } | in between object can have no pool and can not be used |
| Exception& exception() const { return fpool.exception(); } | @see SQL_Driver_manager |
| */ | |
| void set_pool(Pool *apool) { fpool=apool; } | |
| //{ | |
| /// @name useful wrapper around pool | |
| void *malloc(size_t size, int place=0) const { return fpool->malloc(size, place); } | |
| void *calloc(size_t size) const { return fpool->calloc(size); } | |
| void register_cleanup(void (*cleanup) (void *), void *data) { fpool->register_cleanup(cleanup, data); } | |
| #ifdef XML | |
| const char *transcode_cstr(const XalanDOMString& s) { return fpool->transcode_cstr(s); } | |
| String& transcode(const XalanDOMString& s) { return fpool->transcode(s); } | |
| #endif | |
| //} | |
| }; | }; |
| /// useful macro for creating objects on current Pooled object Pooled::pool() | |
| #define NEW new(pool()) | #define NEW new(pool()) |
| class Temp_exception { | |
| Pool pool; | |
| Exception *saved_exception; | |
| public: | |
| Temp_exception(Pool& apool, Exception& exception) : | |
| pool(apool), | |
| saved_exception(apool.set_exception(&exception)) { | |
| } | |
| ~Temp_exception() { | |
| pool.restore_exception(saved_exception); | |
| } | |
| }; | |
| #define TRY \ | |
| { \ | |
| Exception temp_exception; \ | |
| Temp_exception le(pool(), temp_exception); \ | |
| if(setjmp(temp_exception.mark)==0) | |
| #define THROW exception()._throw | |
| #define POOL_THROW pool.exception()._throw | |
| #define CATCH(e) \ | |
| else{ \ | |
| Exception& e=temp_exception; | |
| #define END_CATCH \ | |
| } \ | |
| } | |
| // usage: | |
| // TRY { ...; if(?) RAISE(?); ...; } CATCH(e) { catch-code e.comment() } END_CATCH | |
| #endif | #endif |