|
|
| version 1.12, 2001/10/27 10:14:45 | version 1.62, 2017/02/07 22:00:27 |
|---|---|
| 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-2017 Art. Lebedev Studio (http://www.artlebedev.com) |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | Author: Alexandr Petrosian <paf@design.ru> (http://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) { | DECLARE_CLASS_VAR(hashfile, new MHashfile); |
| Pool& pool=r.pool(); | |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| self.assign( | |
| params->as_string(0, "DB_HOME must be string"), | |
| params->as_string(1, "filename must be string") | |
| ); | |
| } | |
| static void _transaction(Request& r, const String& method_name, MethodParams *params) { | // defines for statics |
| Pool& pool=r.pool(); | |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | #define OPEN_DATA_NAME "HASHFILE-OPEN-DATA" |
| // body code | |
| Value& body_code=params->as_junction(0, "body must be code"); | |
| // table | // methods |
| DB_Table_ptr table_ptr=self.get_table_ptr(&method_name); | |
| // transaction | typedef HashString<bool> HashStringBool; |
| DB_Transaction transaction(pool, *table_ptr, self.current_transaction); | |
| // 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; | |
| } | } |
| const String& file_spec=r.absolute(params.as_string(0, FILE_NAME_MUST_BE_STRING)); | |
| if(file_list->get(file_spec)) | |
| throw Exception(PARSER_RUNTIME, | |
| 0, | |
| "this hashfile is already opened, use existing variable"); | |
| file_list->put(file_spec, true); | |
| 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; |
| // transaction | |
| DB_Transaction transaction(pool, *self.get_table_ptr(&method_name), self.current_transaction); | |
| // execute body | |
| try { | |
| if(expires) { // 'expires' specified? try cached copy... | |
| if(String *cached_body=transaction.get(key)) { // have cached copy? | |
| r.write_assign_lang(*cached_body); | |
| // happy with it | |
| return; | |
| } | |
| } else // 'expires'=0, forget cached copy | |
| transaction.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()) | |
| transaction.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); |
| // remove | if(info.key_var_name){ |
| self.get_table_ptr(&method_name)->remove(self.current_transaction, key); | VString* vkey=new VString(*new String(key, String::L_TAINTED)); |
| info.r->put_element(*info.var_context, *info.key_var_name, vkey); | |
| } | |
| if(info.value_var_name){ | |
| VString* vvalue=new VString(value); | |
| info.r->put_element(*info.var_context, *info.value_var_name, vvalue); | |
| } | } |
| } | |
| static void _clear(Request& r, const String& method_name, MethodParams *) { | Value& sv_processed=info.r->process(*info.body_code); |
| Pool& pool=r.pool(); | TempSkip4Delimiter skip(*info.r); |
| VHashfile& self=*static_cast<VHashfile *>(r.self); | |
| DB_Cursor cursor( | |
| *self.get_table_ptr(&method_name), | |
| self.current_transaction, &method_name); | |
| while(true) { | |
| if(!cursor.move(DB_NEXT)) | |
| break; | |
| cursor.remove(0/*flags*/); | 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[db_home;filename] | VHashfile& self=GET_SELF(r, VHashfile); |
| add_native_method("open", Method::CT_DYNAMIC, _open, 2, 2); | self.for_each(one_foreach_cycle, &info); |
| // ^transaction{code} | |
| add_native_method("transaction", Method::CT_DYNAMIC, _transaction, 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); | |
| // ^hashfile.clear[] | |
| add_native_method("clear", Method::CT_DYNAMIC, _clear, 0, 0); | |
| } | } |
| // 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); | |
| } | } |