|
|
| version 1.82, 2007/12/28 11:23:13 | version 1.88, 2008/07/03 09:17:57 |
|---|---|
| Line 45 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 71 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"); |
| Line 230 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; | Table2hash_value_type value_type=C_HASH; |
| Line 283 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 342 struct Foreach_info { | Line 424 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; |
| }; | }; |
| Line 350 static bool one_foreach_cycle( | Line 433 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); |
| } | |
| 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); | 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; | return lskip==Request::SKIP_BREAK; |
| Line 371 static void _foreach(Request& r, MethodP | Line 460 static void _foreach(Request& r, MethodP |
| Temp_hash_value<const String::Body, void*> | Temp_hash_value<const String::Body, void*> |
| cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r); | 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 |
| }; | }; |