Annotation of parser3/src/classes/hash.C, revision 1.110
1.1 paf 1: /** @file
2: Parser: @b hash parser class.
3:
1.89 misha 4: Copyright (c) 2001-2009 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.110 ! misha 8: static const char * const IDENT_HASH_C="$Date: 2010-11-26 06:40:01 $";
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.103 misha 25: Value* create_new_value(Pool&) { return new VHash(); }
1.2 parser 26:
1.1 paf 27: public:
1.57 paf 28: MHash();
1.1 paf 29: };
30:
1.57 paf 31: // global variable
32:
33: DECLARE_CLASS_VAR(hash, new MHash, 0);
34:
1.1 paf 35: // methods
36:
1.11 parser 37: #ifndef DOXYGEN
1.52 paf 38: class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers {
1.57 paf 39: const String& statement_string; const char* statement_cstr;
40: bool distinct;
41: HashStringValue& rows_hash;
1.86 misha 42: Value* row_value;
1.57 paf 43: int column_index;
1.86 misha 44: ArrayString& columns;
45: bool one_bool_column;
1.67 paf 46: static VBool only_one_column_value;
1.86 misha 47: Table2hash_value_type value_type;
48: int columns_count;
49: public:
50: Table* empty;
1.11 parser 51: public:
1.57 paf 52: Hash_sql_event_handlers(
1.86 misha 53: const String& astatement_string,
54: const char* astatement_cstr,
1.49 paf 55: bool adistinct,
1.86 misha 56: HashStringValue& arows_hash,
57: Table2hash_value_type avalue_type)
58: :
59: statement_string(astatement_string),
60: statement_cstr(astatement_cstr),
1.49 paf 61: distinct(adistinct),
1.11 parser 62: rows_hash(arows_hash),
1.86 misha 63: value_type(avalue_type),
64: row_value(0),
1.67 paf 65: column_index(0),
1.86 misha 66: one_bool_column(false),
67: columns(*new ArrayString),
68: empty(0) {
1.11 parser 69: }
1.86 misha 70:
1.109 misha 71: bool add_column(SQL_Error& error, const char* str, size_t ) {
1.53 paf 72: try {
1.107 moko 73: columns+=new String(str, String::L_TAINTED /* no length as 0x00 can be inside */);
1.53 paf 74: return false;
75: } catch(...) {
76: error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
77: return true;
78: }
1.11 parser 79: }
1.86 misha 80:
1.53 paf 81: bool before_rows(SQL_Error& error) {
1.67 paf 82: if(columns.count()<1) {
1.79 misha 83: error=SQL_Error(PARSER_RUNTIME, "no columns");
1.53 paf 84: return true;
85: }
1.86 misha 86: switch(value_type){
87: case C_STRING: {
88: if(columns.count()>2){
89: error=SQL_Error(PARSER_RUNTIME, "only 2 columns allowed for $.type[string].");
90: return true;
91: }
92: }
93: case C_TABLE: {
94: // create empty table which we'll copy later
95: empty=new Table(&columns);
96: columns_count=columns.count();
97: }
98: case C_HASH: {
99: one_bool_column=columns.count()==1;
100: }
101: }
1.53 paf 102: return false;
1.11 parser 103: }
1.86 misha 104:
1.53 paf 105: bool add_row(SQL_Error& /*error*/) {
1.11 parser 106: column_index=0;
1.53 paf 107: return false;
1.11 parser 108: }
1.86 misha 109:
1.109 misha 110: bool add_row_cell(SQL_Error& error, const char *ptr, size_t ) {
1.53 paf 111: try {
1.107 moko 112: String& cell=*new String(ptr, String::L_TAINTED /* no length as 0x00 can be inside */);
1.86 misha 113:
1.69 paf 114: bool duplicate=false;
1.86 misha 115: if(one_bool_column) {
1.69 paf 116: duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value); // put. existed?
117: } else if(column_index==0) {
1.86 misha 118: switch(value_type){
119: case C_HASH: {
120: VHash* row_vhash=new VHash;
121: row_value=row_vhash;
122: duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?
123: break;
124: }
125: case C_STRING: {
126: VString* row_vstring=new VString();
127: row_value=row_vstring;
128: duplicate=rows_hash.put_dont_replace(cell, row_vstring); // put. existed?
129: break;
130: }
131: case C_TABLE: {
132: VTable* vtable=(VTable*)rows_hash.get(cell);
133: Table* table;
134:
135: if(vtable) { // table with this key exist?
136: if(!distinct) {
137: duplicate=true;
138: break;
139: }
140: table=vtable->get_table();
141: } else {
142: // no? creating table of same structure as source
143: Table::Action_options table_options(0, 0);
144: table=new Table(*empty, table_options/*no rows, just structure*/);
145: vtable=new VTable(table);
146: rows_hash.put(cell, vtable); // put
147: }
148: ArrayString* row=new ArrayString(columns_count);
149: row_value=(Value*)row;
150: *row+=&cell;
151: *table+=row;
152: break;
153: }
154: }
155: } else {
156: switch(value_type) {
157: case C_HASH: {
158: row_value->get_hash()->put(*columns[column_index], new VString(cell));
159: break;
160: }
161: case C_STRING: {
162: VString* row_string=(VString*)row_value;
163: row_string->set_string(cell);
164: break;
165: }
166: case C_TABLE: {
167: ArrayString* row=(ArrayString*)row_value;
168: *row+=&cell;
169: break;
170: }
171: }
172: }
1.69 paf 173:
174: if(duplicate & !distinct) {
1.79 misha 175: error=SQL_Error(PARSER_RUNTIME, "duplicate key");
1.69 paf 176: return true;
177: }
178:
1.53 paf 179: column_index++;
180: return false;
181: } catch(...) {
182: error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell");
183: return true;
184: }
1.11 parser 185: }
186:
187: };
1.67 paf 188: VBool Hash_sql_event_handlers::only_one_column_value(true);
189:
1.11 parser 190: #endif
191:
1.57 paf 192: static void _create_or_add(Request& r, MethodParams& params) {
193: if(params.count()) {
1.66 paf 194: Value& vsrc=params.as_no_junction(0, "param must be hash");
195: if(HashStringValue* src=vsrc.get_hash()) {
1.72 paf 196: VHash& self=GET_SELF(r, VHash);
197: HashStringValue* self_hash=&(self.hash());
198: if(src==self_hash) // same: doing nothing
1.66 paf 199: return;
1.77 paf 200: src->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash);
1.72 paf 201:
1.103 misha 202: if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE)))
1.78 paf 203: {
204: if(Value* vdefault=vhash_src->get_default())
205: {
206: if(vdefault->is_defined())
207: {
208: self.set_default(vdefault);
209: }
210: }
211: }
1.66 paf 212: }
1.20 parser 213: }
214: }
1.22 parser 215:
1.57 paf 216: static void _sub(Request& r, MethodParams& params) {
1.66 paf 217: Value& vsrc=params.as_no_junction(0, "param must be hash");
218: if(HashStringValue* src=vsrc.get_hash()) {
219: HashStringValue* self=&(GET_SELF(r, VHash).hash());
220: if(src==self) { // same: clearing
221: self->clear();
222: return;
223: }
1.77 paf 224: src->for_each<HashStringValue*>(remove_key_from, self);
1.66 paf 225: }
1.57 paf 226: }
227:
228: static void copy_all_dontoverwrite_to(
1.100 misha 229: HashStringValue::key_type key,
230: HashStringValue::value_type value,
231: HashStringValue* dest) {
1.57 paf 232: dest->put_dont_replace(key, value);
1.22 parser 233: }
1.57 paf 234: static void _union(Request& r, MethodParams& params) {
1.22 parser 235: // dest = copy of self
1.57 paf 236: Value& result=*new VHash(GET_SELF(r, VHash).hash());
1.22 parser 237: // dest += b
1.66 paf 238: Value& vsrc=params.as_no_junction(0, "param must be hash");
239: if(HashStringValue* src=vsrc.get_hash())
1.77 paf 240: src->for_each<HashStringValue*>(copy_all_dontoverwrite_to, result.get_hash());
1.22 parser 241:
242: // return result
243: r.write_no_lang(result);
244: }
245:
246: #ifndef DOXYGEN
247: struct Copy_intersection_to_info {
1.57 paf 248: HashStringValue* b;
249: HashStringValue* dest;
1.22 parser 250: };
251: #endif
1.57 paf 252: static void copy_intersection_to(
1.100 misha 253: HashStringValue::key_type key,
254: HashStringValue::value_type value,
255: Copy_intersection_to_info *info) {
1.57 paf 256: if(info->b->get(key))
257: info->dest->put_dont_replace(key, value);
1.22 parser 258: }
1.57 paf 259: static void _intersection(Request& r, MethodParams& params) {
260: Value& result=*new VHash;
1.22 parser 261: // dest += b
1.57 paf 262: Value& vb=params.as_no_junction(0, "param must be hash");
263: if(HashStringValue* b=vb.get_hash()) {
264: Copy_intersection_to_info info={b, result.get_hash()};
1.77 paf 265: GET_SELF(r, VHash).hash().for_each<Copy_intersection_to_info*>(copy_intersection_to, &info);
1.22 parser 266: }
267:
268: // return result
1.57 paf 269: r.write_no_lang(result);
1.22 parser 270: }
271:
1.57 paf 272: static bool intersects(
1.100 misha 273: HashStringValue::key_type key,
274: HashStringValue::value_type /*value*/,
275: HashStringValue* b) {
1.57 paf 276: return b->get(key)!=0;
1.22 parser 277: }
278:
1.57 paf 279: static void _intersects(Request& r, MethodParams& params) {
280: bool result=false;
281:
282: Value& vb=params.as_no_junction(0, "param must be hash");
283: if(HashStringValue* b=vb.get_hash())
1.77 paf 284: result=GET_SELF(r, VHash).hash().first_that<HashStringValue*>(intersects, b)!=0;
1.22 parser 285:
286: // return result
1.89 misha 287: r.write_no_lang(VBool::get(result));
1.22 parser 288: }
289:
290:
1.71 paf 291: extern String sql_bind_name;
1.57 paf 292: extern String sql_limit_name;
293: extern String sql_offset_name;
294: extern String sql_default_name;
295: extern String sql_distinct_name;
1.81 misha 296: extern String sql_value_type_name;
297: extern Table2hash_value_type get_value_type(Value& vvalue_type);
1.71 paf 298: extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);
299: extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
300:
1.57 paf 301: static void _sql(Request& r, MethodParams& params) {
302: Value& statement=params.as_junction(0, "statement must be code");
1.2 parser 303:
1.71 paf 304: HashStringValue* bind=0;
1.87 misha 305: ulong limit=SQL_NO_LIMIT;
1.33 paf 306: ulong offset=0;
1.49 paf 307: bool distinct=false;
1.81 misha 308: Table2hash_value_type value_type=C_HASH;
1.110 ! misha 309: if(params.count()>1)
! 310: if(HashStringValue* options=params.as_hash(1)) {
! 311: int valid_options=0;
! 312: if(Value* vbind=options->get(sql_bind_name)) {
! 313: valid_options++;
! 314: bind=vbind->get_hash();
! 315: }
! 316: if(Value* vlimit=options->get(sql_limit_name)) {
! 317: valid_options++;
! 318: limit=(ulong)r.process_to_value(*vlimit).as_double();
! 319: }
! 320: if(Value* voffset=options->get(sql_offset_name)) {
! 321: valid_options++;
! 322: offset=(ulong)r.process_to_value(*voffset).as_double();
! 323: }
! 324: if(Value* vdistinct=options->get(sql_distinct_name)) {
! 325: valid_options++;
! 326: distinct=r.process_to_value(*vdistinct).as_bool();
! 327: }
! 328: if(Value* vvalue_type=options->get(sql_value_type_name)) {
! 329: valid_options++;
! 330: value_type=get_value_type(r.process_to_value(*vvalue_type));
! 331: }
! 332: if(valid_options!=options->count())
! 333: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
! 334: }
1.2 parser 335:
1.71 paf 336: SQL_Driver::Placeholder* placeholders=0;
337: uint placeholders_count=0;
338: if(bind)
339: placeholders_count=marshal_binds(*bind, placeholders);
340:
1.57 paf 341: Temp_lang temp_lang(r, String::L_SQL);
1.37 paf 342: const String& statement_string=r.process_to_string(statement);
1.99 misha 343: const char* statement_cstr=statement_string.untaint_cstr(r.flang, r.connection());
1.97 misha 344:
1.57 paf 345: HashStringValue& hash=GET_SELF(r, VHash).hash();
1.11 parser 346: hash.clear();
1.57 paf 347: Hash_sql_event_handlers handlers(
1.49 paf 348: statement_string, statement_cstr,
349: distinct,
1.86 misha 350: hash,
351: value_type);
1.87 misha 352:
1.57 paf 353: r.connection()->query(
1.70 paf 354: statement_cstr,
1.73 paf 355: placeholders_count, placeholders,
1.70 paf 356: offset, limit,
1.45 paf 357: handlers,
358: statement_string);
1.71 paf 359:
360: if(bind)
361: unmarshal_bind_updates(*bind, placeholders_count, placeholders);
1.2 parser 362: }
363:
1.57 paf 364: static void keys_collector(
1.100 misha 365: HashStringValue::key_type key,
366: HashStringValue::value_type,
367: Table *table) {
368: Table::element_type row(new ArrayString(1));
1.57 paf 369: *row+=new String(key, String::L_TAINTED);
370: *table+=row;
1.9 parser 371: }
1.68 paf 372: static void _keys(Request& r, MethodParams& params) {
373: const String* keys_column_name;
374: if(params.count()>0)
1.81 misha 375: keys_column_name=¶ms.as_string(0, COLUMN_NAME_MUST_BE_STRING);
1.68 paf 376: else
377: keys_column_name=new String("key");
378:
1.101 misha 379: Table::columns_type columns(new ArrayString(1));
1.68 paf 380: *columns+=keys_column_name;
1.57 paf 381: Table* table=new Table(columns);
1.9 parser 382:
1.77 paf 383: GET_SELF(r, VHash).hash().for_each<Table*>(keys_collector, table);
1.9 parser 384:
1.57 paf 385: r.write_no_lang(*new VTable(table));
1.9 parser 386: }
387:
1.57 paf 388: static void _count(Request& r, MethodParams&) {
389: r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count()));
1.16 parser 390: }
391:
1.57 paf 392: static void _delete(Request& r, MethodParams& params) {
1.25 paf 393:
1.57 paf 394: GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));
1.25 paf 395: }
396:
1.82 misha 397: static void _contains(Request& r, MethodParams& params) {
1.89 misha 398: bool result=GET_SELF(r, VHash).hash().contains(params.as_string(0, "key must be string"));
399: r.write_no_lang(VBool::get(result));
1.80 misha 400: }
401:
1.26 paf 402: #ifndef DOXYGEN
1.59 paf 403: struct Foreach_info {
1.26 paf 404: Request *r;
405: const String* key_var_name;
406: const String* value_var_name;
1.57 paf 407: Value* body_code;
408: Value* delim_maybe_code;
1.26 paf 409:
1.84 misha 410: Value* var_context;
1.26 paf 411: bool need_delim;
412: };
413: #endif
1.90 misha 414: static bool one_foreach_cycle(
415: HashStringValue::key_type akey,
416: HashStringValue::value_type avalue,
417: Foreach_info *info) {
1.84 misha 418: Value& var_context=*info->var_context;
1.85 misha 419: if(info->key_var_name){
1.96 misha 420: VString* vkey=new VString(*new String(akey, String::L_TAINTED));
1.103 misha 421: var_context.put_element(*info->key_var_name, vkey, false);
1.85 misha 422: }
423: if(info->value_var_name)
1.103 misha 424: var_context.put_element(*info->value_var_name, avalue, false);
1.57 paf 425:
1.91 misha 426: if(info->delim_maybe_code){ // delimiter set
1.90 misha 427: StringOrValue sv_processed=info->r->process(*info->body_code);
428: Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
429:
430: const String* s_processed=sv_processed.get_string();
1.91 misha 431: if(s_processed && !s_processed->is_empty()) { // we have body
1.90 misha 432: if(info->need_delim) // need delim & iteration produced string?
433: info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));
434: else
435: info->need_delim=true;
436: }
437: info->r->write_pass_lang(sv_processed);
438: return lskip==Request::SKIP_BREAK;
439: } else {
440: info->r->process_write(*info->body_code);
441: Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
442: return lskip==Request::SKIP_BREAK;
1.26 paf 443: }
444: }
1.57 paf 445: static void _foreach(Request& r, MethodParams& params) {
1.102 misha 446: InCycle temp(r);
1.77 paf 447:
1.85 misha 448: const String& key_var_name=params.as_string(0, "key-var name must be string");
449: const String& value_var_name=params.as_string(1, "value-var name must be string");
450:
1.63 paf 451: Foreach_info info={
452: &r,
1.85 misha 453: key_var_name.is_empty()? 0 : &key_var_name,
454: value_var_name.is_empty()? 0 : &value_var_name,
1.63 paf 455: ¶ms.as_junction(2, "body must be code"),
1.84 misha 456: /*delimiter*/params.count()>3?params.get(3):0,
457: /*var_context*/r.get_method_frame()->caller(),
1.85 misha 458: false
1.63 paf 459: };
1.57 paf 460:
461: VHash& self=GET_SELF(r, VHash);
462: HashStringValue& hash=self.hash();
1.28 paf 463: VHash_lock lock(self);
1.77 paf 464: hash.first_that<Foreach_info*>(one_foreach_cycle, &info);
1.26 paf 465: }
466:
1.105 misha 467: static void _at(Request& r, MethodParams& params) {
468: HashStringValue& hash=GET_SELF(r, VHash).hash();
469: size_t count=hash.count();
470:
471: int pos=0;
472:
473: Value& vwhence=*params.get(0);
474: if(vwhence.is_string()){
475: const String& swhence=*vwhence.get_string();
476: if(swhence == "last")
477: pos=count-1;
478: else if(swhence != "first")
479: throw Exception(PARSER_RUNTIME,
480: &swhence,
481: "whence must be 'first', 'last' or expression");
482: } else {
483: pos=r.process_to_value(vwhence).as_int();
484: if(pos < 0)
485: pos+=count;
486: }
487:
1.106 misha 488: if(count && pos >= 0 && (size_t)pos < count){
1.105 misha 489: if(pos == 0)
490: r.write_assign_lang(*hash.first_value());
1.106 misha 491: else if((size_t)pos == count-1)
1.105 misha 492: r.write_assign_lang(*hash.last_value());
493: else
494: for(HashStringValue::Iterator i(hash); i; i.next(), pos-- )
495: if(!pos){
496: r.write_assign_lang(*i.value());
497: break;
498: }
499: }
500:
501: }
502:
1.1 paf 503: // constructor
504:
1.57 paf 505: MHash::MHash(): Methoded("hash")
1.39 paf 506: {
1.21 parser 507: // ^hash::create[[copy_from]]
1.57 paf 508: add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
1.22 parser 509: // ^hash.add[add_from]
1.57 paf 510: add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
1.22 parser 511: // ^hash.sub[sub_from]
512: add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
513: // ^a.union[b] = hash
514: add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
515: // ^a.intersection[b] = hash
516: add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
517: // ^a.intersects[b] = bool
518: add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
1.25 paf 519:
520: // ^a.delete[key]
521: add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
1.2 parser 522:
1.82 misha 523: // ^a.contains[key]
524: add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1);
525: // backward
526: add_native_method("contain", Method::CT_DYNAMIC, _contains, 1, 1);
1.80 misha 527:
528: // ^hash::sql[query][options hash]
1.33 paf 529: add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.2 parser 530:
1.68 paf 531: // ^hash._keys[[column name]]
532: add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1);
1.16 parser 533:
534: // ^hash._count[]
535: add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);
1.26 paf 536:
537: // ^hash.foreach[key;value]{code}[delim]
538: add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
1.105 misha 539:
540: // ^hash._at[first|last]
541: // ^hash._at([-]offset)
542: add_native_method("_at", Method::CT_DYNAMIC, _at, 1, 1);
1.1 paf 543: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to hash.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / classes