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

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

E-mail: