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?¶ms.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: