|
|
| version 1.59, 2003/11/06 11:13:24 | version 1.85, 2008/05/16 14:28:23 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b hash parser class. | 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 <paf@design.ru> (http://paf.design.ru) | Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) |
| */ | */ |
| static const char* IDENT_HASH_C="$Date$"; | static const char * const IDENT_HASH_C="$Date$"; |
| #include "classes.h" | #include "classes.h" |
| #include "pa_vmethod_frame.h" | #include "pa_vmethod_frame.h" |
| Line 22 static const char* IDENT_HASH_C="$Date$" | Line 22 static const char* IDENT_HASH_C="$Date$" |
| class MHash: public Methoded { | class MHash: public Methoded { |
| public: // VStateless_class | public: // VStateless_class |
| Value* create_new_value() { return new VHash(); } | Value* create_new_value(Pool&, HashStringValue&) { return new VHash(); } |
| public: | public: |
| MHash(); | MHash(); |
| 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 44 class Hash_sql_event_handlers: public SQ | Line 48 class Hash_sql_event_handlers: public SQ |
| HashStringValue* row_hash; | HashStringValue* row_hash; |
| int column_index; | int column_index; |
| ArrayString columns; | ArrayString columns; |
| bool only_one_column; | |
| static VBool only_one_column_value; | |
| 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, |
| Line 53 public: | Line 59 public: |
| distinct(adistinct), | distinct(adistinct), |
| rows_hash(arows_hash), | rows_hash(arows_hash), |
| row_hash(0), | row_hash(0), |
| column_index(0) { | column_index(0), |
| only_one_column(false) { | |
| } | } |
| 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 { |
| Line 65 public: | Line 72 public: |
| } | } |
| } | } |
| 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", | error=SQL_Error(PARSER_RUNTIME, "no columns"); |
| /*method_name,*/ | |
| "column count must be more than 1 to create a hash"); | |
| return true; | return true; |
| } | } |
| only_one_column=columns.count()==1; | |
| return false; | return false; |
| } | } |
| Line 83 public: | Line 89 public: |
| 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); |
| if(column_index==0) { | 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; | VHash* row_vhash=new VHash; |
| row_hash=&row_vhash->hash(); | row_hash=&row_vhash->hash(); |
| if(rows_hash.put_dont_replace(cell, row_vhash)) // put. existed? | duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed? |
| if(!distinct) { | |
| error=SQL_Error("parser.runtime", | |
| /*cell,*/ | |
| "duplicate key"); | |
| return true; | |
| } | |
| } else | } else |
| row_hash->put(*columns[column_index], new VString(cell)); | row_hash->put(*columns[column_index], new VString(cell)); |
| column_index++; | |
| if(duplicate & !distinct) { | |
| error=SQL_Error(PARSER_RUNTIME, "duplicate key"); | |
| return true; | |
| } | |
| column_index++; | |
| return false; | return false; |
| } catch(...) { | } catch(...) { |
| error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell"); | error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell"); |
| Line 105 public: | Line 113 public: |
| } | } |
| }; | }; |
| VBool Hash_sql_event_handlers::only_one_column_value(true); | |
| #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& vb=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) | if(HashStringValue* src=vsrc.get_hash()) { |
| b->for_each(copy_all_overwrite_to, &(GET_SELF(r, VHash).hash())); | VHash& self=GET_SELF(r, VHash); |
| HashStringValue* self_hash=&(self.hash()); | |
| if(src==self_hash) // same: doing nothing | |
| return; | |
| src->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash); | |
| if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false))) | |
| { | |
| 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& vb=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) | if(HashStringValue* src=vsrc.get_hash()) { |
| b->for_each(remove_key_from, &GET_SELF(r, VHash).hash()); | HashStringValue* self=&(GET_SELF(r, VHash).hash()); |
| if(src==self) { // same: clearing | |
| self->clear(); | |
| return; | |
| } | |
| src->for_each<HashStringValue*>(remove_key_from, self); | |
| } | |
| } | } |
| static void copy_all_dontoverwrite_to( | static void copy_all_dontoverwrite_to( |
| Line 143 static void _union(Request& r, MethodPar | Line 163 static void _union(Request& r, MethodPar |
| // dest = copy of self | // dest = copy of self |
| Value& result=*new VHash(GET_SELF(r, VHash).hash()); | Value& result=*new VHash(GET_SELF(r, VHash).hash()); |
| // dest += b | // dest += b |
| Value& vb=params.as_no_junction(0, "param must be hash"); | Value& vsrc=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) | if(HashStringValue* src=vsrc.get_hash()) |
| b->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 170 static void _intersection(Request& r, Me | Line 190 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 189 static void _intersects(Request& r, Meth | Line 209 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(*new VBool(result)); |
| } | } |
| extern String sql_bind_name; | |
| extern String sql_limit_name; | 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 void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders); | |
| static void _sql(Request& r, MethodParams& params) { | static void _sql(Request& r, MethodParams& params) { |
| Value& statement=params.as_junction(0, "statement must be code"); | Value& statement=params.as_junction(0, "statement must be code"); |
| HashStringValue* bind=0; | |
| ulong limit=0; | ulong limit=0; |
| 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)) { | |
| valid_options++; | |
| bind=vbind->get_hash(); | |
| } | |
| if(Value* vlimit=options->get(sql_limit_name)) { | if(Value* vlimit=options->get(sql_limit_name)) { |
| valid_options++; | valid_options++; |
| limit=(ulong)r.process_to_value(*vlimit).as_double(); | limit=(ulong)r.process_to_value(*vlimit).as_double(); |
| Line 223 static void _sql(Request& r, MethodParam | Line 255 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"); |
| } | } |
| 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); | Temp_lang temp_lang(r, String::L_SQL); |
| const String& statement_string=r.process_to_string(statement); | const String& statement_string=r.process_to_string(statement); |
| const char* statement_cstr= | const char* statement_cstr= |
| Line 244 static void _sql(Request& r, MethodParam | Line 285 static void _sql(Request& r, MethodParam |
| distinct, | distinct, |
| hash); | hash); |
| r.connection()->query( | r.connection()->query( |
| statement_cstr, offset, limit, | statement_cstr, |
| placeholders_count, placeholders, | |
| offset, limit, | |
| handlers, | handlers, |
| statement_string); | statement_string); |
| if(bind) | |
| unmarshal_bind_updates(*bind, placeholders_count, placeholders); | |
| } | } |
| static void keys_collector( | static void keys_collector( |
| HashStringValue::key_type key, | HashStringValue::key_type key, |
| HashStringValue::value_type value, | HashStringValue::value_type, |
| Table *table) { | Table *table) { |
| Table::element_type row(new ArrayString); | Table::element_type row(new ArrayString); |
| *row+=new String(key, String::L_TAINTED); | *row+=new String(key, String::L_TAINTED); |
| *table+=row; | *table+=row; |
| } | } |
| static void _keys(Request& r, MethodParams&) { | 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"); | |
| Table::columns_type columns(new ArrayString); | Table::columns_type columns(new ArrayString); |
| *columns+=new String("key"); | *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 276 static void _delete(Request& r, MethodPa | Line 328 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(*new VBool(result)); | |
| } | |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| struct Foreach_info { | struct Foreach_info { |
| Request *r; | Request *r; |
| Line 284 struct Foreach_info { | Line 342 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(*info->key_var_name, info->vkey, false); | info->vkey->set_string(*new String(akey, String::L_TAINTED)); |
| ncontext.put_element(*info->value_var_name, avalue, false); | var_context.put_element(var_context, *info->key_var_name, info->vkey, false); |
| } | |
| if(info->value_var_name) | |
| var_context.put_element(var_context, *info->value_var_name, avalue, false); | |
| StringOrValue sv_processed=info->r->process(*info->body_code); | 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(); | 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->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->need_delim) // need delim & iteration produced string? |
| info->r->write_pass_lang(info->r->process(*info->delim_maybe_code)); | info->r->write_pass_lang(info->r->process(*info->delim_maybe_code)); |
| info->need_delim=true; | else |
| info->need_delim=true; | |
| } | } |
| info->r->write_pass_lang(sv_processed); | info->r->write_pass_lang(sv_processed); |
| return lskip==Request::SKIP_BREAK; | |
| } | } |
| static void _foreach(Request& r, MethodParams& params) { | static void _foreach(Request& r, MethodParams& params) { |
| Foreach_info info={0}; | Temp_hash_value<const String::Body, void*> |
| info.r=&r; | cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r); |
| info.key_var_name=¶ms.as_string(0, "key-var name must be string"); | |
| info.value_var_name=¶ms.as_string(1, "value-var name must be string"); | const String& key_var_name=params.as_string(0, "key-var name must be string"); |
| info.body_code=¶ms.as_junction(2, "body must be code"); | const String& value_var_name=params.as_string(1, "value-var name must be string"); |
| info.delim_maybe_code=params.count()>3?params.get(3):0; | |
| info.vkey=new VString; | 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, | |
| false | |
| }; | |
| 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 341 MHash::MHash(): Methoded("hash") | Line 418 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][$.limit(1) $.offset(2)] | // ^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[] | // ^hash._keys[[column name]] |
| add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0); | add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1); |
| // ^hash._count[] | // ^hash._count[] |
| add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0); | add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0); |