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

1.1       paf         1: /** @file
                      2:        Parser: @b hash parser class.
                      3: 
1.75      paf         4:        Copyright (c) 2001-2005 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.85    ! misha       8: static const char * const IDENT_HASH_C="$Date: 2008-05-15 09:33:37 $";
1.1       paf         9: 
                     10: #include "classes.h"
1.57      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.57      paf        23: class MHash: public Methoded {
1.2       parser     24: public: // VStateless_class
1.74      paf        25:        Value* create_new_value(Pool&, HashStringValue&) { return new VHash(); }
1.2       parser     26: 
1.1       paf        27: public:
1.57      paf        28:        MHash();
1.1       paf        29: public: // Methoded
1.2       parser     30:        bool used_directly() { return true; }
1.1       paf        31: };
                     32: 
1.57      paf        33: // global variable
                     34: 
                     35: DECLARE_CLASS_VAR(hash, new MHash, 0);
                     36: 
1.77      paf        37: // externs
                     38: 
                     39: extern String cycle_data_name;
                     40: 
1.1       paf        41: // methods
                     42: 
1.11      parser     43: #ifndef DOXYGEN
1.52      paf        44: class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers {
1.57      paf        45:        const String& statement_string; const char* statement_cstr;
                     46:        bool distinct;
                     47:        HashStringValue& rows_hash;
                     48:        HashStringValue* row_hash;
                     49:        int column_index;
                     50:        ArrayString columns;
1.67      paf        51:        bool only_one_column;
                     52:        static VBool only_one_column_value;
1.11      parser     53: public:
1.57      paf        54:        Hash_sql_event_handlers(
                     55:                const String& astatement_string, const char* astatement_cstr,
1.49      paf        56:                bool adistinct,
1.57      paf        57:                HashStringValue& arows_hash): 
                     58:                statement_string(astatement_string), statement_cstr(astatement_cstr),
1.49      paf        59:                distinct(adistinct),
1.11      parser     60:                rows_hash(arows_hash),
1.57      paf        61:                row_hash(0),
1.67      paf        62:                column_index(0),
                     63:                only_one_column(false) {
1.11      parser     64:        }
1.57      paf        65:        bool add_column(SQL_Error& error, const char* str, size_t length) {
1.53      paf        66:                try {
1.57      paf        67:                        columns+=new String(str, length, true);
1.53      paf        68:                        return false;
                     69:                } catch(...) {
                     70:                        error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
                     71:                        return true;
                     72:                }
1.11      parser     73:        }
1.53      paf        74:        bool before_rows(SQL_Error& error) { 
1.67      paf        75:                if(columns.count()<1) {
1.79      misha      76:                        error=SQL_Error(PARSER_RUNTIME, "no columns");
1.53      paf        77:                        return true;
                     78:                }
1.67      paf        79:                only_one_column=columns.count()==1;
1.53      paf        80: 
                     81:                return false;
1.11      parser     82:        }
1.53      paf        83:        bool add_row(SQL_Error& /*error*/) {
1.11      parser     84:                column_index=0;
1.53      paf        85:                return false;
1.11      parser     86:        }
1.57      paf        87:        bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) {
1.53      paf        88:                try {
1.57      paf        89:                        String& cell=*new String;
                     90:                        if(length)
                     91:                                cell.append_know_length(ptr, length, String::L_TAINTED);
1.69      paf        92:                        bool duplicate=false;
                     93:                        if(only_one_column) {
                     94:                                duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value);  // put. existed?
                     95:                        } else if(column_index==0) {
1.57      paf        96:                                VHash* row_vhash=new VHash;
                     97:                                row_hash=&row_vhash->hash();
1.69      paf        98:                                duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?
1.53      paf        99:                        } else
1.57      paf       100:                                row_hash->put(*columns[column_index], new VString(cell));
1.69      paf       101: 
                    102:                        if(duplicate & !distinct) {
1.79      misha     103:                                error=SQL_Error(PARSER_RUNTIME, "duplicate key");
1.69      paf       104:                                return true;
                    105:                        }
                    106: 
1.53      paf       107:                        column_index++;
                    108:                        return false;
                    109:                } catch(...) {
                    110:                        error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell");
                    111:                        return true;
                    112:                }
1.11      parser    113:        }
                    114: 
                    115: };
1.67      paf       116: VBool Hash_sql_event_handlers::only_one_column_value(true);
                    117: 
1.11      parser    118: #endif
                    119: 
1.57      paf       120: static void _create_or_add(Request& r, MethodParams& params) {
                    121:        if(params.count()) {
1.66      paf       122:                Value& vsrc=params.as_no_junction(0, "param must be hash");
                    123:                if(HashStringValue* src=vsrc.get_hash()) {
1.72      paf       124:                        VHash& self=GET_SELF(r, VHash);
                    125:                        HashStringValue* self_hash=&(self.hash());
                    126:                        if(src==self_hash) // same: doing nothing
1.66      paf       127:                                return;
1.77      paf       128:                        src->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash);
1.72      paf       129: 
                    130:                        if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false)))
1.78      paf       131:                        {
                    132:                                if(Value* vdefault=vhash_src->get_default())
                    133:                                {
                    134:                                        if(vdefault->is_defined())
                    135:                                        {
                    136:                                                self.set_default(vdefault);
                    137:                                        }
                    138:                                }
                    139:                        }
1.66      paf       140:                }
1.20      parser    141:        }
                    142: }
1.22      parser    143: 
1.57      paf       144: static void _sub(Request& r, MethodParams& params) {
1.66      paf       145:        Value& vsrc=params.as_no_junction(0, "param must be hash");
                    146:        if(HashStringValue* src=vsrc.get_hash()) {
                    147:                HashStringValue* self=&(GET_SELF(r, VHash).hash());
                    148:                if(src==self) { // same: clearing
                    149:                        self->clear();
                    150:                        return;
                    151:                }
1.77      paf       152:                src->for_each<HashStringValue*>(remove_key_from, self);
1.66      paf       153:        }
1.57      paf       154: }
                    155: 
                    156: static void copy_all_dontoverwrite_to(
                    157:                                                                  HashStringValue::key_type key, 
                    158:                                                                  HashStringValue::value_type value, 
                    159:                                                                  HashStringValue* dest) {
                    160:        dest->put_dont_replace(key, value);
1.22      parser    161: }
1.57      paf       162: static void _union(Request& r, MethodParams& params) {
1.22      parser    163:        // dest = copy of self
1.57      paf       164:        Value& result=*new VHash(GET_SELF(r, VHash).hash());
1.22      parser    165:        // dest += b
1.66      paf       166:        Value& vsrc=params.as_no_junction(0, "param must be hash");
                    167:        if(HashStringValue* src=vsrc.get_hash())
1.77      paf       168:                src->for_each<HashStringValue*>(copy_all_dontoverwrite_to, result.get_hash());
1.22      parser    169: 
                    170:        // return result
                    171:        r.write_no_lang(result);
                    172: }
                    173: 
                    174: #ifndef DOXYGEN
                    175: struct Copy_intersection_to_info {
1.57      paf       176:        HashStringValue* b;
                    177:        HashStringValue* dest;
1.22      parser    178: };
                    179: #endif
1.57      paf       180: static void copy_intersection_to(
                    181:                                                                 HashStringValue::key_type key, 
                    182:                                                                 HashStringValue::value_type value, 
                    183:                                                                 Copy_intersection_to_info *info) {
                    184:        if(info->b->get(key))
                    185:                info->dest->put_dont_replace(key, value);
1.22      parser    186: }
1.57      paf       187: static void _intersection(Request& r, MethodParams& params) {
                    188:        Value& result=*new VHash;
1.22      parser    189:        // dest += b
1.57      paf       190:        Value& vb=params.as_no_junction(0, "param must be hash");
                    191:        if(HashStringValue* b=vb.get_hash()) {
                    192:                Copy_intersection_to_info info={b, result.get_hash()};
1.77      paf       193:                GET_SELF(r, VHash).hash().for_each<Copy_intersection_to_info*>(copy_intersection_to, &info);
1.22      parser    194:        }
                    195: 
                    196:        // return result
1.57      paf       197:        r.write_no_lang(result);
1.22      parser    198: }
                    199: 
1.57      paf       200: static bool intersects(
                    201:                                           HashStringValue::key_type key, 
                    202:                                           HashStringValue::value_type /*value*/, 
                    203:                                           HashStringValue* b) {
                    204:        return b->get(key)!=0;
1.22      parser    205: }
                    206: 
1.57      paf       207: static void _intersects(Request& r, MethodParams& params) {
                    208:        bool result=false;
                    209: 
                    210:        Value& vb=params.as_no_junction(0, "param must be hash");
                    211:        if(HashStringValue* b=vb.get_hash())
1.77      paf       212:                result=GET_SELF(r, VHash).hash().first_that<HashStringValue*>(intersects, b)!=0;
1.22      parser    213: 
                    214:        // return result
1.57      paf       215:        r.write_no_lang(*new VBool(result));
1.22      parser    216: }
                    217: 
                    218: 
1.71      paf       219: extern String sql_bind_name;
1.57      paf       220: extern String sql_limit_name;
                    221: extern String sql_offset_name;
                    222: extern String sql_default_name;
                    223: extern String sql_distinct_name;
1.81      misha     224: extern String sql_value_type_name;
                    225: extern Table2hash_value_type get_value_type(Value& vvalue_type);
1.71      paf       226: extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);
                    227: extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
                    228: 
1.57      paf       229: static void _sql(Request& r, MethodParams& params) {
                    230:        Value& statement=params.as_junction(0, "statement must be code");
1.2       parser    231: 
1.71      paf       232:        HashStringValue* bind=0;
1.2       parser    233:        ulong limit=0;
1.33      paf       234:        ulong offset=0;
1.49      paf       235:        bool distinct=false;
1.81      misha     236:        Table2hash_value_type value_type=C_HASH;
1.57      paf       237:        if(params.count()>1) {
                    238:                Value& voptions=params.as_no_junction(1, "options must be hash, not code");
1.76      paf       239:                if(voptions.is_defined() && !voptions.is_string())
1.57      paf       240:                        if(HashStringValue* options=voptions.get_hash()) {
1.49      paf       241:                                int valid_options=0;
1.71      paf       242:                                if(Value* vbind=options->get(sql_bind_name)) {
                    243:                                        valid_options++;
                    244:                                        bind=vbind->get_hash();
                    245:                                }
1.57      paf       246:                                if(Value* vlimit=options->get(sql_limit_name)) {
1.49      paf       247:                                        valid_options++;
1.37      paf       248:                                        limit=(ulong)r.process_to_value(*vlimit).as_double();
1.49      paf       249:                                }
1.57      paf       250:                                if(Value* voffset=options->get(sql_offset_name)) {
1.49      paf       251:                                        valid_options++;
1.37      paf       252:                                        offset=(ulong)r.process_to_value(*voffset).as_double();
1.49      paf       253:                                }
1.57      paf       254:                                if(Value* vdistinct=options->get(sql_distinct_name)) {
1.49      paf       255:                                        valid_options++;
                    256:                                        distinct=r.process_to_value(*vdistinct).as_bool();
                    257:                                }
1.81      misha     258:                                if(Value* vvalue_type=options->get(sql_value_type_name)) {
                    259:                                        valid_options++;
                    260:                                        value_type=get_value_type(r.process_to_value(*vvalue_type));
                    261:                                }
1.57      paf       262:                                if(valid_options!=options->count())
1.79      misha     263:                                        throw Exception(PARSER_RUNTIME,
1.57      paf       264:                                                0,
1.49      paf       265:                                                "called with invalid option");
1.33      paf       266:                        } else
1.79      misha     267:                                throw Exception(PARSER_RUNTIME,
1.57      paf       268:                                        0,
1.33      paf       269:                                        "options must be hash");
1.2       parser    270:        }
                    271: 
1.71      paf       272:        SQL_Driver::Placeholder* placeholders=0;
                    273:        uint placeholders_count=0;
                    274:        if(bind)
                    275:                placeholders_count=marshal_binds(*bind, placeholders);
                    276: 
1.57      paf       277:        Temp_lang temp_lang(r, String::L_SQL);
1.37      paf       278:        const String& statement_string=r.process_to_string(statement);
1.57      paf       279:        const char* statement_cstr=
                    280:                statement_string.cstr(String::L_UNSPECIFIED, r.connection());
                    281:        HashStringValue& hash=GET_SELF(r, VHash).hash();
1.11      parser    282:        hash.clear();   
1.57      paf       283:        Hash_sql_event_handlers handlers(
1.49      paf       284:                statement_string, statement_cstr, 
                    285:                distinct,
                    286:                hash);
1.57      paf       287:        r.connection()->query(
1.70      paf       288:                statement_cstr, 
1.73      paf       289:                placeholders_count, placeholders,
1.70      paf       290:                offset, limit,
1.45      paf       291:                handlers,
                    292:                statement_string);
1.71      paf       293: 
                    294:        if(bind)
                    295:                unmarshal_bind_updates(*bind, placeholders_count, placeholders);
1.2       parser    296: }
                    297: 
1.57      paf       298: static void keys_collector(
                    299:                           HashStringValue::key_type key, 
1.61      paf       300:                           HashStringValue::value_type, 
1.57      paf       301:                           Table *table) {
                    302:        Table::element_type row(new ArrayString);
                    303:        *row+=new String(key, String::L_TAINTED);
                    304:        *table+=row;
1.9       parser    305: }
1.68      paf       306: static void _keys(Request& r, MethodParams& params) {
                    307:        const String* keys_column_name;
                    308:        if(params.count()>0)
1.81      misha     309:                keys_column_name=&params.as_string(0, COLUMN_NAME_MUST_BE_STRING);
1.68      paf       310:        else 
                    311:                keys_column_name=new String("key");
                    312: 
1.57      paf       313:        Table::columns_type columns(new ArrayString);
1.68      paf       314:        *columns+=keys_column_name;
1.57      paf       315:        Table* table=new Table(columns);
1.9       parser    316: 
1.77      paf       317:        GET_SELF(r, VHash).hash().for_each<Table*>(keys_collector, table);
1.9       parser    318: 
1.57      paf       319:        r.write_no_lang(*new VTable(table));
1.9       parser    320: }
                    321: 
1.57      paf       322: static void _count(Request& r, MethodParams&) {
                    323:        r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count()));
1.16      parser    324: }
                    325: 
1.57      paf       326: static void _delete(Request& r, MethodParams& params) {
1.25      paf       327: 
1.57      paf       328:        GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));
1.25      paf       329: }
                    330: 
1.82      misha     331: static void _contains(Request& r, MethodParams& params) {
1.80      misha     332: 
1.82      misha     333:        bool result = GET_SELF(r, VHash).hash().contains(params.as_string(0, "key must be string"));
1.80      misha     334:        r.write_no_lang(*new VBool(result));
                    335: }
                    336: 
1.26      paf       337: #ifndef DOXYGEN
1.59      paf       338: struct Foreach_info {
1.26      paf       339:        Request *r;
                    340:        const String* key_var_name;
                    341:        const String* value_var_name;
1.57      paf       342:        Value* body_code;
                    343:        Value* delim_maybe_code;
1.26      paf       344: 
1.84      misha     345:        Value* var_context;
1.57      paf       346:        VString* vkey;
1.26      paf       347:        bool need_delim;
                    348: };
                    349: #endif
1.77      paf       350: static bool one_foreach_cycle(                                            
1.57      paf       351:                              HashStringValue::key_type akey, 
                    352:                              HashStringValue::value_type avalue, 
1.59      paf       353:                              Foreach_info *info) {
1.84      misha     354:        Value& var_context=*info->var_context;
1.85    ! misha     355:        if(info->key_var_name){
        !           356:                info->vkey->set_string(*new String(akey, String::L_TAINTED));
        !           357:                var_context.put_element(var_context, *info->key_var_name, info->vkey, false);
        !           358:        }
        !           359:        if(info->value_var_name)
        !           360:                var_context.put_element(var_context, *info->value_var_name, avalue, false);
1.57      paf       361: 
                    362:        StringOrValue sv_processed=info->r->process(*info->body_code);
1.77      paf       363:        Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
1.83      misha     364: 
1.57      paf       365:        const String* s_processed=sv_processed.get_string();
                    366:        if(info->delim_maybe_code && s_processed && s_processed->length()) { // delimiter set and we have body
                    367:                if(info->need_delim) // need delim & iteration produced string?
                    368:                        info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));
1.83      misha     369:                else
                    370:                        info->need_delim=true;
1.26      paf       371:        }
1.83      misha     372: 
1.57      paf       373:        info->r->write_pass_lang(sv_processed);
1.77      paf       374: 
                    375:        return lskip==Request::SKIP_BREAK;
1.26      paf       376: }
1.57      paf       377: static void _foreach(Request& r, MethodParams& params) {
1.77      paf       378:        Temp_hash_value<const String::Body, void*> 
                    379:                cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r);
                    380: 
1.85    ! misha     381:        const String& key_var_name=params.as_string(0, "key-var name must be string");
        !           382:        const String& value_var_name=params.as_string(1, "value-var name must be string");
        !           383: 
1.63      paf       384:        Foreach_info info={
                    385:                &r,
1.85    ! misha     386:                key_var_name.is_empty()? 0 : &key_var_name,
        !           387:                value_var_name.is_empty()? 0 : &value_var_name,
1.63      paf       388:                &params.as_junction(2, "body must be code"),
1.84      misha     389:                /*delimiter*/params.count()>3?params.get(3):0,
                    390:                /*var_context*/r.get_method_frame()->caller(),
1.64      paf       391:                /*vkey=*/new VString,
1.85    ! misha     392:                false
1.63      paf       393:        };
1.57      paf       394: 
                    395:        VHash& self=GET_SELF(r, VHash);
                    396:        HashStringValue& hash=self.hash();
1.28      paf       397:        VHash_lock lock(self);
1.77      paf       398:        hash.first_that<Foreach_info*>(one_foreach_cycle, &info);
1.26      paf       399: }
                    400: 
1.1       paf       401: // constructor
                    402: 
1.57      paf       403: MHash::MHash(): Methoded("hash") 
1.39      paf       404: {
1.21      parser    405:        // ^hash::create[[copy_from]]
1.57      paf       406:        add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
1.22      parser    407:        // ^hash.add[add_from]
1.57      paf       408:        add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
1.22      parser    409:        // ^hash.sub[sub_from]
                    410:        add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
                    411:        // ^a.union[b] = hash
                    412:        add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
                    413:        // ^a.intersection[b] = hash
                    414:        add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
                    415:        // ^a.intersects[b] = bool
                    416:        add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
1.25      paf       417: 
                    418:        // ^a.delete[key]
                    419:        add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
1.2       parser    420: 
1.82      misha     421:        // ^a.contains[key]
                    422:        add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1);
                    423:        // backward
                    424:        add_native_method("contain", Method::CT_DYNAMIC, _contains, 1, 1);
1.80      misha     425: 
                    426:        // ^hash::sql[query][options hash]
1.33      paf       427:        add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.2       parser    428: 
1.68      paf       429:        // ^hash._keys[[column name]]
                    430:        add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1);    
1.16      parser    431: 
                    432:        // ^hash._count[]
                    433:        add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);  
1.26      paf       434: 
                    435:        // ^hash.foreach[key;value]{code}[delim]
                    436:        add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
1.1       paf       437: }

E-mail: