--- parser3/src/classes/hash.C 2001/05/21 16:01:10 1.2 +++ parser3/src/classes/hash.C 2001/11/01 14:59:57 1.26 @@ -2,17 +2,19 @@ Parser: @b hash parser class. Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) - $Id: hash.C,v 1.2 2001/05/21 16:01:10 parser Exp $ + $Id: hash.C,v 1.26 2001/11/01 14:59:57 paf Exp $ */ #include "classes.h" #include "pa_request.h" #include "pa_vhash.h" -#include "pa_vunknown.h" +#include "pa_vvoid.h" #include "pa_sql_connection.h" +#include "pa_vtable.h" +#include "pa_vbool.h" +#include "pa_vmethod_frame.h" // defines @@ -32,37 +34,189 @@ public: // Methoded // methods -static void _default(Request& r, const String&, MethodParams *params) { +#ifndef DOXYGEN +class Hash_sql_event_handlers : public SQL_Driver_query_event_handlers { +public: + Hash_sql_event_handlers(Pool& apool, const String& amethod_name, + const String& astatement_string, const char *astatement_cstr, + Hash& arows_hash) : + pool(apool), + method_name(amethod_name), + statement_string(astatement_string), + statement_cstr(astatement_cstr), + rows_hash(arows_hash), + columns(pool), + row_index(0) { + } + void add_column(void *ptr, size_t size) { + String *column=new(pool) String(pool); + column->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, 0); + columns+=column; + } + void before_rows() { + if(columns.size()<=1) + throw Exception(0, 0, + &method_name, + "column count must be more than 1 to create a hash"); + } + void add_row() { + column_index=0; + } + void add_row_cell(void *ptr, size_t size) { + String *cell=new(pool) String(pool); + if(size) + cell->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, row_index++); + if(column_index==0) { + VHash *row_vhash=new(pool) VHash(pool); + row_hash=row_vhash->get_hash(0); + rows_hash.put(*cell, row_vhash); + } else + row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell)); + column_index++; + } + +private: + Pool& pool; + const String& method_name; + const String& statement_string; const char *statement_cstr; + Hash& rows_hash; + Hash *row_hash; + int column_index; + Array columns; + int row_index; +}; +#endif + +static void copy_all_overwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { + Hash& dest=*static_cast(info); + dest.put(key, value); +} + +static void _create_or_add(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); + + if(params->size()) { + Value& vb=params->as_no_junction(0, "param must be hash"); + if(Hash *b=vb.get_hash(&method_name)) + b->for_each(copy_all_overwrite_to, &static_cast(r.self)->hash()); + } +} + +static void remove_key_from(const Hash::Key& key, Hash::Val *value, void *info) { + Hash& dest=*static_cast(info); + dest.remove(key); +} - VHash& vhash=*static_cast(r.self); - if(params->size()) - vhash.set_default(params->get(0)); // info: may be code.. - else { - Value *default_value=vhash.get_default(); - r.write_assign_lang(default_value?*default_value:*new(pool) VUnknown(pool)); +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(r.self)->hash()); +} + +static void copy_all_dontoverwrite_to(const Hash::Key& key, Hash::Val *value, void *info) { + Hash& dest=*static_cast(info); + dest.put_dont_replace(key, value); +} + +static void _union(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + + // dest = copy of self + Hash& dest=*new(pool) Hash(static_cast(r.self)->hash()); + // dest += b + Value& vb=params->as_no_junction(0, "param must be hash"); + if(Hash *b=vb.get_hash(&method_name)) + b->for_each(copy_all_dontoverwrite_to, &dest); + + // return result + Value& result=*new(pool) VHash(pool, dest); + result.set_name(method_name); + r.write_no_lang(result); +} + +#ifndef DOXYGEN +struct Copy_intersection_to_info { + Hash *b; + Hash *dest; +}; +#endif + +static void copy_intersection_to(const Hash::Key& key, Hash::Val *value, void *info) { + Copy_intersection_to_info& i=*static_cast(info); + + if(i.b->get(key)) + i.dest->put_dont_replace(key, value); +} + +static void _intersection(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + + // dest = copy of self + 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)) { + Copy_intersection_to_info info={ + b, + &dest + }; + static_cast(r.self)->hash().for_each(copy_intersection_to, &info); } + + // return result + Value& result=*new(pool) VHash(pool, dest); + result.set_name(method_name); + r.write_no_lang(result); } +static void *intersects(const Hash::Key& key, Hash::Val *value, void *info) { + return static_cast(info)->get(key); +} + +static void _intersects(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + + bool yes=false; + + // dest = copy of self + 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(r.self)->hash().first_that(intersects, b)!=0; + + // return result + Value& result=*new(pool) VBool(pool, yes); + result.set_name(method_name); + r.write_no_lang(result); +} + + static void _sql(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); if(!r.connection) - PTHROW(0, 0, + throw Exception(0, 0, &method_name, "without connect"); - Value& statement=params->get_junction(0, "statement must be code"); + Value& statement=params->as_junction(0, "statement must be code"); ulong limit=0; if(params->size()>1) { - Value& limit_code=params->get_junction(1, "limit must be expression"); + Value& limit_code=params->as_junction(1, "limit must be expression"); limit=(uint)r.process(limit_code).as_double(); } ulong offset=0; if(params->size()>2) { - Value& offset_code=params->get_junction(2, "offset must be expression"); + Value& offset_code=params->as_junction(2, "offset must be expression"); offset=(ulong)r.process(offset_code).as_double(); } @@ -70,58 +224,101 @@ static void _sql(Request& r, const Strin const String& statement_string=r.process(statement).as_string(); const char *statement_cstr= statement_string.cstr(String::UL_UNSPECIFIED, r.connection); - unsigned int sql_column_count; SQL_Driver::Cell *sql_columns; - unsigned long sql_row_count; SQL_Driver::Cell **sql_rows; - bool need_rethrow=false; Exception rethrow_me; - PTRY { - r.connection->query( - statement_cstr, offset, limit, - &sql_column_count, &sql_columns, - &sql_row_count, &sql_rows); - } - PCATCH(e) { // connect/process problem - rethrow_me=e; need_rethrow=true; - } - PEND_CATCH - if(need_rethrow) - PTHROW(rethrow_me.type(), rethrow_me.code(), - &statement_string, // setting more specific source [were url] - rethrow_me.comment()); - - Hash& rows_hash=static_cast(r.self)->hash(); - rows_hash.clear(); + Hash& hash=static_cast(r.self)->hash(); + hash.clear(); + Hash_sql_event_handlers handlers(pool, method_name, + statement_string, statement_cstr, hash); + + r.connection->query( + statement_cstr, offset, limit, + handlers); +} - if(sql_column_count<=1) - return; +static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) { + Table& table=*static_cast(info); + Pool& pool=table.pool(); + + Array& row=*new(pool) Array(pool); + row+=&key; + table+=&row; +} +static void _keys(Request& r, const String& method_name, MethodParams *) { + Pool& pool=r.pool(); Array& columns=*new(pool) Array(pool); - for(unsigned int i=0+1; iAPPEND_TAINTED( - (const char *)sql_cells[i].ptr, sql_cells[i].size, - statement_cstr, row); - if(i==0) - key=cell; - else - row_hash.put(*columns.get_string(i-1), new(pool) VString(*cell)); - } - rows_hash.put(*key, &row_vhash); + columns+=new(pool) String(pool, "key"); + Table& table=*new(pool) Table(pool, &method_name, &columns); + + static_cast(r.self)->hash().for_each(keys_collector, &table); + + VTable& result=*new(pool) VTable(pool, &table); + result.set_name(method_name); + r.write_no_lang(result); +} + +static void _count(Request& r, const String& method_name, MethodParams *) { + Pool& pool=r.pool(); + + Value& result=*new(pool) VInt(pool, static_cast(r.self)->hash().size()); + result.set_name(method_name); + r.write_no_lang(result); +} + +static void _delete(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + + static_cast(r.self)->hash().remove(params->as_string(0, "key must be string")); +} + +#ifndef DOXYGEN +struct Foreach_info { + Request *r; + const String* key_var_name; + const String* value_var_name; + Value *body_code; + Value *delim_maybe_code; + + Value *var_context; + VString *vkey; + bool need_delim; +}; +#endif + +static void one_foreach_cycle(const Hash::Key& akey, Hash::Val *avalue, + void *info) { + Foreach_info& i=*static_cast(info); + + i.vkey->set_string(akey); + i.var_context->put_element(*i.key_var_name, i.vkey); + i.var_context->put_element(*i.value_var_name, static_cast(avalue)); + + Value& processed_body=i.r->process(*i.body_code); + if(i.delim_maybe_code) { // delimiter set? + const String *string=processed_body.get_string(); + if(i.need_delim && string && string->size()) // need delim & iteration produced string? + i.r->write_pass_lang(i.r->process(*i.delim_maybe_code)); + i.need_delim=true; } + i.r->write_pass_lang(processed_body); +} +static void _foreach(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + 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"); + Value& body_code=params->as_junction(2, "body must be code"); + Value *delim_maybe_code=params->size()>3?¶ms->get(3):0; + + Foreach_info info={ + &r, + &key_var_name, &value_var_name, + &body_code, + delim_maybe_code, + + body_code.get_junction()->wcontext, + new(pool) VString(pool), + false + }; + static_cast(r.self)->hash().for_each(one_foreach_cycle, &info); } // constructor @@ -129,13 +326,36 @@ static void _sql(Request& r, const Strin MHash::MHash(Pool& apool) : Methoded(apool) { set_name(*NEW String(pool(), HASH_CLASS_NAME)); - // ^hash.default[] - // ^hash.default[hash] - add_native_method("default", Method::CT_DYNAMIC, _default, 0, 1); + // ^hash::create[[copy_from]] + add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1); + // ^hash.add[add_from] + add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1); + // ^hash.sub[sub_from] + add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1); + // ^a.union[b] = hash + add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1); + // ^a.intersection[b] = hash + add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1); + // ^a.intersects[b] = bool + add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1); + + // ^a.delete[key] + add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1); // ^hash:sql[query][(count[;offset])] add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3); + // ^hash._keys[] + add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0); + + // ^hash._count[] + add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0); + + // ^hash.foreach[key;value]{code}[delim] + Array *for_each_locals=NEW Array(pool(), 2); + *for_each_locals += NEW String(pool(), "key"); + + add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1); } // global variable