--- parser3/src/classes/table.C 2001/04/03 08:23:06 1.47 +++ parser3/src/classes/table.C 2001/04/06 08:55:18 1.57 @@ -5,7 +5,7 @@ Author: Alexander Petrosyan (http://design.ru/paf) - $Id: table.C,v 1.47 2001/04/03 08:23:06 paf Exp $ + $Id: table.C,v 1.57 2001/04/06 08:55:18 paf Exp $ */ #include "pa_config_includes.h" @@ -14,6 +14,7 @@ #include "_table.h" #include "pa_vtable.h" #include "pa_vint.h" +#include "pa_sql_connection.h" // global var @@ -25,7 +26,7 @@ static void _set(Request& r, const Strin // 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"); + r.fail_if_junction_(false, *vdata, method_name, "body must be code"); Temp_lang temp_lang(r, String::UL_PASS_APPENDED); const String& data=r.process(*vdata).as_string(); @@ -70,13 +71,10 @@ static void _load(Request& r, const Stri 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"); + method_name, "file name must not be code"); - // forcing untaint language - String lfilename(pool); - lfilename.append(vfilename->as_string(), String::UL_FILE_NAME, true); // loading text - char *data=file_read_text(pool, r.absolute(lfilename)); + char *data=file_read_text(pool, r.absolute(vfilename->as_string())); // parse columns Array *columns; @@ -123,12 +121,7 @@ static void _save(Request& r, const Stri 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"); - - // forcing untaint language - String ltable_name(pool); - ltable_name.append(vtable_name->as_string(), - String::UL_FILE_NAME, true); + method_name, "file name must not be code"); Table& table=static_cast(r.self)->table(); @@ -168,7 +161,8 @@ static void _save(Request& r, const Stri } // write - file_write(pool, r.absolute(ltable_name), sdata.cstr(), sdata.size(), true); + file_write(pool, r.absolute(vtable_name->as_string()), + sdata.cstr(), sdata.size(), true); } static void _count(Request& r, const String&method_name, Array *) { @@ -194,11 +188,12 @@ static void _offset(Request& r, const St } } +/// @test $a.menu{ $a[123] } 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"); + method_name, "body must be code"); Value *delim_code=params->size()==2?static_cast(params->get(1)):0; @@ -285,13 +280,13 @@ static int sort_cmp_double(const void *a 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"); + r.fail_if_junction_(false, key_maker, method_name, "key-maker must be code"); 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"); + r.fail_if_junction_(true, order, method_name, "order must not be code"); reverse=order.as_string()=="desc"; } else reverse=false; @@ -345,13 +340,13 @@ static void _found(Request& r, const Str 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"); + method_name, "found-parameter must be code"); 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"); + method_name, "not found-parameter must be code"); r.write_pass_lang(r.process(else_code)); } } @@ -381,7 +376,7 @@ static void _append(Request& r, const St // data is last parameter Value *value=static_cast(params->get(0)); // forcing {this body type} - r.fail_if_junction_(false, *value, method_name, "body must be junction"); + r.fail_if_junction_(false, *value, method_name, "body must be code"); const String& string=r.process(*value).as_string(); @@ -397,18 +392,18 @@ static void _join(Request& r, const Stri 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"); + r.fail_if_junction_(true, *value, method_name, "table ref must not be code"); Table *maybe_src=value->get_table(); if(!maybe_src) - RTHROW(0, 0, + PTHROW(0, 0, &method_name, "source is not a table"); Table& src=*maybe_src; Table& dest=static_cast(r.self)->table(); if(&src == &dest) - RTHROW(0, 0, + PTHROW(0, 0, &method_name, "source and destination are same table"); @@ -428,15 +423,97 @@ static void _join(Request& r, const Stri } } +/// ^table:sql{query}[(count[;offset])] +static void _sql(Request& r, const String& method_name, Array *params) { + Pool& pool=r.pool(); + + if(!r.connection) + PTHROW(0, 0, + &method_name, + "without connect"); + + Value& statement=*static_cast(params->get(0)); + // forcing {this query param type} + r.fail_if_junction_(false, statement, method_name, "statement must be code"); + + ulong limit=0; + if(params->size()>1) { + Value& limit_code=*static_cast(params->get(1)); + // forcing (this limit param type) + r.fail_if_junction_(false, limit_code, method_name, "limit must be expression"); + limit=(uint)r.process(limit_code).as_double(); + } + + 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(); + } + + 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; + 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()); + + Array& table_columns=*new(pool) Array(pool); + for(unsigned int i=0; i(r.self)->set_table(table); +} + // initialize void initialize_table_class(Pool& pool, VStateless_class& vclass) { - // ^table.set{data} - // ^table.set[nameless]{data} + // ^table:set{data} + // ^table:set[nameless]{data} vclass.add_native_method("set", Method::CT_DYNAMIC, _set, 1, 2); - // ^table.load[file] - // ^table.load[nameless;file] + // ^table:load[file] + // ^table:load[nameless;file] vclass.add_native_method("load", Method::CT_DYNAMIC, _load, 1, 2); // ^table.save[file] @@ -482,4 +559,8 @@ void initialize_table_class(Pool& pool, // ^table.join[table] vclass.add_native_method("join", Method::CT_DYNAMIC, _join, 1, 1); + + + // ^table:sql[query][(count[;offset])] + vclass.add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3); }