--- parser3/src/classes/hashfile.C 2003/11/06 10:25:40 1.26 +++ parser3/src/classes/hashfile.C 2009/06/16 07:36:01 1.48 @@ -1,12 +1,11 @@ /** @file Parser: @b hashfile parser class. - Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) + Copyright (c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) */ - -static const char* IDENT="$Id: hashfile.C,v 1.26 2003/11/06 10:25:40 paf Exp $"; +static const char * const IDENT="$Id: hashfile.C,v 1.48 2009/06/16 07:36:01 misha Exp $"; #include "classes.h" @@ -19,7 +18,7 @@ static const char* IDENT="$Id: hashfile. class MHashfile : public Methoded { public: // VStateless_class - Value *create_new_value() { return new VHashfile; } + Value *create_new_value(Pool& apool, HashStringValue*) { return new VHashfile(apool); } public: MHashfile(); @@ -31,15 +30,37 @@ public: // Methoded DECLARE_CLASS_VAR(hashfile, new MHashfile, 0); +// externs + +extern String cycle_data_name; + +// defines for statics + +#define OPEN_DATA_NAME "HASHFILE-OPEN-DATA" + // methods +typedef HashString HashStringBool; + static void _open(Request& r, MethodParams& params) { + HashStringBool* file_list=static_cast(r.classes_conf.get(OPEN_DATA_NAME)); + if(!file_list) { + file_list=new HashStringBool(); + r.classes_conf.put(OPEN_DATA_NAME, file_list); + } + + 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(r.absolute(params.as_string(0, "filename must be string"))); + self.open(file_spec); } -static void _hash(Request& r, MethodParams& params) { +static void _hash(Request& r, MethodParams&) { VHashfile& self=GET_SELF(r, VHashfile); // write out result @@ -50,6 +71,11 @@ static void _hash(Request& r, MethodPara static void _delete(Request& r, MethodParams& params) { VHashfile& self=GET_SELF(r, VHashfile); + if(!params.count()) { + // ^hashfile.delete[] asked to delete hashfile itself + self.delete_files(); + return; + } // key const String &key=params.as_string(0, "key must be string"); // remove @@ -58,61 +84,101 @@ static void _delete(Request& r, MethodPa static void _clear(Request& r, MethodParams&) { VHashfile& self=GET_SELF(r, VHashfile); - self.clear(); + self.delete_files(); } -#if LATER +#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; + bool need_delim; +}; +#endif +static bool one_foreach_cycle( + const String::Body key, + const String& value, + void* ainfo) { + Foreach_info& info=*static_cast(ainfo); + if(info.key_var_name){ + VString* vkey=new VString(*new String(key, String::L_TAINTED)); + info.var_context->put_element(*info.var_context, *info.key_var_name, vkey, false); + } + if(info.value_var_name){ + VString* vvalue=new VString(value); + info.var_context->put_element(*info.var_context, *info.value_var_name, vvalue, false); + } + + StringOrValue sv_processed=info.r->process(*info.body_code); + Request::Skip lskip=info.r->get_skip(); info.r->set_skip(Request::SKIP_NOTHING); + + 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; + } + + info.r->write_pass_lang(sv_processed); + + return lskip==Request::SKIP_BREAK; +} static void _foreach(Request& r, MethodParams& params) { - VHashfile& self=GET_SELF(r, VHashfile); + Temp_hash_value + cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r); 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"); - Value& body_code=params.as_junction(2, "body must be code"); - Value *delim_maybe_code=params.size()>3?¶ms.get(3):0; - bool need_delim=false; - Value& var_context=*body_code.get_junction()->wcontext; - VString& vkey=*new(pool) VString(pool); - VString& vvalue=*new(pool) VString(pool); - - DB_Cursor cursor(*self.get_table_ptr(&method_name), &method_name); - while(true) { - String *key; - String *data; - if(!cursor.get(pool, key, data, DB_NEXT)) - break; - - if(!key) - continue; // expired - - vkey.set_string(*key); - vvalue.set_string(*data); - var_context.put_element(key_var_name, &vkey); - var_context.put_element(value_var_name, &vvalue); - - Value& processed_body=r.process(body_code); - if(delim_maybe_code) { // delimiter set? - const String *string=processed_body.get_string(); - if(need_delim && string && string->size()) // need delim & iteration produced string? - r.write_pass_lang(r.process(*delim_maybe_code)); - need_delim=true; - } - r.write_pass_lang(processed_body); - } + Foreach_info info={ + &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(), + false + }; + + VHashfile& self=GET_SELF(r, VHashfile); + self.for_each(one_foreach_cycle, &info); +} + +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); + + self.for_each(one_cleanup_cycle, 0); +} + +static void _release(Request& r, MethodParams&) { + VHashfile& self=GET_SELF(r, VHashfile); + self.close(); } -#endif // constructor MHashfile::MHashfile(): Methoded("hashfile") { - // ^hashfile::open[db_home;filename] + // ^hashfile::open[filename] add_native_method("open", Method::CT_DYNAMIC, _open, 1, 1); - // ^hash[] + // ^hashfile.hash[] add_native_method("hash", Method::CT_DYNAMIC, _hash, 0, 0); // ^hashfile.delete[key] - add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); - // ^hashfile.clear[] + 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); + add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); }