Diff for /parser3/src/classes/hash.C between versions 1.74 and 1.91

version 1.74, 2005/07/28 11:23:01 version 1.91, 2009/05/05 10:06:57
Line 1 Line 1
 /** @file  /** @file
         Parser: @b hash parser class.          Parser: @b hash parser class.
   
         Copyright (c) 2001-2004 ArtLebedev Group (http://www.artlebedev.com)          Copyright (c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com)
         Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)          Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
 */  */
   
Line 34  public: // Methoded Line 34  public: // Methoded
   
 DECLARE_CLASS_VAR(hash, new MHash, 0);  DECLARE_CLASS_VAR(hash, new MHash, 0);
   
   // externs
   
   extern String cycle_data_name;
   
 // methods  // methods
   
 #ifndef DOXYGEN  #ifndef DOXYGEN
Line 41  class Hash_sql_event_handlers: public SQ Line 45  class Hash_sql_event_handlers: public SQ
         const String& statement_string; const char* statement_cstr;          const String& statement_string; const char* statement_cstr;
         bool distinct;          bool distinct;
         HashStringValue& rows_hash;          HashStringValue& rows_hash;
         HashStringValue* row_hash;          Value* row_value;
         int column_index;          int column_index;
         ArrayString columns;          ArrayString& columns;
         bool only_one_column;          bool one_bool_column;
         static VBool only_one_column_value;          static VBool only_one_column_value;
           Table2hash_value_type value_type;
           int columns_count;
   public:
           Table* empty;
 public:  public:
         Hash_sql_event_handlers(          Hash_sql_event_handlers(
                 const String& astatement_string, const char* astatement_cstr,                  const String& astatement_string,
                   const char* astatement_cstr,
                 bool adistinct,                  bool adistinct,
                 HashStringValue& arows_hash):                   HashStringValue& arows_hash,
                 statement_string(astatement_string), statement_cstr(astatement_cstr),                  Table2hash_value_type avalue_type)
           : 
                   statement_string(astatement_string),
                   statement_cstr(astatement_cstr),
                 distinct(adistinct),                  distinct(adistinct),
                 rows_hash(arows_hash),                  rows_hash(arows_hash),
                 row_hash(0),                  value_type(avalue_type),
                   row_value(0),
                 column_index(0),                  column_index(0),
                 only_one_column(false) {                  one_bool_column(false),
                   columns(*new ArrayString),
                   empty(0) {
         }          }
   
         bool add_column(SQL_Error& error, const char* str, size_t length) {          bool add_column(SQL_Error& error, const char* str, size_t length) {
                 try {                  try {
                         columns+=new String(str, length, true);                          columns+=new String(str, length, true);
Line 67  public: Line 83  public:
                         return true;                          return true;
                 }                  }
         }          }
   
         bool before_rows(SQL_Error& error) {           bool before_rows(SQL_Error& error) { 
                 if(columns.count()<1) {                  if(columns.count()<1) {
                         error=SQL_Error("parser.runtime", "no columns");                          error=SQL_Error(PARSER_RUNTIME, "no columns");
                         return true;                          return true;
                 }                  }
                 only_one_column=columns.count()==1;                  switch(value_type){
                           case C_STRING: {
                                   if(columns.count()>2){
                                           error=SQL_Error(PARSER_RUNTIME, "only 2 columns allowed for $.type[string].");
                                           return true;
                                   }
                           }
                           case C_TABLE: {
                                   // create empty table which we'll copy later
                                   empty=new Table(&columns);
                                   columns_count=columns.count();
                           }
                           case C_HASH: {
                                   one_bool_column=columns.count()==1;
                           }
                   }
                 return false;                  return false;
         }          }
   
         bool add_row(SQL_Error& /*error*/) {          bool add_row(SQL_Error& /*error*/) {
                 column_index=0;                  column_index=0;
                 return false;                  return false;
         }          }
   
         bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) {          bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) {
                 try {                  try {
                         String& cell=*new String;                          String& cell=*new String;
                         if(length)                          if(length)
                                 cell.append_know_length(ptr, length, String::L_TAINTED);                                  cell.append_know_length(ptr, length, String::L_TAINTED);
   
                         bool duplicate=false;                          bool duplicate=false;
                         if(only_one_column) {                          if(one_bool_column) {
                                 duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value);  // put. existed?                                  duplicate=rows_hash.put_dont_replace(cell, &only_one_column_value);  // put. existed?
                         } else if(column_index==0) {                          } else if(column_index==0) {
                                 VHash* row_vhash=new VHash;                                  switch(value_type){
                                 row_hash=&row_vhash->hash();                                          case C_HASH: {
                                 duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?                                                  VHash* row_vhash=new VHash;
                         } else                                                  row_value=row_vhash;
                                 row_hash->put(*columns[column_index], new VString(cell));                                                  duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?
                                                   break;
                                           }
                                           case C_STRING: {
                                                   VString* row_vstring=new VString();
                                                   row_value=row_vstring;
                                                   duplicate=rows_hash.put_dont_replace(cell, row_vstring);  // put. existed?
                                                   break;
                                           }
                                           case C_TABLE: {
                                                   VTable* vtable=(VTable*)rows_hash.get(cell);
                                                   Table* table;
   
                                                   if(vtable) { // table with this key exist?
                                                           if(!distinct) {
                                                                   duplicate=true;
                                                                   break;
                                                           }
                                                           table=vtable->get_table();
                                                   } else {
                                                           // no? creating table of same structure as source
                                                           Table::Action_options table_options(0, 0);
                                                           table=new Table(*empty, table_options/*no rows, just structure*/);
                                                           vtable=new VTable(table);
                                                           rows_hash.put(cell, vtable); // put
                                                   }
                                                   ArrayString* row=new ArrayString(columns_count);
                                                   row_value=(Value*)row;
                                                   *row+=&cell;
                                                   *table+=row;
                                                   break;
                                           }
                                   }
                           } else {
                                   switch(value_type) {
                                           case C_HASH: {
                                                   row_value->get_hash()->put(*columns[column_index], new VString(cell));
                                                   break;
                                           }
                                           case C_STRING: {
                                                   VString* row_string=(VString*)row_value;
                                                   row_string->set_string(cell);
                                                   break;
                                           }
                                           case C_TABLE: {
                                                   ArrayString* row=(ArrayString*)row_value;
                                                   *row+=&cell;
                                                   break;
                                           }
                                   }
                           }
   
                         if(duplicate & !distinct) {                          if(duplicate & !distinct) {
                                 error=SQL_Error("parser.runtime", "duplicate key");                                  error=SQL_Error(PARSER_RUNTIME, "duplicate key");
                                 return true;                                  return true;
                         }                          }
   
Line 113  VBool Hash_sql_event_handlers::only_one_ Line 197  VBool Hash_sql_event_handlers::only_one_
   
 #endif  #endif
   
 static void copy_all_overwrite_to(  
                                                                   HashStringValue::key_type key,   
                                                                   HashStringValue::value_type value,   
                                                                   HashStringValue* dest) {  
         dest->put(key, value);  
 }  
 static void _create_or_add(Request& r, MethodParams& params) {  static void _create_or_add(Request& r, MethodParams& params) {
         if(params.count()) {          if(params.count()) {
                 Value& vsrc=params.as_no_junction(0, "param must be hash");                  Value& vsrc=params.as_no_junction(0, "param must be hash");
Line 127  static void _create_or_add(Request& r, M Line 205  static void _create_or_add(Request& r, M
                         HashStringValue* self_hash=&(self.hash());                          HashStringValue* self_hash=&(self.hash());
                         if(src==self_hash) // same: doing nothing                          if(src==self_hash) // same: doing nothing
                                 return;                                  return;
                         src->for_each(copy_all_overwrite_to, self_hash);                          src->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash);
   
                         if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false)))                          if(VHash* vhash_src=static_cast<VHash*>(vsrc.as(VHASH_TYPE, false)))
                                 self.set_default(vhash_src->get_default());                          {
                                   if(Value* vdefault=vhash_src->get_default())
                                   {
                                           if(vdefault->is_defined())
                                           {
                                                   self.set_default(vdefault);
                                           }
                                   }
                           }
                 }                  }
         }          }
 }  }
   
 static void remove_key_from(  
                                                         HashStringValue::key_type key,   
                                                         HashStringValue::value_type /*value*/,   
                                                         HashStringValue* dest) {  
         dest->remove(key);  
 }  
 static void _sub(Request& r, MethodParams& params) {  static void _sub(Request& r, MethodParams& params) {
         Value& vsrc=params.as_no_junction(0, "param must be hash");          Value& vsrc=params.as_no_junction(0, "param must be hash");
         if(HashStringValue* src=vsrc.get_hash()) {          if(HashStringValue* src=vsrc.get_hash()) {
Line 149  static void _sub(Request& r, MethodParam Line 229  static void _sub(Request& r, MethodParam
                         self->clear();                          self->clear();
                         return;                          return;
                 }                  }
                 src->for_each(remove_key_from, self);                  src->for_each<HashStringValue*>(remove_key_from, self);
         }          }
 }  }
   
Line 165  static void _union(Request& r, MethodPar Line 245  static void _union(Request& r, MethodPar
         // dest += b          // dest += b
         Value& vsrc=params.as_no_junction(0, "param must be hash");          Value& vsrc=params.as_no_junction(0, "param must be hash");
         if(HashStringValue* src=vsrc.get_hash())          if(HashStringValue* src=vsrc.get_hash())
                 src->for_each(copy_all_dontoverwrite_to, result.get_hash());                  src->for_each<HashStringValue*>(copy_all_dontoverwrite_to, result.get_hash());
   
         // return result          // return result
         r.write_no_lang(result);          r.write_no_lang(result);
Line 190  static void _intersection(Request& r, Me Line 270  static void _intersection(Request& r, Me
         Value& vb=params.as_no_junction(0, "param must be hash");          Value& vb=params.as_no_junction(0, "param must be hash");
         if(HashStringValue* b=vb.get_hash()) {          if(HashStringValue* b=vb.get_hash()) {
                 Copy_intersection_to_info info={b, result.get_hash()};                  Copy_intersection_to_info info={b, result.get_hash()};
                 GET_SELF(r, VHash).hash().for_each(copy_intersection_to, &info);                  GET_SELF(r, VHash).hash().for_each<Copy_intersection_to_info*>(copy_intersection_to, &info);
         }          }
   
         // return result          // return result
Line 209  static void _intersects(Request& r, Meth Line 289  static void _intersects(Request& r, Meth
   
         Value& vb=params.as_no_junction(0, "param must be hash");          Value& vb=params.as_no_junction(0, "param must be hash");
         if(HashStringValue* b=vb.get_hash())          if(HashStringValue* b=vb.get_hash())
                 result=GET_SELF(r, VHash).hash().first_that(intersects, b)!=0;                  result=GET_SELF(r, VHash).hash().first_that<HashStringValue*>(intersects, b)!=0;
   
         // return result          // return result
         r.write_no_lang(*new VBool(result));          r.write_no_lang(VBool::get(result));
 }  }
   
   
Line 221  extern String sql_limit_name; Line 301  extern String sql_limit_name;
 extern String sql_offset_name;  extern String sql_offset_name;
 extern String sql_default_name;  extern String sql_default_name;
 extern String sql_distinct_name;  extern String sql_distinct_name;
   extern String sql_value_type_name;
   extern Table2hash_value_type get_value_type(Value& vvalue_type);
 extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);  extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);
 extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);  extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
   
Line 228  static void _sql(Request& r, MethodParam Line 310  static void _sql(Request& r, MethodParam
         Value& statement=params.as_junction(0, "statement must be code");          Value& statement=params.as_junction(0, "statement must be code");
   
         HashStringValue* bind=0;          HashStringValue* bind=0;
         ulong limit=0;          ulong limit=SQL_NO_LIMIT;
         ulong offset=0;          ulong offset=0;
         bool distinct=false;          bool distinct=false;
           Table2hash_value_type value_type=C_HASH;
         if(params.count()>1) {          if(params.count()>1) {
                 Value& voptions=params.as_no_junction(1, "options must be hash, not code");                  Value& voptions=params.as_no_junction(1, "options must be hash, not code");
                 if(!voptions.is_string())                  if(voptions.is_defined() && !voptions.is_string())
                         if(HashStringValue* options=voptions.get_hash()) {                          if(HashStringValue* options=voptions.get_hash()) {
                                 int valid_options=0;                                  int valid_options=0;
                                 if(Value* vbind=options->get(sql_bind_name)) {                                  if(Value* vbind=options->get(sql_bind_name)) {
Line 252  static void _sql(Request& r, MethodParam Line 335  static void _sql(Request& r, MethodParam
                                         valid_options++;                                          valid_options++;
                                         distinct=r.process_to_value(*vdistinct).as_bool();                                          distinct=r.process_to_value(*vdistinct).as_bool();
                                 }                                  }
                                   if(Value* vvalue_type=options->get(sql_value_type_name)) {
                                           valid_options++;
                                           value_type=get_value_type(r.process_to_value(*vvalue_type));
                                   }
                                 if(valid_options!=options->count())                                  if(valid_options!=options->count())
                                         throw Exception("parser.runtime",                                          throw Exception(PARSER_RUNTIME,
                                                 0,                                                  0,
                                                 "called with invalid option");                                                  "called with invalid option");
                         } else                          } else
                                 throw Exception("parser.runtime",                                  throw Exception(PARSER_RUNTIME,
                                         0,                                          0,
                                         "options must be hash");                                          "options must be hash");
         }          }
Line 276  static void _sql(Request& r, MethodParam Line 363  static void _sql(Request& r, MethodParam
         Hash_sql_event_handlers handlers(          Hash_sql_event_handlers handlers(
                 statement_string, statement_cstr,                   statement_string, statement_cstr, 
                 distinct,                  distinct,
                 hash);                  hash,
                   value_type);
   
         r.connection()->query(          r.connection()->query(
                 statement_cstr,                   statement_cstr, 
                 placeholders_count, placeholders,                  placeholders_count, placeholders,
Line 299  static void keys_collector( Line 388  static void keys_collector(
 static void _keys(Request& r, MethodParams& params) {  static void _keys(Request& r, MethodParams& params) {
         const String* keys_column_name;          const String* keys_column_name;
         if(params.count()>0)          if(params.count()>0)
                 keys_column_name=&params.as_string(0, "column name must be string");                  keys_column_name=&params.as_string(0, COLUMN_NAME_MUST_BE_STRING);
         else           else 
                 keys_column_name=new String("key");                  keys_column_name=new String("key");
   
Line 307  static void _keys(Request& r, MethodPara Line 396  static void _keys(Request& r, MethodPara
         *columns+=keys_column_name;          *columns+=keys_column_name;
         Table* table=new Table(columns);          Table* table=new Table(columns);
   
         GET_SELF(r, VHash).hash().for_each(keys_collector, table);          GET_SELF(r, VHash).hash().for_each<Table*>(keys_collector, table);
   
         r.write_no_lang(*new VTable(table));          r.write_no_lang(*new VTable(table));
 }  }
Line 321  static void _delete(Request& r, MethodPa Line 410  static void _delete(Request& r, MethodPa
         GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));          GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));
 }  }
   
   static void _contains(Request& r, MethodParams& params) {
           bool result=GET_SELF(r, VHash).hash().contains(params.as_string(0, "key must be string"));
           r.write_no_lang(VBool::get(result));
   }
   
 #ifndef DOXYGEN  #ifndef DOXYGEN
 struct Foreach_info {  struct Foreach_info {
         Request *r;          Request *r;
Line 329  struct Foreach_info { Line 423  struct Foreach_info {
         Value* body_code;          Value* body_code;
         Value* delim_maybe_code;          Value* delim_maybe_code;
   
           Value* var_context;
         VString* vkey;          VString* vkey;
         bool need_delim;          bool need_delim;
 };  };
 #endif  #endif
 static void one_foreach_cycle(                                               static bool one_foreach_cycle(
                               HashStringValue::key_type akey,                                   HashStringValue::key_type akey, 
                               HashStringValue::value_type avalue,                                   HashStringValue::value_type avalue, 
                               Foreach_info *info) {                                  Foreach_info *info) {
         info->vkey->set_string(*new String(akey, String::L_TAINTED));          Value& var_context=*info->var_context;
         Value& ncontext=*info->r->get_method_frame()->caller();          if(info->key_var_name){
         ncontext.put_element(ncontext, *info->key_var_name, info->vkey, false);                  info->vkey->set_string(*new String(akey, String::L_TAINTED));
         ncontext.put_element(ncontext, *info->value_var_name, avalue, false);                  var_context.put_element(var_context, *info->key_var_name, info->vkey, false);
           }
         StringOrValue sv_processed=info->r->process(*info->body_code);          if(info->value_var_name)
         const String* s_processed=sv_processed.get_string();                  var_context.put_element(var_context, *info->value_var_name, avalue, false);
         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?          if(info->delim_maybe_code){ // delimiter set
                         info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));                  StringOrValue sv_processed=info->r->process(*info->body_code);
                 info->need_delim=true;                  Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
   
                   const String* s_processed=sv_processed.get_string();
                   if(s_processed && !s_processed->is_empty()) { // we have body
                           if(info->need_delim) // need delim & iteration produced string?
                                   info->r->write_pass_lang(info->r->process(*info->delim_maybe_code));
                           else
                                   info->need_delim=true;
                   }
                   info->r->write_pass_lang(sv_processed);
                   return lskip==Request::SKIP_BREAK;
           } else {
                   info->r->process_write(*info->body_code);
                   Request::Skip lskip=info->r->get_skip(); info->r->set_skip(Request::SKIP_NOTHING);
                   return lskip==Request::SKIP_BREAK;
         }          }
         info->r->write_pass_lang(sv_processed);  
 }  }
 static void _foreach(Request& r, MethodParams& params) {  static void _foreach(Request& r, MethodParams& params) {
           Temp_hash_value<const String::Body, void*> 
                   cycle_data_setter(r.classes_conf, cycle_data_name, /*any not null flag*/&r);
   
           const String& key_var_name=params.as_string(0, "key-var name must be string");
           const String& value_var_name=params.as_string(1, "value-var name must be string");
   
         Foreach_info info={          Foreach_info info={
                 &r,                  &r,
                 &params.as_string(0, "key-var name must be string"),                  key_var_name.is_empty()? 0 : &key_var_name,
                 &params.as_string(1, "value-var name must be string"),                  value_var_name.is_empty()? 0 : &value_var_name,
                 &params.as_junction(2, "body must be code"),                  &params.as_junction(2, "body must be code"),
                 params.count()>3?params.get(3):0,                  /*delimiter*/params.count()>3?params.get(3):0,
                   /*var_context*/r.get_method_frame()->caller(),
                 /*vkey=*/new VString,                  /*vkey=*/new VString,
                 false                  false
         };          };
Line 365  static void _foreach(Request& r, MethodP Line 480  static void _foreach(Request& r, MethodP
         VHash& self=GET_SELF(r, VHash);          VHash& self=GET_SELF(r, VHash);
         HashStringValue& hash=self.hash();          HashStringValue& hash=self.hash();
         VHash_lock lock(self);          VHash_lock lock(self);
         hash.for_each(one_foreach_cycle, &info);          hash.first_that<Foreach_info*>(one_foreach_cycle, &info);
 }  }
   
 // constructor  // constructor
Line 388  MHash::MHash(): Methoded("hash") Line 503  MHash::MHash(): Methoded("hash")
         // ^a.delete[key]          // ^a.delete[key]
         add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);          add_native_method("delete", Method::CT_DYNAMIC, _delete, 1, 1);
   
         // ^hash:sql[query][options hash]          // ^a.contains[key]
           add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1);
           // backward
           add_native_method("contain", Method::CT_DYNAMIC, _contains, 1, 1);
   
           // ^hash::sql[query][options hash]
         add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);          add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
   
         // ^hash._keys[[column name]]          // ^hash._keys[[column name]]

Removed from v.1.74  
changed lines
  Added in v.1.91


E-mail: