|
|
| version 1.46, 2002/08/29 12:22:46 | version 1.56, 2003/04/25 11:39:54 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b hash parser class. | Parser: @b hash parser class. |
| Copyright (c) 2001, 2002 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) |
| */ | */ |
| Line 31 public: // Methoded | Line 31 public: // Methoded |
| // 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 { |
| public: | public: |
| Hash_sql_event_handlers(Pool& apool, const String& amethod_name, | Hash_sql_event_handlers(Pool& apool, const String& amethod_name, |
| const String& astatement_string, const char *astatement_cstr, | const String& astatement_string, const char *astatement_cstr, |
| Hash& arows_hash) : | bool adistinct, |
| Hash& arows_hash): | |
| pool(apool), | pool(apool), |
| method_name(amethod_name), | method_name(amethod_name), |
| statement_string(astatement_string), | statement_string(astatement_string), |
| statement_cstr(astatement_cstr), | statement_cstr(astatement_cstr), |
| distinct(adistinct), | |
| rows_hash(arows_hash), | rows_hash(arows_hash), |
| columns(pool), | columns(pool), |
| row_index(0) { | row_index(0) { |
| } | } |
| void add_column(void *ptr, size_t size) { | bool add_column(SQL_Error& error, void *ptr, size_t size) { |
| String *column=new(pool) String(pool); | try { |
| column->APPEND_TAINTED( | String *column=new(pool) String(pool); |
| (const char *)ptr, size, | column->APPEND_TAINTED( |
| statement_cstr, 0); | (const char *)ptr, size, |
| columns+=column; | statement_cstr, 0); |
| columns+=column; | |
| return false; | |
| } catch(...) { | |
| error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column"); | |
| return true; | |
| } | |
| } | } |
| void before_rows() { | bool before_rows(SQL_Error& error) { |
| if(columns.size()<=1) | if(columns.size()<=1) { |
| throw Exception("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 false; | |
| } | } |
| void add_row() { | bool add_row(SQL_Error& /*error*/) { |
| column_index=0; | column_index=0; |
| return false; | |
| } | } |
| void add_row_cell(void *ptr, size_t size) { | bool add_row_cell(SQL_Error& error, void *ptr, size_t size) { |
| String *cell=new(pool) String(pool); | try { |
| if(size) | String *cell=new(pool) String(pool); |
| cell->APPEND_TAINTED( | if(size) |
| (const char *)ptr, size, | cell->APPEND_TAINTED( |
| statement_cstr, row_index++); | (const char *)ptr, size, |
| if(column_index==0) { | statement_cstr, row_index++); |
| VHash *row_vhash=new(pool) VHash(pool); | if(column_index==0) { |
| row_hash=row_vhash->get_hash(0); | VHash *row_vhash=new(pool) VHash(pool); |
| rows_hash.put(*cell, row_vhash); | row_hash=row_vhash->get_hash(0); |
| } else | if(rows_hash.put_dont_replace(*cell, row_vhash)) // put. existed? |
| row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); | if(!distinct) { |
| column_index++; | error=SQL_Error("parser.runtime", |
| cell, | |
| "duplicate key"); | |
| return true; | |
| } | |
| } else | |
| row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); | |
| column_index++; | |
| return false; | |
| } catch(...) { | |
| error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell"); | |
| return true; | |
| } | |
| } | } |
| private: | private: |
| Pool& pool; | Pool& pool; |
| const String& method_name; | const String& method_name; |
| const String& statement_string; const char *statement_cstr; | const String& statement_string; const char *statement_cstr; |
| bool distinct; | |
| Hash& rows_hash; | Hash& rows_hash; |
| Hash *row_hash; | Hash *row_hash; |
| int column_index; | int column_index; |
| Line 91 static void copy_all_overwrite_to(const | Line 119 static void copy_all_overwrite_to(const |
| Hash& dest=*static_cast<Hash *>(info); | Hash& dest=*static_cast<Hash *>(info); |
| dest.put(key, value); | dest.put(key, value); |
| } | } |
| static void _create_or_add(Request& r, const String& method_name, MethodParams *params) { | static void _create_or_add(Request& r, const String& method_name, MethodParams *params, |
| bool is_create) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| if(params->size()) { | 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(Hash *b=vb.get_hash(&method_name)) | if(VHash* vbh=static_cast<VHash*>(vb_maybehash.as("hash", false))) { |
| b->for_each(copy_all_overwrite_to, &static_cast<VHash *>(r.self)->hash(&method_name)); | 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 if(!vb_maybehash.is_string()) // allow ^hash::create[^rem{xxx}] | |
| 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) { | static void remove_key_from(const Hash::Key& key, Hash::Val *value, void *info) { |
| Hash& dest=*static_cast<Hash *>(info); | Hash& dest=*static_cast<Hash *>(info); |
| Line 110 static void _sub(Request& r, const Strin | Line 157 static void _sub(Request& r, const Strin |
| 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(Hash *b=vb.get_hash(&method_name)) |
| b->for_each(remove_key_from, &static_cast<VHash *>(r.self)->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 copy_all_dontoverwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { |
| Line 121 static void _union(Request& r, const Str | Line 168 static void _union(Request& r, const Str |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| // dest = copy of self | // dest = copy of self |
| Hash& dest=*new(pool) Hash(static_cast<VHash *>(r.self)->hash(&method_name)); | Hash& dest=*new(pool) Hash(static_cast<VHash *>(r.get_self())->hash(&method_name)); |
| // 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(Hash *b=vb.get_hash(&method_name)) |
| Line 157 static void _intersection(Request& r, co | Line 204 static void _intersection(Request& r, co |
| b, | b, |
| &dest | &dest |
| }; | }; |
| static_cast<VHash *>(r.self)->hash(&method_name).for_each(copy_intersection_to, &info); | static_cast<VHash *>(r.get_self())->hash(&method_name).for_each(copy_intersection_to, &info); |
| } | } |
| // return result | // return result |
| Line 178 static void _intersects(Request& r, cons | Line 225 static void _intersects(Request& r, cons |
| // 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(Hash *b=vb.get_hash(&method_name)) |
| yes=static_cast<VHash *>(r.self)->hash(&method_name).first_that(intersects, b)!=0; | 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(pool) VBool(pool, yes)); |
| Line 192 static void _sql(Request& r, const Strin | Line 239 static void _sql(Request& r, const Strin |
| ulong limit=0; | ulong limit=0; |
| ulong offset=0; | ulong offset=0; |
| bool distinct=false; | |
| if(params->size()>1) { | if(params->size()>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(Hash *options=voptions.get_hash(&method_name)) { |
| if(Value *vlimit=(Value *)options->get(*sql_limit_name)) | int valid_options=0; |
| if(Value *vlimit=(Value *)options->get(*sql_limit_name)) { | |
| 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=(Value *)options->get(*sql_offset_name)) { | |
| 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)) { | |
| valid_options++; | |
| distinct=r.process_to_value(*vdistinct).as_bool(); | |
| } | |
| if(valid_options!=options->size()) | |
| throw Exception("parser.runtime", | |
| &method_name, | |
| "called with invalid option"); | |
| } else | } else |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| &method_name, | &method_name, |
| Line 210 static void _sql(Request& r, const Strin | Line 271 static void _sql(Request& r, const Strin |
| 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::UL_UNSPECIFIED, r.connection(&method_name)); |
| Hash& hash=static_cast<VHash *>(r.self)->hash(&method_name); | Hash& hash=static_cast<VHash *>(r.get_self())->hash(&method_name); |
| hash.clear(); | hash.clear(); |
| Hash_sql_event_handlers handlers(pool, method_name, | Hash_sql_event_handlers handlers(pool, method_name, |
| statement_string, statement_cstr, hash); | statement_string, statement_cstr, |
| distinct, | |
| hash); | |
| r.connection(&method_name)->query( | r.connection(&method_name)->query( |
| statement_cstr, offset, limit, | statement_cstr, offset, limit, |
| Line 236 static void _keys(Request& r, const Stri | Line 299 static void _keys(Request& r, const Stri |
| columns+=new(pool) String(pool, "key"); | columns+=new(pool) String(pool, "key"); |
| Table& table=*new(pool) Table(pool, &method_name, &columns); | Table& table=*new(pool) Table(pool, &method_name, &columns); |
| static_cast<VHash *>(r.self)->hash(&method_name).for_each(keys_collector, &table); | static_cast<VHash *>(r.get_self())->hash(&method_name).for_each(keys_collector, &table); |
| r.write_no_lang(*new(pool) VTable(pool, &table)); | r.write_no_lang(*new(pool) VTable(pool, &table)); |
| } | } |
| Line 245 static void _count(Request& r, const Str | Line 308 static void _count(Request& r, const Str |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| r.write_no_lang( | r.write_no_lang( |
| *new(pool) VInt(pool, static_cast<VHash *>(r.self)->hash(&method_name).size())); | *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, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| static_cast<VHash *>(r.self)->hash(&method_name).remove(params->as_string(0, "key must be string")); | static_cast<VHash *>(r.get_self())->hash(&method_name).remove(params->as_string(0, "key must be string")); |
| } | } |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| Line 272 static void one_foreach_cycle(const Hash | Line 335 static void one_foreach_cycle(const Hash |
| Foreach_info& i=*static_cast<Foreach_info *>(info); | Foreach_info& i=*static_cast<Foreach_info *>(info); |
| i.vkey->set_string(akey); | i.vkey->set_string(akey); |
| i.r->method_frame->put_element(*i.key_var_name, i.vkey, false); | Value& ncontext=*i.r->get_method_frame()->caller(); |
| i.r->method_frame->put_element(*i.value_var_name, static_cast<Value *>(avalue), false); | ncontext.put_element(*i.key_var_name, i.vkey, false); |
| ncontext.put_element(*i.value_var_name, static_cast<Value *>(avalue), false); | |
| StringOrValue processed_body=i.r->process(*i.body_code); | |
| if(i.delim_maybe_code) { // delimiter set? | StringOrValue sv_processed=i.r->process(*i.body_code); |
| const String *string=processed_body.get_string(); | const String *s_processed=sv_processed.get_string(); |
| if(i.need_delim && string && string->size()) // need delim & iteration produced 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.r->write_pass_lang(i.r->process(*i.delim_maybe_code)); |
| i.need_delim=true; | i.need_delim=true; |
| } | } |
| i.r->write_pass_lang(processed_body); | i.r->write_pass_lang(sv_processed); |
| } | } |
| static void _foreach(Request& r, const String& method_name, MethodParams *params) { | static void _foreach(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Line 300 static void _foreach(Request& r, const S | Line 364 static void _foreach(Request& r, const S |
| new(pool) VString(pool), | new(pool) VString(pool), |
| false | false |
| }; | }; |
| VHash& self=*static_cast<VHash *>(r.self); | VHash& self=*static_cast<VHash *>(r.get_self()); |
| Hash& hash=self.hash(&method_name); | 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); |
| Line 311 static void _foreach(Request& r, const S | Line 375 static void _foreach(Request& r, const S |
| MHash::MHash(Pool& apool) : Methoded(apool, "hash") | MHash::MHash(Pool& apool) : Methoded(apool, "hash") |
| { | { |
| // ^hash::create[[copy_from]] | // ^hash::create[[copy_from]] |
| add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1); | add_native_method("create", Method::CT_DYNAMIC, _create, 0, 1); |
| // ^hash.add[add_from] | // ^hash.add[add_from] |
| add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1); | add_native_method("add", Method::CT_DYNAMIC, _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 |