[BACK]Return to hash.C CVS log [TXT] [DIR] Up to [parser3project] / parser3 / src / classes

Diff for /parser3/src/classes/hash.C between versions 1.18 and 1.115

version 1.18, 2001/09/21 15:35:45 version 1.115, 2012/06/13 22:53:47
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-2012 Art. Lebedev Studio (http://www.artlebedev.com)
           Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
         Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)  
 */  */
 static const char *RCSId="$Id$";   
   
 #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"
   #include "pa_vmethod_frame.h"
   
 // defines  volatile const char * IDENT_HASH_C="$Id$";
   
 #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  
         bool used_directly() { return true; }  
 };  };
   
   // global variable
   
   DECLARE_CLASS_VAR(hash, new MHash, 0);
   
 // methods  // methods
   
 #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;
           Value* row_value;
           int column_index;
           ArrayString& columns;
           bool one_bool_column;
           static VBool only_one_column_value;
           Table2hash_value_type value_type;
           int columns_count;
   public:
           Table* empty;
 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,
                 Hash& arows_hash) :                  const char* astatement_cstr,
                 pool(apool),                   bool adistinct,
                 method_name(amethod_name),                  HashStringValue& arows_hash,
                   Table2hash_value_type avalue_type)
           : 
                 statement_string(astatement_string),                  statement_string(astatement_string),
                 statement_cstr(astatement_cstr),                  statement_cstr(astatement_cstr),
                   distinct(adistinct),
                 rows_hash(arows_hash),                  rows_hash(arows_hash),
                 columns(pool),                  value_type(avalue_type),
                 row_index(0) {                  row_value(0),
                   column_index(0),
                   one_bool_column(false),
                   columns(*new ArrayString),
                   empty(0) {
         }          }
         void add_column(void *ptr, size_t size) {  
                 String *column=new(pool) String(pool);          bool add_column(SQL_Error& error, const char* str, size_t ) {
                 column->APPEND_TAINTED(                  try {
                         (const char *)ptr, size,                           columns+=new String(str, String::L_TAINTED /* no length as 0x00 can be inside */);
                         statement_cstr, 0);                          return false;
                 columns+=column;                  } catch(...) {
         }                          error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
         void before_rows() {                           return true;
                 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;
                   }
                   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;
           }
   
           bool add_row(SQL_Error& /*error*/) {
                 column_index=0;                  column_index=0;
                   return false;
         }          }
         void add_row_cell(void *ptr, size_t size) {  
                 String *cell=new(pool) String(pool);          bool add_row_cell(SQL_Error& error, const char *ptr, size_t ) {
                 if(size)                  try {
                         cell->APPEND_TAINTED(                          String& cell=*new String(ptr, String::L_TAINTED /* no length as 0x00 can be inside */);
                                 (const char *)ptr, size,   
                                 statement_cstr, row_index++);                          bool duplicate=false;
                 if(column_index==0) {                          if(one_bool_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);                                  switch(value_type){
                 } else                                          case C_HASH: {
                         row_hash->put(*columns.get_string(column_index), new(pool) VString(*cell));                                                  VHash* row_vhash=new VHash;
                 column_index++;                                                  row_value=row_vhash;
         }                                                  duplicate=rows_hash.put_dont_replace(cell, row_vhash); // put. existed?
                                                   break;
 private:                                          }
         Pool& pool;                                          case C_STRING: {
         const String& method_name;                                                  VString* row_vstring=new VString();
         const String& statement_string; const char *statement_cstr;                                                  row_value=row_vstring;
         Hash& rows_hash;                                                  duplicate=rows_hash.put_dont_replace(cell, row_vstring);  // put. existed?
         Hash *row_hash;                                                  break;
         int column_index;                                          }
         Array columns;                                          case C_TABLE: {
         int row_index;                                                  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) {
                                   error=SQL_Error(PARSER_RUNTIME, "duplicate key");
                                   return true;
                           }
   
                           column_index++;
                           return false;
                   } 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 _create_or_add(Request& r, MethodParams& params) {
         Pool& pool=r.pool();          if(params.count()) {
                   Value& vsrc=params.as_no_junction(0, PARAM_MUST_BE_HASH);
                   VHash& self=GET_SELF(r, VHash);
                   HashStringValue* src_hash;
                   HashStringValue* self_hash=&(self.hash());
   
                   if(VHash* src=static_cast<VHash*>(vsrc.as(VHASH_TYPE))) {
                           src_hash=&(src->hash_ro());
   
                           if(src_hash==self_hash) // same: doing nothing
                                   return;
   
                           if(Value* vdefault=src->get_default())
                                   if(vdefault->is_defined())
                                           self.set_default(vdefault);
                   } else {
                           src_hash=vsrc.get_hash();
                   }
   
         if(!r.connection)                  if(src_hash)
                 PTHROW(0, 0,                          src_hash->for_each<HashStringValue*>(copy_all_overwrite_to, self_hash);
                         &method_name,  
                         "without connect");  
   
         Value& statement=params->as_junction(0, "statement must be code");  
   
         ulong limit=0;  
         if(params->size()>1) {  
                 Value& limit_code=params->as_junction(1, "limit must be expression");  
                 limit=(uint)r.process(limit_code).as_double();  
         }          }
   }
   
         ulong offset=0;  static void _sub(Request& r, MethodParams& params) {
         if(params->size()>2) {          if(HashStringValue* src=params.as_hash(0, "param")) {
                 Value& offset_code=params->as_junction(2, "offset must be expression");                  HashStringValue* self=&(GET_SELF(r, VHash).hash());
                 offset=(ulong)r.process(offset_code).as_double();                  if(src==self) { // same: clearing
                           self->clear();
                           return;
                   }
                   src->for_each<HashStringValue*>(remove_key_from, self);
         }          }
   }
   
         Temp_lang temp_lang(r, String::UL_SQL);  static void copy_all_dontoverwrite_to(
         const String& statement_string=r.process(statement).as_string();                                          HashStringValue::key_type key, 
         const char *statement_cstr=                                          HashStringValue::value_type value, 
                 statement_string.cstr(String::UL_UNSPECIFIED, r.connection);                                          HashStringValue* dest) {
         Hash& hash=static_cast<VHash *>(r.self)->hash();          dest->put_dont_replace(key, value);
         hash.clear();     }
         Hash_sql_event_handlers handlers(pool, method_name,  static void _union(Request& r, MethodParams& params) {
                 statement_string, statement_cstr, hash);          // dest = copy of self
         bool need_rethrow=false; Exception rethrow_me;          Value& result=*new VHash(GET_SELF(r, VHash).hash());
         PTRY {          // dest += b
                 r.connection->query(          if(HashStringValue* src=params.as_hash(0, "param"))
                         statement_cstr, offset, limit,                  src->for_each<HashStringValue*>(copy_all_dontoverwrite_to, result.get_hash());
                         handlers);  
         }  
         PCATCH(e) { // query problem  
                 rethrow_me=e;  need_rethrow=true;  
         }  
         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);          // return result
           r.write_no_lang(result);
   }
   
         VTable& result=*new(pool) VTable(pool, &table);  #ifndef DOXYGEN
         result.set_name(method_name);  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
           if(HashStringValue* b=params.as_hash(0, "param")) {
                   Copy_intersection_to_info info={b, result.get_hash()};
                   GET_SELF(r, VHash).hash().for_each<Copy_intersection_to_info*>(copy_intersection_to, &info);
           }
   
           // return result
         r.write_no_lang(result);          r.write_no_lang(result);
 }  }
   
 static void _count(Request& r, const String& method_name, MethodParams *) {  static bool intersects(
         Pool& pool=r.pool();                                          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& value=*new(pool) VInt(pool, static_cast<VHash *>(r.self)->hash().size());          if(HashStringValue* b=params.as_hash(0, "param"))
         value.set_name(method_name);                  result=GET_SELF(r, VHash).hash().first_that<HashStringValue*>(intersects, b)!=0;
         r.write_no_lang(value);  
           // return result
           r.write_no_lang(VBool::get(result));
 }  }
   
 // constructor  
   
 MHash::MHash(Pool& apool) : Methoded(apool) {  extern String sql_bind_name;
         set_name(*NEW String(pool(), HASH_CLASS_NAME));  extern String sql_limit_name;
   extern String sql_offset_name;
   extern String sql_default_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 void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
   
         // ^hash:sql[query][(count[;offset])]  static void _sql(Request& r, MethodParams& params) {
         add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3);          Value& statement=params.as_junction(0, "statement must be code");
   
         // ^hash._keys[]          HashStringValue* bind=0;
         add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0);              ulong limit=SQL_NO_LIMIT;
           ulong offset=0;
           bool distinct=false;
           Table2hash_value_type value_type=C_HASH;
           if(params.count()>1)
                   if(HashStringValue* options=params.as_hash(1, "sql options")) {
                           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(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())
                                   throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
                   }
   
           SQL_Driver::Placeholder* placeholders=0;
           uint placeholders_count=0;
           if(bind)
                   placeholders_count=marshal_binds(*bind, placeholders);
   
           Temp_lang temp_lang(r, String::L_SQL);
           const String& statement_string=r.process_to_string(statement);
           const char* statement_cstr=statement_string.untaint_cstr(r.flang, r.connection());
   
         // ^hash._count[]          HashStringValue& hash=GET_SELF(r, VHash).hash();
         add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);            hash.clear();   
           Hash_sql_event_handlers handlers(
                   statement_string, statement_cstr, 
                   distinct,
                   hash,
                   value_type);
   
           r.connection()->query(
                   statement_cstr, 
                   placeholders_count, placeholders,
                   offset, limit,
                   handlers,
                   statement_string);
   
           if(bind)
                   unmarshal_bind_updates(*bind, placeholders_count, placeholders);
 }  }
   
 // global variable  static void keys_collector(
                           HashStringValue::key_type key, 
                           HashStringValue::value_type, 
                           Table *table) {
           Table::element_type row(new ArrayString(1));
           *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(1));
           *columns+=keys_column_name;
           Table* table=new Table(columns);
   
           GET_SELF(r, VHash).hash_ro().for_each<Table*>(keys_collector, table);
   
           r.write_no_lang(*new VTable(table));
   }
   
   static void _count(Request& r, MethodParams&) {
           r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash_ro().count()));
   }
   
   static void _delete(Request& r, MethodParams& params) {
   
           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_ro().contains(params.as_string(0, "key must be string"));
           r.write_no_lang(VBool::get(result));
   }
   
   #ifndef DOXYGEN
   struct Foreach_info {
           Request *r;
           const String* key_var_name;
           const String* value_var_name;
           Value* body_code;
           Value* delim_maybe_code;
   
           Value* var_context;
           bool need_delim;
   };
   #endif
   static bool one_foreach_cycle(
                                   HashStringValue::key_type akey, 
                                   HashStringValue::value_type avalue, 
                                   Foreach_info *info) {
           Value& var_context=*info->var_context;
           if(info->key_var_name){
                   VString* vkey=new VString(*new String(akey, String::L_TAINTED));
                   var_context.put_element(*info->key_var_name, vkey, false);
           }
           if(info->value_var_name)
                   var_context.put_element(*info->value_var_name, avalue, false);
   
 Methoded *hash_base_class;          if(info->delim_maybe_code){ // delimiter set
                   StringOrValue sv_processed=info->r->process(*info->body_code);
                   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;
           }
   }
   static void _foreach(Request& r, MethodParams& params) {
           InCycle temp(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={
                   &r,
                   key_var_name.is_empty()? 0 : &key_var_name,
                   value_var_name.is_empty()? 0 : &value_var_name,
                   &params.as_junction(2, "body must be code"),
                   /*delimiter*/params.count()>3?params.get(3):0,
                   /*var_context*/r.get_method_frame()->caller(),
                   false
           };
   
           VHash& self=GET_SELF(r, VHash);
           HashStringValue& hash=self.hash_ro();
           VHash_lock lock(self);
           hash.first_that<Foreach_info*>(one_foreach_cycle, &info);
   }
   
   static void _at(Request& r, MethodParams& params) {
           HashStringValue& hash=GET_SELF(r, VHash).hash_ro();
           size_t count=hash.count();
   
           int pos=0;
   
           Value& vwhence=*params.get(0);
           if(vwhence.is_string()){
                   const String& swhence=*vwhence.get_string();
                   if(swhence == "last")
                           pos=count-1;
                   else if(swhence != "first")
                           throw Exception(PARSER_RUNTIME,
                                   &swhence,
                                   "whence must be 'first', 'last' or expression");
           } else {
                   pos=r.process_to_value(vwhence).as_int();
                   if(pos < 0)
                           pos+=count;
           }
   
           if(count && pos >= 0 && (size_t)pos < count){
                   if(pos == 0)
                           r.write_assign_lang(*hash.first_value());
                   else if((size_t)pos == count-1)
                           r.write_assign_lang(*hash.last_value());
                   else
                           for(HashStringValue::Iterator i(hash); i; i.next(), pos-- )
                                   if(!pos){
                                           r.write_assign_lang(*i.value());
                                           break;
                                   }
           }
           
   }
   
   // 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);
   
           // ^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);
   
           // ^hash._keys[[column name]]
           add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 1);    
   
           // ^hash._count[]
           add_native_method("_count", Method::CT_DYNAMIC, _count, 0, 0);  
   
 // creator          // ^hash.foreach[key;value]{code}[delim]
           add_native_method("foreach", Method::CT_DYNAMIC, _foreach, 2+1, 2+1+1);
   
 Methoded *MHash_create(Pool& pool) {          // ^hash._at[first|last]
         return hash_base_class=new(pool) MHash(pool);          // ^hash._at([-]offset)
           add_native_method("_at", Method::CT_DYNAMIC, _at, 1, 1);
 }  }
Removed from v.1.18  
changed lines
  Added in v.1.115

E-mail: