Annotation of parser3/src/classes/hash.C, revision 1.82
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.82 ! misha 8: static const char * const IDENT_HASH_C="$Date: 2007/09/17 15:14:31 $";
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=¶ms.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.57 paf 345: VString* vkey;
1.26 paf 346: bool need_delim;
347: };
348: #endif
1.77 paf 349: static bool one_foreach_cycle(
1.57 paf 350: HashStringValue::key_type akey,
351: HashStringValue::value_type avalue,
1.59 paf 352: Foreach_info *info) {
1.57 paf 353: info->vkey->set_string(*new String(akey, String::L_TAINTED));
354: Value& ncontext=*info->r->get_method_frame()->caller();
1.74 paf 355: ncontext.put_element(ncontext, *info->key_var_name, info->vkey, false);
356: ncontext.put_element(ncontext, *info->value_var_name, avalue, false);
1.57 paf 357:
358: StringOrValue sv_processed=info->r->process(*info->body_code);
1.77 paf 359: Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
1.57 paf 360: const String* s_processed=sv_processed.get_string();
361: if(info->delim_maybe_code && s_processed && s_processed->length()) { // delimiter set and we have body
362: if(info->need_delim) // need delim & iteration produced string?
363: info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));
364: info->need_delim=true;
1.26 paf 365: }
1.57 paf 366: info->r->write_pass_lang(sv_processed);
1.77 paf 367:
368: return lskip==Request::SKIP_BREAK;
1.26 paf 369: }
1.57 paf 370: static void _foreach(Request& r, MethodParams& params) {
1.77 paf 371: Temp_hash_value<const String::Body, void*>
372: cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r);
373:
1.63 paf 374: Foreach_info info={
375: &r,
376: ¶ms.as_string(0, "key-var name must be string"),
377: ¶ms.as_string(1, "value-var name must be string"),
378: ¶ms.as_junction(2, "body must be code"),
379: params.count()>3?params.get(3):0,
1.64 paf 380: /*vkey=*/new VString,
1.63 paf 381: false
382: };
1.57 paf 383:
384: VHash& self=GET_SELF(r, VHash);
385: HashStringValue& hash=self.hash();
1.28 paf 386: VHash_lock lock(self);
1.77 paf 387: hash.first_that<Foreach_info*>(one_foreach_cycle, &info);
1.26 paf 388: }
389:
1.1 paf 390: // constructor
391:
1.57 paf 392: MHash::MHash(): Methoded("hash")
1.39 paf 393: {
1.21 parser 394: // ^hash::create[[copy_from]]
1.57 paf 395: add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
1.22 parser 396: // ^hash.add[add_from]
1.57 paf 397: add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
1.22 parser 398: // ^hash.sub[sub_from]
399: add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
400: // ^a.union[b] = hash
401: add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
402: // ^a.intersection[b] = hash
403: add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
404: // ^a.intersects[b] = bool
405: add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
1.25 paf 406:
407: // ^a.delete[key]
408: add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
1.2 parser 409:
1.82 ! misha 410: // ^a.contains[key]
! 411: add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1);
! 412: // backward
! 413: add_native_method("contain", Method::CT_DYNAMIC, _contains, 1, 1);
1.80 misha 414:
415: // ^hash::sql[query][options hash]
1.33 paf 416: add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.2 parser 417:
1.68 paf 418: // ^hash._keys[[column name]]
419: add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1);
1.16 parser 420:
421: // ^hash._count[]
422: add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);
1.26 paf 423:
424: // ^hash.foreach[key;value]{code}[delim]
425: add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
1.1 paf 426: }
E-mail: