Diff for /parser3/src/classes/hash.C between versions 1.11 and 1.72

version 1.11, 2001/07/23 11:19:25 version 1.72, 2004/11/24 07:33:51
Line 1 Line 1
 /** @file  /** @file
         Parser: @b hash parser class.          Parser: @b hash parser class.
   
         Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)          Copyright (c) 2001-2004 ArtLebedev Group (http://www.artlebedev.com)
           Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
         Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)  
   
         $Id$  
 */  */
 static const char *RCSId="$Id$";   
   static const char * const IDENT_HASH_C="$Date$";
   
 #include "classes.h"  #include "classes.h"
   #include "pa_vmethod_frame.h"
   
 #include "pa_request.h"  #include "pa_request.h"
 #include "pa_vhash.h"  #include "pa_vhash.h"
 #include "pa_vvoid.h"  #include "pa_vvoid.h"
 #include "pa_sql_connection.h"  #include "pa_sql_connection.h"
 #include "pa_vtable.h"  #include "pa_vtable.h"
   #include "pa_vbool.h"
 // defines  #include "pa_vmethod_frame.h"
   
 #define HASH_CLASS_NAME "hash"  
   
 // class  // class
   
 class MHash : public Methoded {  class MHash: public Methoded {
 public: // VStateless_class  public: // VStateless_class
         Value *create_new_value(Pool& pool) { return new(pool) VHash(pool); }          Value* create_new_value(Pool&) { return new VHash(); }
   
 public:  public:
         MHash(Pool& pool);          MHash();
 public: // Methoded  public: // Methoded
         bool used_directly() { return true; }          bool used_directly() { return true; }
 };  };
   
 // methods  // global variable
   
 static void _default(Request& r, const String&, MethodParams *params) {   DECLARE_CLASS_VAR(hash, new MHash, 0);
         Pool& pool=r.pool();  
   
         VHash& vhash=*static_cast<VHash *>(r.self);  // methods
         if(params->size())  
                 vhash.set_default(params->get(0)); // info: may be code..  
         else {  
                 Value *default_value=vhash.get_default();  
                 r.write_assign_lang(default_value?*default_value:*new(pool) VVoid(pool));  
         }  
 }  
   
 #ifndef DOXYGEN  #ifndef DOXYGEN
 class Hash_sql_event_handlers : public SQL_Driver_query_event_handlers {  class Hash_sql_event_handlers: public SQL_Driver_query_event_handlers {
           const String& statement_string; const char* statement_cstr;
           bool distinct;
           HashStringValue& rows_hash;
           HashStringValue* row_hash;
           int column_index;
           ArrayString columns;
           bool only_one_column;
           static VBool only_one_column_value;
 public:  public:
         Hash_sql_event_handlers(Pool& apool, const String& amethod_name,          Hash_sql_event_handlers(
                 const String& astatement_string, const char *astatement_cstr,                  const String& astatement_string, const char* astatement_cstr,
                 Hash& arows_hash) :                  bool adistinct,
                 pool(apool),                   HashStringValue& arows_hash): 
                 method_name(amethod_name),                  statement_string(astatement_string), statement_cstr(astatement_cstr),
                 statement_string(astatement_string),                  distinct(adistinct),
                 statement_cstr(astatement_cstr),  
                 rows_hash(arows_hash),                  rows_hash(arows_hash),
                 columns(pool),                  row_hash(0),
                 row_index(0) {                  column_index(0),
                   only_one_column(false) {
         }          }
         void add_column(void *ptr, size_t size) {          bool add_column(SQL_Error& error, const char* str, size_t length) {
                 String *column=new(pool) String(pool);                  try {
                 column->APPEND_TAINTED(                          columns+=new String(str, length, true);
                         (const char *)ptr, size,                           return false;
                         statement_cstr, 0);                  } catch(...) {
                 columns+=column;                          error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
         }                          return true;
         void before_rows() {                   }
                 if(columns.size()<=1)  
                         PTHROW(0, 0,  
                                 &method_name,  
                                 "column count must be more than 1 to create a hash");  
         }          }
         void add_row() {          bool before_rows(SQL_Error& error) { 
                   if(columns.count()<1) {
                           error=SQL_Error("parser.runtime", "no columns");
                           return true;
                   }
                   only_one_column=columns.count()==1;
   
                   return false;
           }
           bool add_row(SQL_Error& /*error*/) {
                 column_index=0;                  column_index=0;
                   return false;
         }          }
         void add_row_cell(void *ptr, size_t size) {          bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) {
                 String *cell=new(pool) String(pool);                  try {
                 if(size)                          String& cell=*new String;
                         cell->APPEND_TAINTED(                          if(length)
                                 (const char *)ptr, size,                                   cell.append_know_length(ptr, length, String::L_TAINTED);
                                 statement_cstr, row_index++);                          bool duplicate=false;
                 if(column_index==0) {                          if(only_one_column) {
                         VHash *row_vhash=new(pool) VHash(pool);                                  duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value);  // put. existed?
                         row_hash=row_vhash->get_hash();                          } else if(column_index==0) {
                         rows_hash.put(*cell, row_vhash);                                  VHash* row_vhash=new VHash;
                 } else                                  row_hash=&row_vhash->hash();
                         row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell));                                  duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?
         }                          } else
                                   row_hash->put(*columns[column_index], new VString(cell));
 private:  
         Pool& pool;                          if(duplicate & !distinct) {
         const String& method_name;                                  error=SQL_Error("parser.runtime", "duplicate key");
         const String& statement_string; const char *statement_cstr;                                  return true;
         Hash& rows_hash;                          }
         Hash *row_hash;  
         int column_index;                          column_index++;
         Array columns;                          return false;
         int row_index;                  } catch(...) {
                           error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell");
                           return true;
                   }
           }
   
 };  };
   VBool Hash_sql_event_handlers::only_one_column_value(true);
   
 #endif  #endif
   
 static void _sql(Request& r, const String& method_name, MethodParams *params) {  static void copy_all_overwrite_to(
         Pool& pool=r.pool();                                                                    HashStringValue::key_type key, 
                                                                     HashStringValue::value_type value, 
                                                                     HashStringValue* dest) {
           dest->put(key, value);
   }
   static void _create_or_add(Request& r, MethodParams& params) {
           if(params.count()) {
                   Value& vsrc=params.as_no_junction(0, "param must be hash");
                   if(HashStringValue* src=vsrc.get_hash()) {
                           VHash& self=GET_SELF(r, VHash);
                           HashStringValue* self_hash=&(self.hash());
                           if(src==self_hash) // same: doing nothing
                                   return;
                           src->for_each(copy_all_overwrite_to, self_hash);
   
                           if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false)))
                                   self.set_default(vhash_src->get_default());
                   }
           }
   }
   
         if(!r.connection)  static void remove_key_from(
                 PTHROW(0, 0,                                                          HashStringValue::key_type key, 
                         &method_name,                                                          HashStringValue::value_type /*value*/, 
                         "without connect");                                                          HashStringValue* dest) {
           dest->remove(key);
   }
   static void _sub(Request& r, MethodParams& params) {
           Value& vsrc=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* src=vsrc.get_hash()) {
                   HashStringValue* self=&(GET_SELF(r, VHash).hash());
                   if(src==self) { // same: clearing
                           self->clear();
                           return;
                   }
                   src->for_each(remove_key_from, self);
           }
   }
   
         Value& statement=params->as_junction(0, "statement must be code");  static void copy_all_dontoverwrite_to(
                                                                     HashStringValue::key_type key, 
                                                                     HashStringValue::value_type value, 
                                                                     HashStringValue* dest) {
           dest->put_dont_replace(key, value);
   }
   static void _union(Request& r, MethodParams& params) {
           // dest = copy of self
           Value& result=*new VHash(GET_SELF(r, VHash).hash());
           // dest += b
           Value& vsrc=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* src=vsrc.get_hash())
                   src->for_each(copy_all_dontoverwrite_to, result.get_hash());
   
         ulong limit=0;          // return result
         if(params->size()>1) {          r.write_no_lang(result);
                 Value& limit_code=params->as_junction(1, "limit must be expression");  }
                 limit=(uint)r.process(limit_code).as_double();  
   #ifndef DOXYGEN
   struct Copy_intersection_to_info {
           HashStringValue* b;
           HashStringValue* dest;
   };
   #endif
   static void copy_intersection_to(
                                                                    HashStringValue::key_type key, 
                                                                    HashStringValue::value_type value, 
                                                                    Copy_intersection_to_info *info) {
           if(info->b->get(key))
                   info->dest->put_dont_replace(key, value);
   }
   static void _intersection(Request& r, MethodParams& params) {
           Value& result=*new VHash;
           // dest += b
           Value& vb=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* b=vb.get_hash()) {
                   Copy_intersection_to_info info={b, result.get_hash()};
                   GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info);
         }          }
   
           // return result
           r.write_no_lang(result);
   }
   
   static bool intersects(
                                              HashStringValue::key_type key, 
                                              HashStringValue::value_type /*value*/, 
                                              HashStringValue* b) {
           return b->get(key)!=0;
   }
   
   static void _intersects(Request& r, MethodParams& params) {
           bool result=false;
   
           Value& vb=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* b=vb.get_hash())
                   result=GET_SELF(r, VHash).hash().first_that(intersects, b)!=0;
   
           // return result
           r.write_no_lang(*new VBool(result));
   }
   
   
   extern String sql_bind_name;
   extern String sql_limit_name;
   extern String sql_offset_name;
   extern String sql_default_name;
   extern String sql_distinct_name;
   extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);
   extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
   
   static void _sql(Request& r, MethodParams& params) {
           Value& statement=params.as_junction(0, "statement must be code");
   
           HashStringValue* bind=0;
           ulong limit=0;
         ulong offset=0;          ulong offset=0;
         if(params->size()>2) {          bool distinct=false;
                 Value& offset_code=params->as_junction(2, "offset must be expression");          if(params.count()>1) {
                 offset=(ulong)r.process(offset_code).as_double();                  Value& voptions=params.as_no_junction(1, "options must be hash, not code");
                   if(!voptions.is_string())
                           if(HashStringValue* options=voptions.get_hash()) {
                                   int valid_options=0;
                                   if(Value* vbind=options->get(sql_bind_name)) {
                                           valid_options++;
                                           bind=vbind->get_hash();
                                   }
                                   if(Value* vlimit=options->get(sql_limit_name)) {
                                           valid_options++;
                                           limit=(ulong)r.process_to_value(*vlimit).as_double();
                                   }
                                   if(Value* voffset=options->get(sql_offset_name)) {
                                           valid_options++;
                                           offset=(ulong)r.process_to_value(*voffset).as_double();
                                   }
                                   if(Value* vdistinct=options->get(sql_distinct_name)) {
                                           valid_options++;
                                           distinct=r.process_to_value(*vdistinct).as_bool();
                                   }
                                   if(valid_options!=options->count())
                                           throw Exception("parser.runtime",
                                                   0,
                                                   "called with invalid option");
                           } else
                                   throw Exception("parser.runtime",
                                           0,
                                           "options must be hash");
         }          }
   
         Temp_lang temp_lang(r, String::UL_SQL);          SQL_Driver::Placeholder* placeholders=0;
         const String& statement_string=r.process(statement).as_string();          uint placeholders_count=0;
         const char *statement_cstr=          if(bind)
                 statement_string.cstr(String::UL_UNSPECIFIED, r.connection);                  placeholders_count=marshal_binds(*bind, placeholders);
         Hash& hash=static_cast<VHash *>(r.self)->hash();  
           Temp_lang temp_lang(r, String::L_SQL);
           const String& statement_string=r.process_to_string(statement);
           const char* statement_cstr=
                   statement_string.cstr(String::L_UNSPECIFIED, r.connection());
           HashStringValue& hash=GET_SELF(r, VHash).hash();
         hash.clear();             hash.clear();   
         Hash_sql_event_handlers handlers(pool, method_name,          Hash_sql_event_handlers handlers(
                 statement_string, statement_cstr, hash);                  statement_string, statement_cstr, 
         bool need_rethrow=false; Exception rethrow_me;                  distinct,
         PTRY {                  hash);
                 r.connection->query(          r.connection()->query(
                         statement_cstr, offset, limit,                  statement_cstr, 
                         handlers);                  0, 0,
         }                  offset, limit,
         PCATCH(e) { // query problem                  handlers,
                 rethrow_me=e;  need_rethrow=true;                  statement_string);
         }  
         PEND_CATCH  
         if(need_rethrow)  
                 PTHROW(rethrow_me.type(), rethrow_me.code(),  
                         &statement_string, // setting more specific source [were url]  
                         rethrow_me.comment());  
 }  
   
 static void keys_collector(const Hash::Key& key, Hash::Val *value, void *info) {  
         Table& table=*static_cast<Table *>(info);  
         Pool& pool=table.pool();  
   
         Array& row=*new(pool) Array(pool);  
         row+=&key;  
         table+=&row;  
 }  
 static void _keys(Request& r, const String& method_name, MethodParams *) {  
         Pool& pool=r.pool();  
   
         Array& columns=*new(pool) Array(pool);  
         columns+=new(pool) String(pool, "key");  
         Table& table=*new(pool) Table(pool, &method_name, &columns);  
   
         static_cast<VHash *>(r.self)->hash().for_each(keys_collector, &table);          if(bind)
                   unmarshal_bind_updates(*bind, placeholders_count, placeholders);
   }
   
         VTable& result=*new(pool) VTable(pool, &table);  static void keys_collector(
         result.set_name(method_name);                             HashStringValue::key_type key, 
         r.write_no_lang(result);                             HashStringValue::value_type, 
                              Table *table) {
           Table::element_type row(new ArrayString);
           *row+=new String(key, String::L_TAINTED);
           *table+=row;
 }  }
   static void _keys(Request& r, MethodParams& params) {
           const String* keys_column_name;
           if(params.count()>0)
                   keys_column_name=&params.as_string(0, "column name must be string");
           else 
                   keys_column_name=new String("key");
   
           Table::columns_type columns(new ArrayString);
           *columns+=keys_column_name;
           Table* table=new Table(columns);
   
 // constructor          GET_SELF(r, VHash).hash().for_each(keys_collector, table);
   
 MHash::MHash(Pool& apool) : Methoded(apool) {          r.write_no_lang(*new VTable(table));
         set_name(*NEW String(pool(), HASH_CLASS_NAME));  }
   
         // ^hash.default[]  static void _count(Request& r, MethodParams&) {
         // ^hash.default[hash]          r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count()));
         add_native_method("default", Method::CT_DYNAMIC, _default, 0, 1);  }
   
         // ^hash:sql[query][(count[;offset])]  static void _delete(Request& r, MethodParams& params) {
         add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3);  
   
         // ^hash.keys[]          GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));
         add_native_method("keys", Method::CT_DYNAMIC, _keys, 0, 0);       
 }  }
   
 // global variable  #ifndef DOXYGEN
   struct Foreach_info {
           Request *r;
           const String* key_var_name;
           const String* value_var_name;
           Value* body_code;
           Value* delim_maybe_code;
   
           VString* vkey;
           bool need_delim;
   };
   #endif
   static void one_foreach_cycle(                                             
                                 HashStringValue::key_type akey, 
                                 HashStringValue::value_type avalue, 
                                 Foreach_info *info) {
           info->vkey->set_string(*new String(akey, String::L_TAINTED));
           Value& ncontext=*info->r->get_method_frame()->caller();
           ncontext.put_element(*info->key_var_name, info->vkey, false);
           ncontext.put_element(*info->value_var_name, avalue, false);
   
           StringOrValue sv_processed=info->r->process(*info->body_code);
           const String* s_processed=sv_processed.get_string();
           if(info->delim_maybe_code && s_processed && s_processed->length()) { // delimiter set and we have body
                   if(info->need_delim) // need delim & iteration produced string?
                           info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));
                   info->need_delim=true;
           }
           info->r->write_pass_lang(sv_processed);
   }
   static void _foreach(Request& r, MethodParams& params) {
           Foreach_info info={
                   &r,
                   &params.as_string(0, "key-var name must be string"),
                   &params.as_string(1, "value-var name must be string"),
                   &params.as_junction(2, "body must be code"),
                   params.count()>3?params.get(3):0,
                   /*vkey=*/new VString,
                   false
           };
   
           VHash& self=GET_SELF(r, VHash);
           HashStringValue& hash=self.hash();
           VHash_lock lock(self);
           hash.for_each(one_foreach_cycle, &info);
   }
   
   // constructor
   
   MHash::MHash(): Methoded("hash") 
   {
           // ^hash::create[[copy_from]]
           add_native_method("create", Method::CT_DYNAMIC, _create_or_add, 0, 1);
           // ^hash.add[add_from]
           add_native_method("add", Method::CT_DYNAMIC, _create_or_add, 1, 1);
           // ^hash.sub[sub_from]
           add_native_method("sub", Method::CT_DYNAMIC, _sub, 1, 1);
           // ^a.union[b] = hash
           add_native_method("union", Method::CT_DYNAMIC, _union, 1, 1);
           // ^a.intersection[b] = hash
           add_native_method("intersection", Method::CT_DYNAMIC, _intersection, 1, 1);
           // ^a.intersects[b] = bool
           add_native_method("intersects", Method::CT_DYNAMIC, _intersects, 1, 1);
   
           // ^a.delete[key]
           add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
   
           // ^hash:sql[query][options hash]
           add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
   
 Methoded *hash_base_class;          // ^hash._keys[[column name]]
           add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1);    
   
 // creator          // ^hash._count[]
           add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);  
   
 Methoded *MHash_create(Pool& pool) {          // ^hash.foreach[key;value]{code}[delim]
         return hash_base_class=new(pool) MHash(pool);          add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
 }  }

Removed from v.1.11  
changed lines
  Added in v.1.72


E-mail: