--- parser3/src/classes/hash.C 2003/02/04 15:59:04 1.54.2.6 +++ parser3/src/classes/hash.C 2006/06/09 19:03:49 1.78 @@ -1,11 +1,11 @@ /** @file Parser: @b hash parser class. - Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) + Copyright (c) 2001-2005 ArtLebedev Group (http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) */ -static const char* IDENT_HASH_C="$Date: 2003/02/04 15:59:04 $"; +static const char * const IDENT_HASH_C="$Date: 2006/06/09 19:03:49 $"; #include "classes.h" #include "pa_vmethod_frame.h" @@ -22,7 +22,7 @@ static const char* IDENT_HASH_C="$Date: class MHash: public Methoded { public: // VStateless_class - ValuePtr create_new_value() { return ValuePtr(new VHash()); } + Value* create_new_value(Pool&, HashStringValue&) { return new VHash(); } public: MHash(); @@ -30,43 +30,41 @@ public: // Methoded bool used_directly() { return true; } }; - // global variable -MethodedPtr hash_class(new MHash); +DECLARE_CLASS_VAR(hash, new MHash, 0); + +// externs + +extern String cycle_data_name; // methods #ifndef DOXYGEN class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers { - StringPtr method_name; - StringPtr statement_string; const char* statement_cstr; + const String& statement_string; const char* statement_cstr; bool distinct; HashStringValue& rows_hash; HashStringValue* row_hash; int column_index; ArrayString columns; - int row_index; + bool only_one_column; + static VBool only_one_column_value; public: - Hash_sql_event_handlers(StringPtr amethod_name, - StringPtr astatement_string, const char* astatement_cstr, + Hash_sql_event_handlers( + const String& astatement_string, const char* astatement_cstr, bool adistinct, HashStringValue& arows_hash): - method_name(amethod_name), - statement_string(astatement_string), - statement_cstr(astatement_cstr), + statement_string(astatement_string), statement_cstr(astatement_cstr), distinct(adistinct), rows_hash(arows_hash), - row_index(0) { + row_hash(0), + column_index(0), + only_one_column(false) { } - bool add_column(SQL_Error& error, void *ptr, size_t size) { + bool add_column(SQL_Error& error, const char* str, size_t length) { try { - StringPtr column(new String); - column->APPEND_TAINTED( - (const char* )ptr, size, - statement_cstr, 0); - columns+=column; - + columns+=new String(str, length, true); return false; } catch(...) { error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column"); @@ -74,12 +72,11 @@ public: } } bool before_rows(SQL_Error& error) { - if(columns.count()<=1) { - error=SQL_Error("parser.runtime", - /*method_name,*/ - "column count must be more than 1 to create a hash"); + if(columns.count()<1) { + error=SQL_Error("parser.runtime", "no columns"); return true; } + only_one_column=columns.count()==1; return false; } @@ -87,27 +84,27 @@ public: column_index=0; return false; } - bool add_row_cell(SQL_Error& error, void *ptr, size_t size) { + bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) { try { - StringPtr cell(new String); - if(size) - cell->APPEND_TAINTED( - (const char* )ptr, size, - statement_cstr, row_index++); - if(column_index==0) { - VHashPtr row_vhash(new VHash); + String& cell=*new String; + if(length) + cell.append_know_length(ptr, length, String::L_TAINTED); + bool duplicate=false; + if(only_one_column) { + duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value); // put. existed? + } else if(column_index==0) { + VHash* row_vhash=new VHash; row_hash=&row_vhash->hash(); - if(rows_hash.put_dont_replace(cell, row_vhash)) // put. existed? - if(!distinct) { - error=SQL_Error("parser.runtime", - /*cell,*/ - "duplicate key"); - return true; - } + duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed? } else - row_hash->put(columns[column_index], ValuePtr(new VString(cell))); - column_index++; + row_hash->put(*columns[column_index], new VString(cell)); + + if(duplicate & !distinct) { + error=SQL_Error("parser.runtime", "duplicate key"); + return true; + } + column_index++; return false; } catch(...) { error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell"); @@ -116,6 +113,8 @@ public: } }; +VBool Hash_sql_event_handlers::only_one_column_value(true); + #endif static void copy_all_overwrite_to( @@ -124,11 +123,27 @@ static void copy_all_overwrite_to( HashStringValue* dest) { dest->put(key, value); } -static void _create_or_add(Request& r, StringPtr method_name, MethodParams& params) { +static void _create_or_add(Request& r, MethodParams& params) { if(params.count()) { - ValuePtr vb=params.as_no_junction(0, "param must be hash"); - if(HashStringValue* b=vb->get_hash(method_name)) - b->for_each(copy_all_overwrite_to, &(GET_SELF(r, VHash).hash())); + Value& vsrc=params.as_no_junction(0, "param must be hash"); + if(HashStringValue* src=vsrc.get_hash()) { + VHash& self=GET_SELF(r, VHash); + HashStringValue* self_hash=&(self.hash()); + if(src==self_hash) // same: doing nothing + return; + src->for_each(copy_all_overwrite_to, self_hash); + + if(VHash* vhash_src=static_cast(vsrc.as(VHASH_TYPE, false))) + { + if(Value* vdefault=vhash_src->get_default()) + { + if(vdefault->is_defined()) + { + self.set_default(vdefault); + } + } + } + } } } @@ -138,10 +153,16 @@ static void remove_key_from( HashStringValue* dest) { dest->remove(key); } -static void _sub(Request& r, StringPtr method_name, MethodParams& params) { - ValuePtr vb=params.as_no_junction(0, "param must be hash"); - if(HashStringValue* b=vb->get_hash(method_name)) - b->for_each(remove_key_from, &GET_SELF(r, VHash).hash()); +static void _sub(Request& r, MethodParams& params) { + Value& vsrc=params.as_no_junction(0, "param must be hash"); + if(HashStringValue* src=vsrc.get_hash()) { + HashStringValue* self=&(GET_SELF(r, VHash).hash()); + if(src==self) { // same: clearing + self->clear(); + return; + } + src->for_each(remove_key_from, self); + } } static void copy_all_dontoverwrite_to( @@ -150,13 +171,13 @@ static void copy_all_dontoverwrite_to( HashStringValue* dest) { dest->put_dont_replace(key, value); } -static void _union(Request& r, StringPtr method_name, MethodParams& params) { +static void _union(Request& r, MethodParams& params) { // dest = copy of self - ValuePtr result(new VHash(GET_SELF(r, VHash).hash())); + Value& result=*new VHash(GET_SELF(r, VHash).hash()); // dest += b - ValuePtr vb=params.as_no_junction(0, "param must be hash"); - if(HashStringValue* b=vb->get_hash(method_name)) - b->for_each(copy_all_dontoverwrite_to, result->get_hash(Exception::undefined_source)); + Value& vsrc=params.as_no_junction(0, "param must be hash"); + if(HashStringValue* src=vsrc.get_hash()) + src->for_each(copy_all_dontoverwrite_to, result.get_hash()); // return result r.write_no_lang(result); @@ -175,13 +196,13 @@ static void copy_intersection_to( if(info->b->get(key)) info->dest->put_dont_replace(key, value); } -static void _intersection(Request& r, StringPtr method_name, MethodParams& params) { - ValuePtr result(new VHash); +static void _intersection(Request& r, MethodParams& params) { + Value& result=*new VHash; // dest += b - ValuePtr vb=params.as_no_junction(0, "param must be hash"); - if(HashStringValue* b=vb->get_hash(method_name)) { - Copy_intersection_to_info info={b, result->get_hash(Exception::undefined_source)}; - GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info); + Value& vb=params.as_no_junction(0, "param must be hash"); + if(HashStringValue* b=vb.get_hash()) { + Copy_intersection_to_info info={b, result.get_hash()}; + GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info); } // return result @@ -192,102 +213,124 @@ static bool intersects( HashStringValue::key_type key, HashStringValue::value_type /*value*/, HashStringValue* b) { - return b->get(key); + return b->get(key)!=0; } -static void _intersects(Request& r, StringPtr method_name, MethodParams& params) { +static void _intersects(Request& r, MethodParams& params) { bool result=false; - ValuePtr vb=params.as_no_junction(0, "param must be hash"); - if(HashStringValue* b=vb->get_hash(method_name)) - result=GET_SELF(r, VHash).hash().first_that(intersects, b); + Value& vb=params.as_no_junction(0, "param must be hash"); + if(HashStringValue* b=vb.get_hash()) + result=GET_SELF(r, VHash).hash().first_that(intersects, b)!=0; // return result - r.write_no_lang(ValuePtr(new VBool(result))); + r.write_no_lang(*new VBool(result)); } -static void _sql(Request& r, StringPtr method_name, MethodParams& params) { - ValuePtr statement=params.as_junction(0, "statement must be code"); +extern String sql_bind_name; +extern String sql_limit_name; +extern String sql_offset_name; +extern String sql_default_name; +extern String sql_distinct_name; +extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders); +extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders); + +static void _sql(Request& r, MethodParams& params) { + Value& statement=params.as_junction(0, "statement must be code"); + HashStringValue* bind=0; ulong limit=0; ulong offset=0; bool distinct=false; if(params.count()>1) { - ValuePtr voptions=params.as_no_junction(1, "options must be hash, not code"); - if(!voptions->is_string()) - if(HashStringValuePtr options=voptions->get_hash(method_name)) { + Value& voptions=params.as_no_junction(1, "options must be hash, not code"); + if(voptions.is_defined() && !voptions.is_string()) + if(HashStringValue* options=voptions.get_hash()) { int valid_options=0; - if(ValuePtr vlimit=(ValuePtr )options->get(*sql_limit_name)) { + if(Value* vbind=options->get(sql_bind_name)) { + valid_options++; + bind=vbind->get_hash(); + } + if(Value* vlimit=options->get(sql_limit_name)) { valid_options++; limit=(ulong)r.process_to_value(*vlimit).as_double(); } - if(ValuePtr voffset=(ValuePtr )options->get(*sql_offset_name)) { + if(Value* voffset=options->get(sql_offset_name)) { valid_options++; offset=(ulong)r.process_to_value(*voffset).as_double(); } - if(ValuePtr vdistinct=(ValuePtr )options->get(*sql_distinct_name)) { + if(Value* vdistinct=options->get(sql_distinct_name)) { valid_options++; distinct=r.process_to_value(*vdistinct).as_bool(); } - if(valid_options!=options->size()) + if(valid_options!=options->count()) throw Exception("parser.runtime", - method_name, + 0, "called with invalid option"); } else throw Exception("parser.runtime", - method_name, + 0, "options must be hash"); } - Temp_lang temp_lang(r, String::UL_SQL); + SQL_Driver::Placeholder* placeholders=0; + uint placeholders_count=0; + if(bind) + placeholders_count=marshal_binds(*bind, placeholders); + + Temp_lang temp_lang(r, String::L_SQL); const String& statement_string=r.process_to_string(statement); const char* statement_cstr= - statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); - Hash& hash=static_cast(r.get_self())->hash(&method_name); + statement_string.cstr(String::L_UNSPECIFIED, r.connection()); + HashStringValue& hash=GET_SELF(r, VHash).hash(); hash.clear(); - Hash_sql_event_handlers handlers(pool, method_name, + Hash_sql_event_handlers handlers( statement_string, statement_cstr, distinct, hash); - - r.connection(&method_name)->query( - statement_cstr, offset, limit, + r.connection()->query( + statement_cstr, + placeholders_count, placeholders, + offset, limit, handlers, statement_string); -} -static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) { - Table& table=*static_cast(info); - Pool& pool=table.pool(); + if(bind) + unmarshal_bind_updates(*bind, placeholders_count, placeholders); +} - Array& row=*new(pool) Array(pool); - row+=&key; - table+=&row; +static void keys_collector( + HashStringValue::key_type key, + HashStringValue::value_type, + Table *table) { + Table::element_type row(new ArrayString); + *row+=new String(key, String::L_TAINTED); + *table+=row; } -static void _keys(Request& r, StringPtr method_name, MethodParams& ) { - Pool& pool=r.pool(); +static void _keys(Request& r, MethodParams& params) { + const String* keys_column_name; + if(params.count()>0) + keys_column_name=¶ms.as_string(0, "column name must be string"); + else + keys_column_name=new String("key"); - Array& columns=*new(pool) Array(pool); - columns+=new(pool) String(pool, "key"); - Table& table=*new(pool) Table(pool, method_name, &columns); + Table::columns_type columns(new ArrayString); + *columns+=keys_column_name; + Table* table=new Table(columns); - static_cast(r.get_self())->hash(&method_name).for_each(keys_collector, &table); + GET_SELF(r, VHash).hash().for_each(keys_collector, table); - r.write_no_lang(*new(pool) VTable(pool, &table)); + r.write_no_lang(*new VTable(table)); } -static void _count(Request& r, StringPtr method_name, MethodParams& ) { - Pool& pool=r.pool(); - - r.write_no_lang( - *new(pool) VInt(pool, static_cast(r.get_self())->hash(&method_name).size())); +static void _count(Request& r, MethodParams&) { + r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count())); } -static void _delete(Request& r, StringPtr method_name, MethodParams& params) { - Pool& pool=r.pool(); +static void _delete(Request& r, MethodParams& params) { - static_cast(r.get_self())->hash(&method_name).remove(params.as_string(0, "key must be string")); + GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string")); } #ifndef DOXYGEN @@ -295,52 +338,52 @@ struct Foreach_info { Request *r; const String* key_var_name; const String* value_var_name; - ValuePtr body_code; - ValuePtr delim_maybe_code; + Value* body_code; + Value* delim_maybe_code; - VString *vkey; + VString* vkey; bool need_delim; }; #endif - -static void one_foreach_cycle(const Hash::Key& akey, Hash::Val *avalue, - void *info) { - Foreach_info& i=*static_cast(info); - - i.vkey->set_string(akey); - 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; +static bool one_foreach_cycle( + HashStringValue::key_type akey, + HashStringValue::value_type avalue, + Foreach_info *info) { + info->vkey->set_string(*new String(akey, String::L_TAINTED)); + Value& ncontext=*info->r->get_method_frame()->caller(); + ncontext.put_element(ncontext, *info->key_var_name, info->vkey, false); + ncontext.put_element(ncontext, *info->value_var_name, avalue, 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->length()) { // 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)); + info->need_delim=true; } - i.r->write_pass_lang(sv_processed); + info->r->write_pass_lang(sv_processed); + + return lskip==Request::SKIP_BREAK; } -static void _foreach(Request& r, StringPtr method_name, MethodParams& params) { - Pool& pool=r.pool(); - 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"); - ValuePtr delim_maybe_code=params.count()>3?¶ms.get(3):0; +static void _foreach(Request& r, MethodParams& params) { + Temp_hash_value + cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r); Foreach_info info={ &r, - &key_var_name, &value_var_name, - &body_code, - delim_maybe_code, - - new(pool) VString(pool), + ¶ms.as_string(0, "key-var name must be string"), + ¶ms.as_string(1, "value-var name must be string"), + ¶ms.as_junction(2, "body must be code"), + params.count()>3?params.get(3):0, + /*vkey=*/new VString, false }; - VHash& self=*static_cast(r.get_self()); - Hash& hash=self.hash(&method_name); + + VHash& self=GET_SELF(r, VHash); + HashStringValue& hash=self.hash(); VHash_lock lock(self); - hash.for_each(one_foreach_cycle, &info); + hash.first_that(one_foreach_cycle, &info); } // constructor @@ -363,11 +406,11 @@ MHash::MHash(): Methoded("hash") // ^a.delete[key] add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); - // ^hash:sql[query][$.limit(1) $.offset(2)] + // ^hash:sql[query][options hash] add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); - // ^hash._keys[] - add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0); + // ^hash._keys[[column name]] + add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1); // ^hash._count[] add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);