|
|
| version 1.74, 2005/07/28 11:23:01 | version 1.91, 2009/05/05 10:06:57 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b hash parser class. | Parser: @b hash parser class. |
| Copyright (c) 2001-2004 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) |
| */ | */ |
| Line 34 public: // Methoded | Line 34 public: // Methoded |
| DECLARE_CLASS_VAR(hash, new MHash, 0); | DECLARE_CLASS_VAR(hash, new MHash, 0); |
| // externs | |
| extern String cycle_data_name; | |
| // methods | // methods |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| Line 41 class Hash_sql_event_handlers: public SQ | Line 45 class Hash_sql_event_handlers: public SQ |
| const String& statement_string; const char* statement_cstr; | const String& statement_string; const char* statement_cstr; |
| bool distinct; | bool distinct; |
| HashStringValue& rows_hash; | HashStringValue& rows_hash; |
| HashStringValue* row_hash; | Value* row_value; |
| int column_index; | int column_index; |
| ArrayString columns; | ArrayString& columns; |
| bool only_one_column; | bool one_bool_column; |
| static VBool only_one_column_value; | static VBool only_one_column_value; |
| Table2hash_value_type value_type; | |
| int columns_count; | |
| public: | |
| Table* empty; | |
| public: | public: |
| Hash_sql_event_handlers( | Hash_sql_event_handlers( |
| const String& astatement_string, const char* astatement_cstr, | const String& astatement_string, |
| const char* astatement_cstr, | |
| bool adistinct, | bool adistinct, |
| HashStringValue& arows_hash): | HashStringValue& arows_hash, |
| statement_string(astatement_string), statement_cstr(astatement_cstr), | Table2hash_value_type avalue_type) |
| : | |
| statement_string(astatement_string), | |
| statement_cstr(astatement_cstr), | |
| distinct(adistinct), | distinct(adistinct), |
| rows_hash(arows_hash), | rows_hash(arows_hash), |
| row_hash(0), | value_type(avalue_type), |
| row_value(0), | |
| column_index(0), | column_index(0), |
| only_one_column(false) { | one_bool_column(false), |
| columns(*new ArrayString), | |
| empty(0) { | |
| } | } |
| bool add_column(SQL_Error& error, const char* str, size_t length) { | bool add_column(SQL_Error& error, const char* str, size_t length) { |
| try { | try { |
| columns+=new String(str, length, true); | columns+=new String(str, length, true); |
| Line 67 public: | Line 83 public: |
| return true; | return true; |
| } | } |
| } | } |
| bool before_rows(SQL_Error& error) { | bool before_rows(SQL_Error& error) { |
| if(columns.count()<1) { | if(columns.count()<1) { |
| error=SQL_Error("parser.runtime", "no columns"); | error=SQL_Error(PARSER_RUNTIME, "no columns"); |
| return true; | return true; |
| } | } |
| only_one_column=columns.count()==1; | switch(value_type){ |
| case C_STRING: { | |
| if(columns.count()>2){ | |
| error=SQL_Error(PARSER_RUNTIME, "only 2 columns allowed for $.type[string]."); | |
| return true; | |
| } | |
| } | |
| case C_TABLE: { | |
| // create empty table which we'll copy later | |
| empty=new Table(&columns); | |
| columns_count=columns.count(); | |
| } | |
| case C_HASH: { | |
| one_bool_column=columns.count()==1; | |
| } | |
| } | |
| return false; | return false; |
| } | } |
| bool add_row(SQL_Error& /*error*/) { | bool add_row(SQL_Error& /*error*/) { |
| column_index=0; | column_index=0; |
| return false; | return false; |
| } | } |
| bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) { | bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) { |
| try { | try { |
| String& cell=*new String; | String& cell=*new String; |
| if(length) | if(length) |
| cell.append_know_length(ptr, length, String::L_TAINTED); | cell.append_know_length(ptr, length, String::L_TAINTED); |
| bool duplicate=false; | bool duplicate=false; |
| if(only_one_column) { | if(one_bool_column) { |
| duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value); // put. existed? | duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value); // put. existed? |
| } else if(column_index==0) { | } else if(column_index==0) { |
| VHash* row_vhash=new VHash; | switch(value_type){ |
| row_hash=&row_vhash->hash(); | case C_HASH: { |
| duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed? | VHash* row_vhash=new VHash; |
| } else | row_value=row_vhash; |
| row_hash->put(*columns[column_index], new VString(cell)); | duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed? |
| break; | |
| } | |
| case C_STRING: { | |
| VString* row_vstring=new VString(); | |
| row_value=row_vstring; | |
| duplicate=rows_hash.put_dont_replace(cell, row_vstring); // put. existed? | |
| break; | |
| } | |
| case C_TABLE: { | |
| VTable* vtable=(VTable*)rows_hash.get(cell); | |
| Table* table; | |
| if(vtable) { // table with this key exist? | |
| if(!distinct) { | |
| duplicate=true; | |
| break; | |
| } | |
| table=vtable->get_table(); | |
| } else { | |
| // no? creating table of same structure as source | |
| Table::Action_options table_options(0, 0); | |
| table=new Table(*empty, table_options/*no rows, just structure*/); | |
| vtable=new VTable(table); | |
| rows_hash.put(cell, vtable); // put | |
| } | |
| ArrayString* row=new ArrayString(columns_count); | |
| row_value=(Value*)row; | |
| *row+=&cell; | |
| *table+=row; | |
| break; | |
| } | |
| } | |
| } else { | |
| switch(value_type) { | |
| case C_HASH: { | |
| row_value->get_hash()->put(*columns[column_index], new VString(cell)); | |
| break; | |
| } | |
| case C_STRING: { | |
| VString* row_string=(VString*)row_value; | |
| row_string->set_string(cell); | |
| break; | |
| } | |
| case C_TABLE: { | |
| ArrayString* row=(ArrayString*)row_value; | |
| *row+=&cell; | |
| break; | |
| } | |
| } | |
| } | |
| if(duplicate & !distinct) { | if(duplicate & !distinct) { |
| error=SQL_Error("parser.runtime", "duplicate key"); | error=SQL_Error(PARSER_RUNTIME, "duplicate key"); |
| return true; | return true; |
| } | } |
| Line 113 VBool Hash_sql_event_handlers::only_one_ | Line 197 VBool Hash_sql_event_handlers::only_one_ |
| #endif | #endif |
| static void copy_all_overwrite_to( | |
| HashStringValue::key_type key, | |
| HashStringValue::value_type value, | |
| HashStringValue* dest) { | |
| dest->put(key, value); | |
| } | |
| static void _create_or_add(Request& r, MethodParams& params) { | static void _create_or_add(Request& r, MethodParams& params) { |
| if(params.count()) { | if(params.count()) { |
| Value& vsrc=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| Line 127 static void _create_or_add(Request& r, M | Line 205 static void _create_or_add(Request& r, M |
| HashStringValue* self_hash=&(self.hash()); | HashStringValue* self_hash=&(self.hash()); |
| if(src==self_hash) // same: doing nothing | if(src==self_hash) // same: doing nothing |
| return; | return; |
| src->for_each(copy_all_overwrite_to, self_hash); | src->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash); |
| if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false))) | if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false))) |
| self.set_default(vhash_src->get_default()); | { |
| if(Value* vdefault=vhash_src->get_default()) | |
| { | |
| if(vdefault->is_defined()) | |
| { | |
| self.set_default(vdefault); | |
| } | |
| } | |
| } | |
| } | } |
| } | } |
| } | } |
| static void remove_key_from( | |
| HashStringValue::key_type key, | |
| HashStringValue::value_type /*value*/, | |
| HashStringValue* dest) { | |
| dest->remove(key); | |
| } | |
| static void _sub(Request& r, MethodParams& params) { | static void _sub(Request& r, MethodParams& params) { |
| Value& vsrc=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* src=vsrc.get_hash()) { | if(HashStringValue* src=vsrc.get_hash()) { |
| Line 149 static void _sub(Request& r, MethodParam | Line 229 static void _sub(Request& r, MethodParam |
| self->clear(); | self->clear(); |
| return; | return; |
| } | } |
| src->for_each(remove_key_from, self); | src->for_each<HashStringValue*>(remove_key_from, self); |
| } | } |
| } | } |
| Line 165 static void _union(Request& r, MethodPar | Line 245 static void _union(Request& r, MethodPar |
| // dest += b | // dest += b |
| Value& vsrc=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* src=vsrc.get_hash()) | if(HashStringValue* src=vsrc.get_hash()) |
| src->for_each(copy_all_dontoverwrite_to, result.get_hash()); | src->for_each<HashStringValue*>(copy_all_dontoverwrite_to, result.get_hash()); |
| // return result | // return result |
| r.write_no_lang(result); | r.write_no_lang(result); |
| Line 190 static void _intersection(Request& r, Me | Line 270 static void _intersection(Request& r, Me |
| Value& vb=params.as_no_junction(0, "param must be hash"); | Value& vb=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) { | if(HashStringValue* b=vb.get_hash()) { |
| Copy_intersection_to_info info={b, result.get_hash()}; | Copy_intersection_to_info info={b, result.get_hash()}; |
| GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info); | GET_SELF(r, VHash).hash().for_each<Copy_intersection_to_info*>(copy_intersection_to, &info); |
| } | } |
| // return result | // return result |
| Line 209 static void _intersects(Request& r, Meth | Line 289 static void _intersects(Request& r, Meth |
| Value& vb=params.as_no_junction(0, "param must be hash"); | Value& vb=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) | if(HashStringValue* b=vb.get_hash()) |
| result=GET_SELF(r, VHash).hash().first_that(intersects, b)!=0; | result=GET_SELF(r, VHash).hash().first_that<HashStringValue*>(intersects, b)!=0; |
| // return result | // return result |
| r.write_no_lang(*new VBool(result)); | r.write_no_lang(VBool::get(result)); |
| } | } |
| Line 221 extern String sql_limit_name; | Line 301 extern String sql_limit_name; |
| extern String sql_offset_name; | extern String sql_offset_name; |
| extern String sql_default_name; | extern String sql_default_name; |
| extern String sql_distinct_name; | extern String sql_distinct_name; |
| extern String sql_value_type_name; | |
| extern Table2hash_value_type get_value_type(Value& vvalue_type); | |
| extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders); | extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders); |
| extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders); | extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders); |
| Line 228 static void _sql(Request& r, MethodParam | Line 310 static void _sql(Request& r, MethodParam |
| Value& statement=params.as_junction(0, "statement must be code"); | Value& statement=params.as_junction(0, "statement must be code"); |
| HashStringValue* bind=0; | HashStringValue* bind=0; |
| ulong limit=0; | ulong limit=SQL_NO_LIMIT; |
| ulong offset=0; | ulong offset=0; |
| bool distinct=false; | bool distinct=false; |
| Table2hash_value_type value_type=C_HASH; | |
| if(params.count()>1) { | if(params.count()>1) { |
| Value& voptions=params.as_no_junction(1, "options must be hash, not code"); | Value& voptions=params.as_no_junction(1, "options must be hash, not code"); |
| if(!voptions.is_string()) | if(voptions.is_defined() && !voptions.is_string()) |
| if(HashStringValue* options=voptions.get_hash()) { | if(HashStringValue* options=voptions.get_hash()) { |
| int valid_options=0; | int valid_options=0; |
| if(Value* vbind=options->get(sql_bind_name)) { | if(Value* vbind=options->get(sql_bind_name)) { |
| Line 252 static void _sql(Request& r, MethodParam | Line 335 static void _sql(Request& r, MethodParam |
| valid_options++; | valid_options++; |
| distinct=r.process_to_value(*vdistinct).as_bool(); | distinct=r.process_to_value(*vdistinct).as_bool(); |
| } | } |
| if(Value* vvalue_type=options->get(sql_value_type_name)) { | |
| valid_options++; | |
| value_type=get_value_type(r.process_to_value(*vvalue_type)); | |
| } | |
| if(valid_options!=options->count()) | if(valid_options!=options->count()) |
| throw Exception("parser.runtime", | throw Exception(PARSER_RUNTIME, |
| 0, | 0, |
| "called with invalid option"); | "called with invalid option"); |
| } else | } else |
| throw Exception("parser.runtime", | throw Exception(PARSER_RUNTIME, |
| 0, | 0, |
| "options must be hash"); | "options must be hash"); |
| } | } |
| Line 276 static void _sql(Request& r, MethodParam | Line 363 static void _sql(Request& r, MethodParam |
| Hash_sql_event_handlers handlers( | Hash_sql_event_handlers handlers( |
| statement_string, statement_cstr, | statement_string, statement_cstr, |
| distinct, | distinct, |
| hash); | hash, |
| value_type); | |
| r.connection()->query( | r.connection()->query( |
| statement_cstr, | statement_cstr, |
| placeholders_count, placeholders, | placeholders_count, placeholders, |
| Line 299 static void keys_collector( | Line 388 static void keys_collector( |
| static void _keys(Request& r, MethodParams& params) { | static void _keys(Request& r, MethodParams& params) { |
| const String* keys_column_name; | const String* keys_column_name; |
| if(params.count()>0) | if(params.count()>0) |
| keys_column_name=¶ms.as_string(0, "column name must be string"); | keys_column_name=¶ms.as_string(0, COLUMN_NAME_MUST_BE_STRING); |
| else | else |
| keys_column_name=new String("key"); | keys_column_name=new String("key"); |
| Line 307 static void _keys(Request& r, MethodPara | Line 396 static void _keys(Request& r, MethodPara |
| *columns+=keys_column_name; | *columns+=keys_column_name; |
| Table* table=new Table(columns); | Table* table=new Table(columns); |
| GET_SELF(r, VHash).hash().for_each(keys_collector, table); | GET_SELF(r, VHash).hash().for_each<Table*>(keys_collector, table); |
| r.write_no_lang(*new VTable(table)); | r.write_no_lang(*new VTable(table)); |
| } | } |
| Line 321 static void _delete(Request& r, MethodPa | Line 410 static void _delete(Request& r, MethodPa |
| GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string")); | GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string")); |
| } | } |
| static void _contains(Request& r, MethodParams& params) { | |
| bool result=GET_SELF(r, VHash).hash().contains(params.as_string(0, "key must be string")); | |
| r.write_no_lang(VBool::get(result)); | |
| } | |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| struct Foreach_info { | struct Foreach_info { |
| Request *r; | Request *r; |
| Line 329 struct Foreach_info { | Line 423 struct Foreach_info { |
| Value* body_code; | Value* body_code; |
| Value* delim_maybe_code; | Value* delim_maybe_code; |
| Value* var_context; | |
| VString* vkey; | VString* vkey; |
| bool need_delim; | bool need_delim; |
| }; | }; |
| #endif | #endif |
| static void one_foreach_cycle( | static bool one_foreach_cycle( |
| HashStringValue::key_type akey, | HashStringValue::key_type akey, |
| HashStringValue::value_type avalue, | HashStringValue::value_type avalue, |
| Foreach_info *info) { | Foreach_info *info) { |
| info->vkey->set_string(*new String(akey, String::L_TAINTED)); | Value& var_context=*info->var_context; |
| Value& ncontext=*info->r->get_method_frame()->caller(); | if(info->key_var_name){ |
| ncontext.put_element(ncontext, *info->key_var_name, info->vkey, false); | info->vkey->set_string(*new String(akey, String::L_TAINTED)); |
| ncontext.put_element(ncontext, *info->value_var_name, avalue, false); | var_context.put_element(var_context, *info->key_var_name, info->vkey, false); |
| } | |
| StringOrValue sv_processed=info->r->process(*info->body_code); | if(info->value_var_name) |
| const String* s_processed=sv_processed.get_string(); | var_context.put_element(var_context, *info->value_var_name, avalue, false); |
| 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? | if(info->delim_maybe_code){ // delimiter set |
| info->r->write_pass_lang(info->r->process(*info->delim_maybe_code)); | StringOrValue sv_processed=info->r->process(*info->body_code); |
| info->need_delim=true; | Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING); |
| const String* s_processed=sv_processed.get_string(); | |
| if(s_processed && !s_processed->is_empty()) { // 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; | |
| } else { | |
| info->r->process_write(*info->body_code); | |
| Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING); | |
| return lskip==Request::SKIP_BREAK; | |
| } | } |
| info->r->write_pass_lang(sv_processed); | |
| } | } |
| static void _foreach(Request& r, MethodParams& params) { | 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); | |
| 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"); | |
| Foreach_info info={ | Foreach_info info={ |
| &r, | &r, |
| ¶ms.as_string(0, "key-var name must be string"), | key_var_name.is_empty()? 0 : &key_var_name, |
| ¶ms.as_string(1, "value-var name must be string"), | value_var_name.is_empty()? 0 : &value_var_name, |
| ¶ms.as_junction(2, "body must be code"), | ¶ms.as_junction(2, "body must be code"), |
| params.count()>3?params.get(3):0, | /*delimiter*/params.count()>3?params.get(3):0, |
| /*var_context*/r.get_method_frame()->caller(), | |
| /*vkey=*/new VString, | /*vkey=*/new VString, |
| false | false |
| }; | }; |
| Line 365 static void _foreach(Request& r, MethodP | Line 480 static void _foreach(Request& r, MethodP |
| VHash& self=GET_SELF(r, VHash); | VHash& self=GET_SELF(r, VHash); |
| HashStringValue& hash=self.hash(); | HashStringValue& hash=self.hash(); |
| VHash_lock lock(self); | VHash_lock lock(self); |
| hash.for_each(one_foreach_cycle, &info); | hash.first_that<Foreach_info*>(one_foreach_cycle, &info); |
| } | } |
| // constructor | // constructor |
| Line 388 MHash::MHash(): Methoded("hash") | Line 503 MHash::MHash(): Methoded("hash") |
| // ^a.delete[key] | // ^a.delete[key] |
| add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); | add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); |
| // ^hash:sql[query][options hash] | // ^a.contains[key] |
| add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1); | |
| // backward | |
| add_native_method("contain", Method::CT_DYNAMIC, _contains, 1, 1); | |
| // ^hash::sql[query][options hash] | |
| add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); | add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); |
| // ^hash._keys[[column name]] | // ^hash._keys[[column name]] |