--- parser3/src/classes/table.C 2001/04/05 11:36:51 1.52 +++ parser3/src/classes/table.C 2003/04/25 10:54:54 1.180 @@ -1,35 +1,108 @@ /** @file Parser: @b table parser class. - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - - Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: table.C,v 1.52 2001/04/05 11:36:51 paf Exp $ + Copyright (c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ -#include "pa_config_includes.h" +static const char* IDENT_TABLE_C="$Date: 2003/04/25 10:54:54 $"; + +#include "classes.h" #include "pa_common.h" #include "pa_request.h" -#include "_table.h" #include "pa_vtable.h" #include "pa_vint.h" #include "pa_sql_connection.h" +#include "pa_vbool.h" -// global var +// class -VStateless_class *table_class; +class MTable : public Methoded { +public: // VStateless_class + Value *create_new_value(Pool& pool) { return new(pool) VTable(pool); } + +public: + MTable(Pool& pool); + +public: // Methoded + bool used_directly() { return true; } +}; // methods -static void _set(Request& r, const String& method_name, Array *params) { + +static Table::Action_options get_action_options(Request& r, + const String& method_name, MethodParams *params, + const Table& source) { + Table::Action_options result; + + if(!params->size()) + return result; + + Hash* options=params->get(params->size()-1).get_hash(&method_name); + if(!options) + return result; + + result.defined=true; + bool defined_offset=false; + + int valid_options=0; + if(Value *voffset=(Value *)options->get(*sql_offset_name)) { + valid_options++; + defined_offset=true; + if(voffset->is_string()) { + const String& soffset=*voffset->get_string(); + if(soffset == "cur") + result.offset=source.current(); + else + throw Exception("parser.runtime", + &soffset, + "must be 'cur' string or expression"); + } else + result.offset=r.process_to_value(*voffset).as_int(); + } + if(Value *vlimit=(Value *)options->get(*sql_limit_name)) { + valid_options++; + result.limit=r.process_to_value(*vlimit).as_int(); + } + if(Value *vreverse=(Value *)options->get(*table_reverse_name)) { + valid_options++; + result.reverse=r.process_to_value(*vreverse).as_bool(); + if(result.reverse && !defined_offset) + result.offset=source.size()-1; + } + if(valid_options!=options->size()) + throw Exception("parser.runtime", + &method_name, + "called with invalid option"); + + return result; +} +static void check_option_param(bool options_defined, + const String& method_name, MethodParams *params, + int next_param_index, + const char *msg) { + if(next_param_index+(options_defined?1:0) != params->size()) + throw Exception("parser.runtime", + &method_name, + "%s", msg); +} + +static void _create(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - // data is last parameter - Value *vdata=static_cast(params->get(params->size()-1)); - // forcing {this body type} - r.fail_if_junction_(false, *vdata, method_name, "body must be junction"); + // clone/copy part? + if(const Table *source=params->get(0).get_table()) { + Table::Action_options o=get_action_options(r, method_name, params, *source); + check_option_param(o.defined, method_name, params, 1, + "too many parameters"); + static_cast(r.get_self())-> + set_table(*new(pool) Table(pool, *source, o)); + return; + } + // data is last parameter Temp_lang temp_lang(r, String::UL_PASS_APPENDED); - const String& data=r.process(*vdata).as_string(); + const String& data= + r.process_to_string(params->as_junction(params->size()-1, "body must be code")); size_t pos_after=0; // parse columns @@ -40,41 +113,46 @@ static void _set(Request& r, const Strin columns=new(pool) Array(pool); Array head(pool); - data.split(head, &pos_after, "\n", 1, String::UL_CLEAN, 1); + data.split(head, &pos_after, "\n", 1, String::UL_AS_IS, 1); if(head.size()) - head.get_string(0)->split(*columns, 0, "\t", 1, String::UL_CLEAN); + head.get_string(0)->split(*columns, 0, "\t", 1, String::UL_AS_IS); } Table& table=*new(pool) Table(pool, &method_name, columns); // parse cells Array rows(pool); - data.split(rows, &pos_after, "\n", 1, String::UL_CLEAN); - int size=rows.quick_size(); - for(int i=0; i(r.self)->set_table(table); + static_cast(r.get_self())->set_table(table); } -static void _load(Request& r, const String& method_name, Array *params) { +static void _load(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - // filename is last parameter - Value *vfilename=static_cast(params->get(params->size()-1)); - // forcing [this file name type] - r.fail_if_junction_(true, *vfilename, - method_name, "file name must not be junction"); - + const String& first_param=params->as_string(0, "file name must be string"); + int filename_param_index=0; + bool nameless=first_param=="nameless"; + if(nameless) + filename_param_index++; + int options_param_index=filename_param_index+1; + // loading text - char *data=file_read_text(pool, r.absolute(vfilename->as_string())); + char *data=file_read_text(pool, + r.absolute(params->as_string(filename_param_index, "file name must be string")), + true, + options_param_indexsize()?params->as_no_junction(options_param_index, "additional params must be hash").get_hash(&method_name):0 + ); // parse columns Array *columns; @@ -83,24 +161,31 @@ static void _load(Request& r, const Stri const char *file=origin.file; uint line=origin.line; #endif - if(params->size()==2) { - columns=0; + if(nameless) { + columns=0; // nameless } else { columns=new(pool) Array(pool); - if(char *row_chars=getrow(&data)) + while(char *row_chars=getrow(&data)) { + // remove empty&comment lines + if(!*row_chars || *row_chars == '#') + continue; do { String *name=new(pool) String(pool); name->APPEND_TAINTED(lsplit(&row_chars, '\t'), 0, file, line++); *columns+=name; } while(row_chars); + + break; + } } // parse cells Table& table=*new(pool) Table(pool, &method_name, columns); char *row_chars; while(row_chars=getrow(&data)) { - if(!*row_chars) // remove empty lines + // remove empty&comment lines + if(!*row_chars || *row_chars == '#') continue; Array *row=new(pool) Array(pool); while(char *cell_chars=lsplit(&row_chars, '\t')) { @@ -108,38 +193,40 @@ static void _load(Request& r, const Stri cell->APPEND_TAINTED(cell_chars, 0, file, line); *row+=cell; } +#ifndef NO_STRING_ORIGIN line++; +#endif table+=row; }; // replace any previous table value - static_cast(r.self)->set_table(table); + static_cast(r.get_self())->set_table(table); } -static void _save(Request& r, const String& method_name, Array *params) { +/// @todo "x\nx" "xxx""xx" +static void _save(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - Value *vtable_name=static_cast(params->get(params->size()-1)); - // forcing this body type] - r.fail_if_junction_(true, *vtable_name, - method_name, "file name must not be junction"); + Value& vfile_name=params->as_no_junction(params->size()-1, + "file name must not be code"); - Table& table=static_cast(r.self)->table(); + Table& table=static_cast(r.get_self())->table(&method_name); + bool do_append=false; String sdata(pool); - if(params->size()==1) { // not nameless=named output + if(params->size()==1) { // 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; columnas_string(0, "mode must be string"); + if(mode=="append") + do_append=true; + else if(mode=="nameless") + /*ok, already skipped names output*/; + else + throw Exception("parser.runtime", + &mode, + "unknown mode, must be 'append'"); + } // data lines - for(int index=0; index(table.quick_get(index)); - for(int column=0; columnsize(); column++) { - if(column) + Array_iter i(table); + while(i.has_next()) { + Array_iter c(*static_cast(i.next())); + while(c.has_next()) { + if(const String *s=c.next_string()) + sdata.append(*s, + String::UL_TABLE); + if(c.has_next()) sdata.APPEND_CONST("\t"); - sdata.append(*static_cast(row->quick_get(column)), - String::UL_TABLE); } sdata.APPEND_CONST("\n"); } // write - file_write(pool, r.absolute(vtable_name->as_string()), - sdata.cstr(), sdata.size(), true); + file_write(r.absolute(vfile_name.as_string()), + sdata.cstr(), sdata.size(), true, do_append); } -static void _count(Request& r, const String&method_name, Array *) { +static void _count(Request& r, const String& method_name, MethodParams *) { Pool& pool=r.pool(); - Value& value=*new(pool) VInt(pool, static_cast(r.self)->table().size()); - r.write_no_lang(value); + int result=static_cast(r.get_self())->table(&method_name).size(); + r.write_no_lang(*new(pool) VInt(pool, result)); } -static void _line(Request& r, const String& method_name, Array *) { +static void _line(Request& r, const String& method_name, MethodParams *) { Pool& pool=r.pool(); - Value& value=*new(pool) VInt(pool, 1+static_cast(r.self)->table().current()); - r.write_no_lang(value); + int result=1+static_cast(r.get_self())->table(&method_name).current(); + r.write_no_lang(*new(pool) VInt(pool, result)); } -static void _offset(Request& r, const String& method_name, Array *params) { +static void _offset(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - Table& table=static_cast(r.self)->table(); - if(params->size()) - table.shift((int)r.process(*static_cast(params->get(0))).as_double()); - else { - Value& value=*new(pool) VInt(pool, table.current()); - r.write_no_lang(value); - } + Table& table=static_cast(r.get_self())->table(&method_name); + if(params->size()) { + bool absolute=false; + if(params->size()>1) { + const String& whence=params->as_string(0, "whence must be string"); + if(whence=="cur") + absolute=false; + else if(whence=="set") + absolute=true; + else + throw Exception("parser.runtime", + &whence, + "is invalid whence, valid are 'cur' or 'set'"); + } + + Value& offset_expr=params->as_junction(params->size()-1, "offset must be expression"); + table.offset(absolute, r.process_to_value(offset_expr).as_int()); + } else + r.write_no_lang(*new(pool) VInt(pool, table.current())); } -static void _menu(Request& r, const String& method_name, Array *params) { - Value& body_code=*static_cast(params->get(0)); - // forcing ^menu{this param type} - r.fail_if_junction_(false, body_code, - method_name, "body must be junction"); +static void _menu(Request& r, const String& method_name, MethodParams *params) { + Value& body_code=params->as_junction(0, "body must be code"); - Value *delim_code=params->size()==2?static_cast(params->get(1)):0; + Value *delim_maybe_code=params->size()>1?¶ms->get(1):0; - Table& table=static_cast(r.self)->table(); + Table& table=static_cast(r.get_self())->table(&method_name); bool need_delim=false; int saved_current=table.current(); - for(int row=0; rowsize()) // need delim & iteration produced string? - r.write_pass_lang(r.process(*delim_code)); + StringOrValue sv_processed=r.process(body_code); + const String *s_processed=sv_processed.get_string(); + if(delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body + if(need_delim) // need delim & iteration produced string? + r.write_pass_lang(r.process(*delim_maybe_code)); need_delim=true; } - r.write_pass_lang(processed_body); + r.write_pass_lang(sv_processed); } table.set_current(saved_current); } -static void _empty(Request& r, const String& method_name, Array *params) { - Table& table=static_cast(r.self)->table(); - if(table.size()==0) { - Value& value=r.process(*static_cast(params->get(0))); - r.write_pass_lang(value); - } else if(params->size()==2) { - Value& value=r.process(*static_cast(params->get(1))); - r.write_pass_lang(value); - } -} - -struct Record_info { - Pool *pool; +#ifndef DOXYGEN +enum Table2hash_distint { D_ILLEGAL, D_FIRST, D_TABLES }; +struct Row_info { + Request *r; Table *table; + Value *key_code; + int key_field; + Array *value_fields; Hash *hash; + Table2hash_distint distinct; + int row; }; -static void store_column_item_to_hash(Array::Item *item, void *info) { - Record_info& ri=*static_cast(info); - String& column_name=*static_cast(item); - const String *column_item=ri.table->item(column_name); - Value *value; - if(column_item) - value=new(*ri.pool) VString(*column_item); - else - value=new(*ri.pool) VUnknown(*ri.pool); - ri.hash->put(column_name, value); -} -static void _record(Request& r, const String& method_name, Array *params) { - Table& table=static_cast(r.self)->table(); - if(const Array *columns=table.columns()) { - Pool& pool=r.pool(); - Value& value=*new(pool) VHash(pool); - Record_info record_info={&pool, &table, value.get_hash()}; - columns->for_each(store_column_item_to_hash, &record_info); +#endif +static void table_row_to_hash(Array::Item *value, void *info) { + Array& row=*static_cast(value); + Row_info& ri=*static_cast(info); + Pool& pool=ri.table->pool(); + + const String *key; + if(ri.key_code) { + ri.table->set_current(ri.row++); // change context row + StringOrValue sv_processed=ri.r->process(*ri.key_code); + key=&sv_processed.as_string(); + } else + key=ri.key_fieldsize(); i++) { + int value_field=ri.value_fields->get_int(i); + if(value_fieldcolumns()->get_string(value_field), + new(pool) VString(*row.get_string(value_field))); + } + + if(ri.hash->put_dont_replace(*key, &result)) // put. existed? + if(ri.distinct==D_ILLEGAL) + throw Exception("parser.runtime", + key, + "duplicate key"); + } + break; + case D_TABLES: + { + VTable* vtable=(VTable*)ri.hash->get(*key); // put. table existed? + Table* table; + if(vtable) + table=vtable->get_table(); + else { + // no? creating table of same structure as source + Table::Action_options table_options(0, 0); + table=new(pool) Table(pool, *ri.table, table_options/*no rows, just structure*/); + ri.hash->put(*key, new(pool) VTable(pool, table)); + } + *table+=&row; + } + break; + default: + throw Exception(0, + 0, + "invalid distinct code (#%d)", ri.distinct); } + } +static void _hash(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + Table& self_table=static_cast(r.get_self())->table(&method_name); + Value& result=*new(pool) VHash(pool); + if(const Array *columns=self_table.columns()) + if(columns->size()>0) { + Table2hash_distint distinct=D_ILLEGAL; + int param_index=params->size()-1; + if(param_index>0) { + if(Hash *options= + params->as_no_junction(param_index, "param must not be code").get_hash(0)) { + --param_index; + int valid_options=0; + if(Value *vdistinct_code=(Value *)options->get(*sql_distinct_name)) { + valid_options++; + Value& vdistinct_value=r.process_to_value(*vdistinct_code); + if(vdistinct_value.is_string()) { + const String& sdistinct=*vdistinct_value.get_string(); + if(sdistinct=="tables") + distinct=D_TABLES; + else + throw Exception("parser.runtime", + &sdistinct, + "must be 'tables' or true/false"); + } else + distinct=vdistinct_value.as_bool()?D_FIRST:D_ILLEGAL; + } + if(valid_options!=options->size()) + throw Exception("parser.runtime", + &method_name, + "called with invalid option"); + } + } + if(param_index==2) // bad options param type + throw Exception("parser.runtime", + &method_name, + "options must be hash"); + + Array value_fields(pool); + if(param_index>0) { + if(distinct!=D_ILLEGAL && distinct!=D_FIRST) + throw Exception("parser.runtime", + 0, + "in distinct[tables] mode you may not specify value field(s)"); + Value& value_fields_param=params->as_no_junction(param_index, "value field(s) must not be code"); + if(value_fields_param.is_string()) { + value_fields+=self_table.column_name2index(value_fields_param.as_string(), true); + } else if(Table *value_fields_table=value_fields_param.get_table()) { + for(int i=0; isize(); i++) { + const String& value_field_name= + *static_cast(value_fields_table->get(i))->get_string(0); + value_fields+=self_table.column_name2index(value_field_name, true); + } + } else + throw Exception("parser.runtime", + &method_name, + "value field(s) must be string or self_table" + ); + } else { // by all columns, including key + if(!(distinct!=D_ILLEGAL && distinct!=D_FIRST)) + for(int i=0; isize(); i++) + value_fields+=i; + } -struct Seq_item { + Value& key_param=params->get(0); + Value *key_code=key_param.get_junction()?&key_param:0; + int key_field=key_code?-1 + :self_table.column_name2index(key_param.as_string(), true); + + Row_info row_info={&r, &self_table, + key_code, key_field, &value_fields, + result.get_hash(0), distinct}; + + int saved_current=self_table.current(); + self_table.for_each(table_row_to_hash, &row_info); + self_table.set_current(saved_current); + } + r.write_no_lang(result); +} + +#ifndef DOXYGEN +struct Table_seq_item { Array *row; union { char *c_str; double d; } value; }; +#endif static int sort_cmp_string(const void *a, const void *b) { return strcmp( - static_cast(a)->value.c_str, - static_cast(b)->value.c_str + static_cast(a)->value.c_str, + static_cast(b)->value.c_str ); } static int sort_cmp_double(const void *a, const void *b) { - double va=static_cast(a)->value.d; - double vb=static_cast(b)->value.d; + double va=static_cast(a)->value.d; + double vb=static_cast(b)->value.d; if(vavb) @@ -276,36 +500,30 @@ static int sort_cmp_double(const void *a else return 0; } -static void _sort(Request& r, const String& method_name, Array *params) { - Value& key_maker=*(Value *)params->get(0); - // forcing ^sort{this} ^sort(or this) param type - r.fail_if_junction_(false, key_maker, method_name, "key-maker must be junction"); - - bool reverse; - if(params->size()==2) { // ..[asc|desc] - Value& order=*(Value *)params->get(1); - // forcing ..[this param-type] - r.fail_if_junction_(true, order, method_name, "order must not be junction"); - reverse=order.as_string()=="desc"; - } else - reverse=false; +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"); - Table& table=static_cast(r.self)->table(); + bool reverse=params->size()>1/*..[desc|asc|]*/? + reverse=params->as_no_junction(1, "order must not be code").as_string()=="desc": + false; // default=asc - // anything to sort? - if(!table.size()) + Table& old_table=static_cast(r.get_self())->table(&method_name); + if(old_table.size()==0) return; - Seq_item *seq=(Seq_item *)malloc(sizeof(Seq_item)*table.size()); + Table& new_table=*new(pool) Table(pool, &method_name, old_table.columns()); + + 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.get_self())->set_table(new_table); } -static void _locate(Request& r, const String& method_name, Array *params) { - VTable& vtable=*static_cast(r.self); - Table& table=vtable.table(); - vtable.last_locate_was_successful=table.locate( - static_cast(params->get(0))->as_string(), - static_cast(params->get(1))->as_string()); +#ifndef DOXYGEN +struct Locate_expression_func_info { + Request* r; + Value* expression_code; +}; +#endif +bool locate_expression_func(Table& self, void* ainfo) { + Locate_expression_func_info& info=*static_cast(ainfo); + return info.r->process_to_value(*info.expression_code).as_bool(); +} +static bool _locate_expression(Table& table, Table::Action_options o, + Request& r, const String& method_name, MethodParams *params) { + check_option_param(o.defined, method_name, params, 1, + "locate by expression only has parameters: expression and, maybe, options"); + Value& expression_code=params->as_junction(0, "must be expression"); + + Locate_expression_func_info info={&r, &expression_code}; + return table.locate(locate_expression_func, &info, o); +} +static bool _locate_name_value(Table& table, Table::Action_options o, + Request& r, const String& method_name, MethodParams *params) { + check_option_param(o.defined, method_name, params, 2, + "locate by locate by name has parameters: name, value and, maybe, options"); + const String& name=params->as_string(0, "column name must be string"); + const String& value=params->as_string(1, "value must be string"); + + return table.locate(name, value, o); } +static void _locate(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + Table& table=static_cast(r.get_self())->table(&method_name); -static void _found(Request& r, const String& method_name, Array *params) { - if(static_cast(r.self)->last_locate_was_successful) { - Value& then_code=*static_cast(params->get(0)); - // forcing ^found{this param type} - r.fail_if_junction_(false, then_code, - method_name, "found-parameter must be junction"); - r.write_pass_lang(r.process(then_code)); - } else if(params->size()==2) { - Value& else_code=*static_cast(params->get(1)); - // forcing ^found{this param type} - r.fail_if_junction_(false, else_code, - method_name, "not found-parameter must be junction"); - r.write_pass_lang(r.process(else_code)); - } + Table::Action_options o=get_action_options(r, method_name, params, table); + + bool result=params->get(0).get_junction()? + _locate_expression(table, o, r, method_name, params) : + _locate_name_value(table, o, r, method_name, params); + r.write_no_lang(*new(pool) VBool(pool, result)); } -static void _flip(Request& r, const String& method_name, Array *params) { +static void _flip(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - VTable& vtable=*static_cast(r.self); - - Table& old_table=*vtable.get_table(); + Table& old_table=static_cast(r.get_self())->table(&method_name); Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/); if(old_table.size()) if(int old_cols=old_table.at(0).size()) @@ -362,57 +594,65 @@ static void _flip(Request& r, const Stri Array& new_row=*new(pool) Array(pool, old_table.size()); for(int i=0; i(params->get(0)); - // forcing {this body type} - r.fail_if_junction_(false, *value, method_name, "body must be junction"); - - const String& string=r.process(*value).as_string(); + // data + Temp_lang temp_lang(r, String::UL_PASS_APPENDED); + const String& string= + r.process_to_string(params->as_junction(0, "body must be code")); // parse cells Array& row=*new(pool) Array(pool); - string.split(row, 0, "\t", 1, String::UL_CLEAN); + string.split(row, 0, "\t", 1, String::UL_AS_IS); - static_cast(r.self)->table()+=&row; + static_cast(r.get_self())->table(&method_name)+=&row; } -static void _join(Request& r, const String& method_name, Array *params) { +static void _join(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - Value *value=static_cast(params->get(0)); - // forcing [this table ref type] - r.fail_if_junction_(true, *value, method_name, "table ref must not be junction"); - - Table *maybe_src=value->get_table(); + Table* maybe_src=params->as_no_junction(0, "table ref must not be code").get_table(); if(!maybe_src) - PTHROW(0, 0, - &method_name, + throw Exception("parser.runtime", + &method_name, "source is not a table"); - Table& src=*maybe_src; - Table& dest=static_cast(r.self)->table(); + + Table::Action_options o=get_action_options(r, method_name, params, src); + check_option_param(o.defined, method_name, params, 1, + "invalid extra parameter"); + + Table& dest=static_cast(r.get_self())->table(&method_name); if(&src == &dest) - PTHROW(0, 0, - &method_name, + throw Exception("parser.runtime", + &method_name, "source and destination are same table"); + if(o.reverse) + throw Exception(0, + &method_name, + "not implemented in this version, please upgrade"); if(const Array *dest_columns=dest.columns()) { // dest is named int saved_src_current=src.current(); - for(int src_row=0; src_rowm) + o.limit=m; + int end=o.offset+o.limit; + for(int src_row=o.offset; src_rowsize(); dest_column++) - dest_row+=src.item(*dest_columns->get_string(dest_column)); + for(int dest_column=0; dest_columnsize(); dest_column++) { + const String *src_item=src.item(*dest_columns->get_string(dest_column)); + dest_row+=src_item?src_item:new(pool) String(pool); + } dest+=&dest_row; } src.set_current(saved_src_current); @@ -422,142 +662,261 @@ static void _join(Request& r, const Stri } } -/// ^table:sql{query}[(count[;offset])] @test limit/offset -static void _sql(Request& r, const String& method_name, Array *params) { - Pool& pool=r.pool(); +#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), + table(0) + { + } - if(!r.connection) - PTHROW(0, 0, - &method_name, - "without connect"); + bool add_column(SQL_Error& error, void *ptr, size_t size) { + try { + String *column=new(pool) String(pool); + column->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, 0); + columns+=column; + return false; + } catch(...) { + error=SQL_Error("exception occured in Table_sql_event_handlers::add_column"); + return true; + } + } + bool before_rows(SQL_Error& error) { + try { + table=new(pool) Table(pool, &method_name, &columns); + return false; + } catch(...) { + error=SQL_Error("exception occured in Table_sql_event_handlers::before_rows"); + return true; + } + } + bool add_row(SQL_Error& error) { + try { + (*table)+=(row=new(pool) Array(pool)); + return false; + } catch(...) { + error=SQL_Error("exception occured in Table_sql_event_handlers::add_row"); + return true; + } + } + bool add_row_cell(SQL_Error& error, void *ptr, size_t size) { + try { + String *cell=new(pool) String(pool); + if(size) + cell->APPEND_TAINTED( + (const char *)ptr, size, + statement_cstr, table->size()-1); + (*row)+=cell; + return false; + } catch(...) { + error=SQL_Error("exception occured in Table_sql_event_handlers::add_row_cell"); + return true; + } + } - Value& statement=*static_cast(params->get(0)); - // forcing {this query param type} - r.fail_if_junction_(true, statement, method_name, "statement must not be junction"); +private: + Pool& pool; + const String& method_name; + const String& statement_string; const char *statement_cstr; + Array& columns; + Array *row; +public: + Table *table; +}; +#endif +static void _sql(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); - ulong count=0; - if(params->size()>1) { - Value& count_code=*static_cast(params->get(1)); - // forcing (this count param type) - r.fail_if_junction_(false, count_code, method_name, "count must be expression"); - count=(uint)r.process(count_code).as_double(); - } + Value& statement=params->as_junction(0, "statement must be code"); + ulong limit=0; ulong offset=0; - if(params->size()>2) { - Value& offset_code=*static_cast(params->get(2)); - // forcing (this limit param type) - r.fail_if_junction_(false, offset_code, method_name, "offset must be expression"); - offset=(ulong)r.process(offset_code).as_double(); - } - - const String& statement_string=statement.as_string(); - const char *statement_cstr=statement_string.cstr(String::UL_SQL); - 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, - &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, - rethrow_me.comment()); + if(params->size()>1) { + Value& voptions=params->as_no_junction(1, "options must be hash, not code"); + if(!voptions.is_string()) + if(Hash *options=voptions.get_hash(&method_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(); + } + if(Value *voffset=(Value *)options->get(*sql_offset_name)) { + valid_options++; + offset=(ulong)r.process_to_value(*voffset).as_double(); + } + if(valid_options!=options->size()) + throw Exception("parser.runtime", + &method_name, + "called with invalid option"); + } else + throw Exception("parser.runtime", + &method_name, + "options must be hash"); + } + + Temp_lang temp_lang(r, String::UL_SQL); + const String& statement_string=r.process_to_string(statement); + const char *statement_cstr= + statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); + Table_sql_event_handlers handlers(pool, method_name, + statement_string, statement_cstr); +#ifdef RESOURCES_DEBUG + struct timeval mt[2]; + //measure:before + gettimeofday(&mt[0],NULL); +#endif + r.connection(&method_name)->query( + statement_cstr, offset, limit, + handlers, + statement_string); - Array& table_columns=*new(pool) Array(pool); - for(unsigned int i=0; i(r.get_self())->set_table(*result); +} - Table& table=*new(pool) Table(pool, &method_name, &table_columns); +static void _columns(Request& r, const String& method_name, MethodParams *) { + Pool& pool=r.pool(); - { - for(unsigned long r=0; r(r.get_self())->table(&method_name); + if(const Array *source_columns=source_table.columns()) { + Array_iter i(*source_columns); + while(i.has_next()) { + Array& result_row=*new(pool) Array(pool); + result_row+=i.next(); + result_table+=&result_row; } } - // replace any previous table value - static_cast(r.self)->set_table(table); + r.write_no_lang(*new(pool) VTable(pool, &result_table)); +} + +static void _select(Request& r, const String& method_name, MethodParams *params) { + Pool& pool=r.pool(); + + Value& vcondition=params->as_junction(0, "condition must be expression"); + + Table& source_table=static_cast(r.get_self())->table(&method_name); + Table& result_table=*new(pool) Table(pool, + source_table.origin_string(), + source_table.columns() + ); + + int saved_current=source_table.current(); + int size=source_table.size(); + for(int row=0; row