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