|
|
| version 1.4, 2001/10/23 12:53:22 | version 1.47, 2009/06/14 00:33:36 |
|---|---|
| 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-2009 ArtLebedev Group (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" | static const char * const IDENT="$Id$"; |
| #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 | |
| #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, HashStringValue*) { return new VHashfile(apool); } |
| public: | public: |
| MHashfile(Pool& pool); | MHashfile(); |
| public: // Methoded | public: // Methoded |
| bool used_directly() { return true; } | 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, 0); |
| 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( | // externs |
| DB_manager->get_connection( | |
| r.absolute(file_spec), | |
| method_name) | |
| ); | |
| } | |
| extern String cycle_data_name; | |
| 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"); | |
| // 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; | |
| } | } |
| 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, MethodParams&) { | |
| VHashfile& self=GET_SELF(r, VHashfile); | |
| // write out result | |
| VHash& result=*new VHash(*self.get_hash()); | |
| r.write_no_lang(result); | |
| } | } |
| static void _delete(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); | |
| if(!params.count()) { | |
| // ^hashfile.delete[] asked to delete hashfile itself | |
| self.delete_files(); | |
| return; | |
| } | |
| // key | // key |
| const String &key=params->as_string(0, "key must be string"); | 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; | |
| Value* var_context; | |
| VString* vkey; | |
| VString* vvalue; | |
| bool need_delim; | |
| }; | |
| #endif | |
| static bool one_foreach_cycle( | |
| const String::Body key, | |
| const String& value, | |
| void* ainfo) { | |
| Foreach_info& info=*static_cast<Foreach_info*>(ainfo); | |
| if(info.key_var_name){ | |
| info.vkey->set_string(*new String(key, String::L_TAINTED)); | |
| info.var_context->put_element(*info.var_context, *info.key_var_name, info.vkey, false); | |
| } | |
| if(info.value_var_name){ | |
| info.vvalue->set_string(value); | |
| info.var_context->put_element(*info.var_context, *info.value_var_name, info.vvalue, false); | |
| } | |
| // connection | StringOrValue sv_processed=info.r->process(*info.body_code); |
| DB_Connection& connection=self.get_connection(&method_name); | Request::Skip lskip=info.r->get_skip(); info.r->set_skip(Request::SKIP_NOTHING); |
| connection._delete(key); | 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_pass_lang(info.r->process(*info.delim_maybe_code)); | |
| else | |
| info.need_delim=true; | |
| } | |
| static void _clear(Request& r, const String& method_name, MethodParams *params) { | info.r->write_pass_lang(sv_processed); |
| Pool& pool=r.pool(); | |
| // file_spec | |
| const String &file_spec=params->as_string(0, "filename must be string"); | |
| DB_manager->clear_dbfile(r.absolute(file_spec)); | return lskip==Request::SKIP_BREAK; |
| } | } |
| static void _foreach(Request& r, MethodParams& params) { | |
| Temp_hash_value<const String::Body, void*> | |
| cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&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 ? params.get(3) : 0, | |
| /*var_context*/r.get_method_frame()->caller(), | |
| /*vkey=*/new VString, | |
| /*vvalue=*/new VString, | |
| 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.delete[key] | |
| add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); | |
| // ^hashfile:clear[filename] | |
| add_native_method("clear", Method::CT_STATIC, _clear, 1, 1); | |
| // ^pack[] | |
| add_native_method("pack", Method::CT_DYNAMIC, _pack, 0, 0); | |
| /* | |
| // ^cache[key](seconds){code} | |
| add_native_method("cache", Method::CT_DYNAMIC, _cache, 3, 3); | |
| // ^cancel[] | |
| add_native_method("cancel", Method::CT_DYNAMIC, _cancel, 0, 0); | |
| // ^hash[] | |
| add_native_method("hash", Method::CT_DYNAMIC, _hash, 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); | |
| } | } |