--- parser3/src/classes/hash.C 2001/11/01 14:59:57 1.26 +++ parser3/src/classes/hash.C 2003/04/25 11:39:54 1.56 @@ -1,12 +1,12 @@ /** @file Parser: @b hash parser class. - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: hash.C,v 1.26 2001/11/01 14:59:57 paf Exp $ + Copyright (c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ +static const char* IDENT_HASH_C="$Date: 2003/04/25 11:39:54 $"; + #include "classes.h" #include "pa_request.h" #include "pa_vhash.h" @@ -16,10 +16,6 @@ #include "pa_vbool.h" #include "pa_vmethod_frame.h" -// defines - -#define HASH_CLASS_NAME "hash" - // class class MHash : public Methoded { @@ -35,54 +31,82 @@ public: // Methoded // methods #ifndef DOXYGEN -class Hash_sql_event_handlers : public SQL_Driver_query_event_handlers { +class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers { public: Hash_sql_event_handlers(Pool& apool, const String& amethod_name, const String& astatement_string, const char *astatement_cstr, - Hash& arows_hash) : + bool adistinct, + Hash& arows_hash): pool(apool), method_name(amethod_name), statement_string(astatement_string), statement_cstr(astatement_cstr), + distinct(adistinct), rows_hash(arows_hash), columns(pool), row_index(0) { } - void add_column(void *ptr, size_t size) { - String *column=new(pool) String(pool); - column->APPEND_TAINTED( - (const char *)ptr, size, - statement_cstr, 0); - columns+=column; + bool add_column(SQL_Error& error, void *ptr, size_t size) { + try { + String *column=new(pool) String(pool); + column->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, 0); + columns+=column; + + return false; + } catch(...) { + error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column"); + return true; + } } - void before_rows() { - if(columns.size()<=1) - throw Exception(0, 0, + bool before_rows(SQL_Error& error) { + if(columns.size()<=1) { + error=SQL_Error("parser.runtime", &method_name, "column count must be more than 1 to create a hash"); + return true; + } + + return false; } - void add_row() { + bool add_row(SQL_Error& /*error*/) { column_index=0; + return false; } - void add_row_cell(void *ptr, size_t size) { - String *cell=new(pool) String(pool); - if(size) - cell->APPEND_TAINTED( - (const char *)ptr, size, - statement_cstr, row_index++); - if(column_index==0) { - VHash *row_vhash=new(pool) VHash(pool); - row_hash=row_vhash->get_hash(0); - rows_hash.put(*cell, row_vhash); - } else - row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); - column_index++; + bool add_row_cell(SQL_Error& error, void *ptr, size_t size) { + try { + String *cell=new(pool) String(pool); + if(size) + cell->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, row_index++); + if(column_index==0) { + VHash *row_vhash=new(pool) VHash(pool); + row_hash=row_vhash->get_hash(0); + if(rows_hash.put_dont_replace(*cell, row_vhash)) // put. existed? + if(!distinct) { + error=SQL_Error("parser.runtime", + cell, + "duplicate key"); + return true; + } + } else + row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); + column_index++; + + return false; + } catch(...) { + error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell"); + return true; + } } private: Pool& pool; const String& method_name; const String& statement_string; const char *statement_cstr; + bool distinct; Hash& rows_hash; Hash *row_hash; int column_index; @@ -95,40 +119,56 @@ static void copy_all_overwrite_to(const Hash& dest=*static_cast(info); dest.put(key, value); } - -static void _create_or_add(Request& r, const String& method_name, MethodParams *params) { +static void _create_or_add(Request& r, const String& method_name, MethodParams *params, + bool is_create) { Pool& pool=r.pool(); if(params->size()) { - Value& vb=params->as_no_junction(0, "param must be hash"); - if(Hash *b=vb.get_hash(&method_name)) - b->for_each(copy_all_overwrite_to, &static_cast(r.self)->hash()); + Value& vb_maybehash=params->as_no_junction(0, "param must be hash"); + if(VHash* vbh=static_cast(vb_maybehash.as("hash", false))) { + VHash& vah=*static_cast(r.get_self()); + + Hash& a=vah.hash(&method_name); + Hash& b=vbh->hash(&method_name); + b.for_each(copy_all_overwrite_to, &a); + + if(is_create) + vah.set_default(vbh->get_default()); + } else if(!vb_maybehash.is_string()) // allow ^hash::create[^rem{xxx}] + throw Exception("parser.runtime", + &method_name, + "param must be hash"); } } +static void _create(Request& r, const String& method_name, MethodParams *params) { + _create_or_add(r, method_name, params, true); +} +static void _add(Request& r, const String& method_name, MethodParams *params) { + _create_or_add(r, method_name, params, false); +} + static void remove_key_from(const Hash::Key& key, Hash::Val *value, void *info) { Hash& dest=*static_cast(info); dest.remove(key); } - static void _sub(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); Value& vb=params->as_no_junction(0, "param must be hash"); if(Hash *b=vb.get_hash(&method_name)) - b->for_each(remove_key_from, &static_cast(r.self)->hash()); + b->for_each(remove_key_from, &static_cast(r.get_self())->hash(&method_name)); } static void copy_all_dontoverwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { Hash& dest=*static_cast(info); dest.put_dont_replace(key, value); } - static void _union(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); // dest = copy of self - Hash& dest=*new(pool) Hash(static_cast(r.self)->hash()); + Hash& dest=*new(pool) Hash(static_cast(r.get_self())->hash(&method_name)); // dest += b Value& vb=params->as_no_junction(0, "param must be hash"); if(Hash *b=vb.get_hash(&method_name)) @@ -136,7 +176,6 @@ static void _union(Request& r, const Str // return result Value& result=*new(pool) VHash(pool, dest); - result.set_name(method_name); r.write_no_lang(result); } @@ -153,7 +192,6 @@ static void copy_intersection_to(const H if(i.b->get(key)) i.dest->put_dont_replace(key, value); } - static void _intersection(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); @@ -166,13 +204,11 @@ static void _intersection(Request& r, co b, &dest }; - static_cast(r.self)->hash().for_each(copy_intersection_to, &info); + static_cast(r.get_self())->hash(&method_name).for_each(copy_intersection_to, &info); } // return result - Value& result=*new(pool) VHash(pool, dest); - result.set_name(method_name); - r.write_no_lang(result); + r.write_no_lang(*new(pool) VHash(pool, dest)); } static void *intersects(const Hash::Key& key, Hash::Val *value, void *info) { @@ -189,49 +225,63 @@ static void _intersects(Request& r, cons // dest += b Value& vb=params->as_no_junction(0, "param must be hash"); if(Hash *b=vb.get_hash(&method_name)) - yes=static_cast(r.self)->hash().first_that(intersects, b)!=0; + yes=static_cast(r.get_self())->hash(&method_name).first_that(intersects, b)!=0; // return result - Value& result=*new(pool) VBool(pool, yes); - result.set_name(method_name); - r.write_no_lang(result); + r.write_no_lang(*new(pool) VBool(pool, yes)); } static void _sql(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - if(!r.connection) - throw Exception(0, 0, - &method_name, - "without connect"); - Value& statement=params->as_junction(0, "statement must be code"); ulong limit=0; - if(params->size()>1) { - Value& limit_code=params->as_junction(1, "limit must be expression"); - limit=(uint)r.process(limit_code).as_double(); - } - ulong offset=0; - if(params->size()>2) { - Value& offset_code=params->as_junction(2, "offset must be expression"); - offset=(ulong)r.process(offset_code).as_double(); + bool distinct=false; + if(params->size()>1) { + Value& voptions=params->as_no_junction(1, "options must be hash, not code"); + if(!voptions.is_string()) + if(Hash *options=voptions.get_hash(&method_name)) { + int valid_options=0; + if(Value *vlimit=(Value *)options->get(*sql_limit_name)) { + valid_options++; + limit=(ulong)r.process_to_value(*vlimit).as_double(); + } + if(Value *voffset=(Value *)options->get(*sql_offset_name)) { + valid_options++; + offset=(ulong)r.process_to_value(*voffset).as_double(); + } + if(Value *vdistinct=(Value *)options->get(*sql_distinct_name)) { + valid_options++; + distinct=r.process_to_value(*vdistinct).as_bool(); + } + if(valid_options!=options->size()) + throw Exception("parser.runtime", + &method_name, + "called with invalid option"); + } else + throw Exception("parser.runtime", + &method_name, + "options must be hash"); } Temp_lang temp_lang(r, String::UL_SQL); - const String& statement_string=r.process(statement).as_string(); + const String& statement_string=r.process_to_string(statement); const char *statement_cstr= - statement_string.cstr(String::UL_UNSPECIFIED, r.connection); - Hash& hash=static_cast(r.self)->hash(); + statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); + Hash& hash=static_cast(r.get_self())->hash(&method_name); hash.clear(); Hash_sql_event_handlers handlers(pool, method_name, - statement_string, statement_cstr, hash); + statement_string, statement_cstr, + distinct, + hash); - r.connection->query( + r.connection(&method_name)->query( statement_cstr, offset, limit, - handlers); + handlers, + statement_string); } static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) { @@ -249,25 +299,22 @@ static void _keys(Request& r, const Stri columns+=new(pool) String(pool, "key"); Table& table=*new(pool) Table(pool, &method_name, &columns); - static_cast(r.self)->hash().for_each(keys_collector, &table); + static_cast(r.get_self())->hash(&method_name).for_each(keys_collector, &table); - VTable& result=*new(pool) VTable(pool, &table); - result.set_name(method_name); - r.write_no_lang(result); + r.write_no_lang(*new(pool) VTable(pool, &table)); } static void _count(Request& r, const String& method_name, MethodParams *) { Pool& pool=r.pool(); - Value& result=*new(pool) VInt(pool, static_cast(r.self)->hash().size()); - result.set_name(method_name); - r.write_no_lang(result); + r.write_no_lang( + *new(pool) VInt(pool, static_cast(r.get_self())->hash(&method_name).size())); } static void _delete(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - static_cast(r.self)->hash().remove(params->as_string(0, "key must be string")); + static_cast(r.get_self())->hash(&method_name).remove(params->as_string(0, "key must be string")); } #ifndef DOXYGEN @@ -278,7 +325,6 @@ struct Foreach_info { Value *body_code; Value *delim_maybe_code; - Value *var_context; VString *vkey; bool need_delim; }; @@ -289,17 +335,18 @@ static void one_foreach_cycle(const Hash Foreach_info& i=*static_cast(info); i.vkey->set_string(akey); - i.var_context->put_element(*i.key_var_name, i.vkey); - i.var_context->put_element(*i.value_var_name, static_cast(avalue)); - - Value& processed_body=i.r->process(*i.body_code); - if(i.delim_maybe_code) { // delimiter set? - const String *string=processed_body.get_string(); - if(i.need_delim && string && string->size()) // need delim & iteration produced string? + Value& ncontext=*i.r->get_method_frame()->caller(); + ncontext.put_element(*i.key_var_name, i.vkey, false); + ncontext.put_element(*i.value_var_name, static_cast(avalue), false); + + StringOrValue sv_processed=i.r->process(*i.body_code); + const String *s_processed=sv_processed.get_string(); + if(i.delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body + if(i.need_delim) // need delim & iteration produced string? i.r->write_pass_lang(i.r->process(*i.delim_maybe_code)); i.need_delim=true; } - i.r->write_pass_lang(processed_body); + i.r->write_pass_lang(sv_processed); } static void _foreach(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); @@ -314,22 +361,23 @@ static void _foreach(Request& r, const S &body_code, delim_maybe_code, - body_code.get_junction()->wcontext, new(pool) VString(pool), false }; - static_cast(r.self)->hash().for_each(one_foreach_cycle, &info); + VHash& self=*static_cast(r.get_self()); + Hash& hash=self.hash(&method_name); + VHash_lock lock(self); + hash.for_each(one_foreach_cycle, &info); } // constructor -MHash::MHash(Pool& apool) : Methoded(apool) { - set_name(*NEW String(pool(), HASH_CLASS_NAME)); - +MHash::MHash(Pool& apool) : Methoded(apool, "hash") +{ // ^hash::create[[copy_from]] - add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1); + add_native_method("create", Method::CT_DYNAMIC, _create, 0, 1); // ^hash.add[add_from] - add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1); + add_native_method("add", Method::CT_DYNAMIC, _add, 1, 1); // ^hash.sub[sub_from] add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1); // ^a.union[b] = hash @@ -342,8 +390,8 @@ MHash::MHash(Pool& apool) : Methoded(apo // ^a.delete[key] add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); - // ^hash:sql[query][(count[;offset])] - add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3); + // ^hash:sql[query][$.limit(1) $.offset(2)] + add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); // ^hash._keys[] add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0); @@ -352,18 +400,15 @@ MHash::MHash(Pool& apool) : Methoded(apo add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0); // ^hash.foreach[key;value]{code}[delim] - Array *for_each_locals=NEW Array(pool(), 2); - *for_each_locals += NEW String(pool(), "key"); - add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); } // global variable -Methoded *hash_base_class; +Methoded *hash_class; // creator Methoded *MHash_create(Pool& pool) { - return hash_base_class=new(pool) MHash(pool); + return hash_class=new(pool) MHash(pool); }