Annotation of parser3/src/classes/hash.C, revision 1.39
1.1 paf 1: /** @file
2: Parser: @b hash parser class.
3:
1.34 paf 4: Copyright (c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com)
1.35 paf 5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1 paf 6:
1.39 ! paf 7: $Id: hash.C,v 1.38 2002/04/15 12:03:31 paf Exp $
1.1 paf 8: */
9:
10: #include "classes.h"
11: #include "pa_request.h"
12: #include "pa_vhash.h"
1.6 parser 13: #include "pa_vvoid.h"
1.2 parser 14: #include "pa_sql_connection.h"
1.9 parser 15: #include "pa_vtable.h"
1.22 parser 16: #include "pa_vbool.h"
1.26 paf 17: #include "pa_vmethod_frame.h"
1.2 parser 18:
1.1 paf 19: // class
20:
21: class MHash : public Methoded {
1.2 parser 22: public: // VStateless_class
23: Value *create_new_value(Pool& pool) { return new(pool) VHash(pool); }
24:
1.1 paf 25: public:
26: MHash(Pool& pool);
27: public: // Methoded
1.2 parser 28: bool used_directly() { return true; }
1.1 paf 29: };
30:
31: // methods
32:
1.11 parser 33: #ifndef DOXYGEN
34: class Hash_sql_event_handlers : public SQL_Driver_query_event_handlers {
35: public:
36: Hash_sql_event_handlers(Pool& apool, const String& amethod_name,
37: const String& astatement_string, const char *astatement_cstr,
38: Hash& arows_hash) :
39: pool(apool),
40: method_name(amethod_name),
41: statement_string(astatement_string),
42: statement_cstr(astatement_cstr),
43: rows_hash(arows_hash),
44: columns(pool),
45: row_index(0) {
46: }
47: void add_column(void *ptr, size_t size) {
48: String *column=new(pool) String(pool);
49: column->APPEND_TAINTED(
50: (const char *)ptr, size,
51: statement_cstr, 0);
52: columns+=column;
53: }
54: void before_rows() {
55: if(columns.size()<=1)
1.36 paf 56: throw Exception("parser.runtime",
1.11 parser 57: &method_name,
58: "column count must be more than 1 to create a hash");
59: }
60: void add_row() {
61: column_index=0;
62: }
63: void add_row_cell(void *ptr, size_t size) {
64: String *cell=new(pool) String(pool);
65: if(size)
66: cell->APPEND_TAINTED(
67: (const char *)ptr, size,
68: statement_cstr, row_index++);
69: if(column_index==0) {
70: VHash *row_vhash=new(pool) VHash(pool);
1.24 parser 71: row_hash=row_vhash->get_hash(0);
1.11 parser 72: rows_hash.put(*cell, row_vhash);
73: } else
74: row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell));
1.12 parser 75: column_index++;
1.11 parser 76: }
77:
78: private:
79: Pool& pool;
80: const String& method_name;
81: const String& statement_string; const char *statement_cstr;
82: Hash& rows_hash;
83: Hash *row_hash;
84: int column_index;
85: Array columns;
86: int row_index;
87: };
88: #endif
89:
1.22 parser 90: static void copy_all_overwrite_to(const Hash::Key& key, Hash::Val *value, void *info) {
1.20 parser 91: Hash& dest=*static_cast<Hash *>(info);
92: dest.put(key, value);
93: }
1.22 parser 94: static void _create_or_add(Request& r, const String& method_name, MethodParams *params) {
1.20 parser 95: Pool& pool=r.pool();
96:
97: if(params->size()) {
1.22 parser 98: Value& vb=params->as_no_junction(0, "param must be hash");
1.24 parser 99: if(Hash *b=vb.get_hash(&method_name))
1.28 paf 100: b->for_each(copy_all_overwrite_to, &static_cast<VHash *>(r.self)->hash(&method_name));
1.20 parser 101: }
102: }
103:
1.22 parser 104: static void remove_key_from(const Hash::Key& key, Hash::Val *value, void *info) {
105: Hash& dest=*static_cast<Hash *>(info);
1.25 paf 106: dest.remove(key);
1.22 parser 107: }
108: static void _sub(Request& r, const String& method_name, MethodParams *params) {
109: Pool& pool=r.pool();
110:
111: Value& vb=params->as_no_junction(0, "param must be hash");
1.24 parser 112: if(Hash *b=vb.get_hash(&method_name))
1.28 paf 113: b->for_each(remove_key_from, &static_cast<VHash *>(r.self)->hash(&method_name));
1.22 parser 114: }
115:
116: static void copy_all_dontoverwrite_to(const Hash::Key& key, Hash::Val *value, void *info) {
117: Hash& dest=*static_cast<Hash *>(info);
118: dest.put_dont_replace(key, value);
119: }
120: static void _union(Request& r, const String& method_name, MethodParams *params) {
121: Pool& pool=r.pool();
122:
123: // dest = copy of self
1.28 paf 124: Hash& dest=*new(pool) Hash(static_cast<VHash *>(r.self)->hash(&method_name));
1.22 parser 125: // dest += b
126: Value& vb=params->as_no_junction(0, "param must be hash");
1.24 parser 127: if(Hash *b=vb.get_hash(&method_name))
1.22 parser 128: b->for_each(copy_all_dontoverwrite_to, &dest);
129:
130: // return result
131: Value& result=*new(pool) VHash(pool, dest);
132: r.write_no_lang(result);
133: }
134:
135: #ifndef DOXYGEN
136: struct Copy_intersection_to_info {
137: Hash *b;
138: Hash *dest;
139: };
140: #endif
141:
142: static void copy_intersection_to(const Hash::Key& key, Hash::Val *value, void *info) {
143: Copy_intersection_to_info& i=*static_cast<Copy_intersection_to_info *>(info);
144:
145: if(i.b->get(key))
146: i.dest->put_dont_replace(key, value);
147: }
148: static void _intersection(Request& r, const String& method_name, MethodParams *params) {
149: Pool& pool=r.pool();
150:
151: // dest = copy of self
152: Hash& dest=*new(pool) Hash(pool);
153: // dest += b
154: Value& vb=params->as_no_junction(0, "param must be hash");
1.24 parser 155: if(Hash *b=vb.get_hash(&method_name)) {
1.22 parser 156: Copy_intersection_to_info info={
157: b,
158: &dest
159: };
1.28 paf 160: static_cast<VHash *>(r.self)->hash(&method_name).for_each(copy_intersection_to, &info);
1.22 parser 161: }
162:
163: // return result
1.39 ! paf 164: r.write_no_lang(*new(pool) VHash(pool, dest));
1.22 parser 165: }
166:
167: static void *intersects(const Hash::Key& key, Hash::Val *value, void *info) {
168: return static_cast<Hash *>(info)->get(key);
169: }
170:
171: static void _intersects(Request& r, const String& method_name, MethodParams *params) {
172: Pool& pool=r.pool();
173:
174: bool yes=false;
175:
176: // dest = copy of self
177: Hash& dest=*new(pool) Hash(pool);
178: // dest += b
179: Value& vb=params->as_no_junction(0, "param must be hash");
1.24 parser 180: if(Hash *b=vb.get_hash(&method_name))
1.28 paf 181: yes=static_cast<VHash *>(r.self)->hash(&method_name).first_that(intersects, b)!=0;
1.22 parser 182:
183: // return result
1.39 ! paf 184: r.write_no_lang(*new(pool) VBool(pool, yes));
1.22 parser 185: }
186:
187:
1.2 parser 188: static void _sql(Request& r, const String& method_name, MethodParams *params) {
189: Pool& pool=r.pool();
190:
1.10 parser 191: Value& statement=params->as_junction(0, "statement must be code");
1.2 parser 192:
193: ulong limit=0;
1.33 paf 194: ulong offset=0;
1.2 parser 195: if(params->size()>1) {
1.33 paf 196: Value& voptions=params->as_no_junction(1, "options must be hash, not code");
197: if(voptions.is_defined())
198: if(Hash *options=voptions.get_hash(&method_name)) {
199: if(Value *vlimit=(Value *)options->get(*sql_limit_name))
1.37 paf 200: limit=(ulong)r.process_to_value(*vlimit).as_double();
1.33 paf 201: if(Value *voffset=(Value *)options->get(*sql_offset_name))
1.37 paf 202: offset=(ulong)r.process_to_value(*voffset).as_double();
1.33 paf 203: } else
1.36 paf 204: throw Exception("parser.runtime",
1.33 paf 205: &method_name,
206: "options must be hash");
1.2 parser 207: }
208:
209: Temp_lang temp_lang(r, String::UL_SQL);
1.37 paf 210: const String& statement_string=r.process_to_string(statement);
1.2 parser 211: const char *statement_cstr=
1.32 paf 212: statement_string.cstr(String::UL_UNSPECIFIED, r.connection(&method_name));
1.28 paf 213: Hash& hash=static_cast<VHash *>(r.self)->hash(&method_name);
1.11 parser 214: hash.clear();
215: Hash_sql_event_handlers handlers(pool, method_name,
216: statement_string, statement_cstr, hash);
1.23 parser 217:
1.32 paf 218: r.connection(&method_name)->query(
1.23 parser 219: statement_cstr, offset, limit,
220: handlers);
1.2 parser 221: }
222:
1.9 parser 223: static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) {
224: Table& table=*static_cast<Table *>(info);
225: Pool& pool=table.pool();
226:
227: Array& row=*new(pool) Array(pool);
228: row+=&key;
229: table+=&row;
230: }
231: static void _keys(Request& r, const String& method_name, MethodParams *) {
232: Pool& pool=r.pool();
233:
234: Array& columns=*new(pool) Array(pool);
235: columns+=new(pool) String(pool, "key");
236: Table& table=*new(pool) Table(pool, &method_name, &columns);
237:
1.28 paf 238: static_cast<VHash *>(r.self)->hash(&method_name).for_each(keys_collector, &table);
1.9 parser 239:
1.39 ! paf 240: r.write_no_lang(*new(pool) VTable(pool, &table));
1.9 parser 241: }
242:
1.16 parser 243: static void _count(Request& r, const String& method_name, MethodParams *) {
244: Pool& pool=r.pool();
245:
1.39 ! paf 246: r.write_no_lang(
! 247: *new(pool) VInt(pool, static_cast<VHash *>(r.self)->hash(&method_name).size()));
1.16 parser 248: }
249:
1.25 paf 250: static void _delete(Request& r, const String& method_name, MethodParams *params) {
251: Pool& pool=r.pool();
252:
1.28 paf 253: static_cast<VHash *>(r.self)->hash(&method_name).remove(params->as_string(0, "key must be string"));
1.25 paf 254: }
255:
1.26 paf 256: #ifndef DOXYGEN
257: struct Foreach_info {
258: Request *r;
259: const String* key_var_name;
260: const String* value_var_name;
261: Value *body_code;
262: Value *delim_maybe_code;
263:
264: VString *vkey;
265: bool need_delim;
266: };
267: #endif
268:
269: static void one_foreach_cycle(const Hash::Key& akey, Hash::Val *avalue,
270: void *info) {
271: Foreach_info& i=*static_cast<Foreach_info *>(info);
272:
273: i.vkey->set_string(akey);
1.29 paf 274: i.r->root->put_element(*i.key_var_name, i.vkey);
275: i.r->root->put_element(*i.value_var_name, static_cast<Value *>(avalue));
1.26 paf 276:
1.38 paf 277: StringOrValue processed_body=i.r->process(*i.body_code);
1.26 paf 278: if(i.delim_maybe_code) { // delimiter set?
279: const String *string=processed_body.get_string();
280: if(i.need_delim && string && string->size()) // need delim & iteration produced string?
1.38 paf 281: i.r->write_pass_lang(i.r->process(*i.delim_maybe_code));
1.26 paf 282: i.need_delim=true;
283: }
284: i.r->write_pass_lang(processed_body);
285: }
286: static void _foreach(Request& r, const String& method_name, MethodParams *params) {
287: Pool& pool=r.pool();
288: const String& key_var_name=params->as_string(0, "key-var name must be string");
289: const String& value_var_name=params->as_string(1, "value-var name must be string");
290: Value& body_code=params->as_junction(2, "body must be code");
291: Value *delim_maybe_code=params->size()>3?¶ms->get(3):0;
292:
293: Foreach_info info={
294: &r,
295: &key_var_name, &value_var_name,
296: &body_code,
297: delim_maybe_code,
298:
299: new(pool) VString(pool),
300: false
301: };
1.28 paf 302: VHash& self=*static_cast<VHash *>(r.self);
303: Hash& hash=self.hash(&method_name);
304: VHash_lock lock(self);
305: hash.for_each(one_foreach_cycle, &info);
1.26 paf 306: }
307:
1.1 paf 308: // constructor
309:
1.39 ! paf 310: MHash::MHash(Pool& apool) : Methoded(apool, "hash")
! 311: {
1.21 parser 312: // ^hash::create[[copy_from]]
1.22 parser 313: add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
314: // ^hash.add[add_from]
315: add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
316: // ^hash.sub[sub_from]
317: add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
318: // ^a.union[b] = hash
319: add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
320: // ^a.intersection[b] = hash
321: add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
322: // ^a.intersects[b] = bool
323: add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
1.25 paf 324:
325: // ^a.delete[key]
326: add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
1.2 parser 327:
1.33 paf 328: // ^hash:sql[query][$.limit(1) $.offset(2)]
329: add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
1.2 parser 330:
1.14 parser 331: // ^hash._keys[]
1.13 parser 332: add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0);
1.16 parser 333:
334: // ^hash._count[]
335: add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);
1.26 paf 336:
337: // ^hash.foreach[key;value]{code}[delim]
338: add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
1.1 paf 339: }
340:
341: // global variable
342:
343: Methoded *hash_base_class;
344:
345: // creator
346:
347: Methoded *MHash_create(Pool& pool) {
348: return hash_base_class=new(pool) MHash(pool);
349: }
E-mail: