--- parser3/src/classes/table.C 2001/07/18 10:06:04 1.93 +++ parser3/src/classes/table.C 2001/08/10 07:10:17 1.102 @@ -5,7 +5,7 @@ Author: Alexander Petrosyan (http://design.ru/paf) */ -static const char *RCSId="$Id: table.C,v 1.93 2001/07/18 10:06:04 parser Exp $"; +static const char *RCSId="$Id: table.C,v 1.102 2001/08/10 07:10:17 parser Exp $"; #include "pa_config_includes.h" @@ -65,10 +65,10 @@ static void _set(Request& r, const Strin // parse cells Array rows(pool); data.split(rows, &pos_after, "\n", 1, String::UL_CLEAN); - int size=rows.quick_size(); - for(int i=0; isize()==1) { // not nameless=named output // write out names line if(table.columns()) { // named table - for(int column=0; columnsize(); column++) { - if(column) - sdata.APPEND_CONST("\t"); - sdata.append(*static_cast(table.columns()->quick_get(column)), + Array_iter i(*table.columns()); + while(i.has_next()) { + sdata.append(*i.next_string(), //*static_cast(table.columns()->quick_get(column)), String::UL_TABLE); + if(i.has_next()) + sdata.APPEND_CONST("\t"); } } else { // nameless table - int lsize=table.size()?static_cast(table.get(0))->size():0; - if(lsize) + if(int lsize=table.size()?static_cast(table.get(0))->size():0) for(int column=0; column(table.quick_get(index)); - for(int column=0; columnsize(); column++) { - if(column) - sdata.APPEND_CONST("\t"); - sdata.append(*static_cast(row->quick_get(column)), + Array_iter i(table); + while(i.has_next()) { + Array_iter c(*static_cast(i.next())); + while(c.has_next()) { + sdata.append(*c.next_string(), //*static_cast(row->quick_get(column)), String::UL_TABLE); + if(c.has_next()) + sdata.APPEND_CONST("\t"); } sdata.APPEND_CONST("\n"); } @@ -326,8 +327,9 @@ static void _menu(Request& r, const Stri VTable& vtable=*static_cast(r.self); Table& table=vtable.table(); bool need_delim=false; - vtable.lock(); int saved_current=table.current(); - for(int row=0; row(info); - String& column_name=*static_cast(item); - Value *value; - if(const String *column_item=ri.table->item(column_name)) - value=new(*ri.pool) VString(*column_item); - else - value=new(*ri.pool) VVoid(*ri.pool); - ri.hash->put(column_name, value); -} -static void _record(Request& r, const String& method_name, MethodParams *) { - Table& table=static_cast(r.self)->table(); - if(const Array *columns=table.columns()) { - Pool& pool=r.pool(); - Value& result=*new(pool) VHash(pool); - Record_info record_info={&pool, &table, result.get_hash()}; - columns->for_each(store_column_item_to_hash, &record_info); - result.set_name(method_name); - r.write_no_lang(result); - } -} - /// used by table: _hash / table_row_to_hash struct Row_info { Table *table; @@ -404,10 +378,11 @@ static void table_row_to_hash(Array::Ite } } static void _hash(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); Table& table=static_cast(r.self)->table(); + Value& result=*new(pool) VHash(pool); if(const Array *columns=table.columns()) if(columns->size()>1) { - Pool& pool=r.pool(); const String& key_field_name=params->as_no_junction(0, "key field name must not be code").as_string(); @@ -430,12 +405,11 @@ static void _hash(Request& r, const Stri } // integers: key_field & value_fields - Value& result=*new(pool) VHash(pool); Row_info row_info={&table, key_field, &value_fields, result.get_hash()}; table.for_each(table_row_to_hash, &row_info); - result.set_name(method_name); - r.write_no_lang(result); } + result.set_name(method_name); + r.write_no_lang(result); } /// used by table: _sort / sort_cmp_string|sort_cmp_double @@ -463,27 +437,27 @@ static int sort_cmp_double(const void *a return 0; } static void _sort(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); Value& key_maker=params->as_junction(0, "key-maker must be code"); - bool reverse=params->size()==2/*..[asc|desc]*/? - reverse=params->as_no_junction(1, "order must not be code").as_string()=="desc": + bool reverse=params->size()==2/*..[desc|asc|]*/? + reverse=params->as_no_junction(1, "order must not be code").as_string()=="asc": false; - Table& table=static_cast(r.self)->table(); - - // anything to sort? - if(!table.size()) - return; + Table& old_table=static_cast(r.self)->table(); + Table& new_table=*new(pool) Table(pool, &method_name, old_table.columns()); - Table_seq_item *seq=(Table_seq_item *)malloc(sizeof(Table_seq_item)*table.size()); + Table_seq_item *seq=(Table_seq_item *)pool.malloc(sizeof(Table_seq_item)*old_table.size()); int i; // calculate key values bool key_values_are_strings=true; - for(i=0; i(r.self); - // disable ^a.menu{^a.append[]} - vtable.lock(); vtable.table()+=&row; - vtable.unlock(); } static void _join(Request& r, const String& method_name, MethodParams *params) { @@ -586,6 +558,54 @@ static void _join(Request& r, const Stri } } +#ifndef DOXYGEN +class Table_sql_event_handlers : public SQL_Driver_query_event_handlers { +public: + Table_sql_event_handlers(Pool& apool, const String& amethod_name, + const String& astatement_string, const char *astatement_cstr) : + pool(apool), + method_name(amethod_name), + statement_string(astatement_string), + statement_cstr(astatement_cstr), + columns(*new(pool) Array(pool)), + row(0), row_index(0), + table(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() { + table=new(pool) Table(pool, &method_name, &columns); + } + void add_row() { + (*table)+=(row=new(pool) Array(pool)); + } + 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++); + (*row)+=cell; + } + +private: + Pool& pool; + const String& method_name; + const String& statement_string; const char *statement_cstr; + Array& columns; + Array *row; + uint row_index; +public: + Table *table; +}; +#endif static void _sql(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); @@ -597,29 +617,31 @@ static void _sql(Request& r, const Strin Value& statement=params->as_junction(0, "statement must be code"); ulong limit=0; - if(params->size()>1) { - 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->as_junction(2, "offset must be expression"); - offset=(ulong)r.process(offset_code).as_double(); + if(params->size()>1) { + Value& options_param=params->as_no_junction(1, "options must be hash, not code"); + if(Hash *options=options_param.get_hash()) { + if(Value *vlimit=(Value *)options->get(*sql_limit_name)) + limit=(ulong)r.process(*vlimit).as_double(); + if(Value *voffset=(Value *)options->get(*sql_offset_name)) + offset=(ulong)r.process(*voffset).as_double(); + } else + PTHROW(0, 0, + &method_name, + "options must be hash"); } Temp_lang temp_lang(r, String::UL_SQL); 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; + Table_sql_event_handlers handlers(pool, method_name, + statement_string, statement_cstr); 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); + handlers); } PCATCH(e) { // query problem rethrow_me=e; need_rethrow=true; @@ -629,35 +651,13 @@ static void _sql(Request& r, const Strin PTHROW(rethrow_me.type(), rethrow_me.code(), &statement_string, // setting more specific source [were url] rethrow_me.comment()); - - Array& table_columns=*new(pool) Array(pool); - for(unsigned int i=0; i(r.self)->set_table(table); + static_cast(r.self)->set_table(*result); } static void _dir(Request& r, const String& method_name, MethodParams *params) { @@ -744,10 +744,10 @@ static void _columns(Request& r, const S Table& source_table=static_cast(r.self)->table(); if(const Array *source_columns=source_table.columns()) { - int size=source_columns->quick_size(); - for(int i=0; iquick_get(i); + result_row+=i.next(); result_table+=&result_row; } } @@ -795,15 +795,12 @@ MTable::MTable(Pool& apool) : Methoded(a // ^table.empty[] add_native_method("empty", Method::CT_DYNAMIC, _empty, 0, 0); - // ^table.record[] - add_native_method("record", Method::CT_DYNAMIC, _record, 0, 0); - // ^table:hash[key field name] // ^table:hash[key field name][value field name;...] add_native_method("hash", Method::CT_DYNAMIC, _hash, 1, 1000); - // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[asc|desc] - // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[asc|desc] + // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[desc|asc] + // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[desc|asc] add_native_method("sort", Method::CT_DYNAMIC, _sort, 1, 2); // ^table.locate[field;value] @@ -819,8 +816,9 @@ MTable::MTable(Pool& apool) : Methoded(a add_native_method("join", Method::CT_DYNAMIC, _join, 1, 1); - // ^table:sql[query][(count[;offset])] - add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3); + // ^table:sql[query] + // ^table:sql[query][$.limit(1) $.offset(2)] + add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); // ^table:dir[path] // ^table:dir[path][regexp]