Annotation of parser3/src/classes/hash.C, revision 1.54.2.9

1.1       paf         1: /** @file
                      2:        Parser: @b hash parser class.
                      3: 
1.54.2.2  paf         4:        Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com)
1.35      paf         5:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.41      paf         6: */
1.1       paf         7: 
1.54.2.9! paf         8: static const char* IDENT_HASH_C="$Date: 2003/02/06 11:56:31 $";
1.1       paf         9: 
                     10: #include "classes.h"
1.54.2.4  paf        11: #include "pa_vmethod_frame.h"
                     12: 
1.1       paf        13: #include "pa_request.h"
                     14: #include "pa_vhash.h"
1.6       parser     15: #include "pa_vvoid.h"
1.2       parser     16: #include "pa_sql_connection.h"
1.9       parser     17: #include "pa_vtable.h"
1.22      parser     18: #include "pa_vbool.h"
1.26      paf        19: #include "pa_vmethod_frame.h"
1.2       parser     20: 
1.1       paf        21: // class
                     22: 
1.54.2.6  paf        23: class MHash: public Methoded {
1.2       parser     24: public: // VStateless_class
1.54.2.1  paf        25:        ValuePtr create_new_value() { return ValuePtr(new VHash()); }
1.2       parser     26: 
1.1       paf        27: public:
1.54.2.6  paf        28:        MHash();
1.1       paf        29: public: // Methoded
1.2       parser     30:        bool used_directly() { return true; }
1.1       paf        31: };
1.54.2.6  paf        32: 
                     33: // global variable
                     34: 
                     35: MethodedPtr hash_class(new MHash);
                     36: 
1.1       paf        37: // methods
                     38: 
1.11      parser     39: #ifndef DOXYGEN
1.52      paf        40: class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers {
1.54.2.6  paf        41:        StringPtr method_name;
                     42:        StringPtr statement_string; const char* statement_cstr;
                     43:        bool distinct;
                     44:        HashStringValue& rows_hash;
                     45:        HashStringValue* row_hash;
                     46:        int column_index;
                     47:        ArrayString columns;
                     48:        int row_index;
1.11      parser     49: public:
1.54.2.6  paf        50:        Hash_sql_event_handlers(StringPtr amethod_name,
                     51:                StringPtr astatement_string, const char* astatement_cstr,
1.49      paf        52:                bool adistinct,
1.54.2.6  paf        53:                HashStringValue& arows_hash): 
1.11      parser     54:                method_name(amethod_name),
                     55:                statement_string(astatement_string),
                     56:                statement_cstr(astatement_cstr),
1.49      paf        57:                distinct(adistinct),
1.11      parser     58:                rows_hash(arows_hash),
                     59:                row_index(0) {
                     60:        }
1.53      paf        61:        bool add_column(SQL_Error& error, void *ptr, size_t size) {
                     62:                try {
1.54.2.6  paf        63:                        StringPtr column(new String);
1.53      paf        64:                        column->APPEND_TAINTED(
1.54.2.2  paf        65:                                (const char* )ptr, size, 
1.53      paf        66:                                statement_cstr, 0);
                     67:                        columns+=column;
                     68: 
                     69:                        return false;
                     70:                } catch(...) {
                     71:                        error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
                     72:                        return true;
                     73:                }
1.11      parser     74:        }
1.53      paf        75:        bool before_rows(SQL_Error& error) { 
1.54.2.6  paf        76:                if(columns.count()<=1) {
1.53      paf        77:                        error=SQL_Error("parser.runtime",
1.54.2.6  paf        78:                                /*method_name,*/
1.11      parser     79:                                "column count must be more than 1 to create a hash");
1.53      paf        80:                        return true;
                     81:                }
                     82: 
                     83:                return false;
1.11      parser     84:        }
1.53      paf        85:        bool add_row(SQL_Error& /*error*/) {
1.11      parser     86:                column_index=0;
1.53      paf        87:                return false;
1.11      parser     88:        }
1.53      paf        89:        bool add_row_cell(SQL_Error& error, void *ptr, size_t size) {
                     90:                try {
1.54.2.6  paf        91:                        StringPtr cell(new String);
1.53      paf        92:                        if(size)
                     93:                                cell->APPEND_TAINTED(
1.54.2.2  paf        94:                                        (const char* )ptr, size, 
1.53      paf        95:                                        statement_cstr, row_index++);
                     96:                        if(column_index==0) {
1.54.2.6  paf        97:                                VHashPtr row_vhash(new VHash);
                     98:                                row_hash=&row_vhash->hash();
                     99:                                if(rows_hash.put_dont_replace(cell, row_vhash)) // put. existed?
1.53      paf       100:                                        if(!distinct) {
                    101:                                                error=SQL_Error("parser.runtime",
1.54.2.6  paf       102:                                                        /*cell,*/
1.53      paf       103:                                                        "duplicate key");
                    104:                                                return true;
                    105:                                        }
                    106:                        } else
1.54.2.6  paf       107:                                row_hash->put(columns[column_index], ValuePtr(new VString(cell)));
1.53      paf       108:                        column_index++;
                    109: 
                    110:                        return false;
                    111:                } catch(...) {
                    112:                        error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell");
                    113:                        return true;
                    114:                }
1.11      parser    115:        }
                    116: 
                    117: };
                    118: #endif
                    119: 
1.54.2.6  paf       120: static void copy_all_overwrite_to(
                    121:                                                                  HashStringValue::key_type key, 
                    122:                                                                  HashStringValue::value_type value, 
                    123:                                                                  HashStringValue* dest) {
                    124:        dest->put(key, value);
1.20      parser    125: }
1.54.2.3  paf       126: static void _create_or_add(Request& r, StringPtr method_name, MethodParams& params) {
                    127:        if(params.count()) {
1.54.2.6  paf       128:                ValuePtr vb=params.as_no_junction(0, "param must be hash");
                    129:                if(HashStringValue* b=vb->get_hash(method_name))
                    130:                        b->for_each(copy_all_overwrite_to, &(GET_SELF(r, VHash).hash()));
1.20      parser    131:        }
                    132: }
                    133: 
1.54.2.6  paf       134: static void remove_key_from(
                    135:                                                        HashStringValue::key_type key, 
                    136:                                                        HashStringValue::value_type /*value*/, 
                    137:                                                        HashStringValue* dest) {
                    138:        dest->remove(key);
1.22      parser    139: }
1.54.2.3  paf       140: static void _sub(Request& r, StringPtr method_name, MethodParams& params) {
1.54.2.6  paf       141:        ValuePtr vb=params.as_no_junction(0, "param must be hash");
                    142:        if(HashStringValue* b=vb->get_hash(method_name))
                    143:                b->for_each(remove_key_from, &GET_SELF(r, VHash).hash());
1.22      parser    144: }
                    145: 
1.54.2.6  paf       146: static void copy_all_dontoverwrite_to(
                    147:                                                                  HashStringValue::key_type key, 
                    148:                                                                  HashStringValue::value_type value, 
                    149:                                                                  HashStringValue* dest) {
                    150:        dest->put_dont_replace(key, value);
1.22      parser    151: }
1.54.2.3  paf       152: static void _union(Request& r, StringPtr method_name, MethodParams& params) {
1.22      parser    153:        // dest = copy of self
1.54.2.6  paf       154:        ValuePtr result(new VHash(GET_SELF(r, VHash).hash()));
1.22      parser    155:        // dest += b
1.54.2.6  paf       156:        ValuePtr vb=params.as_no_junction(0, "param must be hash");
                    157:        if(HashStringValue* b=vb->get_hash(method_name))
                    158:                b->for_each(copy_all_dontoverwrite_to, result->get_hash(Exception::undefined_source));
1.22      parser    159: 
                    160:        // return result
                    161:        r.write_no_lang(result);
                    162: }
                    163: 
                    164: #ifndef DOXYGEN
                    165: struct Copy_intersection_to_info {
1.54.2.6  paf       166:        HashStringValue* b;
                    167:        HashStringValue* dest;
1.22      parser    168: };
                    169: #endif
1.54.2.6  paf       170: static void copy_intersection_to(
                    171:                                                                 HashStringValue::key_type key, 
                    172:                                                                 HashStringValue::value_type value, 
                    173:                                                                 Copy_intersection_to_info *info) {
                    174:        if(info->b->get(key))
                    175:                info->dest->put_dont_replace(key, value);
1.22      parser    176: }
1.54.2.3  paf       177: static void _intersection(Request& r, StringPtr method_name, MethodParams& params) {
1.54.2.6  paf       178:        ValuePtr result(new VHash);
1.22      parser    179:        // dest += b
1.54.2.6  paf       180:        ValuePtr vb=params.as_no_junction(0, "param must be hash");
                    181:        if(HashStringValue* b=vb->get_hash(method_name)) {
                    182:                Copy_intersection_to_info info={b, result->get_hash(Exception::undefined_source)};
                    183:                GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info);
1.22      parser    184:        }
                    185: 
                    186:        // return result
1.54.2.6  paf       187:        r.write_no_lang(result);
1.22      parser    188: }
                    189: 
1.54.2.6  paf       190: static bool intersects(
                    191:                                           HashStringValue::key_type key, 
                    192:                                           HashStringValue::value_type /*value*/, 
                    193:                                           HashStringValue* b) {
                    194:        return b->get(key);
1.22      parser    195: }
                    196: 
1.54.2.3  paf       197: static void _intersects(Request& r, StringPtr method_name, MethodParams& params) {
1.54.2.6  paf       198:        bool result=false;
1.22      parser    199: 
1.54.2.6  paf       200:        ValuePtr vb=params.as_no_junction(0, "param must be hash");
                    201:        if(HashStringValue* b=vb->get_hash(method_name))
                    202:                result=GET_SELF(r, VHash).hash().first_that(intersects, b);
1.22      parser    203: 
                    204:        // return result
1.54.2.6  paf       205:        r.write_no_lang(ValuePtr(new VBool(result)));
1.22      parser    206: }
                    207: 
                    208: 
1.54.2.9! paf       209: extern StringPtr sql_limit_name;
        !           210: extern StringPtr sql_offset_name;
        !           211: extern StringPtr sql_default_name;
        !           212: extern StringPtr sql_distinct_name;
1.54.2.3  paf       213: static void _sql(Request& r, StringPtr method_name, MethodParams& params) {
1.54.2.6  paf       214:        ValuePtr statement=params.as_junction(0, "statement must be code");
1.2       parser    215: 
                    216:        ulong limit=0;
1.33      paf       217:        ulong offset=0;
1.49      paf       218:        bool distinct=false;
1.54.2.3  paf       219:        if(params.count()>1) {
1.54.2.6  paf       220:                ValuePtr voptions=params.as_no_junction(1, "options must be hash, not code");
                    221:                if(!voptions->is_string())
1.54.2.7  paf       222:                        if(HashStringValue* options=voptions->get_hash(method_name)) {
1.49      paf       223:                                int valid_options=0;
1.54.2.7  paf       224:                                if(ValuePtr vlimit=options->get(sql_limit_name)) {
1.49      paf       225:                                        valid_options++;
1.54.2.7  paf       226:                                        limit=(ulong)r.process_to_value(vlimit)->as_double();
1.49      paf       227:                                }
1.54.2.7  paf       228:                                if(ValuePtr voffset=options->get(sql_offset_name)) {
1.49      paf       229:                                        valid_options++;
1.54.2.7  paf       230:                                        offset=(ulong)r.process_to_value(voffset)->as_double();
1.49      paf       231:                                }
1.54.2.7  paf       232:                                if(ValuePtr vdistinct=options->get(sql_distinct_name)) {
1.49      paf       233:                                        valid_options++;
1.54.2.7  paf       234:                                        distinct=r.process_to_value(vdistinct)->as_bool();
1.49      paf       235:                                }
1.54.2.7  paf       236:                                if(valid_options!=options->count())
1.49      paf       237:                                        throw Exception("parser.runtime",
1.54.2.5  paf       238:                                                method_name,
1.49      paf       239:                                                "called with invalid option");
1.33      paf       240:                        } else
1.36      paf       241:                                throw Exception("parser.runtime",
1.54.2.5  paf       242:                                        method_name,
1.33      paf       243:                                        "options must be hash");
1.2       parser    244:        }
                    245: 
                    246:        Temp_lang temp_lang(r, String::UL_SQL);
1.54.2.7  paf       247:        StringPtr statement_string=r.process_to_string(statement);
1.54.2.2  paf       248:        const char* statement_cstr=
1.54.2.7  paf       249:                statement_string->cstr(r.pool(), String::UL_UNSPECIFIED, r.connection(method_name));
                    250:        HashStringValue& hash=GET_SELF(r, VHash).hash(method_name);
1.11      parser    251:        hash.clear();   
1.54.2.7  paf       252:        Hash_sql_event_handlers handlers(method_name,
1.49      paf       253:                statement_string, statement_cstr, 
                    254:                distinct,
                    255:                hash);
1.54.2.7  paf       256:        r.connection(method_name)->query(
1.23      parser    257:                statement_cstr, offset, limit,
1.45      paf       258:                handlers,
                    259:                statement_string);
1.2       parser    260: }
                    261: 
1.54.2.7  paf       262: static void keys_collector(
                    263:                                                   HashStringValue::key_type key, 
                    264:                                                   HashStringValue::value_type value, 
                    265:                                                   Table *table) {
                    266:        Table::element_type row(new ArrayString);
                    267:        *row+=key;
                    268:        table+=row;
1.9       parser    269: }
1.54.2.3  paf       270: static void _keys(Request& r, StringPtr method_name, MethodParams& ) {
1.54.2.7  paf       271:        Table::columns_type columns(new ArrayString);
                    272:        *columns+=StringPtr(new String("key"));
                    273:        TablePtr table(new Table(method_name, columns));
1.9       parser    274: 
1.54.2.7  paf       275:        GET_SELF(r, VHash).hash(method_name).for_each(keys_collector, table.get());
1.9       parser    276: 
1.54.2.7  paf       277:        r.write_no_lang(ValuePtr(new VTable(table)));
1.9       parser    278: }
                    279: 
1.54.2.3  paf       280: static void _count(Request& r, StringPtr method_name, MethodParams& ) {
1.54.2.7  paf       281:        r.write_no_lang(ValuePtr(new VInt(GET_SELF(r, VHash).hash(method_name).count())));
1.16      parser    282: }
                    283: 
1.54.2.3  paf       284: static void _delete(Request& r, StringPtr method_name, MethodParams& params) {
1.25      paf       285:        Pool& pool=r.pool();
                    286: 
1.54.2.7  paf       287:        GET_SELF(r, VHash).hash(method_name).remove(params.as_string(0, "key must be string"));
1.25      paf       288: }
                    289: 
1.26      paf       290: #ifndef DOXYGEN
                    291: struct Foreach_info {
                    292:        Request *r;
1.54.2.7  paf       293:        StringPtr key_var_name;
                    294:        StringPtr value_var_name;
1.54.2.6  paf       295:        ValuePtr body_code;
                    296:        ValuePtr delim_maybe_code;
1.26      paf       297: 
1.54.2.7  paf       298:        VStringPtr vkey;
1.26      paf       299:        bool need_delim;
                    300: };
                    301: #endif
1.54.2.7  paf       302: static void one_foreach_cycle(                                            
                    303:                                                          HashStringValue::key_type akey, 
                    304:                                                          HashStringValue::value_type avalue, 
                    305:                                                          Foreach_info *info) {
                    306:        info->vkey->set_string(akey);
                    307:        Value& ncontext=*info->r->get_method_frame()->caller();
                    308:        ncontext.put_element(info->key_var_name, info->vkey, false);
                    309:        ncontext.put_element(info->value_var_name, avalue, false);
                    310: 
                    311:        StringOrValue sv_processed=info->r->process(info->body_code);
                    312:        StringPtr s_processed=sv_processed.get_string();
                    313:        if(info->delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body
                    314:                if(info->need_delim) // need delim & iteration produced string?
                    315:                        info->r->write_pass_lang(info->r->process(info->delim_maybe_code));
                    316:                info->need_delim=true;
1.26      paf       317:        }
1.54.2.7  paf       318:        info->r->write_pass_lang(sv_processed);
1.26      paf       319: }
1.54.2.3  paf       320: static void _foreach(Request& r, StringPtr method_name, MethodParams& params) {
1.54.2.7  paf       321:        Foreach_info info;
                    322:        info.r=&r;
                    323:        info.key_var_name=params.as_string(0, "key-var name must be string");
                    324:        info.value_var_name=params.as_string(1, "value-var name must be string");
                    325:        info.body_code=params.as_junction(2, "body must be code");
                    326:        info.delim_maybe_code=params.count()>3?params.get(3):ValuePtr(0);
                    327:        info.vkey=VStringPtr(new VString);
                    328: 
                    329:        VHash& self=GET_SELF(r, VHash);
                    330:        HashStringValue& hash=self.hash(method_name);
1.28      paf       331:        VHash_lock lock(self);
                    332:        hash.for_each(one_foreach_cycle, &info);
1.26      paf       333: }
                    334: 
1.1       paf       335: // constructor
                    336: 
1.54.2.6  paf       337: MHash::MHash(): Methoded("hash") 
1.39      paf       338: {
1.21      parser    339:        // ^hash::create[[copy_from]]
1.22      parser    340:        add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
                    341:        // ^hash.add[add_from]
                    342:        add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
                    343:        // ^hash.sub[sub_from]
                    344:        add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
                    345:        // ^a.union[b] = hash
                    346:        add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
                    347:        // ^a.intersection[b] = hash
                    348:        add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
                    349:        // ^a.intersects[b] = bool
                    350:        add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
1.25      paf       351: 
                    352:        // ^a.delete[key]
                    353:        add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
1.2       parser    354: 
1.33      paf       355:        // ^hash:sql[query][$.limit(1) $.offset(2)]
                    356:        add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.2       parser    357: 
1.14      parser    358:        // ^hash._keys[]
1.13      parser    359:        add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0);    
1.16      parser    360: 
                    361:        // ^hash._count[]
                    362:        add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);  
1.26      paf       363: 
                    364:        // ^hash.foreach[key;value]{code}[delim]
                    365:        add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
1.1       paf       366: }

E-mail: