--- parser3/src/classes/hashfile.C 2001/10/23 12:41:05 1.3 +++ parser3/src/classes/hashfile.C 2009/05/05 10:06:57 1.45 @@ -1,131 +1,188 @@ /** @file Parser: @b hashfile parser class. - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: hashfile.C,v 1.3 2001/10/23 12:41:05 parser Exp $ + Copyright (c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ -#include "pa_config_includes.h" -#ifdef HAVE_LIBDB +static const char * const IDENT="$Id: hashfile.C,v 1.45 2009/05/05 10:06:57 misha Exp $"; #include "classes.h" + #include "pa_request.h" +#include "pa_vmethod_frame.h" #include "pa_vhashfile.h" #include "pa_vhash.h" -// defines - -#define HASH_CLASS_NAME "hashfile" - // class class MHashfile : public Methoded { 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: - 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(r.self); - - // file_spec - const String &file_spec=params->as_string(0, "filename must be string"); +DECLARE_CLASS_VAR(hashfile, new MHashfile, 0); - self.set_connection( - DB_manager->get_connection( - r.absolute(file_spec), - method_name) - ); -} +// externs +extern String cycle_data_name; -static void _transaction(Request& r, const String& method_name, MethodParams *params) { - Pool& pool=r.pool(); - VHashfile& self=*static_cast(r.self); - - // body code - Value& body_code=params->as_junction(0, "body must be code"); +// defines for statics - // connection - DB_Connection& connection=self.get_connection(&method_name); +#define OPEN_DATA_NAME "HASHFILE-OPEN-DATA" + +// methods - // transaction - Auto_transaction transaction(connection); +typedef Hash HashStringBool; - // execute body - try { - r.write_assign_lang(r.process(body_code)); - } catch(...) { // process/commit problem - transaction.mark_to_rollback(); - - /*re*/throw; +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(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) { - Pool& pool=r.pool(); - VHashfile& self=*static_cast(r.self); +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"); + 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(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 - DB_Connection& connection=self.get_connection(&method_name); + StringOrValue sv_processed=info.r->process(*info.body_code); + 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) { - Pool& pool=r.pool(); - - // file_spec - const String &file_spec=params->as_string(0, "filename must be string"); + info.r->write_pass_lang(sv_processed); - DB_manager->clear_dbfile(r.absolute(file_spec)); + return lskip==Request::SKIP_BREAK; } +static void _foreach(Request& r, MethodParams& params) { + Temp_hash_value + 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) { - set_name(*NEW String(pool(), HASH_CLASS_NAME)); + 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(), + /*vkey=*/new VString, + /*vvalue=*/new VString, + false + }; - // ^hashfile::open[filename] - add_native_method("open", Method::CT_DYNAMIC, _open, 1, 1); - // ^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); -/* - // ^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); - // ^pack[] - add_native_method("pack", Method::CT_DYNAMIC, _pack, 0, 0); - // ^hash[] - add_native_method("hash", Method::CT_DYNAMIC, _hash, 0, 0); - */ + VHashfile& self=GET_SELF(r, VHashfile); + self.for_each(one_foreach_cycle, &info); } -// global variable - -Methoded *hashfile_base_class; +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); -// creator + self.for_each(one_cleanup_cycle, 0); +} -Methoded *MHashfile_create(Pool& pool) { - return hashfile_base_class=new(pool) MHashfile(pool); +static void _release(Request& r, MethodParams&) { + VHashfile& self=GET_SELF(r, VHashfile); + self.close(); } -#endif +// constructor + +MHashfile::MHashfile(): Methoded("hashfile") { + // ^hashfile::open[filename] + add_native_method("open", Method::CT_DYNAMIC, _open, 1, 1); + // ^hashfile.hash[] + add_native_method("hash", Method::CT_DYNAMIC, _hash, 0, 0); + // ^hashfile.delete[key] + 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); +}