item.cc
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:15k
源码类别:

MySQL数据库

开发平台:

Visual C++

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