Diff for /parser3/src/classes/hash.C between versions 1.7 and 1.58

version 1.7, 2001/06/28 07:41:59 version 1.58, 2003/11/06 11:12:44
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-2003 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 char *RCSId="$Id$";   
   static const char* 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"
 // defines  #include "pa_vbool.h"
   #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() { return new VHash(); }
   
 public:  public:
         MHash(Pool& pool);          MHash();
 public: // Methoded  public: // Methoded
         bool used_directly() { return true; }          bool used_directly() { return true; }
 };  };
   
   // global variable
   
   DECLARE_CLASS_VAR(hash, new MHash, 0);
   
 // methods  // methods
   
 static void _default(Request& r, const String&, MethodParams *params) {   #ifndef DOXYGEN
         Pool& pool=r.pool();  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;
   public:
           Hash_sql_event_handlers(
                   const String& astatement_string, const char* astatement_cstr,
                   bool adistinct,
                   HashStringValue& arows_hash): 
                   statement_string(astatement_string), statement_cstr(astatement_cstr),
                   distinct(adistinct),
                   rows_hash(arows_hash),
                   row_hash(0),
                   column_index(0) {
           }
           bool add_column(SQL_Error& error, const char* str, size_t length) {
                   try {
                           columns+=new String(str, length, true);
                           return false;
                   } catch(...) {
                           error=SQL_Error("exception occured in Hash_sql_event_handlers::add_column");
                           return true;
                   }
           }
           bool before_rows(SQL_Error& error) { 
                   if(columns.count()<=1) {
                           error=SQL_Error("parser.runtime",
                                   /*method_name,*/
                                   "column count must be more than 1 to create a hash");
                           return true;
                   }
   
                   return false;
           }
           bool add_row(SQL_Error& /*error*/) {
                   column_index=0;
                   return false;
           }
           bool add_row_cell(SQL_Error& error, const char *ptr, size_t length) {
                   try {
                           String& cell=*new String;
                           if(length)
                                   cell.append_know_length(ptr, length, String::L_TAINTED);
                           if(column_index==0) {
                                   VHash* row_vhash=new VHash;
                                   row_hash=&row_vhash->hash();
                                   if(rows_hash.put_dont_replace(cell, row_vhash)) // put. existed?
                                           if(!distinct) {
                                                   error=SQL_Error("parser.runtime",
                                                           /*cell,*/
                                                           "duplicate key");
                                                   return true;
                                           }
                           } else
                                   row_hash->put(*columns[column_index], new VString(cell));
                           column_index++;
   
                           return false;
                   } catch(...) {
                           error=SQL_Error("exception occured in Hash_sql_event_handlers::add_row_cell");
                           return true;
                   }
           }
   
   };
   #endif
   
         VHash& vhash=*static_cast<VHash *>(r.self);  static void copy_all_overwrite_to(
         if(params->size())                                                                    HashStringValue::key_type key, 
                 vhash.set_default(params->get(0)); // info: may be code..                                                                    HashStringValue::value_type value, 
         else {                                                                    HashStringValue* dest) {
                 Value *default_value=vhash.get_default();          dest->put(key, value);
                 r.write_assign_lang(default_value?*default_value:*new(pool) VVoid(pool));  }
   static void _create_or_add(Request& r, MethodParams& params) {
           if(params.count()) {
                   Value& vb=params.as_no_junction(0, "param must be hash");
                   if(HashStringValue* b=vb.get_hash())
                           b->for_each(copy_all_overwrite_to, &(GET_SELF(r, VHash).hash()));
         }          }
 }  }
   
 static void _sql(Request& r, const String& method_name, MethodParams *params) {  static void remove_key_from(
         Pool& pool=r.pool();                                                          HashStringValue::key_type key, 
                                                           HashStringValue::value_type /*value*/, 
                                                           HashStringValue* dest) {
           dest->remove(key);
   }
   static void _sub(Request& r, MethodParams& params) {
           Value& vb=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* b=vb.get_hash())
                   b->for_each(remove_key_from, &GET_SELF(r, VHash).hash());
   }
   
         if(!r.connection)  static void copy_all_dontoverwrite_to(
                 PTHROW(0, 0,                                                                    HashStringValue::key_type key, 
                         &method_name,                                                                    HashStringValue::value_type value, 
                         "without connect");                                                                    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& vb=params.as_no_junction(0, "param must be hash");
           if(HashStringValue* b=vb.get_hash())
                   b->for_each(copy_all_dontoverwrite_to, result.get_hash());
   
         Value& statement=params->get_junction(0, "statement must be code");          // return result
           r.write_no_lang(result);
   }
   
         ulong limit=0;  #ifndef DOXYGEN
         if(params->size()>1) {  struct Copy_intersection_to_info {
                 Value& limit_code=params->get_junction(1, "limit must be expression");          HashStringValue* b;
                 limit=(uint)r.process(limit_code).as_double();          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_limit_name;
   extern String sql_offset_name;
   extern String sql_default_name;
   extern String sql_distinct_name;
   static void _sql(Request& r, MethodParams& params) {
           Value& statement=params.as_junction(0, "statement must be code");
   
           ulong limit=0;
         ulong offset=0;          ulong offset=0;
         if(params->size()>2) {          bool distinct=false;
                 Value& offset_code=params->get_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()) {
         Temp_lang temp_lang(r, String::UL_SQL);                                  int valid_options=0;
         const String& statement_string=r.process(statement).as_string();                                  if(Value* vlimit=options->get(sql_limit_name)) {
         const char *statement_cstr=                                          valid_options++;
                 statement_string.cstr(String::UL_UNSPECIFIED, r.connection);                                          limit=(ulong)r.process_to_value(*vlimit).as_double();
         unsigned int sql_column_count; SQL_Driver::Cell *sql_columns;                                  }
         unsigned long sql_row_count; SQL_Driver::Cell **sql_rows;                                  if(Value* voffset=options->get(sql_offset_name)) {
         bool need_rethrow=false; Exception rethrow_me;                                          valid_options++;
         PTRY {                                          offset=(ulong)r.process_to_value(*voffset).as_double();
                 r.connection->query(                                  }
                         statement_cstr, offset, limit,                                  if(Value* vdistinct=options->get(sql_distinct_name)) {
                         &sql_column_count, &sql_columns,                                          valid_options++;
                         &sql_row_count, &sql_rows);                                          distinct=r.process_to_value(*vdistinct).as_bool();
         }                                  }
         PCATCH(e) { // query problem                                  if(valid_options!=options->count())
                 rethrow_me=e;  need_rethrow=true;                                          throw Exception("parser.runtime",
         }                                                  0,
         PEND_CATCH                                                  "called with invalid option");
         if(need_rethrow)                          } else
                 PTHROW(rethrow_me.type(), rethrow_me.code(),                                  throw Exception("parser.runtime",
                         &statement_string, // setting more specific source [were url]                                          0,
                         rethrow_me.comment());                                          "options must be hash");
   
         Hash& rows_hash=static_cast<VHash *>(r.self)->hash();  
         rows_hash.clear();        
   
         if(sql_column_count<=1)  
                 return;  
   
         Array& columns=*new(pool) Array(pool);  
         for(unsigned int i=0+1; i<sql_column_count; i++) {  
                 String& column=*new(pool) String(pool);  
                 column.APPEND_TAINTED(  
                         (const char *)sql_columns[i].ptr, sql_columns[i].size,  
                         statement_cstr, 0);  
                 columns+=&column;  
         }  
           
         for(unsigned long row=0; row<sql_row_count; row++) {  
                 SQL_Driver::Cell *sql_cells=sql_rows[row];                        
                   
                 VHash& row_vhash=*new(pool) VHash(pool);  
                 Hash& row_hash=*row_vhash.get_hash();  
                 String *key=0; // calm, compiler  
                 String *cell;  
                 for(unsigned int i=0; i<sql_column_count; i++) {  
                         cell=new(pool) String(pool);  
                         cell->APPEND_TAINTED(  
                                 (const char *)sql_cells[i].ptr, sql_cells[i].size,  
                                 statement_cstr, row);  
                         if(i==0)  
                                 key=cell;  
                         else  
                                 row_hash.put(*columns.get_string(i-1), new(pool) VString(*cell));  
                 }  
                 rows_hash.put(*key, &row_vhash);  
         }          }
   
           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_sql_event_handlers handlers(
                   statement_string, statement_cstr, 
                   distinct,
                   hash);
           r.connection()->query(
                   statement_cstr, offset, limit,
                   handlers,
                   statement_string);
 }  }
   
 // constructor  static void keys_collector(
                              HashStringValue::key_type key, 
                              HashStringValue::value_type value, 
                              Table *table) {
           Table::element_type row(new ArrayString);
           *row+=new String(key, String::L_TAINTED);
           *table+=row;
   }
   static void _keys(Request& r, MethodParams&) {
           Table::columns_type columns(new ArrayString);
           *columns+=new String("key");
           Table* table=new Table(columns);
   
 MHash::MHash(Pool& apool) : Methoded(apool) {          GET_SELF(r, VHash).hash().for_each(keys_collector, table);
         set_name(*NEW String(pool(), HASH_CLASS_NAME));  
   
         // ^hash.default[]          r.write_no_lang(*new VTable(table));
         // ^hash.default[hash]  }
         add_native_method("default", Method::CT_DYNAMIC, _default, 0, 1);  
   
         // ^hash:sql[query][(count[;offset])]  static void _count(Request& r, MethodParams&) {
         add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3);          r.write_no_lang(*new VInt(GET_SELF(r, VHash).hash().count()));
   }
   
   static void _delete(Request& r, MethodParams& params) {
   
           GET_SELF(r, VHash).hash().remove(params.as_string(0, "key must be string"));
 }  }
   
 // global variable  #ifndef DOXYGEN
   struct Hash_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, 
                                 Hash_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) {
           Hash_foreach_info info={0};
           info.r=&r;
           info.key_var_name=&params.as_string(0, "key-var name must be string");
           info.value_var_name=&params.as_string(1, "value-var name must be string");
           info.body_code=&params.as_junction(2, "body must be code");
           info.delim_maybe_code=params.count()>3?params.get(3):0;
           info.vkey=new VString;
   
           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][$.limit(1) $.offset(2)]
           add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2);
   
 Methoded *hash_base_class;          // ^hash._keys[]
           add_native_method("_keys", Method::CT_DYNAMIC, _keys, 0, 0);    
   
 // 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.7  
changed lines
  Added in v.1.58


E-mail: