Annotation of parser3/src/classes/table.C, revision 1.152

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

E-mail: