|
|
| version 1.55, 2003/04/04 10:38:26 | version 1.62, 2003/11/20 16:34: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-2003 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_request.h" | #include "pa_request.h" |
| #include "pa_vhash.h" | #include "pa_vhash.h" |
| #include "pa_vvoid.h" | #include "pa_vvoid.h" |
| Line 18 static const char* IDENT_HASH_C="$Date$" | Line 20 static const char* IDENT_HASH_C="$Date$" |
| // class | // class |
| class MHash : public Methoded { | class MHash: public Methoded { |
| public: // VStateless_class | public: // VStateless_class |
| Value *create_new_value(Pool& pool) { return new(pool) VHash(pool); } | Value* create_new_value(Pool&) { return new VHash(); } |
| public: | public: |
| MHash(Pool& pool); | MHash(); |
| public: // Methoded | public: // Methoded |
| bool used_directly() { return true; } | bool used_directly() { return true; } |
| }; | }; |
| // global variable | |
| DECLARE_CLASS_VAR(hash, new MHash, 0); | |
| // methods | // methods |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers { | class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers { |
| const String& statement_string; const char* statement_cstr; | |
| bool distinct; | |
| HashStringValue& rows_hash; | |
| HashStringValue* row_hash; | |
| int column_index; | |
| ArrayString columns; | |
| public: | public: |
| Hash_sql_event_handlers(Pool& apool, const String& amethod_name, | Hash_sql_event_handlers( |
| const String& astatement_string, const char *astatement_cstr, | const String& astatement_string, const char* astatement_cstr, |
| bool adistinct, | bool adistinct, |
| Hash& arows_hash): | HashStringValue& arows_hash): |
| pool(apool), | statement_string(astatement_string), statement_cstr(astatement_cstr), |
| method_name(amethod_name), | |
| statement_string(astatement_string), | |
| statement_cstr(astatement_cstr), | |
| distinct(adistinct), | distinct(adistinct), |
| rows_hash(arows_hash), | rows_hash(arows_hash), |
| columns(pool), | row_hash(0), |
| row_index(0) { | column_index(0) { |
| } | } |
| bool add_column(SQL_Error& error, void *ptr, size_t size) { | bool add_column(SQL_Error& error, const char* str, size_t length) { |
| try { | try { |
| String *column=new(pool) String(pool); | columns+=new String(str, length, true); |
| column->APPEND_TAINTED( | |
| (const char *)ptr, size, | |
| statement_cstr, 0); | |
| columns+=column; | |
| return false; | return false; |
| } catch(...) { | } catch(...) { |
| error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column"); | error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column"); |
| Line 61 public: | Line 65 public: |
| } | } |
| } | } |
| bool before_rows(SQL_Error& error) { | bool before_rows(SQL_Error& error) { |
| if(columns.size()<=1) { | if(columns.count()<=1) { |
| error=SQL_Error("parser.runtime", | error=SQL_Error("parser.runtime", |
| &method_name, | /*method_name,*/ |
| "column count must be more than 1 to create a hash"); | "column count must be more than 1 to create a hash"); |
| return true; | return true; |
| } | } |
| Line 74 public: | Line 78 public: |
| column_index=0; | column_index=0; |
| return false; | 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 { | try { |
| String *cell=new(pool) String(pool); | String& cell=*new String; |
| if(size) | if(length) |
| cell->APPEND_TAINTED( | cell.append_know_length(ptr, length, String::L_TAINTED); |
| (const char *)ptr, size, | |
| statement_cstr, row_index++); | |
| if(column_index==0) { | if(column_index==0) { |
| VHash *row_vhash=new(pool) VHash(pool); | VHash* row_vhash=new VHash; |
| row_hash=row_vhash->get_hash(0); | row_hash=&row_vhash->hash(); |
| if(rows_hash.put_dont_replace(*cell, row_vhash)) // put. existed? | if(rows_hash.put_dont_replace(cell, row_vhash)) // put. existed? |
| if(!distinct) { | if(!distinct) { |
| error=SQL_Error("parser.runtime", | error=SQL_Error("parser.runtime", |
| cell, | /*cell,*/ |
| "duplicate key"); | "duplicate key"); |
| return true; | return true; |
| } | } |
| } else | } else |
| row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); | row_hash->put(*columns[column_index], new VString(cell)); |
| column_index++; | column_index++; |
| return false; | return false; |
| Line 102 public: | Line 104 public: |
| } | } |
| } | } |
| 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; | |
| Array columns; | |
| int row_index; | |
| }; | }; |
| #endif | #endif |
| static void copy_all_overwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { | static void copy_all_overwrite_to( |
| Hash& dest=*static_cast<Hash *>(info); | HashStringValue::key_type key, |
| dest.put(key, value); | HashStringValue::value_type value, |
| } | HashStringValue* dest) { |
| static void _create_or_add(Request& r, const String& method_name, MethodParams *params, | dest->put(key, value); |
| bool is_create) { | } |
| Pool& pool=r.pool(); | static void _create_or_add(Request& r, MethodParams& params) { |
| if(params.count()) { | |
| if(params->size()) { | Value& vb=params.as_no_junction(0, "param must be hash"); |
| Value& vb_maybehash=params->as_no_junction(0, "param must be hash"); | if(HashStringValue* b=vb.get_hash()) |
| if(VHash* vbh=static_cast<VHash*>(vb_maybehash.as("hash", false))) { | b->for_each(copy_all_overwrite_to, &(GET_SELF(r, VHash).hash())); |
| VHash& vah=*static_cast<VHash *>(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 | |
| 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<Hash *>(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<VHash *>(r.get_self())->hash(&method_name)); | |
| } | |
| static void copy_all_dontoverwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { | static void remove_key_from( |
| Hash& dest=*static_cast<Hash *>(info); | HashStringValue::key_type key, |
| dest.put_dont_replace(key, value); | HashStringValue::value_type /*value*/, |
| HashStringValue* dest) { | |
| dest->remove(key); | |
| } | |
| static void _sub(Request& r, MethodParams& params) { | |
| Value& vb=params.as_no_junction(0, "param must be hash"); | |
| if(HashStringValue* b=vb.get_hash()) | |
| b->for_each(remove_key_from, &GET_SELF(r, VHash).hash()); | |
| } | |
| static void copy_all_dontoverwrite_to( | |
| HashStringValue::key_type key, | |
| HashStringValue::value_type value, | |
| HashStringValue* dest) { | |
| dest->put_dont_replace(key, value); | |
| } | } |
| static void _union(Request& r, const String& method_name, MethodParams *params) { | static void _union(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | |
| // dest = copy of self | // dest = copy of self |
| Hash& dest=*new(pool) Hash(static_cast<VHash *>(r.get_self())->hash(&method_name)); | 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& vb=params.as_no_junction(0, "param must be hash"); |
| if(Hash *b=vb.get_hash(&method_name)) | if(HashStringValue* b=vb.get_hash()) |
| b->for_each(copy_all_dontoverwrite_to, &dest); | b->for_each(copy_all_dontoverwrite_to, result.get_hash()); |
| // return result | // return result |
| Value& result=*new(pool) VHash(pool, dest); | |
| r.write_no_lang(result); | r.write_no_lang(result); |
| } | } |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| struct Copy_intersection_to_info { | struct Copy_intersection_to_info { |
| Hash *b; | HashStringValue* b; |
| Hash *dest; | HashStringValue* dest; |
| }; | }; |
| #endif | #endif |
| static void copy_intersection_to( | |
| static void copy_intersection_to(const Hash::Key& key, Hash::Val *value, void *info) { | HashStringValue::key_type key, |
| Copy_intersection_to_info& i=*static_cast<Copy_intersection_to_info *>(info); | HashStringValue::value_type value, |
| Copy_intersection_to_info *info) { | |
| if(i.b->get(key)) | if(info->b->get(key)) |
| i.dest->put_dont_replace(key, value); | info->dest->put_dont_replace(key, value); |
| } | } |
| static void _intersection(Request& r, const String& method_name, MethodParams *params) { | static void _intersection(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | Value& result=*new VHash; |
| // dest = copy of self | |
| Hash& dest=*new(pool) Hash(pool); | |
| // dest += b | // dest += b |
| Value& vb=params->as_no_junction(0, "param must be hash"); | Value& vb=params.as_no_junction(0, "param must be hash"); |
| if(Hash *b=vb.get_hash(&method_name)) { | if(HashStringValue* b=vb.get_hash()) { |
| Copy_intersection_to_info info={ | Copy_intersection_to_info info={b, result.get_hash()}; |
| b, | GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info); |
| &dest | |
| }; | |
| static_cast<VHash *>(r.get_self())->hash(&method_name).for_each(copy_intersection_to, &info); | |
| } | } |
| // return result | // return result |
| r.write_no_lang(*new(pool) VHash(pool, dest)); | r.write_no_lang(result); |
| } | } |
| static void *intersects(const Hash::Key& key, Hash::Val *value, void *info) { | static bool intersects( |
| return static_cast<Hash *>(info)->get(key); | HashStringValue::key_type key, |
| HashStringValue::value_type /*value*/, | |
| HashStringValue* b) { | |
| return b->get(key)!=0; | |
| } | } |
| static void _intersects(Request& r, const String& method_name, MethodParams *params) { | static void _intersects(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | bool result=false; |
| bool yes=false; | Value& vb=params.as_no_junction(0, "param must be hash"); |
| if(HashStringValue* b=vb.get_hash()) | |
| // dest = copy of self | result=GET_SELF(r, VHash).hash().first_that(intersects, b)!=0; |
| Hash& dest=*new(pool) Hash(pool); | |
| // dest += b | |
| Value& vb=params->as_no_junction(0, "param must be hash"); | |
| if(Hash *b=vb.get_hash(&method_name)) | |
| yes=static_cast<VHash *>(r.get_self())->hash(&method_name).first_that(intersects, b)!=0; | |
| // return result | // return result |
| r.write_no_lang(*new(pool) VBool(pool, yes)); | r.write_no_lang(*new VBool(result)); |
| } | } |
| static void _sql(Request& r, const String& method_name, MethodParams *params) { | extern String sql_limit_name; |
| Pool& pool=r.pool(); | extern String sql_offset_name; |
| extern String sql_default_name; | |
| Value& statement=params->as_junction(0, "statement must be code"); | extern String sql_distinct_name; |
| static void _sql(Request& r, MethodParams& params) { | |
| Value& statement=params.as_junction(0, "statement must be code"); | |
| ulong limit=0; | ulong limit=0; |
| ulong offset=0; | ulong offset=0; |
| bool distinct=false; | bool distinct=false; |
| if(params->size()>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_string()) |
| if(Hash *options=voptions.get_hash(&method_name)) { | if(HashStringValue* options=voptions.get_hash()) { |
| int valid_options=0; | int valid_options=0; |
| if(Value *vlimit=(Value *)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(); |
| } | } |
| if(Value *voffset=(Value *)options->get(*sql_offset_name)) { | if(Value* voffset=options->get(sql_offset_name)) { |
| valid_options++; | valid_options++; |
| offset=(ulong)r.process_to_value(*voffset).as_double(); | offset=(ulong)r.process_to_value(*voffset).as_double(); |
| } | } |
| if(Value *vdistinct=(Value *)options->get(*sql_distinct_name)) { | if(Value* vdistinct=options->get(sql_distinct_name)) { |
| valid_options++; | valid_options++; |
| distinct=r.process_to_value(*vdistinct).as_bool(); | distinct=r.process_to_value(*vdistinct).as_bool(); |
| } | } |
| if(valid_options!=options->size()) | if(valid_options!=options->count()) |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| &method_name, | 0, |
| "called with invalid option"); | "called with invalid option"); |
| } else | } else |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| &method_name, | 0, |
| "options must be hash"); | "options must be hash"); |
| } | } |
| Temp_lang temp_lang(r, String::UL_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= |
| statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); | statement_string.cstr(String::L_UNSPECIFIED, r.connection()); |
| Hash& hash=static_cast<VHash *>(r.get_self())->hash(&method_name); | HashStringValue& hash=GET_SELF(r, VHash).hash(); |
| hash.clear(); | hash.clear(); |
| Hash_sql_event_handlers handlers(pool, method_name, | Hash_sql_event_handlers handlers( |
| statement_string, statement_cstr, | statement_string, statement_cstr, |
| distinct, | distinct, |
| hash); | hash); |
| r.connection()->query( | |
| r.connection(&method_name)->query( | |
| statement_cstr, offset, limit, | statement_cstr, offset, limit, |
| handlers, | handlers, |
| statement_string); | statement_string); |
| } | } |
| static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) { | static void keys_collector( |
| Table& table=*static_cast<Table *>(info); | HashStringValue::key_type key, |
| Pool& pool=table.pool(); | HashStringValue::value_type, |
| Table *table) { | |
| Array& row=*new(pool) Array(pool); | Table::element_type row(new ArrayString); |
| row+=&key; | *row+=new String(key, String::L_TAINTED); |
| table+=&row; | *table+=row; |
| } | } |
| static void _keys(Request& r, const String& method_name, MethodParams *) { | static void _keys(Request& r, MethodParams&) { |
| Pool& pool=r.pool(); | Table::columns_type columns(new ArrayString); |
| *columns+=new String("key"); | |
| Table* table=new Table(columns); | |
| Array& columns=*new(pool) Array(pool); | GET_SELF(r, VHash).hash().for_each(keys_collector, table); |
| columns+=new(pool) String(pool, "key"); | |
| Table& table=*new(pool) Table(pool, &method_name, &columns); | |
| static_cast<VHash *>(r.get_self())->hash(&method_name).for_each(keys_collector, &table); | r.write_no_lang(*new VTable(table)); |
| r.write_no_lang(*new(pool) VTable(pool, &table)); | |
| } | } |
| static void _count(Request& r, const String& method_name, MethodParams *) { | static void _count(Request& r, MethodParams&) { |
| Pool& pool=r.pool(); | r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count())); |
| r.write_no_lang( | |
| *new(pool) VInt(pool, static_cast<VHash *>(r.get_self())->hash(&method_name).size())); | |
| } | } |
| static void _delete(Request& r, const String& method_name, MethodParams *params) { | static void _delete(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | |
| static_cast<VHash *>(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 | #ifndef DOXYGEN |
| Line 324 struct Foreach_info { | Line 281 struct Foreach_info { |
| Request *r; | Request *r; |
| const String* key_var_name; | const String* key_var_name; |
| const String* value_var_name; | const String* value_var_name; |
| Value *body_code; | Value* body_code; |
| Value *delim_maybe_code; | Value* delim_maybe_code; |
| VString *vkey; | VString* vkey; |
| bool need_delim; | bool need_delim; |
| }; | }; |
| #endif | #endif |
| static void one_foreach_cycle( | |
| static void one_foreach_cycle(const Hash::Key& akey, Hash::Val *avalue, | HashStringValue::key_type akey, |
| void *info) { | HashStringValue::value_type avalue, |
| Foreach_info& i=*static_cast<Foreach_info *>(info); | Foreach_info *info) { |
| info->vkey->set_string(*new String(akey, String::L_TAINTED)); | |
| i.vkey->set_string(akey); | Value& ncontext=*info->r->get_method_frame()->caller(); |
| Value& ncontext=*i.r->get_method_frame()->caller(); | ncontext.put_element(*info->key_var_name, info->vkey, false); |
| ncontext.put_element(*i.key_var_name, i.vkey, false); | ncontext.put_element(*info->value_var_name, avalue, false); |
| ncontext.put_element(*i.value_var_name, static_cast<Value *>(avalue), false); | |
| StringOrValue sv_processed=info->r->process(*info->body_code); | |
| StringOrValue sv_processed=i.r->process(*i.body_code); | 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(i.delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body | if(info->need_delim) // need delim & iteration produced string? |
| if(i.need_delim) // need delim & iteration produced string? | info->r->write_pass_lang(info->r->process(*info->delim_maybe_code)); |
| i.r->write_pass_lang(i.r->process(*i.delim_maybe_code)); | info->need_delim=true; |
| i.need_delim=true; | |
| } | } |
| i.r->write_pass_lang(sv_processed); | info->r->write_pass_lang(sv_processed); |
| } | } |
| static void _foreach(Request& r, const String& method_name, MethodParams *params) { | static void _foreach(Request& r, MethodParams& params) { |
| Pool& pool=r.pool(); | Foreach_info info={0}; |
| const String& key_var_name=params->as_string(0, "key-var name must be string"); | info.r=&r; |
| const String& value_var_name=params->as_string(1, "value-var name must be string"); | info.key_var_name=¶ms.as_string(0, "key-var name must be string"); |
| Value& body_code=params->as_junction(2, "body must be code"); | info.value_var_name=¶ms.as_string(1, "value-var name must be string"); |
| Value *delim_maybe_code=params->size()>3?¶ms->get(3):0; | info.body_code=¶ms.as_junction(2, "body must be code"); |
| info.delim_maybe_code=params.count()>3?params.get(3):0; | |
| Foreach_info info={ | info.vkey=new VString; |
| &r, | |
| &key_var_name, &value_var_name, | VHash& self=GET_SELF(r, VHash); |
| &body_code, | HashStringValue& hash=self.hash(); |
| delim_maybe_code, | |
| new(pool) VString(pool), | |
| false | |
| }; | |
| VHash& self=*static_cast<VHash *>(r.get_self()); | |
| Hash& hash=self.hash(&method_name); | |
| VHash_lock lock(self); | VHash_lock lock(self); |
| hash.for_each(one_foreach_cycle, &info); | hash.for_each(one_foreach_cycle, &info); |
| } | } |
| // constructor | // constructor |
| MHash::MHash(Pool& apool) : Methoded(apool, "hash") | MHash::MHash(): Methoded("hash") |
| { | { |
| // ^hash::create[[copy_from]] | // ^hash::create[[copy_from]] |
| add_native_method("create", Method::CT_DYNAMIC, _create, 0, 1); | add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1); |
| // ^hash.add[add_from] | // ^hash.add[add_from] |
| add_native_method("add", Method::CT_DYNAMIC, _add, 1, 1); | add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1); |
| // ^hash.sub[sub_from] | // ^hash.sub[sub_from] |
| add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1); | add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1); |
| // ^a.union[b] = hash | // ^a.union[b] = hash |
| Line 404 MHash::MHash(Pool& apool) : Methoded(apo | Line 353 MHash::MHash(Pool& apool) : Methoded(apo |
| // ^hash.foreach[key;value]{code}[delim] | // ^hash.foreach[key;value]{code}[delim] |
| add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); | add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); |
| } | } |
| // global variable | |
| Methoded *hash_class; | |
| // creator | |
| Methoded *MHash_create(Pool& pool) { | |
| return hash_class=new(pool) MHash(pool); | |
| } |