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?&params.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: