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