|
|
| version 1.8, 2001/10/24 11:06:26 | version 1.66, 2024/11/04 03:53:25 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b hashfile parser class. | Parser: @b hashfile parser class. |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2024 Art. Lebedev Studio (http://www.artlebedev.com) |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | Authors: Konstantin Morshnev <moko@design.ru>, Alexandr Petrosian <paf@design.ru> |
| $Id$ | |
| */ | */ |
| #include "pa_config_includes.h" | |
| #include "classes.h" | #include "classes.h" |
| #ifdef HAVE_LIBDB | |
| #include "pa_request.h" | #include "pa_request.h" |
| #include "pa_vmethod_frame.h" | |
| #include "pa_vhashfile.h" | #include "pa_vhashfile.h" |
| #include "pa_vhash.h" | #include "pa_vhash.h" |
| // defines | volatile const char * IDENT_HASHFILE_C="$Id$"; |
| #define HASH_CLASS_NAME "hashfile" | |
| // class | // class |
| class MHashfile : public Methoded { | class MHashfile : public Methoded { |
| public: // VStateless_class | public: // VStateless_class |
| Value *create_new_value(Pool& pool) { return new(pool) VHashfile(pool); } | Value *create_new_value(Pool& apool) { return new VHashfile(apool); } |
| public: | public: |
| MHashfile(Pool& pool); | MHashfile(); |
| public: // Methoded | |
| bool used_directly() { return true; } | |
| }; | }; |
| // methods | // global variable |
| static void _open(Request& r, const String& method_name, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| // file_spec | |
| const String &file_spec=params->as_string(0, "filename must be string"); | |
| self.set_connection( | DECLARE_CLASS_VAR(hashfile, new MHashfile); |
| DB_manager->get_connection( | |
| r.absolute(file_spec), | |
| method_name) | |
| ); | |
| } | |
| // defines for statics | |
| static void _transaction(Request& r, const String& method_name, MethodParams *params) { | #define OPEN_DATA_NAME "HASHFILE-OPEN-DATA" |
| Pool& pool=r.pool(); | |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| // body code | |
| Value& body_code=params->as_junction(0, "body must be code"); | |
| // connection | // methods |
| DB_Connection& connection=self.get_connection(&method_name); | |
| // transaction | typedef HashString<bool> HashStringBool; |
| Auto_transaction transaction(connection); | |
| // execute body | static void _open(Request& r, MethodParams& params) { |
| try { | HashStringBool* file_list=static_cast<HashStringBool*>(r.classes_conf.get(OPEN_DATA_NAME)); |
| r.write_assign_lang(r.process(body_code)); | if(!file_list) { |
| } catch(...) { // process/commit problem | file_list=new HashStringBool(); |
| transaction.mark_to_rollback(); | r.classes_conf.put(OPEN_DATA_NAME, file_list); |
| /*re*/throw; | |
| } | } |
| } | |
| static void _clear(Request& r, const String& method_name, MethodParams *params) { | const String& file_spec=r.full_disk_path(params.as_string(0, FILE_NAME_MUST_BE_STRING)); |
| Pool& pool=r.pool(); | if(file_list->get(file_spec)) |
| throw Exception(PARSER_RUNTIME, | |
| // file_spec | 0, |
| const String &file_spec=params->as_string(0, "filename must be string"); | "this hashfile is already opened, use existing variable"); |
| file_list->put(file_spec, true); | |
| DB_manager->clear_dbfile(r.absolute(file_spec)); | VHashfile& self=GET_SELF(r, VHashfile); |
| self.open(file_spec); | |
| } | } |
| static void _hash(Request& r, const String& method_name, MethodParams *params) { | static void _hash(Request& r, MethodParams&) { |
| Pool& pool=r.pool(); | VHashfile& self=GET_SELF(r, VHashfile); |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| // write out result | // write out result |
| VHash& result=*new(pool) VHash(pool, *self.get_hash(&method_name)); | VHash& result=*new VHash(*self.get_hash()); |
| result.set_name(method_name); | r.write(result); |
| r.write_no_lang(result); | |
| } | } |
| static void _cache(Request& r, const String& method_name, MethodParams *params) { | static void _delete(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | VHashfile& self=GET_SELF(r, VHashfile); |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| // key, expires, body code | if(!params.count()) { |
| const String &key=params->as_string(0, "key must be string"); | // ^hashfile.delete[] asked to delete hashfile itself |
| double expires=params->as_double(1, "expires must be number", r); | self.delete_files(); |
| Value& body_code=params->as_junction(2, "body must be code"); | return; |
| // connection | |
| DB_Connection& connection=self.get_connection(&method_name); | |
| // transaction | |
| Auto_transaction transaction(connection); | |
| // execute body | |
| try { | |
| if(expires) { // 'expires' specified? try cached copy... | |
| if(String *cached_body=connection.get(key)) { // have cached copy? | |
| r.write_assign_lang(*cached_body); | |
| // happy with it | |
| return; | |
| } | |
| } else // 'expires'=0, forget cached copy | |
| connection.remove(key); | |
| // save | |
| Autosave_marked_to_cancel_cache saved(self); | |
| // process | |
| Value& processed_body=r.process(body_code); | |
| r.write_assign_lang(processed_body); | |
| // put it to cache if 'expires' specified & never called ^delete[] | |
| if(expires && !self.marked_to_cancel_cache()) | |
| connection.put(key, processed_body.as_string(), time(0)+(time_t)expires); | |
| } catch(...) { // process/commit problem | |
| transaction.mark_to_rollback(); | |
| /*re*/throw; | |
| } | } |
| } | // key |
| const String &key=params.as_string(0, "key must be string"); | |
| // remove | |
| self.remove(key); | |
| } | |
| static void _clear(Request& r, MethodParams&) { | |
| VHashfile& self=GET_SELF(r, VHashfile); | |
| self.delete_files(); | |
| } | |
| #ifndef DOXYGEN | |
| struct Foreach_info { | |
| Request* r; | |
| const String* key_var_name; | |
| const String* value_var_name; | |
| Value* body_code; | |
| Value* delim_maybe_code; | |
| static void _delete(Request& r, const String& method_name, MethodParams *params) { | Value* var_context; |
| Pool& pool=r.pool(); | bool need_delim; |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | }; |
| #endif | |
| if(params->size()==0) | static bool one_foreach_cycle( |
| self.mark_to_cancel_cache(); | const String::Body key, |
| else { | const String& value, |
| // key | void* ainfo) { |
| const String &key=params->as_string(0, "key must be string"); | Foreach_info& info=*static_cast<Foreach_info*>(ainfo); |
| // connection | if(info.key_var_name){ |
| DB_Connection& connection=self.get_connection(&method_name); | VString* vkey=new VString(*new String(key, String::L_TAINTED)); |
| // remove | info.r->put_element(*info.var_context, *info.key_var_name, vkey); |
| connection.remove(key); | } |
| if(info.value_var_name){ | |
| VString* vvalue=new VString(value); | |
| info.r->put_element(*info.var_context, *info.value_var_name, vvalue); | |
| } | |
| Value& sv_processed=info.r->process(*info.body_code); | |
| TempSkip4Delimiter skip(*info.r); | |
| const String* s_processed=sv_processed.get_string(); | |
| if(info.delim_maybe_code && s_processed && !s_processed->is_empty()) { // delimiter set and we have body | |
| if(info.need_delim) // need delim & iteration produced string? | |
| info.r->write(info.r->process(*info.delim_maybe_code)); | |
| else | |
| info.need_delim=true; | |
| } | } |
| info.r->write(sv_processed); | |
| return skip.check_break(); | |
| } | } |
| static void _foreach(Request& r, MethodParams& params) { | |
| InCycle temp(r); | |
| // constructor | const String& key_var_name=params.as_string(0, "key-var name must be string"); |
| const String& value_var_name=params.as_string(1, "value-var name must be string"); | |
| MHashfile::MHashfile(Pool& apool) : Methoded(apool) { | Foreach_info info={ |
| set_name(*NEW String(pool(), HASH_CLASS_NAME)); | &r, |
| key_var_name.is_empty()? 0 : &key_var_name, | |
| value_var_name.is_empty()? 0 : &value_var_name, | |
| ¶ms.as_junction(2, "body must be code"), | |
| /*delimiter*/params.count()>3 ? ¶ms[3] : 0, | |
| /*var_context*/r.get_method_frame()->caller(), | |
| false | |
| }; | |
| // ^hashfile::open[filename] | VHashfile& self=GET_SELF(r, VHashfile); |
| add_native_method("open", Method::CT_DYNAMIC, _open, 1, 1); | self.for_each(one_foreach_cycle, &info); |
| // ^transaction{code} | |
| add_native_method("transaction", Method::CT_DYNAMIC, _transaction, 1, 1); | |
| // ^hashfile:clear[filename] | |
| add_native_method("clear", Method::CT_STATIC, _clear, 1, 1); | |
| // ^hash[] | |
| add_native_method("hash", Method::CT_DYNAMIC, _hash, 0, 0); | |
| // ^cache[key](seconds){code} | |
| add_native_method("cache", Method::CT_DYNAMIC, _cache, 3, 3); | |
| // ^hashfile.delete[key] | |
| add_native_method("delete", Method::CT_DYNAMIC, _delete, 0, 1); | |
| } | } |
| // global variable | static bool one_cleanup_cycle(const String::Body, const String&, void*) { |
| return false; | |
| } | |
| static void _cleanup(Request& r, MethodParams&) { | |
| VHashfile& self=GET_SELF(r, VHashfile); | |
| Methoded *hashfile_base_class; | self.for_each(one_cleanup_cycle, 0); |
| } | |
| #endif | static void _release(Request& r, MethodParams&) { |
| VHashfile& self=GET_SELF(r, VHashfile); | |
| self.close(); | |
| } | |
| // creator | // constructor |
| Methoded *MHashfile_create(Pool& pool) { | MHashfile::MHashfile(): Methoded("hashfile") { |
| return | // ^hashfile::open[filename] |
| #ifdef HAVE_LIBDB | add_native_method("open", Method::CT_DYNAMIC, _open, 1, 1); |
| hashfile_base_class=new(pool) MHashfile(pool) | // ^hashfile.hash[] |
| #else | add_native_method("hash", Method::CT_DYNAMIC, _hash, 0, 0); |
| 0 | // ^hashfile.delete[key] |
| #endif | add_native_method("delete", Method::CT_DYNAMIC, _delete, 0, 1); |
| ; | // ^hashfile.clear[] -- for backward compatibility. use .delete[] instead. |
| add_native_method("clear", Method::CT_DYNAMIC, _clear, 0, 0); | |
| // ^hashfile.release[] | |
| add_native_method("release", Method::CT_DYNAMIC, _release, 0, 0); | |
| // ^hashfile.cleanup[] | |
| add_native_method("cleanup", Method::CT_DYNAMIC, _cleanup, 0, 0); | |
| add_native_method("defecate", Method::CT_DYNAMIC, _cleanup, 0, 0); | |
| // ^hashfile.foreach[key;value]{code}[delim] | |
| add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); | |
| } | } |