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

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

E-mail: