Annotation of parser3/src/classes/table.C, revision 1.172.2.4
1.20 paf 1: /** @file
1.47 paf 2: Parser: @b table parser class.
1.20 paf 3:
1.172.2.2 paf 4: Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com)
1.144 paf 5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.154 paf 6: */
1.20 paf 7:
1.172.2.4! paf 8: static const char* IDENT_TABLE_C="$Date: 2003/02/04 14:04:46 $";
1.1 paf 9:
1.105 parser 10: #include "classes.h"
1.172.2.4! paf 11: #include "pa_vmethod_frame.h"
! 12:
1.16 paf 13: #include "pa_common.h"
1.1 paf 14: #include "pa_request.h"
15: #include "pa_vtable.h"
1.6 paf 16: #include "pa_vint.h"
1.51 paf 17: #include "pa_sql_connection.h"
1.72 paf 18: #include "pa_vbool.h"
1.1 paf 19:
1.66 paf 20: // class
21:
22: class MTable : public Methoded {
23: public: // VStateless_class
1.172.2.1 paf 24: ValuePtr create_new_value() { return ValuePtr(new VTable()); }
1.66 paf 25:
26: public:
27: MTable(Pool& pool);
1.70 paf 28:
29: public: // Methoded
1.66 paf 30: bool used_directly() { return true; }
31: };
1.1 paf 32:
33: // methods
1.66 paf 34:
1.172.2.3 paf 35: static void get_copy_options(Request& r, StringPtr method_name, MethodParams& params, int param_index,
1.157 paf 36: const Table& source,
37: int& offset,
38: int& limit) {
39: offset=0;
40: limit=0;
1.172.2.3 paf 41: if(params.count()<=param_index)
1.157 paf 42: return;
43:
1.172.2.3 paf 44: Value& voptions=params.as_no_junction(param_index, "options must be hash, not code");
1.158 paf 45: if(!voptions.is_string()) {
1.157 paf 46: if(Hash *options=voptions.get_hash(&method_name)) {
1.164 paf 47: int valid_options=0;
48: if(Value *voffset=(Value *)options->get(*sql_offset_name)) {
49: valid_options++;
1.157 paf 50: if(voffset->is_string()) {
51: const String& soffset=*voffset->get_string();
52: if(soffset == "cur")
53: offset=source.current();
54: else
55: throw Exception("parser.runtime",
56: &soffset,
57: "must be 'cur' string or expression");
58: } else
59: offset=r.process_to_value(*voffset).as_int();
1.164 paf 60: }
61: if(Value *vlimit=(Value *)options->get(*sql_limit_name)) {
62: valid_options++;
1.157 paf 63: limit=r.process_to_value(*vlimit).as_int();
1.164 paf 64: }
65: if(valid_options!=options->size())
66: throw Exception("parser.runtime",
67: &method_name,
68: "called with invalid option");
1.157 paf 69: } else
70: throw Exception("parser.runtime",
71: &method_name,
72: "options must be hash");
1.158 paf 73: }
74: if(!limit) // zero limit = sould be 'nothing to copy', for methods zero means 'all'
75: limit=-1; // thus fixing
1.157 paf 76: }
77:
1.172.2.3 paf 78: static void _create(Request& r, StringPtr method_name, MethodParams& params) {
1.1 paf 79: Pool& pool=r.pool();
1.157 paf 80: // clone/copy part?
1.172.2.3 paf 81: if(const Table *source=params.get(0).get_table()) {
1.157 paf 82: int offset, limit;
83: get_copy_options(r, method_name, params, 1, *source,
84: offset, limit);
1.165 paf 85: static_cast<VTable *>(r.get_self())->set_table(*new(pool) Table(pool, *source, offset, limit));
1.157 paf 86: return;
87: }
1.142 paf 88:
1.1 paf 89: // data is last parameter
1.45 paf 90: Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
1.134 paf 91: const String& data=
1.172.2.3 paf 92: r.process_to_string(params.as_junction(params.count()-1, "body must be code"));
1.44 paf 93:
94: size_t pos_after=0;
95: // parse columns
96: Array *columns;
1.172.2.3 paf 97: if(params.count()==2) {
1.44 paf 98: columns=0;
1.21 paf 99: } else {
1.44 paf 100: columns=new(pool) Array(pool);
101:
102: Array head(pool);
1.137 paf 103: data.split(head, &pos_after, "\n", 1, String::UL_AS_IS, 1);
1.44 paf 104: if(head.size())
1.137 paf 105: head.get_string(0)->split(*columns, 0, "\t", 1, String::UL_AS_IS);
1.44 paf 106: }
107:
108: Table& table=*new(pool) Table(pool, &method_name, columns);
109: // parse cells
110: Array rows(pool);
1.137 paf 111: data.split(rows, &pos_after, "\n", 1, String::UL_AS_IS);
1.98 parser 112: Array_iter i(rows);
113: while(i.has_next()) {
1.44 paf 114: Array& row=*new(pool) Array(pool);
1.98 parser 115: const String& string=*i.next_string();
1.118 parser 116: // remove comment lines
117: if(!string.size())
1.44 paf 118: continue;
119:
1.137 paf 120: string.split(row, 0, "\t", 1, String::UL_AS_IS);
1.44 paf 121: table+=&row;
1.17 paf 122: }
1.1 paf 123:
1.44 paf 124: // replace any previous table value
1.165 paf 125: static_cast<VTable *>(r.get_self())->set_table(table);
1.44 paf 126: }
127:
1.172.2.3 paf 128: static void _load(Request& r, StringPtr method_name, MethodParams& params) {
1.44 paf 129: Pool& pool=r.pool();
1.172.2.3 paf 130: const String& first_param=params.as_string(0, "file name must be string");
1.168 paf 131: int filename_param_index=0;
132: bool nameless=first_param=="nameless";
133: if(nameless)
134: filename_param_index++;
135: int options_param_index=filename_param_index+1;
136:
1.44 paf 137: // loading text
1.168 paf 138: char *data=file_read_text(pool,
1.172.2.3 paf 139: r.absolute(params.as_string(filename_param_index, "file name must be string")),
1.168 paf 140: true,
1.172.2.3 paf 141: options_param_index<params.count()?params.as_no_junction(options_param_index, "additional params must be hash").get_hash(&method_name):0
1.168 paf 142: );
1.44 paf 143:
1.1 paf 144: // parse columns
145: Array *columns;
146: #ifndef NO_STRING_ORIGIN
147: const Origin& origin=method_name.origin();
1.172.2.2 paf 148: const char* file=origin.file;
1.1 paf 149: uint line=origin.line;
150: #endif
1.168 paf 151: if(nameless) {
1.139 paf 152: columns=0; // nameless
1.1 paf 153: } else {
154: columns=new(pool) Array(pool);
155:
1.139 paf 156: while(char *row_chars=getrow(&data)) {
157: // remove empty&comment lines
158: if(!*row_chars || *row_chars == '#')
159: continue;
1.1 paf 160: do {
161: String *name=new(pool) String(pool);
1.45 paf 162: name->APPEND_TAINTED(lsplit(&row_chars, '\t'), 0, file, line++);
1.1 paf 163: *columns+=name;
164: } while(row_chars);
1.139 paf 165:
166: break;
167: }
1.1 paf 168: }
169:
170: // parse cells
1.27 paf 171: Table& table=*new(pool) Table(pool, &method_name, columns);
1.1 paf 172: char *row_chars;
173: while(row_chars=getrow(&data)) {
1.118 parser 174: // remove empty&comment lines
175: if(!*row_chars || *row_chars == '#')
1.28 paf 176: continue;
1.1 paf 177: Array *row=new(pool) Array(pool);
178: while(char *cell_chars=lsplit(&row_chars, '\t')) {
179: String *cell=new(pool) String(pool);
1.44 paf 180: cell->APPEND_TAINTED(cell_chars, 0, file, line);
1.1 paf 181: *row+=cell;
182: }
1.107 parser 183: #ifndef NO_STRING_ORIGIN
1.1 paf 184: line++;
1.107 parser 185: #endif
1.1 paf 186: table+=row;
187: };
188:
189: // replace any previous table value
1.165 paf 190: static_cast<VTable *>(r.get_self())->set_table(table);
1.1 paf 191: }
1.2 paf 192:
1.146 paf 193: /// @todo "x\nx" "xxx""xx"
1.172.2.3 paf 194: static void _save(Request& r, StringPtr method_name, MethodParams& params) {
1.22 paf 195: Pool& pool=r.pool();
1.172.2.3 paf 196: Value& vfile_name=params.as_no_junction(params.count()-1,
1.61 paf 197: "file name must not be code");
1.22 paf 198:
1.165 paf 199: Table& table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.31 paf 200:
1.129 paf 201: bool do_append=false;
1.31 paf 202: String sdata(pool);
1.172.2.3 paf 203: if(params.count()==1) { // named output
1.31 paf 204: // write out names line
205: if(table.columns()) { // named table
1.98 parser 206: Array_iter i(*table.columns());
207: while(i.has_next()) {
208: sdata.append(*i.next_string(), //*static_cast<String *>(table.columns()->quick_get(column)),
209: String::UL_TABLE);
210: if(i.has_next())
1.31 paf 211: sdata.APPEND_CONST("\t");
212: }
213: } else { // nameless table
1.98 parser 214: if(int lsize=table.size()?static_cast<Array *>(table.get(0))->size():0)
1.31 paf 215: for(int column=0; column<lsize; column++) {
1.106 parser 216: char *cindex_tab=(char *)pool.malloc(MAX_NUMBER);
1.31 paf 217: snprintf(cindex_tab, MAX_NUMBER, "%d\t", column);
218: sdata.APPEND_CONST(cindex_tab);
219: }
220: else
221: sdata.APPEND_CONST("empty nameless table");
222: }
223: sdata.APPEND_CONST("\n");
1.129 paf 224: } else { // mode specified
1.172.2.3 paf 225: const String& mode=params.as_string(0, "mode must be string");
1.129 paf 226: if(mode=="append")
227: do_append=true;
228: else if(mode=="nameless")
229: /*ok, already skipped names output*/;
230: else
1.146 paf 231: throw Exception("parser.runtime",
1.129 paf 232: &mode,
233: "unknown mode, must be 'append'");
234:
1.31 paf 235: }
236: // data lines
1.98 parser 237: Array_iter i(table);
238: while(i.has_next()) {
239: Array_iter c(*static_cast<Array *>(i.next()));
240: while(c.has_next()) {
1.153 paf 241: if(const String *s=c.next_string())
242: sdata.append(*s,
243: String::UL_TABLE);
1.98 parser 244: if(c.has_next())
1.31 paf 245: sdata.APPEND_CONST("\t");
246: }
247: sdata.APPEND_CONST("\n");
248: }
249:
250: // write
1.141 paf 251: file_write(r.absolute(vfile_name.as_string()),
1.129 paf 252: sdata.cstr(), sdata.size(), true, do_append);
1.22 paf 253: }
254:
1.172.2.3 paf 255: static void _count(Request& r, StringPtr method_name, MethodParams& ) {
1.6 paf 256: Pool& pool=r.pool();
1.165 paf 257: int result=static_cast<VTable *>(r.get_self())->table(&method_name).size();
1.151 paf 258: r.write_no_lang(*new(pool) VInt(pool, result));
1.6 paf 259: }
260:
1.172.2.3 paf 261: static void _line(Request& r, StringPtr method_name, MethodParams& ) {
1.6 paf 262: Pool& pool=r.pool();
1.165 paf 263: int result=1+static_cast<VTable *>(r.get_self())->table(&method_name).current();
1.151 paf 264: r.write_no_lang(*new(pool) VInt(pool, result));
1.6 paf 265: }
266:
1.172.2.3 paf 267: static void _offset(Request& r, StringPtr method_name, MethodParams& params) {
1.6 paf 268: Pool& pool=r.pool();
1.165 paf 269: Table& table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.172.2.3 paf 270: if(params.count()) {
1.133 paf 271: bool absolute=false;
1.172.2.3 paf 272: if(params.count()>1) {
273: const String& whence=params.as_string(0, "whence must be string");
1.133 paf 274: if(whence=="cur")
1.134 paf 275: absolute=false;
1.133 paf 276: else if(whence=="set")
1.134 paf 277: absolute=true;
1.133 paf 278: else
1.146 paf 279: throw Exception("parser.runtime",
1.134 paf 280: &whence,
281: "is invalid whence, valid are 'cur' or 'set'");
1.157 paf 282: }
1.133 paf 283:
1.172.2.3 paf 284: Value& offset_expr=params.as_junction(params.count()-1, "offset must be expression");
1.147 paf 285: table.offset(absolute, r.process_to_value(offset_expr).as_int());
1.151 paf 286: } else
287: r.write_no_lang(*new(pool) VInt(pool, table.current()));
1.6 paf 288: }
289:
1.172.2.3 paf 290: static void _menu(Request& r, StringPtr method_name, MethodParams& params) {
291: Value& body_code=params.as_junction(0, "body must be code");
1.7 paf 292:
1.172.2.3 paf 293: Value *delim_maybe_code=params.count()>1?¶ms.get(1):0;
1.7 paf 294:
1.165 paf 295: Table& table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.7 paf 296: bool need_delim=false;
1.99 parser 297: int saved_current=table.current();
298: int size=table.size();
299: for(int row=0; row<size; row++) {
1.22 paf 300: table.set_current(row);
1.7 paf 301:
1.161 paf 302: StringOrValue sv_processed=r.process(body_code);
303: const String *s_processed=sv_processed.get_string();
304: if(delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body
305: if(need_delim) // need delim & iteration produced string?
1.150 paf 306: r.write_pass_lang(r.process(*delim_maybe_code));
1.7 paf 307: need_delim=true;
308: }
1.161 paf 309: r.write_pass_lang(sv_processed);
1.7 paf 310: }
1.99 parser 311: table.set_current(saved_current);
1.7 paf 312: }
313:
1.110 parser 314: #ifndef DOXYGEN
1.74 paf 315: struct Row_info {
1.166 paf 316: Request *r;
1.74 paf 317: Table *table;
1.166 paf 318: Value *key_code;
1.74 paf 319: int key_field;
320: Array *value_fields;
321: Hash *hash;
1.163 paf 322: bool distinct;
1.167 paf 323: int row;
1.74 paf 324: };
1.110 parser 325: #endif
1.74 paf 326: static void table_row_to_hash(Array::Item *value, void *info) {
327: Array& row=*static_cast<Array *>(value);
328: Row_info& ri=*static_cast<Row_info *>(info);
329: Pool& pool=ri.table->pool();
330:
1.166 paf 331: const String *key;
332: if(ri.key_code) {
1.167 paf 333: ri.table->set_current(ri.row++); // change context row
1.166 paf 334: StringOrValue sv_processed=ri.r->process(*ri.key_code);
335: key=&sv_processed.as_string();
336: } else
337: key=ri.key_field<row.size()?row.get_string(ri.key_field):0;
338:
339: if(!key)
340: return; // ignore rows without key [too-short-record_array if-indexed]
1.74 paf 341:
1.166 paf 342: VHash& result=*new(pool) VHash(pool);
343: Hash& hash=*result.get_hash(0);
344: for(int i=0; i<ri.value_fields->size(); i++) {
345: int value_field=ri.value_fields->get_int(i);
346: if(value_field<row.size())
347: hash.put(
348: *ri.table->columns()->get_string(value_field),
349: new(pool) VString(*row.get_string(value_field)));
1.74 paf 350: }
1.166 paf 351:
352: if(ri.hash->put_dont_replace(*key, &result)) // put. existed?
353: if(!ri.distinct)
354: throw Exception("parser.runtime",
355: key,
356: "duplicate key");
1.74 paf 357: }
1.172.2.3 paf 358: static void _hash(Request& r, StringPtr method_name, MethodParams& params) {
1.97 parser 359: Pool& pool=r.pool();
1.165 paf 360: Table& self_table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.97 parser 361: Value& result=*new(pool) VHash(pool);
1.152 paf 362: if(const Array *columns=self_table.columns())
363: if(columns->size()>0) {
1.163 paf 364: bool distinct=false;
1.172.2.3 paf 365: int param_index=params.count()-1;
1.166 paf 366: if(param_index>0) {
367: if(Hash *options=
1.172.2.3 paf 368: params.as_no_junction(param_index, "param must not be code").get_hash(0)) {
1.166 paf 369: --param_index;
370: int valid_options=0;
371: if(Value *vdistinct=(Value *)options->get(*sql_distinct_name)) {
372: valid_options++;
373: distinct=r.process_to_value(*vdistinct).as_bool();
374: }
375: if(valid_options!=options->size())
376: throw Exception("parser.runtime",
377: &method_name,
378: "called with invalid option");
1.163 paf 379: }
380: }
381: if(param_index==2) // bad options param type
382: throw Exception("parser.runtime",
383: &method_name,
384: "options must be hash");
385:
1.123 parser 386: Array value_fields(pool);
1.163 paf 387: if(param_index>0) {
1.172.2.3 paf 388: Value& value_fields_param=params.as_no_junction(param_index, "value field(s) must not be code");
1.123 parser 389: if(value_fields_param.is_string()) {
390: value_fields+=self_table.column_name2index(value_fields_param.as_string(), true);
391: } else if(Table *value_fields_table=value_fields_param.get_table()) {
392: for(int i=0; i<value_fields_table->size(); i++) {
393: const String& value_field_name=
394: *static_cast<Array *>(value_fields_table->get(i))->get_string(0);
395: value_fields+=self_table.column_name2index(value_field_name, true);
396: }
397: } else
1.146 paf 398: throw Exception("parser.runtime",
1.123 parser 399: &method_name,
400: "value field(s) must be string or self_table"
401: );
402: } else { // by all columns, including key
1.74 paf 403: for(int i=0; i<columns->size(); i++)
1.123 parser 404: value_fields+=i;
1.74 paf 405: }
406:
1.172.2.3 paf 407: Value& key_param=params.get(0);
1.166 paf 408: Value *key_code=key_param.get_junction()?&key_param:0;
409: int key_field=key_code?-1
410: :self_table.column_name2index(key_param.as_string(), true);
411:
412: Row_info row_info={&r, &self_table, key_code, key_field, &value_fields, result.get_hash(0), distinct};
413:
414: int saved_current=self_table.current();
1.123 parser 415: self_table.for_each(table_row_to_hash, &row_info);
1.166 paf 416: self_table.set_current(saved_current);
1.74 paf 417: }
1.97 parser 418: r.write_no_lang(result);
1.74 paf 419: }
420:
1.110 parser 421: #ifndef DOXYGEN
1.78 paf 422: struct Table_seq_item {
1.34 paf 423: Array *row;
424: union {
425: char *c_str;
426: double d;
427: } value;
1.32 paf 428: };
1.110 parser 429: #endif
1.34 paf 430: static int sort_cmp_string(const void *a, const void *b) {
431: return strcmp(
1.78 paf 432: static_cast<const Table_seq_item *>(a)->value.c_str,
433: static_cast<const Table_seq_item *>(b)->value.c_str
1.34 paf 434: );
435: }
436: static int sort_cmp_double(const void *a, const void *b) {
1.78 paf 437: double va=static_cast<const Table_seq_item *>(a)->value.d;
438: double vb=static_cast<const Table_seq_item *>(b)->value.d;
1.34 paf 439: if(va<vb)
440: return -1;
441: else if(va>vb)
442: return +1;
443: else
444: return 0;
445: }
1.172.2.3 paf 446: static void _sort(Request& r, StringPtr method_name, MethodParams& params) {
1.101 parser 447: Pool& pool=r.pool();
1.172.2.3 paf 448: Value& key_maker=params.as_junction(0, "key-maker must be code");
1.61 paf 449:
1.172.2.3 paf 450: bool reverse=params.count()>1/*..[desc|asc|]*/?
451: reverse=params.as_no_junction(1, "order must not be code").as_string()=="desc":
1.104 parser 452: false; // default=asc
1.32 paf 453:
1.165 paf 454: Table& old_table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.101 parser 455: Table& new_table=*new(pool) Table(pool, &method_name, old_table.columns());
1.34 paf 456:
1.101 parser 457: Table_seq_item *seq=(Table_seq_item *)pool.malloc(sizeof(Table_seq_item)*old_table.size());
1.34 paf 458: int i;
459:
460: // calculate key values
461: bool key_values_are_strings=true;
1.101 parser 462: for(i=0; i<old_table.size(); i++) {
463: old_table.set_current(i);
1.32 paf 464: // calculate key value
1.101 parser 465: seq[i].row=(MethodParams *)old_table.get(i);
1.147 paf 466: Value& value=*r.process_to_value(key_maker).as_expr_result(true/*return string as-is*/);
1.34 paf 467: if(i==0) // determining key values type by first one
468: key_values_are_strings=value.is_string();
469:
470: if(key_values_are_strings)
471: seq[i].value.c_str=value.as_string().cstr();
472: else
473: seq[i].value.d=value.as_double();
1.32 paf 474: }
475: // sort keys
1.101 parser 476: _qsort(seq, old_table.size(), sizeof(Table_seq_item),
1.34 paf 477: key_values_are_strings?sort_cmp_string:sort_cmp_double);
1.32 paf 478:
1.34 paf 479: // reorder table as they require in 'seq'
1.101 parser 480: for(i=0; i<old_table.size(); i++)
481: new_table+=seq[reverse?old_table.size()-1-i:i].row;
1.32 paf 482:
1.116 parser 483: // replace any previous table value
1.165 paf 484: static_cast<VTable *>(r.get_self())->set_table(new_table);
1.32 paf 485: }
486:
1.172.2.3 paf 487: static bool _locate_expression(Request& r, StringPtr method_name, MethodParams& params) {
488: if(params.count()>1)
1.146 paf 489: throw Exception("parser.runtime",
1.145 paf 490: &method_name,
491: "locate by expression has only one parameter - expression");
492:
1.172.2.3 paf 493: Value& expression_code=params.as_junction(0, "must be expression");
1.145 paf 494:
1.165 paf 495: Table& table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.145 paf 496: int saved_current=table.current();
497: int size=table.size();
498: for(int row=0; row<size; row++) {
499: table.set_current(row);
500:
1.147 paf 501: if(r.process_to_value(expression_code).as_bool())
1.145 paf 502: return true;
503: }
504: table.set_current(saved_current);
505: return false;
506: }
1.172.2.3 paf 507: static bool _locate_name_value(Request& r, StringPtr method_name, MethodParams& params) {
508: if(params.count()>2)
1.146 paf 509: throw Exception("parser.runtime",
1.145 paf 510: &method_name,
511: "locate by name and value has only two parameters - name and value");
1.72 paf 512:
1.165 paf 513: Table& table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.145 paf 514: return table.locate(
1.172.2.3 paf 515: params.as_string(0, "column name must be string"),
516: params.as_string(1, "value must be string")
1.145 paf 517: );
518: }
1.172.2.3 paf 519: static void _locate(Request& r, StringPtr method_name, MethodParams& params) {
1.145 paf 520: Pool& pool=r.pool();
1.172.2.3 paf 521: bool result=params.get(0).get_junction()?
1.145 paf 522: _locate_expression(r, method_name, params) :
1.151 paf 523: _locate_name_value(r, method_name, params);
524: r.write_no_lang(*new(pool) VBool(pool, result));
1.37 paf 525: }
526:
1.172.2.3 paf 527: static void _flip(Request& r, StringPtr method_name, MethodParams& params) {
1.39 paf 528: Pool& pool=r.pool();
1.165 paf 529: Table& old_table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.39 paf 530: Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/);
531: if(old_table.size())
532: if(int old_cols=old_table.at(0).size())
533: for(int column=0; column<old_cols; column++) {
534: Array& new_row=*new(pool) Array(pool, old_table.size());
535: for(int i=0; i<old_table.size(); i++) {
536: const Array& old_row=old_table.at(i);
1.126 parser 537: new_row+=column<old_row.size()?old_row.get(column):new(pool) String(pool);
1.39 paf 538: }
539: new_table+=&new_row;
540: }
541:
1.99 parser 542: r.write_no_lang(*new(pool) VTable(pool, &new_table));
1.39 paf 543: }
544:
1.172.2.3 paf 545: static void _append(Request& r, StringPtr method_name, MethodParams& params) {
1.41 paf 546: Pool& pool=r.pool();
1.134 paf 547: // data
548: Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
1.61 paf 549: const String& string=
1.172.2.3 paf 550: r.process_to_string(params.as_junction(0, "body must be code"));
1.41 paf 551:
552: // parse cells
553: Array& row=*new(pool) Array(pool);
1.137 paf 554: string.split(row, 0, "\t", 1, String::UL_AS_IS);
1.41 paf 555:
1.165 paf 556: static_cast<VTable *>(r.get_self())->table(&method_name)+=&row;
1.41 paf 557: }
558:
1.172.2.3 paf 559: static void _join(Request& r, StringPtr method_name, MethodParams& params) {
1.42 paf 560: Pool& pool=r.pool();
561:
1.172.2.3 paf 562: Table *maybe_src=params.as_no_junction(0, "table ref must not be code").get_table();
1.42 paf 563: if(!maybe_src)
1.146 paf 564: throw Exception("parser.runtime",
1.91 parser 565: &method_name,
1.42 paf 566: "source is not a table");
567:
568: Table& src=*maybe_src;
1.165 paf 569: Table& dest=static_cast<VTable *>(r.get_self())->table(&method_name);
1.42 paf 570: if(&src == &dest)
1.146 paf 571: throw Exception("parser.runtime",
1.91 parser 572: &method_name,
1.42 paf 573: "source and destination are same table");
574:
1.157 paf 575: int offset, limit;
576: get_copy_options(r, method_name, params, 1, src,
577: offset, limit);
578:
1.42 paf 579: if(const Array *dest_columns=dest.columns()) { // dest is named
580: int saved_src_current=src.current();
1.157 paf 581: int m=src.size()-offset;
582: if(!limit || limit>m)
583: limit=m;
584: int end=offset+limit;
585: for(int src_row=offset; src_row<end; src_row++) {
1.42 paf 586: src.set_current(src_row);
587: Array& dest_row=*new(pool) Array(pool);
1.140 paf 588: for(int dest_column=0; dest_column<dest_columns->size(); dest_column++) {
589: const String *src_item=src.item(*dest_columns->get_string(dest_column));
590: dest_row+=src_item?src_item:new(pool) String(pool);
591: }
1.42 paf 592: dest+=&dest_row;
593: }
594: src.set_current(saved_src_current);
595: } else { // dest is nameless
596: for(int src_row=0; src_row<src.size(); src_row++)
597: dest+=&src.at(src_row);
598: }
599: }
600:
1.94 parser 601: #ifndef DOXYGEN
1.169 paf 602: class Table_sql_event_handlers: public SQL_Driver_query_event_handlers {
1.94 parser 603: public:
604: Table_sql_event_handlers(Pool& apool, const String& amethod_name,
1.172.2.2 paf 605: const String& astatement_string, const char* astatement_cstr) :
1.94 parser 606: pool(apool),
607: method_name(amethod_name),
608: statement_string(astatement_string),
609: statement_cstr(astatement_cstr),
610: columns(*new(pool) Array(pool)),
1.134 paf 611: row(0),
1.94 parser 612: table(0)
613: {
614: }
615:
1.170 paf 616: bool add_column(SQL_Error& error, void *ptr, size_t size) {
617: try {
618: String *column=new(pool) String(pool);
619: column->APPEND_TAINTED(
1.172.2.2 paf 620: (const char* )ptr, size,
1.170 paf 621: statement_cstr, 0);
622: columns+=column;
623: return false;
624: } catch(...) {
625: error=SQL_Error("exception occured in Table_sql_event_handlers::add_column");
626: return true;
627: }
628: }
629: bool before_rows(SQL_Error& error) {
630: try {
631: table=new(pool) Table(pool, &method_name, &columns);
632: return false;
633: } catch(...) {
634: error=SQL_Error("exception occured in Table_sql_event_handlers::before_rows");
635: return true;
636: }
637: }
638: bool add_row(SQL_Error& error) {
639: try {
640: (*table)+=(row=new(pool) Array(pool));
641: return false;
642: } catch(...) {
643: error=SQL_Error("exception occured in Table_sql_event_handlers::add_row");
644: return true;
645: }
646: }
647: bool add_row_cell(SQL_Error& error, void *ptr, size_t size) {
648: try {
649: String *cell=new(pool) String(pool);
650: if(size)
651: cell->APPEND_TAINTED(
1.172.2.2 paf 652: (const char* )ptr, size,
1.170 paf 653: statement_cstr, table->size()-1);
654: (*row)+=cell;
655: return false;
656: } catch(...) {
657: error=SQL_Error("exception occured in Table_sql_event_handlers::add_row_cell");
658: return true;
659: }
1.94 parser 660: }
661:
662: private:
663: Pool& pool;
664: const String& method_name;
1.172.2.2 paf 665: const String& statement_string; const char* statement_cstr;
1.94 parser 666: Array& columns;
667: Array *row;
668: public:
669: Table *table;
670: };
671: #endif
1.172.2.3 paf 672: static void _sql(Request& r, StringPtr method_name, MethodParams& params) {
1.49 paf 673: Pool& pool=r.pool();
674:
1.172.2.3 paf 675: Value& statement=params.as_junction(0, "statement must be code");
1.49 paf 676:
1.53 paf 677: ulong limit=0;
1.100 parser 678: ulong offset=0;
1.172.2.3 paf 679: if(params.count()>1) {
680: Value& voptions=params.as_no_junction(1, "options must be hash, not code");
1.156 paf 681: if(!voptions.is_string())
1.128 parser 682: if(Hash *options=voptions.get_hash(&method_name)) {
1.164 paf 683: int valid_options=0;
684: if(Value *vlimit=(Value *)options->get(*sql_limit_name)) {
685: valid_options++;
1.147 paf 686: limit=(ulong)r.process_to_value(*vlimit).as_double();
1.164 paf 687: }
688: if(Value *voffset=(Value *)options->get(*sql_offset_name)) {
689: valid_options++;
1.147 paf 690: offset=(ulong)r.process_to_value(*voffset).as_double();
1.164 paf 691: }
692: if(valid_options!=options->size())
693: throw Exception("parser.runtime",
694: &method_name,
695: "called with invalid option");
1.109 parser 696: } else
1.146 paf 697: throw Exception("parser.runtime",
1.109 parser 698: &method_name,
699: "options must be hash");
1.49 paf 700: }
701:
1.54 paf 702: Temp_lang temp_lang(r, String::UL_SQL);
1.147 paf 703: const String& statement_string=r.process_to_string(statement);
1.172.2.2 paf 704: const char* statement_cstr=
1.138 paf 705: statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name));
1.94 parser 706: Table_sql_event_handlers handlers(pool, method_name,
707: statement_string, statement_cstr);
1.135 paf 708: #ifdef RESOURCES_DEBUG
709: struct timeval mt[2];
710: //measure:before
711: gettimeofday(&mt[0],NULL);
712: #endif
1.160 paf 713: r.connection(&method_name)->query(
714: statement_cstr, offset, limit,
715: handlers,
716: statement_string);
1.135 paf 717:
718: #ifdef RESOURCES_DEBUG
719: //measure:after connect
720: gettimeofday(&mt[1],NULL);
721:
722: double t[2];
723: for(int i=0;i<2;i++)
724: t[i]=mt[i].tv_sec+mt[i].tv_usec/1000000.0;
725:
726: r.sql_request_time+=t[1]-t[0];
727: #endif
1.96 parser 728:
729: Table *result=
730: handlers.table?handlers.table: // query resulted in table? return it
731: new(pool) Table(pool, &method_name, 0); // query returned no table, fake it
1.49 paf 732:
733: // replace any previous table value
1.165 paf 734: static_cast<VTable *>(r.get_self())->set_table(*result);
1.49 paf 735: }
736:
1.172.2.3 paf 737: static void _columns(Request& r, StringPtr method_name, MethodParams& ) {
1.88 parser 738: Pool& pool=r.pool();
739:
740: Array& result_columns=*new(pool) Array(pool);
1.89 parser 741: result_columns+=new(pool) String(pool, "column");
1.88 parser 742: Table& result_table=*new(pool) Table(pool, &method_name, &result_columns);
743:
1.165 paf 744: Table& source_table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.88 parser 745: if(const Array *source_columns=source_table.columns()) {
1.98 parser 746: Array_iter i(*source_columns);
747: while(i.has_next()) {
1.88 parser 748: Array& result_row=*new(pool) Array(pool);
1.98 parser 749: result_row+=i.next();
1.88 parser 750: result_table+=&result_row;
751: }
752: }
753:
1.151 paf 754: r.write_no_lang(*new(pool) VTable(pool, &result_table));
1.88 parser 755: }
756:
1.172.2.3 paf 757: static void _select(Request& r, StringPtr method_name, MethodParams& params) {
1.148 paf 758: Pool& pool=r.pool();
759:
1.172.2.3 paf 760: Value& vcondition=params.as_junction(0, "condition must be expression");
1.148 paf 761:
1.165 paf 762: Table& source_table=static_cast<VTable *>(r.get_self())->table(&method_name);
1.148 paf 763: Table& result_table=*new(pool) Table(pool,
764: source_table.origin_string(),
765: source_table.columns()
766: );
767:
768: int saved_current=source_table.current();
769: int size=source_table.size();
770: for(int row=0; row<size; row++) {
771: source_table.set_current(row);
772:
1.150 paf 773: bool condition=r.process_to_value(vcondition,
1.149 paf 774: /*0/*no name* /,*/
1.148 paf 775: false/*don't intercept string*/).as_bool();
776:
777: if(condition) // ...condition is true=
778: result_table+=&source_table.at(row); // =green light to go to result
779: }
780: source_table.set_current(saved_current);
781:
1.151 paf 782: r.write_no_lang(*new(pool) VTable(pool, &result_table));
1.148 paf 783: }
784:
1.66 paf 785: // constructor
786:
1.151 paf 787: MTable::MTable(Pool& apool) : Methoded(apool, "table") {
1.142 paf 788: // ^table::create{data}
789: // ^table::create[nameless]{data}
790: // ^table::create[table]
791: add_native_method("create", Method::CT_DYNAMIC, _create, 1, 2);
792: // old name for compatibility with <= v 1.141 2002/01/25 11:33:45 paf
793: add_native_method("set", Method::CT_DYNAMIC, _create, 1, 2);
1.2 paf 794:
1.142 paf 795: // ^table::load[file]
796: // ^table::load[nameless;file]
1.168 paf 797: add_native_method("load", Method::CT_DYNAMIC, _load, 1, 3);
1.22 paf 798:
799: // ^table.save[file]
800: // ^table.save[nameless;file]
1.66 paf 801: add_native_method("save", Method::CT_DYNAMIC, _save, 1, 2);
1.6 paf 802:
803: // ^table.count[]
1.66 paf 804: add_native_method("count", Method::CT_DYNAMIC, _count, 0, 0);
1.6 paf 805:
806: // ^table.line[]
1.66 paf 807: add_native_method("line", Method::CT_DYNAMIC, _line, 0, 0);
1.6 paf 808:
1.10 paf 809: // ^table.offset[]
1.133 paf 810: // ^table.offset(offset)
811: // ^table.offset[cur|set](offset)
812: add_native_method("offset", Method::CT_DYNAMIC, _offset, 0, 2);
1.7 paf 813:
1.10 paf 814: // ^table.menu{code}
815: // ^table.menu{code}[delim]
1.66 paf 816: add_native_method("menu", Method::CT_DYNAMIC, _menu, 1, 2);
1.74 paf 817:
1.79 paf 818: // ^table:hash[key field name]
1.123 parser 819: // ^table:hash[key field name][value field name(s) string/table]
1.163 paf 820: add_native_method("hash", Method::CT_DYNAMIC, _hash, 1, 3);
1.32 paf 821:
1.102 parser 822: // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[desc|asc]
823: // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[desc|asc]
1.66 paf 824: add_native_method("sort", Method::CT_DYNAMIC, _sort, 1, 2);
1.8 paf 825:
1.36 paf 826: // ^table.locate[field;value]
1.145 paf 827: add_native_method("locate", Method::CT_DYNAMIC, _locate, 1, 2);
1.39 paf 828:
829: // ^table.flip[]
1.66 paf 830: add_native_method("flip", Method::CT_DYNAMIC, _flip, 0, 0);
1.41 paf 831:
832: // ^table.append{r{tab}e{tab}c{tab}o{tab}r{tab}d}
1.66 paf 833: add_native_method("append", Method::CT_DYNAMIC, _append, 1, 1);
1.42 paf 834:
1.157 paf 835: // ^table.join[table][$.limit(10) $.offset(1) $.offset[cur] ]
836: add_native_method("join", Method::CT_DYNAMIC, _join, 1, 2);
1.49 paf 837:
838:
1.100 parser 839: // ^table:sql[query]
840: // ^table:sql[query][$.limit(1) $.offset(2)]
841: add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.88 parser 842:
843: // ^table:columns[]
844: add_native_method("columns", Method::CT_DYNAMIC, _columns, 0, 0);
1.148 paf 845:
846: // ^table.select(expression) = table
847: add_native_method("select", Method::CT_DYNAMIC, _select, 1, 1);
1.66 paf 848: }
849:
850: // global variable
851:
852: Methoded *table_class;
853:
854: // creator
855:
856: Methoded *MTable_create(Pool& pool) {
857: return table_class=new(pool) MTable(pool);
1.40 paf 858: }
E-mail: