home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / sql / item.cpp < prev    next >
C/C++ Source or Header  |  2000-08-31  |  15KB  |  682 lines
  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  7.    
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  16.  
  17.  
  18. #ifdef __GNUC__
  19. #pragma implementation                // gcc: Class implementation
  20. #endif
  21.  
  22. #include "mysql_priv.h"
  23. #include <m_ctype.h>
  24. #include "my_dir.h"
  25.  
  26. /*****************************************************************************
  27. ** Item functions
  28. *****************************************************************************/
  29.  
  30. /* Init all special items */
  31.  
  32. void item_init(void)
  33. {
  34.   item_user_lock_init();
  35. }
  36.  
  37. Item::Item()
  38. {
  39.   marker=0;
  40.   binary=maybe_null=null_value=with_sum_func=0;
  41.   name=0;
  42.   decimals=0; max_length=0;
  43.   next=current_thd->free_list;            // Put in free list
  44.   current_thd->free_list=this;
  45. }
  46.  
  47. void Item::set_name(char *str,uint length)
  48. {
  49.   if (!length)
  50.     name=str;                    // Used by AS
  51.   else
  52.   {
  53.     while (length && !isgraph(*str))
  54.     {                        // Fix problem with yacc
  55.       length--;
  56.       str++;
  57.     }
  58.     name=sql_strmake(str,min(length,MAX_FIELD_WIDTH));
  59.   }
  60. }
  61.  
  62. bool Item::eq(const Item *item) const        // Only doing this on conds
  63. {
  64.   return type() == item->type() && name && item->name &&
  65.     !my_strcasecmp(name,item->name);
  66. }
  67.  
  68. /*
  69.   Get the value of the function as a TIME structure.
  70.   As a extra convenience the time structure is reset on error!
  71.  */
  72.  
  73. bool Item::get_date(TIME *ltime,bool fuzzydate)
  74. {
  75.   char buff[40];
  76.   String tmp(buff,sizeof(buff)),*res;
  77.   if (!(res=val_str(&tmp)) ||
  78.       str_to_TIME(res->ptr(),res->length(),ltime,0) == TIMESTAMP_NONE)
  79.   {
  80.     bzero((char*) ltime,sizeof(*ltime));
  81.     return 1;
  82.   }
  83.   return 0;
  84. }
  85.  
  86. /*
  87.   Get time of first argument.
  88.   As a extra convenience the time structure is reset on error!
  89.  */
  90.  
  91. bool Item::get_time(TIME *ltime)
  92. {
  93.   char buff[40];
  94.   String tmp(buff,sizeof(buff)),*res;
  95.   if (!(res=val_str(&tmp)) ||
  96.       str_to_time(res->ptr(),res->length(),ltime))
  97.   {
  98.     bzero((char*) ltime,sizeof(*ltime));
  99.     return 1;
  100.   }
  101.   return 0;
  102. }
  103.  
  104. Item_field::Item_field(Field *f) :Item_ident(NullS,f->table_name,f->field_name)
  105. {
  106.   set_field(f);
  107. }
  108.  
  109.  
  110. void Item_field::set_field(Field *field_par)
  111. {
  112.   field=result_field=field_par;            // for easy coding with fields
  113.   maybe_null=field->maybe_null();
  114.   max_length=field_par->field_length;
  115.   decimals= field->decimals();
  116.   table_name=field_par->table_name;
  117.   field_name=field_par->field_name;
  118.   binary=field_par->binary();
  119. }
  120.  
  121. const char *Item_ident::full_name() const
  122. {
  123.   char *tmp;
  124.   if (!table_name)
  125.     return field_name ? field_name : name ? name : "tmp_field";
  126.   if (db_name)
  127.   {
  128.     tmp=(char*) sql_alloc((uint) strlen(db_name)+(uint) strlen(table_name)+
  129.               (uint) strlen(field_name)+3);
  130.     strxmov(tmp,db_name,".",table_name,".",field_name,NullS);
  131.   }
  132.   else
  133.   {
  134.     tmp=(char*) sql_alloc((uint) strlen(table_name)+
  135.               (uint) strlen(field_name)+2);
  136.     strxmov(tmp,table_name,".",field_name,NullS);
  137.   }
  138.   return tmp;
  139. }
  140.  
  141. /* ARGSUSED */
  142. String *Item_field::val_str(String *str)
  143. {
  144.   if ((null_value=field->is_null()))
  145.     return 0;
  146.   return field->val_str(str,&str_value);
  147. }
  148.  
  149. double Item_field::val()
  150. {
  151.   if ((null_value=field->is_null()))
  152.     return 0.0;
  153.   return field->val_real();
  154. }
  155.  
  156. longlong Item_field::val_int()
  157. {
  158.   if ((null_value=field->is_null()))
  159.     return 0;
  160.   return field->val_int();
  161. }
  162.  
  163.  
  164. String *Item_field::str_result(String *str)
  165. {
  166.   if ((null_value=result_field->is_null()))
  167.     return 0;
  168.   return result_field->val_str(str,&str_value);
  169. }
  170.  
  171. bool Item_field::get_date(TIME *ltime,bool fuzzydate)
  172. {
  173.   if ((null_value=field->is_null()) || field->get_date(ltime,fuzzydate))
  174.   {
  175.     bzero((char*) ltime,sizeof(*ltime));
  176.     return 1;
  177.   }
  178.   return 0;
  179. }
  180.  
  181. bool Item_field::get_time(TIME *ltime)
  182. {
  183.   if ((null_value=field->is_null()) || field->get_time(ltime))
  184.   {
  185.     bzero((char*) ltime,sizeof(*ltime));
  186.     return 1;
  187.   }
  188.   return 0;
  189. }
  190.  
  191. double Item_field::val_result()
  192. {
  193.   if ((null_value=result_field->is_null()))
  194.     return 0.0;
  195.   return result_field->val_real();
  196. }
  197.  
  198. longlong Item_field::val_int_result()
  199. {
  200.   if ((null_value=result_field->is_null()))
  201.     return 0;
  202.   return result_field->val_int();
  203. }
  204.  
  205. bool Item_field::eq(const Item *item) const
  206. {
  207.   return item->type() == FIELD_ITEM && ((Item_field*) item)->field == field;
  208. }
  209.  
  210. table_map Item_field::used_tables() const
  211. {
  212.   if (field->table->const_table)
  213.     return 0;                    // const item
  214.   return field->table->map;
  215. }
  216.  
  217.  
  218. String *Item_int::val_str(String *str)
  219. {
  220.   str->set(value);
  221.   return str;
  222. }
  223.  
  224. void Item_int::print(String *str)
  225. {
  226.   if (!name)
  227.   {
  228.     str_value.set(value);
  229.     name=str_value.c_ptr();
  230.   }
  231.   str->append(name);
  232. }
  233.  
  234.  
  235. String *Item_real::val_str(String *str)
  236. {
  237.   str->set(value,decimals);
  238.   return str;
  239. }
  240.  
  241. void Item_string::print(String *str)
  242. {
  243.   str->append('\'');
  244.   str->append(full_name());
  245.   str->append('\'');
  246. }
  247.  
  248. bool Item_null::eq(const Item *item) const { return item->type() == type(); }
  249. double Item_null::val() { null_value=1; return 0.0; }
  250. longlong Item_null::val_int() { null_value=1; return 0; }
  251. /* ARGSUSED */
  252. String *Item_null::val_str(String *str)
  253. { null_value=1; return 0;}
  254.  
  255.  
  256. void Item_copy_string::copy()
  257. {
  258.   String *res=item->val_str(&str_value);
  259.   if (res && res != &str_value)
  260.     str_value.copy(*res);
  261.   null_value=item->null_value;
  262. }
  263.  
  264. /* ARGSUSED */
  265. String *Item_copy_string::val_str(String *str)
  266. {
  267.   if (null_value)
  268.     return (String*) 0;
  269.   return &str_value;
  270. }
  271.  
  272. /*
  273. ** Functions to convert item to field (for send_fields)
  274. */
  275.  
  276. /* ARGSUSED */
  277. bool Item::fix_fields(THD *thd,
  278.               struct st_table_list *list)
  279. {
  280.   return 0;
  281. }
  282.  
  283. bool Item_field::fix_fields(THD *thd,TABLE_LIST *tables)
  284. {
  285.   if (!field)
  286.   {
  287.     Field *tmp;
  288.     if (!(tmp=find_field_in_tables(thd,this,tables)))
  289.       return 1;
  290.     set_field(tmp);
  291.   }
  292.   return 0;
  293. }
  294.  
  295.  
  296. void Item::init_make_field(Send_field *tmp_field,
  297.                enum enum_field_types field_type)
  298. {
  299.   tmp_field->table_name=(char*) "";
  300.   tmp_field->col_name=name;
  301.   tmp_field->flags=maybe_null ? 0 : NOT_NULL_FLAG;
  302.   tmp_field->type=field_type;
  303.   tmp_field->length=max_length;
  304.   tmp_field->decimals=decimals;
  305. }
  306.  
  307. /* ARGSUSED */
  308. void Item_field::make_field(Send_field *tmp_field)
  309. {
  310.   field->make_field(tmp_field);
  311.   if (name)
  312.     tmp_field->col_name=name;            // Use user supplied name
  313. }
  314.  
  315. void Item_int::make_field(Send_field *tmp_field)
  316. {
  317.   init_make_field(tmp_field,FIELD_TYPE_LONGLONG);
  318. }
  319.  
  320. void Item_real::make_field(Send_field *tmp_field)
  321. {
  322.   init_make_field(tmp_field,FIELD_TYPE_DOUBLE);
  323. }
  324.  
  325. void Item_string::make_field(Send_field *tmp_field)
  326. {
  327.   init_make_field(tmp_field,FIELD_TYPE_STRING);
  328. }
  329.  
  330. void Item_datetime::make_field(Send_field *tmp_field)
  331. {
  332.   init_make_field(tmp_field,FIELD_TYPE_DATETIME);
  333. }
  334.  
  335. void Item_null::make_field(Send_field *tmp_field)
  336. {
  337.   init_make_field(tmp_field,FIELD_TYPE_NULL);
  338.   tmp_field->length=4;
  339. }
  340.  
  341. void Item_func::make_field(Send_field *tmp_field)
  342. {
  343.   init_make_field(tmp_field, ((result_type() == STRING_RESULT) ?
  344.                   FIELD_TYPE_VAR_STRING :
  345.                   (result_type() == INT_RESULT) ?
  346.                   FIELD_TYPE_LONGLONG : FIELD_TYPE_DOUBLE));
  347. }
  348.  
  349. void Item_avg_field::make_field(Send_field *tmp_field)
  350. {
  351.   init_make_field(tmp_field,FIELD_TYPE_DOUBLE);
  352. }
  353.  
  354. void Item_std_field::make_field(Send_field *tmp_field)
  355. {
  356.   init_make_field(tmp_field,FIELD_TYPE_DOUBLE);
  357. }
  358.  
  359. /*
  360. ** Set a field:s value from a item
  361. */
  362.  
  363.  
  364. void Item_field::save_org_in_field(Field *to)
  365. {
  366.   if (field->is_null())
  367.   {
  368.     null_value=1;
  369.     set_field_to_null(to);
  370.   }
  371.   else
  372.   {
  373.     to->set_notnull();
  374.     field_conv(to,field);
  375.     null_value=0;
  376.   }
  377. }
  378.  
  379. bool Item_field::save_in_field(Field *to)
  380. {
  381.   if (result_field->is_null())
  382.   {
  383.     null_value=1;
  384.     return set_field_to_null(to);
  385.   }
  386.   else
  387.   {
  388.     to->set_notnull();
  389.     field_conv(to,result_field);
  390.     null_value=0;
  391.   }
  392.   return 0;
  393. }
  394.  
  395.  
  396. bool Item_null::save_in_field(Field *field)
  397. {
  398.   return set_field_to_null(field);
  399. }
  400.  
  401.  
  402. bool Item::save_in_field(Field *field)
  403. {
  404.   if (result_type() == STRING_RESULT ||
  405.       result_type() == REAL_RESULT &&
  406.       field->result_type() == STRING_RESULT)
  407.   {
  408.     String *result;
  409.     char buff[MAX_FIELD_WIDTH];        // Alloc buffer for small columns
  410.     str_value.set_quick(buff,sizeof(buff));
  411.     result=val_str(&str_value);
  412.     if (null_value)
  413.       return set_field_to_null(field);
  414.     field->set_notnull();
  415.     field->store(result->ptr(),result->length());
  416.     str_value.set_quick(0, 0);
  417.   }
  418.   else if (result_type() == REAL_RESULT)
  419.   {
  420.     double nr=val();
  421.     if (null_value)
  422.       return set_field_to_null(field);
  423.     field->set_notnull();
  424.     field->store(nr);
  425.   }
  426.   else
  427.   {
  428.     longlong nr=val_int();
  429.     if (null_value)
  430.       return set_field_to_null(field);
  431.     field->set_notnull();
  432.     field->store(nr);
  433.   }
  434.   return 0;
  435. }
  436.  
  437. bool Item_string::save_in_field(Field *field)
  438. {
  439.   String *result;
  440.   result=val_str(&str_value);
  441.   if (null_value)
  442.     return set_field_to_null(field);
  443.   field->set_notnull();
  444.   field->store(result->ptr(),result->length());
  445.   return 0;
  446. }
  447.  
  448. bool Item_int::save_in_field(Field *field)
  449. {
  450.   longlong nr=val_int();
  451.   if (null_value)
  452.     return set_field_to_null(field);
  453.   field->set_notnull();
  454.   field->store(nr);
  455.   return 0;
  456. }
  457.  
  458. bool Item_real::save_in_field(Field *field)
  459. {
  460.   double nr=val();
  461.   if (null_value)
  462.     return set_field_to_null(field);
  463.   field->set_notnull();
  464.   field->store(nr);
  465.   return 0;
  466. }
  467.  
  468. /****************************************************************************
  469. ** varbinary item
  470. ** In string context this is a binary string
  471. ** In number context this is a longlong value.
  472. ****************************************************************************/
  473.  
  474. static inline uint char_val(char X)
  475. {
  476.   return (uint) (X >= '0' && X <= '9' ? X-'0' :
  477.          X >= 'A' && X <= 'Z' ? X-'A'+10 :
  478.          X-'a'+10);
  479. }
  480.  
  481. Item_varbinary::Item_varbinary(const char *str, uint str_length)
  482. {
  483.   name=(char*) str-2;                // Lex makes this start with 0x
  484.   max_length=(str_length+1)/2;
  485.   char *ptr=(char*) sql_alloc(max_length+1);
  486.   if (!ptr)
  487.     return;
  488.   str_value.set(ptr,max_length);
  489.   char *end=ptr+max_length;
  490.   if (max_length*2 != str_length)
  491.     *ptr++=char_val(*str++);            // Not even, assume 0 prefix
  492.   while (ptr != end)
  493.   {
  494.     *ptr++= (char) (char_val(str[0])*16+char_val(str[1]));
  495.     str+=2;
  496.   }
  497.   *ptr=0;                    // Keep purify happy
  498.   binary=1;                    // Binary is default
  499. }
  500.  
  501. longlong Item_varbinary::val_int()
  502. {
  503.   char *end=(char*) str_value.ptr()+str_value.length(),
  504.        *ptr=end-min(str_value.length(),sizeof(longlong));
  505.  
  506.   ulonglong value=0;
  507.   for (; ptr != end ; ptr++)
  508.     value=(value << 8)+ (ulonglong) (uchar) *ptr;
  509.   return (longlong) value;
  510. }
  511.  
  512.  
  513. bool Item_varbinary::save_in_field(Field *field)
  514. {
  515.   field->set_notnull();
  516.   if (field->result_type() == STRING_RESULT)
  517.   {
  518.     field->store(str_value.ptr(),str_value.length());
  519.   }
  520.   else
  521.   {
  522.     longlong nr=val_int();
  523.     field->store(nr);
  524.   }
  525.   return 0;
  526. }
  527.  
  528.  
  529. void Item_varbinary::make_field(Send_field *tmp_field)
  530. {
  531.   init_make_field(tmp_field,FIELD_TYPE_STRING);
  532. }
  533.  
  534. /*
  535. ** pack data in buffer for sending
  536. */
  537.  
  538. bool Item::send(String *packet)
  539. {
  540.   char buff[MAX_FIELD_WIDTH];
  541.   String s(buff,sizeof(buff)),*res;
  542.   if (!(res=val_str(&s)))
  543.     return net_store_null(packet);
  544.   CONVERT *convert;
  545.   if ((convert=current_thd->convert_set))
  546.     return convert->store(packet,res->ptr(),res->length());
  547.   return net_store_data(packet,res->ptr(),res->length());
  548. }
  549.  
  550. bool Item_null::send(String *packet)
  551. {
  552.   return net_store_null(packet);
  553. }
  554.  
  555. /*
  556.   This is used for HAVING clause
  557.   Find field in select list having the same name
  558.  */
  559.  
  560. bool Item_ref::fix_fields(THD *thd,TABLE_LIST *tables)
  561. {
  562.   if (!ref)
  563.   {
  564.     if (!(ref=find_item_in_list(this,thd->lex.item_list)))
  565.       return 1;
  566.     max_length= (*ref)->max_length;
  567.     maybe_null= (*ref)->maybe_null;
  568.     decimals=    (*ref)->decimals;
  569.     binary=    (*ref)->binary;
  570.   }
  571.   return 0;
  572. }
  573.  
  574. /*
  575. ** If item is a const function, calculate it and return a const item
  576. ** The original item is freed if not returned
  577. */
  578.  
  579. Item_result item_cmp_type(Item_result a,Item_result b)
  580. {
  581.   if (a == STRING_RESULT && b == STRING_RESULT)
  582.     return STRING_RESULT;
  583.   else if (a == INT_RESULT && b == INT_RESULT)
  584.     return INT_RESULT;
  585.   else
  586.     return REAL_RESULT;
  587. }
  588.  
  589.  
  590. Item *resolve_const_item(Item *item,Item *comp_item)
  591. {
  592.   if (item->basic_const_item())
  593.     return item;                // Can't be better
  594.   Item_result res_type=item_cmp_type(comp_item->result_type(),
  595.                      item->result_type());
  596.   char *name=item->name;            // Alloced by sql_alloc
  597.  
  598.   if (res_type == STRING_RESULT)
  599.   {
  600.     char buff[MAX_FIELD_WIDTH];
  601.     String tmp(buff,sizeof(buff)),*result;
  602.     result=item->val_str(&tmp);
  603.     if (item->null_value)
  604.     {
  605. #ifdef DELETE_ITEMS
  606.       delete item;
  607. #endif
  608.       return new Item_null(name);
  609.     }
  610.     uint length=result->length();
  611.     char *tmp_str=sql_strmake(result->ptr(),length);
  612. #ifdef DELETE_ITEMS
  613.     delete item;
  614. #endif
  615.     return new Item_string(name,tmp_str,length);
  616.   }
  617.   if (res_type == INT_RESULT)
  618.   {
  619.     longlong result=item->val_int();
  620.     uint length=item->max_length;
  621.     bool null_value=item->null_value;
  622. #ifdef DELETE_ITEMS
  623.     delete item;
  624. #endif
  625.     return (null_value ? (Item*) new Item_null(name) :
  626.         (Item*) new Item_int(name,result,length));
  627.   }
  628.   else
  629.   {                        // It must REAL_RESULT
  630.     double result=item->val();
  631.     uint length=item->max_length,decimals=item->decimals;
  632.     bool null_value=item->null_value;
  633. #ifdef DELETE_ITEMS
  634.     delete item;
  635. #endif
  636.     return (null_value ? (Item*) new Item_null(name) :
  637.         (Item*) new Item_real(name,result,decimals,length));
  638.   }
  639. }
  640.  
  641. /*
  642.   Return true if the value stored in the field is equal to the const item
  643.   We need to use this on the range optimizer because in some cases
  644.   we can't store the value in the field without some precision/character loss.
  645. */
  646.  
  647. bool field_is_equal_to_item(Field *field,Item *item)
  648. {
  649.  
  650.   Item_result res_type=item_cmp_type(field->result_type(),
  651.                      item->result_type());
  652.   if (res_type == STRING_RESULT)
  653.   {
  654.     char item_buff[MAX_FIELD_WIDTH];
  655.     char field_buff[MAX_FIELD_WIDTH];
  656.     String item_tmp(item_buff,sizeof(item_buff)),*item_result;
  657.     String field_tmp(field_buff,sizeof(field_buff));
  658.     item_result=item->val_str(&item_tmp);
  659.     if (item->null_value)
  660.       return 1;                    // This must be true
  661.     field->val_str(&field_tmp,&field_tmp);
  662.     return !stringcmp(&field_tmp,item_result);
  663.   }
  664.   if (res_type == INT_RESULT)
  665.     return 1;                    // Both where of type int
  666.   double result=item->val();
  667.   if (item->null_value)
  668.     return 1;
  669.   return result == field->val_real();
  670. }
  671.  
  672.  
  673. /*****************************************************************************
  674. ** Instantiate templates
  675. *****************************************************************************/
  676.  
  677. #ifdef __GNUC__
  678. template class List<Item>;
  679. template class List_iterator<Item>;
  680. template class List<List_item>;
  681. #endif
  682.