|
|
| version 1.151.2.1, 2002/05/17 12:18:12 | version 1.163, 2002/09/17 10:58:24 |
|---|---|
| Line 3 | Line 3 |
| Copyright (c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001, 2002 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) |
| $Id$ | |
| */ | */ |
| static const char* IDENT_TABLE_C="$Date$"; | |
| #include "classes.h" | #include "classes.h" |
| #include "pa_common.h" | #include "pa_common.h" |
| #include "pa_request.h" | #include "pa_request.h" |
| Line 30 public: // Methoded | Line 30 public: // Methoded |
| // methods | // methods |
| static void get_copy_options(Request& r, const String& method_name, MethodParams *params, int param_index, | |
| const Table& source, | |
| int& offset, | |
| int& limit) { | |
| offset=0; | |
| limit=0; | |
| if(params->size()<=param_index) | |
| return; | |
| Value& voptions=params->as_no_junction(param_index, "options must be hash, not code"); | |
| if(!voptions.is_string()) { | |
| if(Hash *options=voptions.get_hash(&method_name)) { | |
| if(Value *voffset=(Value *)options->get(*sql_offset_name)) | |
| if(voffset->is_string()) { | |
| const String& soffset=*voffset->get_string(); | |
| if(soffset == "cur") | |
| offset=source.current(); | |
| else | |
| throw Exception("parser.runtime", | |
| &soffset, | |
| "must be 'cur' string or expression"); | |
| } else | |
| offset=r.process_to_value(*voffset).as_int(); | |
| if(Value *vlimit=(Value *)options->get(*sql_limit_name)) | |
| limit=r.process_to_value(*vlimit).as_int(); | |
| } else | |
| throw Exception("parser.runtime", | |
| &method_name, | |
| "options must be hash"); | |
| } | |
| if(!limit) // zero limit = sould be 'nothing to copy', for methods zero means 'all' | |
| limit=-1; // thus fixing | |
| } | |
| static void _create(Request& r, const String& method_name, MethodParams *params) { | static void _create(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| // clone? | // clone/copy part? |
| if(params->size()==1) | if(const Table *source=params->get(0).get_table()) { |
| if(const Table *source=params->get(0).get_table()) { | int offset, limit; |
| static_cast<VTable *>(r.self)->set_table(*new(pool) Table(*source)); | get_copy_options(r, method_name, params, 1, *source, |
| return; | offset, limit); |
| } | static_cast<VTable *>(r.self)->set_table(*new(pool) Table(pool, *source, offset, limit)); |
| return; | |
| } | |
| // data is last parameter | // data is last parameter |
| Temp_lang temp_lang(r, String::UL_PASS_APPENDED); | Temp_lang temp_lang(r, String::UL_PASS_APPENDED); |
| Line 142 static void _save(Request& r, const Stri | Line 178 static void _save(Request& r, const Stri |
| Value& vfile_name=params->as_no_junction(params->size()-1, | Value& vfile_name=params->as_no_junction(params->size()-1, |
| "file name must not be code"); | "file name must not be code"); |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(&method_name); |
| bool do_append=false; | bool do_append=false; |
| String sdata(pool); | String sdata(pool); |
| Line 200 static void _save(Request& r, const Stri | Line 236 static void _save(Request& r, const Stri |
| static void _count(Request& r, const String& method_name, MethodParams *) { | static void _count(Request& r, const String& method_name, MethodParams *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| int result=static_cast<VTable *>(r.self)->table().size(); | int result=static_cast<VTable *>(r.self)->table(&method_name).size(); |
| r.write_no_lang(*new(pool) VInt(pool, result)); | r.write_no_lang(*new(pool) VInt(pool, result)); |
| } | } |
| static void _line(Request& r, const String& method_name, MethodParams *) { | static void _line(Request& r, const String& method_name, MethodParams *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| int result=1+static_cast<VTable *>(r.self)->table().current(); | int result=1+static_cast<VTable *>(r.self)->table(&method_name).current(); |
| r.write_no_lang(*new(pool) VInt(pool, result)); | r.write_no_lang(*new(pool) VInt(pool, result)); |
| } | } |
| static void _offset(Request& r, const String& method_name, MethodParams *params) { | static void _offset(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(&method_name); |
| if(params->size()) { | if(params->size()) { |
| bool absolute=false; | bool absolute=false; |
| if(params->size()>1) { | if(params->size()>1) { |
| Line 225 static void _offset(Request& r, const St | Line 261 static void _offset(Request& r, const St |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| &whence, | &whence, |
| "is invalid whence, valid are 'cur' or 'set'"); | "is invalid whence, valid are 'cur' or 'set'"); |
| } | } |
| Value& offset_expr=params->as_junction(params->size()-1, "offset must be expression"); | Value& offset_expr=params->as_junction(params->size()-1, "offset must be expression"); |
| table.offset(absolute, r.process_to_value(offset_expr).as_int()); | table.offset(absolute, r.process_to_value(offset_expr).as_int()); |
| Line 238 static void _menu(Request& r, const Stri | Line 274 static void _menu(Request& r, const Stri |
| Value *delim_maybe_code=params->size()>1?¶ms->get(1):0; | Value *delim_maybe_code=params->size()>1?¶ms->get(1):0; |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(&method_name); |
| bool need_delim=false; | bool need_delim=false; |
| int saved_current=table.current(); | int saved_current=table.current(); |
| int size=table.size(); | int size=table.size(); |
| for(int row=0; row<size; row++) { | for(int row=0; row<size; row++) { |
| table.set_current(row); | table.set_current(row); |
| StringOrValue processed_body=r.process(body_code); | StringOrValue sv_processed=r.process(body_code); |
| if(delim_maybe_code) { // delimiter set? | const String *s_processed=sv_processed.get_string(); |
| const String *string=processed_body.get_string(); | if(delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body |
| if(need_delim && string && string->size()) // need delim & iteration produced string? | if(need_delim) // need delim & iteration produced string? |
| r.write_pass_lang(r.process(*delim_maybe_code)); | r.write_pass_lang(r.process(*delim_maybe_code)); |
| need_delim=true; | need_delim=true; |
| } | } |
| r.write_pass_lang(processed_body); | r.write_pass_lang(sv_processed); |
| } | } |
| table.set_current(saved_current); | table.set_current(saved_current); |
| } | } |
| Line 263 struct Row_info { | Line 299 struct Row_info { |
| int key_field; | int key_field; |
| Array *value_fields; | Array *value_fields; |
| Hash *hash; | Hash *hash; |
| bool distinct; | |
| }; | }; |
| #endif | #endif |
| static void table_row_to_hash(Array::Item *value, void *info) { | static void table_row_to_hash(Array::Item *value, void *info) { |
| Line 281 static void table_row_to_hash(Array::Ite | Line 318 static void table_row_to_hash(Array::Ite |
| new(pool) VString(*row.get_string(value_field))); | new(pool) VString(*row.get_string(value_field))); |
| } | } |
| ri.hash->put(*row.get_string(ri.key_field), &result); | const String& key=*row.get_string(ri.key_field); |
| if(ri.hash->put_dont_replace(key, &result)) // put. existed? | |
| if(!ri.distinct) | |
| throw Exception("parser.runtime", | |
| &key, | |
| "duplicate key"); | |
| } | } |
| } | } |
| static void _hash(Request& r, const String& method_name, MethodParams *params) { | static void _hash(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Table& self_table=static_cast<VTable *>(r.self)->table(); | Table& self_table=static_cast<VTable *>(r.self)->table(&method_name); |
| Value& result=*new(pool) VHash(pool); | Value& result=*new(pool) VHash(pool); |
| if(const Array *columns=self_table.columns()) | if(const Array *columns=self_table.columns()) |
| if(columns->size()>1) { | if(columns->size()>0) { |
| const String& key_field_name=params->as_no_junction(0, | const String& key_field_name=params->as_no_junction(0, |
| "key field name must not be code").as_string(); | "key field name must not be code").as_string(); |
| int key_field=self_table.column_name2index(key_field_name, true); | int key_field=self_table.column_name2index(key_field_name, true); |
| bool distinct=false; | |
| int param_index=params->size()-1; | |
| 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=(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"); | |
| } | |
| if(param_index==2) // bad options param type | |
| throw Exception("parser.runtime", | |
| &method_name, | |
| "options must be hash"); | |
| Array value_fields(pool); | Array value_fields(pool); |
| if(params->size()>1) { | if(param_index>0) { |
| Value& value_fields_param=params->as_no_junction(1, "value field(s) must not be code"); | Value& value_fields_param=params->as_no_junction(param_index, "value field(s) must not be code"); |
| if(value_fields_param.is_string()) { | if(value_fields_param.is_string()) { |
| value_fields+=self_table.column_name2index(value_fields_param.as_string(), true); | value_fields+=self_table.column_name2index(value_fields_param.as_string(), true); |
| } else if(Table *value_fields_table=value_fields_param.get_table()) { | } else if(Table *value_fields_table=value_fields_param.get_table()) { |
| Line 316 static void _hash(Request& r, const Stri | Line 378 static void _hash(Request& r, const Stri |
| } | } |
| // integers: key_field & value_fields | // integers: key_field & value_fields |
| Row_info row_info={&self_table, key_field, &value_fields, result.get_hash(0)}; | Row_info row_info={&self_table, key_field, &value_fields, result.get_hash(0), distinct}; |
| self_table.for_each(table_row_to_hash, &row_info); | self_table.for_each(table_row_to_hash, &row_info); |
| } | } |
| r.write_no_lang(result); | r.write_no_lang(result); |
| Line 355 static void _sort(Request& r, const Stri | Line 417 static void _sort(Request& r, const Stri |
| reverse=params->as_no_junction(1, "order must not be code").as_string()=="desc": | reverse=params->as_no_junction(1, "order must not be code").as_string()=="desc": |
| false; // default=asc | false; // default=asc |
| Table& old_table=static_cast<VTable *>(r.self)->table(); | Table& old_table=static_cast<VTable *>(r.self)->table(&method_name); |
| Table& new_table=*new(pool) Table(pool, &method_name, old_table.columns()); | 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()); | Table_seq_item *seq=(Table_seq_item *)pool.malloc(sizeof(Table_seq_item)*old_table.size()); |
| Line 400 static bool _locate_expression(Request& | Line 462 static bool _locate_expression(Request& |
| Value& expression_code=params->as_junction(0, "must be expression"); | Value& expression_code=params->as_junction(0, "must be expression"); |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(&method_name); |
| int saved_current=table.current(); | int saved_current=table.current(); |
| int size=table.size(); | int size=table.size(); |
| for(int row=0; row<size; row++) { | for(int row=0; row<size; row++) { |
| Line 418 static bool _locate_name_value(Request& | Line 480 static bool _locate_name_value(Request& |
| &method_name, | &method_name, |
| "locate by name and value has only two parameters - name and value"); | "locate by name and value has only two parameters - name and value"); |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(&method_name); |
| return table.locate( | return table.locate( |
| params->as_string(0, "column name must be string"), | params->as_string(0, "column name must be string"), |
| params->as_string(1, "value must be string") | params->as_string(1, "value must be string") |
| Line 434 static void _locate(Request& r, const St | Line 496 static void _locate(Request& r, const St |
| static void _flip(Request& r, const String& method_name, MethodParams *params) { | static void _flip(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Table& old_table=static_cast<VTable *>(r.self)->table(); | Table& old_table=static_cast<VTable *>(r.self)->table(&method_name); |
| Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/); | Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/); |
| if(old_table.size()) | if(old_table.size()) |
| if(int old_cols=old_table.at(0).size()) | if(int old_cols=old_table.at(0).size()) |
| Line 461 static void _append(Request& r, const St | Line 523 static void _append(Request& r, const St |
| Array& row=*new(pool) Array(pool); | Array& row=*new(pool) Array(pool); |
| string.split(row, 0, "\t", 1, String::UL_AS_IS); | string.split(row, 0, "\t", 1, String::UL_AS_IS); |
| static_cast<VTable *>(r.self)->table()+=&row; | static_cast<VTable *>(r.self)->table(&method_name)+=&row; |
| } | } |
| static void _join(Request& r, const String& method_name, MethodParams *params) { | static void _join(Request& r, const String& method_name, MethodParams *params) { |
| Line 474 static void _join(Request& r, const Stri | Line 536 static void _join(Request& r, const Stri |
| "source is not a table"); | "source is not a table"); |
| Table& src=*maybe_src; | Table& src=*maybe_src; |
| Table& dest=static_cast<VTable *>(r.self)->table(); | Table& dest=static_cast<VTable *>(r.self)->table(&method_name); |
| if(&src == &dest) | if(&src == &dest) |
| throw Exception("parser.runtime", | throw Exception("parser.runtime", |
| &method_name, | &method_name, |
| "source and destination are same table"); | "source and destination are same table"); |
| int offset, limit; | |
| get_copy_options(r, method_name, params, 1, src, | |
| offset, limit); | |
| if(const Array *dest_columns=dest.columns()) { // dest is named | if(const Array *dest_columns=dest.columns()) { // dest is named |
| int saved_src_current=src.current(); | int saved_src_current=src.current(); |
| for(int src_row=0; src_row<src.size(); src_row++) { | int m=src.size()-offset; |
| if(!limit || limit>m) | |
| limit=m; | |
| int end=offset+limit; | |
| for(int src_row=offset; src_row<end; src_row++) { | |
| src.set_current(src_row); | src.set_current(src_row); |
| Array& dest_row=*new(pool) Array(pool); | Array& dest_row=*new(pool) Array(pool); |
| for(int dest_column=0; dest_column<dest_columns->size(); dest_column++) { | for(int dest_column=0; dest_column<dest_columns->size(); dest_column++) { |
| Line 554 static void _sql(Request& r, const Strin | Line 624 static void _sql(Request& r, const Strin |
| ulong offset=0; | ulong offset=0; |
| 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_defined()) | 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)) | if(Value *vlimit=(Value *)options->get(*sql_limit_name)) |
| limit=(ulong)r.process_to_value(*vlimit).as_double(); | limit=(ulong)r.process_to_value(*vlimit).as_double(); |
| Line 572 static void _sql(Request& r, const Strin | Line 642 static void _sql(Request& r, const Strin |
| statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); | statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name)); |
| Table_sql_event_handlers handlers(pool, method_name, | Table_sql_event_handlers handlers(pool, method_name, |
| statement_string, statement_cstr); | statement_string, statement_cstr); |
| try { | |
| #ifdef RESOURCES_DEBUG | #ifdef RESOURCES_DEBUG |
| struct timeval mt[2]; | struct timeval mt[2]; |
| //measure:before | //measure:before |
| gettimeofday(&mt[0],NULL); | gettimeofday(&mt[0],NULL); |
| #endif | #endif |
| r.connection(&method_name)->query( | r.connection(&method_name)->query( |
| statement_cstr, offset, limit, | statement_cstr, offset, limit, |
| handlers); | handlers, |
| statement_string); | |
| #ifdef RESOURCES_DEBUG | #ifdef RESOURCES_DEBUG |
| //measure:after connect | //measure:after connect |
| Line 592 static void _sql(Request& r, const Strin | Line 662 static void _sql(Request& r, const Strin |
| r.sql_request_time+=t[1]-t[0]; | r.sql_request_time+=t[1]-t[0]; |
| #endif | #endif |
| } catch(const Exception& e) { // query problem | |
| // more specific source [were url] | |
| throw Exception("sql.execute", | |
| &statement_string, | |
| "%s", e.comment()); | |
| } | |
| Table *result= | Table *result= |
| handlers.table?handlers.table: // query resulted in table? return it | handlers.table?handlers.table: // query resulted in table? return it |
| Line 614 static void _columns(Request& r, const S | Line 678 static void _columns(Request& r, const S |
| result_columns+=new(pool) String(pool, "column"); | result_columns+=new(pool) String(pool, "column"); |
| Table& result_table=*new(pool) Table(pool, &method_name, &result_columns); | Table& result_table=*new(pool) Table(pool, &method_name, &result_columns); |
| Table& source_table=static_cast<VTable *>(r.self)->table(); | Table& source_table=static_cast<VTable *>(r.self)->table(&method_name); |
| if(const Array *source_columns=source_table.columns()) { | if(const Array *source_columns=source_table.columns()) { |
| Array_iter i(*source_columns); | Array_iter i(*source_columns); |
| while(i.has_next()) { | while(i.has_next()) { |
| Line 632 static void _select(Request& r, const St | Line 696 static void _select(Request& r, const St |
| Value& vcondition=params->as_junction(0, "condition must be expression"); | Value& vcondition=params->as_junction(0, "condition must be expression"); |
| Table& source_table=static_cast<VTable *>(r.self)->table(); | Table& source_table=static_cast<VTable *>(r.self)->table(&method_name); |
| Table& result_table=*new(pool) Table(pool, | Table& result_table=*new(pool) Table(pool, |
| source_table.origin_string(), | source_table.origin_string(), |
| source_table.columns() | source_table.columns() |
| Line 690 MTable::MTable(Pool& apool) : Methoded(a | Line 754 MTable::MTable(Pool& apool) : Methoded(a |
| // ^table:hash[key field name] | // ^table:hash[key field name] |
| // ^table:hash[key field name][value field name(s) string/table] | // ^table:hash[key field name][value field name(s) string/table] |
| add_native_method("hash", Method::CT_DYNAMIC, _hash, 1, 2); | add_native_method("hash", Method::CT_DYNAMIC, _hash, 1, 3); |
| // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[desc|asc] | // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[desc|asc] |
| // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[desc|asc] | // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[desc|asc] |
| Line 705 MTable::MTable(Pool& apool) : Methoded(a | Line 769 MTable::MTable(Pool& apool) : Methoded(a |
| // ^table.append{r{tab}e{tab}c{tab}o{tab}r{tab}d} | // ^table.append{r{tab}e{tab}c{tab}o{tab}r{tab}d} |
| add_native_method("append", Method::CT_DYNAMIC, _append, 1, 1); | add_native_method("append", Method::CT_DYNAMIC, _append, 1, 1); |
| // ^table.join[table] | // ^table.join[table][$.limit(10) $.offset(1) $.offset[cur] ] |
| add_native_method("join", Method::CT_DYNAMIC, _join, 1, 1); | add_native_method("join", Method::CT_DYNAMIC, _join, 1, 2); |
| // ^table:sql[query] | // ^table:sql[query] |