item_sum.cc 84.5 KB
Newer Older
unknown's avatar
unknown committed
1
/* Copyright (C) 2000-2003 MySQL AB
unknown's avatar
unknown committed
2

unknown's avatar
unknown committed
3 4
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
unknown's avatar
unknown committed
5
   the Free Software Foundation; version 2 of the License.
unknown's avatar
unknown committed
6

unknown's avatar
unknown committed
7 8 9 10
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
unknown's avatar
unknown committed
11

unknown's avatar
unknown committed
12 13 14 15 16 17 18
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */


/* Sum functions (COUNT, MIN...) */

19
#ifdef USE_PRAGMA_IMPLEMENTATION
unknown's avatar
unknown committed
20 21 22 23
#pragma implementation				// gcc: Class implementation
#endif

#include "mysql_priv.h"
24
#include "sql_select.h"
unknown's avatar
unknown committed
25

unknown's avatar
unknown committed
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
/* 
  Prepare an aggregate function item for checking context conditions

  SYNOPSIS
    init_sum_func_check()
    thd      reference to the thread context info

  DESCRIPTION
    The function initializes the members of the Item_sum object created
    for a set function that are used to check validity of the set function
    occurrence.
    If the set function is not allowed in any subquery where it occurs
    an error is reported immediately.

  NOTES
    This function is to be called for any item created for a set function
    object when the traversal of trees built for expressions used in the query
    is performed at the phase of context analysis. This function is to
    be invoked at the descent of this traversal.
  
  RETURN
    TRUE   if an error is reported     
    FALSE  otherwise
*/
 
bool Item_sum::init_sum_func_check(THD *thd)
{
  if (!thd->lex->allow_sum_func)
  {
    my_message(ER_INVALID_GROUP_FUNC_USE, ER(ER_INVALID_GROUP_FUNC_USE),
               MYF(0));
    return TRUE;
  }
  /* Set a reference to the nesting set function if there is  any */
  in_sum_func= thd->lex->in_sum_func;
  /* Save a pointer to object to be used in items for nested set functions */
  thd->lex->in_sum_func= this;
  nest_level= thd->lex->current_select->nest_level;
  ref_by= 0;
  aggr_level= -1;
66
  aggr_sel= NULL;
unknown's avatar
unknown committed
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
  max_arg_level= -1;
  max_sum_func_level= -1;
  return FALSE;
}

/* 
  Check constraints imposed on a usage of a set function

  SYNOPSIS
    check_sum_func()
    thd      reference to the thread context info
    ref      location of the pointer to this item in the embedding expression 

  DESCRIPTION
    The method verifies whether context conditions imposed on a usage
    of any set function are met for this occurrence.
    It checks whether the set function occurs in the position where it
    can be aggregated and, when it happens to occur in argument of another
    set function, the method checks that these two functions are aggregated in
    different subqueries.
    If the context conditions are not met the method reports an error.
    If the set function is aggregated in some outer subquery the method
    adds it to the chain of items for such set functions that is attached
    to the the st_select_lex structure for this subquery.

  NOTES
    This function is to be called for any item created for a set function
    object when the traversal of trees built for expressions used in the query
    is performed at the phase of context analysis. This function is to
    be invoked at the ascent of this traversal.

  IMPLEMENTATION
    A number of designated members of the object are used to check the
    conditions. They are specified in the comment before the Item_sum
    class declaration.
    Additionally a bitmap variable called allow_sum_func is employed.
    It is included into the thd->lex structure.
    The bitmap contains 1 at n-th position if the set function happens
    to occur under a construct of the n-th level subquery where usage
    of set functions are allowed (i.e either in the SELECT list or
    in the HAVING clause of the corresponding subquery)
    Consider the query:
      SELECT SUM(t1.b) FROM t1 GROUP BY t1.a
        HAVING t1.a IN (SELECT t2.c FROM t2 WHERE AVG(t1.b) > 20) AND
               t1.a > (SELECT MIN(t2.d) FROM t2);
    allow_sum_func will contain: 
    for SUM(t1.b) - 1 at the first position 
    for AVG(t1.b) - 1 at the first position, 0 at the second position
    for MIN(t2.d) - 1 at the first position, 1 at the second position.

  RETURN
    TRUE   if an error is reported     
    FALSE  otherwise
*/
 
bool Item_sum::check_sum_func(THD *thd, Item **ref)
{
  bool invalid= FALSE;
  nesting_map allow_sum_func= thd->lex->allow_sum_func;
  /*  
    The value of max_arg_level is updated if an argument of the set function
    contains a column reference resolved  against a subquery whose level is
    greater than the current value of max_arg_level.
    max_arg_level cannot be greater than nest level.
    nest level is always >= 0  
  */ 
  if (nest_level == max_arg_level)
  {
    /*
      The function must be aggregated in the current subquery, 
      If it is there under a construct where it is not allowed 
      we report an error. 
    */ 
    invalid= !(allow_sum_func & (1 << max_arg_level));
  }
  else if (max_arg_level >= 0 || !(allow_sum_func & (1 << nest_level)))
  {
    /*
      The set function can be aggregated only in outer subqueries.
      Try to find a subquery where it can be aggregated;
      If we fail to find such a subquery report an error.
    */
    if (register_sum_func(thd, ref))
      return TRUE;
    invalid= aggr_level < 0 && !(allow_sum_func & (1 << nest_level));
unknown's avatar
unknown committed
152 153
    if (!invalid && thd->variables.sql_mode & MODE_ANSI)
      invalid= aggr_level < 0 && max_arg_level < nest_level;
unknown's avatar
unknown committed
154 155
  }
  if (!invalid && aggr_level < 0)
156
  {
unknown's avatar
unknown committed
157
    aggr_level= nest_level;
158 159
    aggr_sel= thd->lex->current_select;
  }
unknown's avatar
unknown committed
160 161 162 163 164 165 166 167 168
  /*
    By this moment we either found a subquery where the set function is
    to be aggregated  and assigned a value that is  >= 0 to aggr_level,
    or set the value of 'invalid' to TRUE to report later an error. 
  */
  /* 
    Additionally we have to check whether possible nested set functions
    are acceptable here: they are not, if the level of aggregation of
    some of them is less than aggr_level.
unknown's avatar
unknown committed
169 170 171
  */
  if (!invalid) 
    invalid= aggr_level <= max_sum_func_level;
unknown's avatar
unknown committed
172 173 174 175 176 177
  if (invalid)  
  {
    my_message(ER_INVALID_GROUP_FUNC_USE, ER(ER_INVALID_GROUP_FUNC_USE),
               MYF(0));
    return TRUE;
  }
unknown's avatar
unknown committed
178
  if (in_sum_func)
unknown's avatar
unknown committed
179 180 181 182
  {
    /*
      If the set function is nested adjust the value of
      max_sum_func_level for the nesting set function.
unknown's avatar
unknown committed
183 184 185 186 187 188 189 190 191 192
      We take into account only enclosed set functions that are to be 
      aggregated on the same level or above of the nest level of 
      the enclosing set function.
      But we must always pass up the max_sum_func_level because it is
      the maximum nested level of all directly and indirectly enclosed
      set functions. We must do that even for set functions that are
      aggregated inside of their enclosing set function's nest level
      because the enclosing function may contain another enclosing
      function that is to be aggregated outside or on the same level
      as its parent's nest level.
unknown's avatar
unknown committed
193
    */
unknown's avatar
unknown committed
194 195 196
    if (in_sum_func->nest_level >= aggr_level)
      set_if_bigger(in_sum_func->max_sum_func_level, aggr_level);
    set_if_bigger(in_sum_func->max_sum_func_level, max_sum_func_level);
unknown's avatar
unknown committed
197
  }
unknown's avatar
unknown committed
198
  update_used_tables();
unknown's avatar
unknown committed
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241
  thd->lex->in_sum_func= in_sum_func;
  return FALSE;
}

/* 
  Attach a set function to the subquery where it must be aggregated

  SYNOPSIS
    register_sum_func()
    thd      reference to the thread context info
    ref      location of the pointer to this item in the embedding expression
 
  DESCRIPTION
    The function looks for an outer subquery where the set function must be
    aggregated. If it finds such a subquery then aggr_level is set to
    the nest level of this subquery and the item for the set function
    is added to the list of set functions used in nested subqueries
    inner_sum_func_list defined for each subquery. When the item is placed 
    there the field 'ref_by' is set to ref.

  NOTES.
    Now we 'register' only set functions that are aggregated in outer
    subqueries. Actually it makes sense to link all set function for
    a subquery in one chain. It would simplify the process of 'splitting'
    for set functions.

  RETURN
    FALSE  if the executes without failures (currently always)
    TRUE   otherwise
*/  

bool Item_sum::register_sum_func(THD *thd, Item **ref)
{
  SELECT_LEX *sl;
  nesting_map allow_sum_func= thd->lex->allow_sum_func;
  for (sl= thd->lex->current_select->master_unit()->outer_select() ;
       sl && sl->nest_level > max_arg_level;
       sl= sl->master_unit()->outer_select() )
  {
    if (aggr_level < 0 && (allow_sum_func & (1 << sl->nest_level)))
    {
      /* Found the most nested subquery where the function can be aggregated */
      aggr_level= sl->nest_level;
242
      aggr_sel= sl;
unknown's avatar
unknown committed
243 244 245 246 247 248 249 250 251 252
    }
  }
  if (sl && (allow_sum_func & (1 << sl->nest_level)))
  {
    /* 
      We reached the subquery of level max_arg_level and checked
      that the function can be aggregated here. 
      The set function will be aggregated in this subquery.
    */   
    aggr_level= sl->nest_level;
253 254
    aggr_sel= sl;

unknown's avatar
unknown committed
255 256 257 258
  }
  if (aggr_level >= 0)
  {
    ref_by= ref;
259 260
    /* Add the object to the list of registered objects assigned to aggr_sel */
    if (!aggr_sel->inner_sum_func_list)
unknown's avatar
unknown committed
261 262 263
      next= this;
    else
    {
264 265
      next= aggr_sel->inner_sum_func_list->next;
      aggr_sel->inner_sum_func_list->next= this;
unknown's avatar
unknown committed
266
    }
267 268
    aggr_sel->inner_sum_func_list= this;
    aggr_sel->with_sum_func= 1;
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285

    /* 
      Mark Item_subselect(s) as containing aggregate function all the way up
      to aggregate function's calculation context.
      Note that we must not mark the Item of calculation context itself
      because with_sum_func on the calculation context st_select_lex is
      already set above.

      with_sum_func being set for an Item means that this Item refers 
      (somewhere in it, e.g. one of its arguments if it's a function) directly
      or through intermediate items to an aggregate function that is calculated
      in a context "outside" of the Item (e.g. in the current or outer select).

      with_sum_func being set for an st_select_lex means that this st_select_lex
      has aggregate functions directly referenced (i.e. not through a sub-select).
    */
    for (sl= thd->lex->current_select; 
286
         sl && sl != aggr_sel && sl->master_unit()->item;
287 288
         sl= sl->master_unit()->outer_select() )
      sl->master_unit()->item->with_sum_func= 1;
unknown's avatar
unknown committed
289
  }
290
  thd->lex->current_select->mark_as_dependent(aggr_sel);
unknown's avatar
unknown committed
291 292 293 294
  return FALSE;
}


unknown's avatar
unknown committed
295 296
Item_sum::Item_sum(List<Item> &list) :arg_count(list.elements), 
  forced_const(FALSE)
unknown's avatar
unknown committed
297 298 299 300
{
  if ((args=(Item**) sql_alloc(sizeof(Item*)*arg_count)))
  {
    uint i=0;
unknown's avatar
unknown committed
301
    List_iterator_fast<Item> li(list);
unknown's avatar
unknown committed
302 303 304 305 306 307 308
    Item *item;

    while ((item=li++))
    {
      args[i++]= item;
    }
  }
309
  mark_as_sum_func();
unknown's avatar
unknown committed
310 311 312
  list.empty();					// Fields are used
}

unknown's avatar
unknown committed
313 314 315 316 317

/*
  Constructor used in processing select with temporary tebles
*/

318
Item_sum::Item_sum(THD *thd, Item_sum *item):
unknown's avatar
unknown committed
319
  Item_result_field(thd, item), arg_count(item->arg_count),
320
  aggr_sel(item->aggr_sel),
unknown's avatar
unknown committed
321 322
  nest_level(item->nest_level), aggr_level(item->aggr_level),
  quick_group(item->quick_group), used_tables_cache(item->used_tables_cache),
323
  forced_const(item->forced_const) 
324 325 326 327
{
  if (arg_count <= 2)
    args=tmp_args;
  else
unknown's avatar
unknown committed
328
    if (!(args= (Item**) thd->alloc(sizeof(Item*)*arg_count)))
329
      return;
unknown's avatar
unknown committed
330 331 332 333
  memcpy(args, item->args, sizeof(Item*)*arg_count);
}


334
void Item_sum::mark_as_sum_func()
335
{
336 337 338
  SELECT_LEX *cur_select= current_thd->lex->current_select;
  cur_select->n_sum_items++;
  cur_select->with_sum_func= 1;
unknown's avatar
unknown committed
339
  with_sum_func= 1;
340
}
unknown's avatar
unknown committed
341

342

unknown's avatar
unknown committed
343 344 345 346
void Item_sum::make_field(Send_field *tmp_field)
{
  if (args[0]->type() == Item::FIELD_ITEM && keep_field_type())
  {
347
    ((Item_field*) args[0])->field->make_field(tmp_field);
348 349 350 351 352 353 354
    /* For expressions only col_name should be non-empty string. */
    char *empty_string= (char*)"";
    tmp_field->db_name= empty_string;
    tmp_field->org_table_name= empty_string;
    tmp_field->table_name= empty_string;
    tmp_field->org_col_name= empty_string;
    tmp_field->col_name= name;
355 356
    if (maybe_null)
      tmp_field->flags&= ~NOT_NULL_FLAG;
unknown's avatar
unknown committed
357
  }
358 359
  else
    init_make_field(tmp_field, field_type());
unknown's avatar
unknown committed
360 361
}

362

unknown's avatar
unknown committed
363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382
void Item_sum::print(String *str)
{
  str->append(func_name());
  for (uint i=0 ; i < arg_count ; i++)
  {
    if (i)
      str->append(',');
    args[i]->print(str);
  }
  str->append(')');
}

void Item_sum::fix_num_length_and_dec()
{
  decimals=0;
  for (uint i=0 ; i < arg_count ; i++)
    set_if_bigger(decimals,args[i]->decimals);
  max_length=float_length(decimals);
}

383
Item *Item_sum::get_tmp_table_item(THD *thd)
384
{
385
  Item_sum* sum_item= (Item_sum *) copy_or_same(thd);
386 387
  if (sum_item && sum_item->result_field)	   // If not a const sum func
  {
388
    Field *result_field_tmp= sum_item->result_field;
389 390 391 392 393 394
    for (uint i=0 ; i < sum_item->arg_count ; i++)
    {
      Item *arg= sum_item->args[i];
      if (!arg->const_item())
      {
	if (arg->type() == Item::FIELD_ITEM)
395
	  ((Item_field*) arg)->field= result_field_tmp++;
396
	else
397
	  sum_item->args[i]= new Item_field(result_field_tmp++);
398 399 400 401 402
      }
    }
  }
  return sum_item;
}
unknown's avatar
unknown committed
403

unknown's avatar
unknown committed
404

405
bool Item_sum::walk (Item_processor processor, bool walk_subquery,
406
                     uchar *argument)
407 408 409 410 411 412
{
  if (arg_count)
  {
    Item **arg,**arg_end;
    for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
    {
413
      if ((*arg)->walk(processor, walk_subquery, argument))
414 415 416 417 418 419
	return 1;
    }
  }
  return (this->*processor)(argument);
}

420

unknown's avatar
unknown committed
421 422 423
Field *Item_sum::create_tmp_field(bool group, TABLE *table,
                                  uint convert_blob_length)
{
unknown's avatar
unknown committed
424
  Field *field;
unknown's avatar
unknown committed
425 426
  switch (result_type()) {
  case REAL_RESULT:
427
    field= new Field_double(max_length, maybe_null, name, decimals, TRUE);
unknown's avatar
unknown committed
428
    break;
unknown's avatar
unknown committed
429
  case INT_RESULT:
unknown's avatar
unknown committed
430 431
    field= new Field_longlong(max_length, maybe_null, name, unsigned_flag);
    break;
unknown's avatar
unknown committed
432
  case STRING_RESULT:
433
    if (max_length/collation.collation->mbmaxlen <= 255 ||
434
        convert_blob_length > Field_varstring::MAX_SIZE ||
435
        !convert_blob_length)
unknown's avatar
unknown committed
436 437 438 439
      return make_string_field(table);
    field= new Field_varstring(convert_blob_length, maybe_null,
                               name, table->s, collation.collation);
    break;
unknown's avatar
unknown committed
440
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
441
    field= new Field_new_decimal(max_length, maybe_null, name,
unknown's avatar
unknown committed
442
                                 decimals, unsigned_flag);
unknown's avatar
unknown committed
443
    break;
unknown's avatar
unknown committed
444 445 446 447 448 449
  case ROW_RESULT:
  default:
    // This case should never be choosen
    DBUG_ASSERT(0);
    return 0;
  }
unknown's avatar
unknown committed
450 451 452
  if (field)
    field->init(table);
  return field;
unknown's avatar
unknown committed
453 454 455
}


unknown's avatar
unknown committed
456 457 458 459 460 461 462 463 464 465 466 467 468 469
void Item_sum::update_used_tables ()
{
  if (!forced_const)
  {
    used_tables_cache= 0;
    for (uint i=0 ; i < arg_count ; i++)
    {
      args[i]->update_used_tables();
      used_tables_cache|= args[i]->used_tables();
    }

    used_tables_cache&= PSEUDO_TABLE_BITS;

    /* the aggregate function is aggregated into its local context */
470
    used_tables_cache |=  (1 << aggr_sel->join->tables) - 1;
unknown's avatar
unknown committed
471 472 473 474
  }
}


unknown's avatar
unknown committed
475 476 477
String *
Item_sum_num::val_str(String *str)
{
478
  return val_string_from_real(str);
unknown's avatar
unknown committed
479 480 481
}


unknown's avatar
unknown committed
482 483
my_decimal *Item_sum_num::val_decimal(my_decimal *decimal_value)
{
484
  return val_decimal_from_real(decimal_value);
unknown's avatar
unknown committed
485 486 487
}


unknown's avatar
unknown committed
488 489 490
String *
Item_sum_int::val_str(String *str)
{
491 492 493 494 495 496 497
  return val_string_from_int(str);
}


my_decimal *Item_sum_int::val_decimal(my_decimal *decimal_value)
{
  return val_decimal_from_int(decimal_value);
unknown's avatar
unknown committed
498 499 500 501
}


bool
502
Item_sum_num::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
503
{
504
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
505

unknown's avatar
unknown committed
506
  if (init_sum_func_check(thd))
unknown's avatar
unknown committed
507
    return TRUE;
unknown's avatar
unknown committed
508

unknown's avatar
unknown committed
509 510 511 512
  decimals=0;
  maybe_null=0;
  for (uint i=0 ; i < arg_count ; i++)
  {
513
    if (args[i]->fix_fields(thd, args + i) || args[i]->check_cols(1))
unknown's avatar
unknown committed
514
      return TRUE;
unknown's avatar
unknown committed
515
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
516 517 518 519 520 521
    maybe_null |= args[i]->maybe_null;
  }
  result_field=0;
  max_length=float_length(decimals);
  null_value=1;
  fix_length_and_dec();
unknown's avatar
unknown committed
522 523 524 525

  if (check_sum_func(thd, ref))
    return TRUE;

526
  fixed= 1;
unknown's avatar
unknown committed
527
  return FALSE;
unknown's avatar
unknown committed
528 529 530
}


unknown's avatar
unknown committed
531 532 533
Item_sum_hybrid::Item_sum_hybrid(THD *thd, Item_sum_hybrid *item)
  :Item_sum(thd, item), value(item->value), hybrid_type(item->hybrid_type),
  hybrid_field_type(item->hybrid_field_type), cmp_sign(item->cmp_sign),
unknown's avatar
unknown committed
534
  was_values(item->was_values)
unknown's avatar
unknown committed
535
{
536 537
  /* copy results from old value */
  switch (hybrid_type) {
unknown's avatar
unknown committed
538 539 540 541 542 543 544 545 546
  case INT_RESULT:
    sum_int= item->sum_int;
    break;
  case DECIMAL_RESULT:
    my_decimal2decimal(&item->sum_dec, &sum_dec);
    break;
  case REAL_RESULT:
    sum= item->sum;
    break;
unknown's avatar
unknown committed
547 548 549 550 551 552
  case STRING_RESULT:
    /*
      This can happen with ROLLUP. Note that the value is already
      copied at function call.
    */
    break;
unknown's avatar
unknown committed
553 554 555 556 557 558 559
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
  collation.set(item->collation);
}

unknown's avatar
unknown committed
560
bool
561
Item_sum_hybrid::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
562
{
563
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
564

565
  Item *item= args[0];
unknown's avatar
unknown committed
566 567

  if (init_sum_func_check(thd))
unknown's avatar
unknown committed
568
    return TRUE;
569 570

  // 'item' can be changed during fix_fields
571
  if (!item->fixed &&
572
      item->fix_fields(thd, args) ||
573
      (item= args[0])->check_cols(1))
unknown's avatar
unknown committed
574
    return TRUE;
unknown's avatar
unknown committed
575
  decimals=item->decimals;
576

577
  switch (hybrid_type= item->result_type()) {
unknown's avatar
unknown committed
578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596
  case INT_RESULT:
    max_length= 20;
    sum_int= 0;
    break;
  case DECIMAL_RESULT:
    max_length= item->max_length;
    my_decimal_set_zero(&sum_dec);
    break;
  case REAL_RESULT:
    max_length= float_length(decimals);
    sum= 0.0;
    break;
  case STRING_RESULT:
    max_length= item->max_length;
    break;
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  };
597 598
  /* MIN/MAX can return NULL for empty set indepedent of the used column */
  maybe_null= 1;
599
  unsigned_flag=item->unsigned_flag;
600
  collation.set(item->collation);
unknown's avatar
unknown committed
601 602 603
  result_field=0;
  null_value=1;
  fix_length_and_dec();
604 605 606 607
  if (item->type() == Item::FIELD_ITEM)
    hybrid_field_type= ((Item_field*) item)->field->type();
  else
    hybrid_field_type= Item::field_type();
unknown's avatar
unknown committed
608 609 610 611

  if (check_sum_func(thd, ref))
    return TRUE;

612
  fixed= 1;
unknown's avatar
unknown committed
613
  return FALSE;
unknown's avatar
unknown committed
614 615
}

616 617 618
Field *Item_sum_hybrid::create_tmp_field(bool group, TABLE *table,
					 uint convert_blob_length)
{
unknown's avatar
unknown committed
619
  Field *field;
620 621
  if (args[0]->type() == Item::FIELD_ITEM)
  {
unknown's avatar
unknown committed
622
    field= ((Item_field*) args[0])->field;
623
    
unknown's avatar
unknown committed
624 625
    if ((field= create_tmp_field_from_field(current_thd, field, name, table,
					    NULL, convert_blob_length)))
626 627 628
      field->flags&= ~NOT_NULL_FLAG;
    return field;
  }
629 630 631 632 633 634 635
  /*
    DATE/TIME fields have STRING_RESULT result types.
    In order to preserve field type, it's needed to handle DATE/TIME
    fields creations separately.
  */
  switch (args[0]->field_type()) {
  case MYSQL_TYPE_DATE:
unknown's avatar
unknown committed
636 637
    field= new Field_date(maybe_null, name, collation.collation);
    break;
638
  case MYSQL_TYPE_TIME:
unknown's avatar
unknown committed
639 640
    field= new Field_time(maybe_null, name, collation.collation);
    break;
641 642
  case MYSQL_TYPE_TIMESTAMP:
  case MYSQL_TYPE_DATETIME:
unknown's avatar
unknown committed
643
    field= new Field_datetime(maybe_null, name, collation.collation);
644
    break;
unknown's avatar
unknown committed
645 646
  default:
    return Item_sum::create_tmp_field(group, table, convert_blob_length);
647
  }
unknown's avatar
unknown committed
648 649 650
  if (field)
    field->init(table);
  return field;
651 652
}

unknown's avatar
unknown committed
653 654 655 656 657

/***********************************************************************
** reset and add of sum_func
***********************************************************************/

unknown's avatar
unknown committed
658 659 660 661
Item_sum_sum::Item_sum_sum(THD *thd, Item_sum_sum *item) 
  :Item_sum_num(thd, item), hybrid_type(item->hybrid_type),
   curr_dec_buff(item->curr_dec_buff)
{
662
  /* TODO: check if the following assignments are really needed */
unknown's avatar
unknown committed
663 664 665 666 667 668 669 670 671
  if (hybrid_type == DECIMAL_RESULT)
  {
    my_decimal2decimal(item->dec_buffs, dec_buffs);
    my_decimal2decimal(item->dec_buffs + 1, dec_buffs + 1);
  }
  else
    sum= item->sum;
}

672
Item *Item_sum_sum::copy_or_same(THD* thd)
unknown's avatar
unknown committed
673
{
unknown's avatar
unknown committed
674
  return new (thd->mem_root) Item_sum_sum(thd, this);
unknown's avatar
unknown committed
675 676
}

677

678
void Item_sum_sum::clear()
679
{
unknown's avatar
unknown committed
680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
  DBUG_ENTER("Item_sum_sum::clear");
  null_value=1;
  if (hybrid_type == DECIMAL_RESULT)
  {
    curr_dec_buff= 0;
    my_decimal_set_zero(dec_buffs);
  }
  else
    sum= 0.0;
  DBUG_VOID_RETURN;
}


void Item_sum_sum::fix_length_and_dec()
{
  DBUG_ENTER("Item_sum_sum::fix_length_and_dec");
  maybe_null=null_value=1;
  decimals= args[0]->decimals;
698
  switch (args[0]->result_type()) {
unknown's avatar
unknown committed
699 700 701 702 703 704 705
  case REAL_RESULT:
  case STRING_RESULT:
    hybrid_type= REAL_RESULT;
    sum= 0.0;
    break;
  case INT_RESULT:
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
706
  {
unknown's avatar
unknown committed
707
    /* SUM result can't be longer than length(arg) + length(MAX_ROWS) */
unknown's avatar
unknown committed
708 709 710
    int precision= args[0]->decimal_precision() + DECIMAL_LONGLONG_DIGITS;
    max_length= my_decimal_precision_to_length(precision, decimals,
                                               unsigned_flag);
unknown's avatar
unknown committed
711 712 713 714
    curr_dec_buff= 0;
    hybrid_type= DECIMAL_RESULT;
    my_decimal_set_zero(dec_buffs);
    break;
unknown's avatar
unknown committed
715
  }
unknown's avatar
unknown committed
716 717 718 719 720 721 722 723 724 725 726 727
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
  DBUG_PRINT("info", ("Type: %s (%d, %d)",
                      (hybrid_type == REAL_RESULT ? "REAL_RESULT" :
                       hybrid_type == DECIMAL_RESULT ? "DECIMAL_RESULT" :
                       hybrid_type == INT_RESULT ? "INT_RESULT" :
                       "--ILLEGAL!!!--"),
                      max_length,
                      (int)decimals));
  DBUG_VOID_RETURN;
728 729 730
}


unknown's avatar
unknown committed
731 732
bool Item_sum_sum::add()
{
unknown's avatar
unknown committed
733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764
  DBUG_ENTER("Item_sum_sum::add");
  if (hybrid_type == DECIMAL_RESULT)
  {
    my_decimal value, *val= args[0]->val_decimal(&value);
    if (!args[0]->null_value)
    {
      my_decimal_add(E_DEC_FATAL_ERROR, dec_buffs + (curr_dec_buff^1),
                     val, dec_buffs + curr_dec_buff);
      curr_dec_buff^= 1;
      null_value= 0;
    }
  }
  else
  {
    sum+= args[0]->val_real();
    if (!args[0]->null_value)
      null_value= 0;
  }
  DBUG_RETURN(0);
}


longlong Item_sum_sum::val_int()
{
  DBUG_ASSERT(fixed == 1);
  if (hybrid_type == DECIMAL_RESULT)
  {
    longlong result;
    my_decimal2int(E_DEC_FATAL_ERROR, dec_buffs + curr_dec_buff, unsigned_flag,
                   &result);
    return result;
  }
765
  return (longlong) rint(val_real());
unknown's avatar
unknown committed
766 767
}

768

769
double Item_sum_sum::val_real()
unknown's avatar
unknown committed
770
{
771
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
772 773
  if (hybrid_type == DECIMAL_RESULT)
    my_decimal2double(E_DEC_FATAL_ERROR, dec_buffs + curr_dec_buff, &sum);
unknown's avatar
unknown committed
774 775 776 777
  return sum;
}


778
String *Item_sum_sum::val_str(String *str)
unknown's avatar
unknown committed
779 780
{
  if (hybrid_type == DECIMAL_RESULT)
781 782
    return val_string_from_decimal(str);
  return val_string_from_real(str);
unknown's avatar
unknown committed
783 784 785 786 787
}


my_decimal *Item_sum_sum::val_decimal(my_decimal *val)
{
788 789 790
  if (hybrid_type == DECIMAL_RESULT)
    return (dec_buffs + curr_dec_buff);
  return val_decimal_from_real(val);
unknown's avatar
unknown committed
791 792
}

793
/***************************************************************************/
794

795 796 797
C_MODE_START

/* Declarations for auxilary C-callbacks */
798

799 800 801 802 803 804 805 806 807
static int simple_raw_key_cmp(void* arg, const void* key1, const void* key2)
{
    return memcmp(key1, key2, *(uint *) arg);
}


static int item_sum_distinct_walk(void *element, element_count num_of_dups,
                                  void *item)
{
808
  return ((Item_sum_distinct*) (item))->unique_walk_function(element);
809 810 811 812 813 814 815 816
}

C_MODE_END

/* Item_sum_distinct */

Item_sum_distinct::Item_sum_distinct(Item *item_arg)
  :Item_sum_num(item_arg), tree(0)
817 818 819 820 821 822 823 824 825 826 827
{
  /*
    quick_group is an optimizer hint, which means that GROUP BY can be
    handled with help of index on grouped columns.
    By setting quick_group to zero we force creation of temporary table
    to perform GROUP BY.
  */
  quick_group= 0;
}


828 829 830
Item_sum_distinct::Item_sum_distinct(THD *thd, Item_sum_distinct *original)
  :Item_sum_num(thd, original), val(original->val), tree(0),
  table_field_type(original->table_field_type)
831 832 833 834 835
{
  quick_group= 0;
}


836 837 838 839 840 841 842 843 844
/*
  Behaves like an Integer except to fix_length_and_dec().
  Additionally div() converts val with this traits to a val with true
  decimal traits along with conversion of integer value to decimal value.
  This is to speedup SUM/AVG(DISTINCT) evaluation for 8-32 bit integer
  values.
*/
struct Hybrid_type_traits_fast_decimal: public
       Hybrid_type_traits_integer
unknown's avatar
unknown committed
845
{
846 847 848 849 850
  virtual Item_result type() const { return DECIMAL_RESULT; }
  virtual void fix_length_and_dec(Item *item, Item *arg) const
  { Hybrid_type_traits_decimal::instance()->fix_length_and_dec(item, arg); }

  virtual void div(Hybrid_type *val, ulonglong u) const
unknown's avatar
unknown committed
851
  {
852 853 854 855
    int2my_decimal(E_DEC_FATAL_ERROR, val->integer, 0, val->dec_buf);
    val->used_dec_buf_no= 0;
    val->traits= Hybrid_type_traits_decimal::instance();
    val->traits->div(val, u);
unknown's avatar
unknown committed
856
  }
unknown's avatar
unknown committed
857
  static const Hybrid_type_traits_fast_decimal *instance();
858
  Hybrid_type_traits_fast_decimal() {};
859
};
unknown's avatar
unknown committed
860

unknown's avatar
unknown committed
861 862 863 864 865 866 867
static const Hybrid_type_traits_fast_decimal fast_decimal_traits_instance;

const Hybrid_type_traits_fast_decimal
  *Hybrid_type_traits_fast_decimal::instance()
{
  return &fast_decimal_traits_instance;
}
unknown's avatar
unknown committed
868

869
void Item_sum_distinct::fix_length_and_dec()
870
{
871
  DBUG_ASSERT(args[0]->fixed);
872

873
  table_field_type= args[0]->field_type();
874

875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909
  /* Adjust tmp table type according to the chosen aggregation type */
  switch (args[0]->result_type()) {
  case STRING_RESULT:
  case REAL_RESULT:
    val.traits= Hybrid_type_traits::instance();
    if (table_field_type != MYSQL_TYPE_FLOAT)
      table_field_type= MYSQL_TYPE_DOUBLE;
    break;
  case INT_RESULT:
  /*
    Preserving int8, int16, int32 field types gives ~10% performance boost
    as the size of result tree becomes significantly smaller.
    Another speed up we gain by using longlong for intermediate
    calculations. The range of int64 is enough to hold sum 2^32 distinct
    integers each <= 2^32.
  */
  if (table_field_type == MYSQL_TYPE_INT24 ||
      table_field_type >= MYSQL_TYPE_TINY &&
      table_field_type <= MYSQL_TYPE_LONG)
  {
    val.traits= Hybrid_type_traits_fast_decimal::instance();
    break;
  }
  table_field_type= MYSQL_TYPE_LONGLONG;
  /* fallthrough */
  case DECIMAL_RESULT:
    val.traits= Hybrid_type_traits_decimal::instance();
    if (table_field_type != MYSQL_TYPE_LONGLONG)
      table_field_type= MYSQL_TYPE_NEWDECIMAL;
    break;
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
  val.traits->fix_length_and_dec(this, args[0]);
910 911 912
}


913
bool Item_sum_distinct::setup(THD *thd)
914
{
unknown's avatar
unknown committed
915 916
  List<Create_field> field_list;
  Create_field field_def;                              /* field definition */
917
  DBUG_ENTER("Item_sum_distinct::setup");
unknown's avatar
unknown committed
918
  DBUG_ASSERT(tree == 0);
919 920

  /*
921 922 923
    Virtual table and the tree are created anew on each re-execution of
    PS/SP. Hence all further allocations are performed in the runtime
    mem_root.
924
  */
925 926 927
  if (field_list.push_back(&field_def))
    return TRUE;

928
  null_value= maybe_null= 1;
929 930 931 932 933 934 935 936 937
  quick_group= 0;

  DBUG_ASSERT(args[0]->fixed);

  field_def.init_for_tmp_table(table_field_type, args[0]->max_length,
                               args[0]->decimals, args[0]->maybe_null,
                               args[0]->unsigned_flag);

  if (! (table= create_virtual_tmp_table(thd, field_list)))
unknown's avatar
unknown committed
938
    return TRUE;
939 940 941

  /* XXX: check that the case of CHAR(0) works OK */
  tree_key_length= table->s->reclength - table->s->null_bytes;
942 943

  /*
944 945 946 947
    Unique handles all unique elements in a tree until they can't fit
    in.  Then the tree is dumped to the temporary file. We can use
    simple_raw_key_cmp because the table contains numbers only; decimals
    are converted to binary representation as well.
948
  */
949
  tree= new Unique(simple_raw_key_cmp, &tree_key_length, tree_key_length,
950 951
                   thd->variables.max_heap_table_size);

952
  is_evaluated= FALSE;
953
  DBUG_RETURN(tree == 0);
954 955 956
}


957
bool Item_sum_distinct::add()
958
{
959
  args[0]->save_in_field(table->field[0], FALSE);
960
  is_evaluated= FALSE;
961
  if (!table->field[0]->is_null())
962
  {
963 964 965 966 967 968 969
    DBUG_ASSERT(tree);
    null_value= 0;
    /*
      '0' values are also stored in the tree. This doesn't matter
      for SUM(DISTINCT), but is important for AVG(DISTINCT)
    */
    return tree->unique_add(table->field[0]->ptr);
970
  }
971
  return 0;
unknown's avatar
unknown committed
972 973 974
}


975
bool Item_sum_distinct::unique_walk_function(void *element)
unknown's avatar
unknown committed
976
{
977 978 979 980
  memcpy(table->field[0]->ptr, element, tree_key_length);
  ++count;
  val.traits->add(&val, table->field[0]);
  return 0;
unknown's avatar
unknown committed
981 982 983
}


984
void Item_sum_distinct::clear()
unknown's avatar
unknown committed
985
{
986 987 988 989
  DBUG_ENTER("Item_sum_distinct::clear");
  DBUG_ASSERT(tree != 0);                        /* we always have a tree */
  null_value= 1;
  tree->reset();
990
  is_evaluated= FALSE;
991
  DBUG_VOID_RETURN;
992 993
}

994 995 996 997 998 999
void Item_sum_distinct::cleanup()
{
  Item_sum_num::cleanup();
  delete tree;
  tree= 0;
  table= 0;
1000
  is_evaluated= FALSE;
unknown's avatar
unknown committed
1001 1002
}

1003 1004 1005 1006
Item_sum_distinct::~Item_sum_distinct()
{
  delete tree;
  /* no need to free the table */
1007 1008
}

1009

1010
void Item_sum_distinct::calculate_val_and_count()
1011
{
1012
  if (!is_evaluated)
unknown's avatar
unknown committed
1013
  {
1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025
    count= 0;
    val.traits->set_zero(&val);
    /*
      We don't have a tree only if 'setup()' hasn't been called;
      this is the case of sql_select.cc:return_zero_rows.
     */
    if (tree)
    {
      table->field[0]->set_notnull();
      tree->walk(item_sum_distinct_walk, (void*) this);
    }
    is_evaluated= TRUE;
unknown's avatar
unknown committed
1026
  }
1027 1028
}

unknown's avatar
unknown committed
1029

1030
double Item_sum_distinct::val_real()
unknown's avatar
unknown committed
1031
{
1032 1033
  calculate_val_and_count();
  return val.traits->val_real(&val);
unknown's avatar
unknown committed
1034 1035 1036
}


1037
my_decimal *Item_sum_distinct::val_decimal(my_decimal *to)
unknown's avatar
unknown committed
1038
{
1039 1040 1041 1042
  calculate_val_and_count();
  if (null_value)
    return 0;
  return val.traits->val_decimal(&val, to);
unknown's avatar
unknown committed
1043 1044 1045
}


1046
longlong Item_sum_distinct::val_int()
unknown's avatar
unknown committed
1047
{
1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058
  calculate_val_and_count();
  return val.traits->val_int(&val, unsigned_flag);
}


String *Item_sum_distinct::val_str(String *str)
{
  calculate_val_and_count();
  if (null_value)
    return 0;
  return val.traits->val_str(&val, str, decimals);
unknown's avatar
unknown committed
1059 1060
}

1061 1062 1063 1064
/* end of Item_sum_distinct */

/* Item_sum_avg_distinct */

1065 1066 1067 1068
void
Item_sum_avg_distinct::fix_length_and_dec()
{
  Item_sum_distinct::fix_length_and_dec();
unknown's avatar
unknown committed
1069
  prec_increment= current_thd->variables.div_precincrement;
1070 1071 1072 1073
  /*
    AVG() will divide val by count. We need to reserve digits
    after decimal point as the result can be fractional.
  */
unknown's avatar
unknown committed
1074
  decimals= min(decimals + prec_increment, NOT_FIXED_DEC);
1075 1076 1077
}


1078 1079 1080
void
Item_sum_avg_distinct::calculate_val_and_count()
{
1081 1082 1083 1084 1085 1086 1087
  if (!is_evaluated)
  {
    Item_sum_distinct::calculate_val_and_count();
    if (count)
      val.traits->div(&val, count);
    is_evaluated= TRUE;
  }
1088
}
unknown's avatar
unknown committed
1089

1090

1091
Item *Item_sum_count::copy_or_same(THD* thd)
unknown's avatar
unknown committed
1092
{
unknown's avatar
unknown committed
1093
  return new (thd->mem_root) Item_sum_count(thd, this);
unknown's avatar
unknown committed
1094 1095
}

1096

1097
void Item_sum_count::clear()
1098
{
1099
  count= 0;
1100 1101 1102
}


unknown's avatar
unknown committed
1103 1104
bool Item_sum_count::add()
{
1105
  if (!args[0]->maybe_null || !args[0]->is_null())
unknown's avatar
unknown committed
1106 1107 1108 1109 1110 1111
    count++;
  return 0;
}

longlong Item_sum_count::val_int()
{
1112
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1113 1114 1115
  return (longlong) count;
}

1116 1117 1118 1119

void Item_sum_count::cleanup()
{
  DBUG_ENTER("Item_sum_count::cleanup");
1120
  count= 0;
1121 1122 1123 1124 1125
  Item_sum_int::cleanup();
  DBUG_VOID_RETURN;
}


unknown's avatar
unknown committed
1126
/*
1127
  Avgerage
unknown's avatar
unknown committed
1128
*/
unknown's avatar
unknown committed
1129 1130 1131 1132
void Item_sum_avg::fix_length_and_dec()
{
  Item_sum_sum::fix_length_and_dec();
  maybe_null=null_value=1;
unknown's avatar
unknown committed
1133
  prec_increment= current_thd->variables.div_precincrement;
unknown's avatar
unknown committed
1134 1135
  if (hybrid_type == DECIMAL_RESULT)
  {
unknown's avatar
unknown committed
1136 1137 1138 1139 1140 1141
    int precision= args[0]->decimal_precision() + prec_increment;
    decimals= min(args[0]->decimals + prec_increment, DECIMAL_MAX_SCALE);
    max_length= my_decimal_precision_to_length(precision, decimals,
                                               unsigned_flag);
    f_precision= min(precision+DECIMAL_LONGLONG_DIGITS, DECIMAL_MAX_PRECISION);
    f_scale=  args[0]->decimals;
unknown's avatar
unknown committed
1142 1143
    dec_bin_size= my_decimal_get_binary_size(f_precision, f_scale);
  }
1144
  else {
unknown's avatar
unknown committed
1145
    decimals= min(args[0]->decimals + prec_increment, NOT_FIXED_DEC);
1146 1147
    max_length= args[0]->max_length + prec_increment;
  }
unknown's avatar
unknown committed
1148 1149
}

unknown's avatar
unknown committed
1150

1151
Item *Item_sum_avg::copy_or_same(THD* thd)
unknown's avatar
unknown committed
1152
{
unknown's avatar
unknown committed
1153
  return new (thd->mem_root) Item_sum_avg(thd, this);
unknown's avatar
unknown committed
1154 1155
}

1156

unknown's avatar
unknown committed
1157 1158 1159
Field *Item_sum_avg::create_tmp_field(bool group, TABLE *table,
                                      uint convert_blob_len)
{
unknown's avatar
unknown committed
1160
  Field *field;
1161
  if (group)
unknown's avatar
unknown committed
1162
  {
1163 1164
    /*
      We must store both value and counter in the temporary table in one field.
1165
      The easiest way is to do this is to store both value in a string
1166 1167
      and unpack on access.
    */
unknown's avatar
unknown committed
1168
    field= new Field_string(((hybrid_type == DECIMAL_RESULT) ?
1169
                             dec_bin_size : sizeof(double)) + sizeof(longlong),
unknown's avatar
unknown committed
1170
                            0, name, &my_charset_bin);
unknown's avatar
unknown committed
1171
  }
unknown's avatar
unknown committed
1172 1173
  else if (hybrid_type == DECIMAL_RESULT)
    field= new Field_new_decimal(max_length, maybe_null, name,
unknown's avatar
unknown committed
1174
                                 decimals, unsigned_flag);
unknown's avatar
unknown committed
1175
  else
1176
    field= new Field_double(max_length, maybe_null, name, decimals, TRUE);
unknown's avatar
unknown committed
1177 1178 1179
  if (field)
    field->init(table);
  return field;
unknown's avatar
unknown committed
1180 1181 1182
}


1183
void Item_sum_avg::clear()
1184
{
unknown's avatar
unknown committed
1185 1186
  Item_sum_sum::clear();
  count=0;
1187 1188 1189
}


unknown's avatar
unknown committed
1190 1191
bool Item_sum_avg::add()
{
unknown's avatar
unknown committed
1192 1193
  if (Item_sum_sum::add())
    return TRUE;
unknown's avatar
unknown committed
1194 1195
  if (!args[0]->null_value)
    count++;
unknown's avatar
unknown committed
1196
  return FALSE;
unknown's avatar
unknown committed
1197 1198
}

1199
double Item_sum_avg::val_real()
unknown's avatar
unknown committed
1200
{
1201
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1202 1203 1204 1205 1206
  if (!count)
  {
    null_value=1;
    return 0.0;
  }
unknown's avatar
unknown committed
1207 1208 1209 1210 1211 1212
  return Item_sum_sum::val_real() / ulonglong2double(count);
}


my_decimal *Item_sum_avg::val_decimal(my_decimal *val)
{
1213
  my_decimal sum_buff, cnt;
1214
  const my_decimal *sum_dec;
unknown's avatar
unknown committed
1215 1216 1217 1218 1219 1220
  DBUG_ASSERT(fixed == 1);
  if (!count)
  {
    null_value=1;
    return NULL;
  }
1221
  sum_dec= Item_sum_sum::val_decimal(&sum_buff);
unknown's avatar
unknown committed
1222
  int2my_decimal(E_DEC_FATAL_ERROR, count, 0, &cnt);
unknown's avatar
unknown committed
1223
  my_decimal_div(E_DEC_FATAL_ERROR, val, sum_dec, &cnt, prec_increment);
unknown's avatar
unknown committed
1224 1225 1226 1227 1228 1229 1230
  return val;
}


String *Item_sum_avg::val_str(String *str)
{
  if (hybrid_type == DECIMAL_RESULT)
1231 1232
    return val_string_from_decimal(str);
  return val_string_from_real(str);
unknown's avatar
unknown committed
1233 1234 1235 1236
}


/*
1237
  Standard deviation
unknown's avatar
unknown committed
1238 1239
*/

1240
double Item_sum_std::val_real()
unknown's avatar
unknown committed
1241
{
1242
  DBUG_ASSERT(fixed == 1);
1243 1244 1245
  double nr= Item_sum_variance::val_real();
  DBUG_ASSERT(nr >= 0.0);
  return sqrt(nr);
unknown's avatar
unknown committed
1246 1247
}

1248 1249
Item *Item_sum_std::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1250
  return new (thd->mem_root) Item_sum_std(thd, this);
1251 1252 1253
}


unknown's avatar
unknown committed
1254
/*
1255
  Variance
unknown's avatar
unknown committed
1256 1257
*/

unknown's avatar
unknown committed
1258

1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
/**
  Variance implementation for floating-point implementations, without
  catastrophic cancellation, from Knuth's _TAoCP_, 3rd ed, volume 2, pg232.
  This alters the value at m, s, and increments count.
*/

/*
  These two functions are used by the Item_sum_variance and the
  Item_variance_field classes, which are unrelated, and each need to calculate
  variance.  The difference between the two classes is that the first is used
  for a mundane SELECT, while the latter is used in a GROUPing SELECT.
*/
static void variance_fp_recurrence_next(double *m, double *s, ulonglong *count, double nr)
unknown's avatar
unknown committed
1272
{
1273 1274 1275
  *count += 1;

  if (*count == 1) 
unknown's avatar
unknown committed
1276
  {
1277 1278
    *m= nr;
    *s= 0;
unknown's avatar
unknown committed
1279 1280 1281
  }
  else
  {
1282 1283 1284
    double m_kminusone= *m;
    *m= m_kminusone + (nr - m_kminusone) / (double) *count;
    *s= *s + (nr - m_kminusone) * (nr - *m);
unknown's avatar
unknown committed
1285 1286 1287 1288
  }
}


1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
static double variance_fp_recurrence_result(double s, ulonglong count, bool is_sample_variance)
{
  if (count == 1)
    return 0.0;

  if (is_sample_variance)
    return s / (count - 1);

  /* else, is a population variance */
  return s / count;
}


Item_sum_variance::Item_sum_variance(THD *thd, Item_sum_variance *item):
  Item_sum_num(thd, item), hybrid_type(item->hybrid_type),
    count(item->count), sample(item->sample),
    prec_increment(item->prec_increment)
{
  recurrence_m= item->recurrence_m;
  recurrence_s= item->recurrence_s;
}


unknown's avatar
unknown committed
1312 1313
void Item_sum_variance::fix_length_and_dec()
{
1314 1315
  DBUG_ENTER("Item_sum_variance::fix_length_and_dec");
  maybe_null= null_value= 1;
unknown's avatar
unknown committed
1316
  prec_increment= current_thd->variables.div_precincrement;
1317 1318 1319 1320 1321 1322 1323 1324 1325

  /*
    According to the SQL2003 standard (Part 2, Foundations; sec 10.9,
    aggregate function; paragraph 7h of Syntax Rules), "the declared 
    type of the result is an implementation-defined aproximate numeric
    type.
  */
  hybrid_type= REAL_RESULT;

1326
  switch (args[0]->result_type()) {
unknown's avatar
unknown committed
1327 1328
  case REAL_RESULT:
  case STRING_RESULT:
unknown's avatar
unknown committed
1329
    decimals= min(args[0]->decimals + 4, NOT_FIXED_DEC);
unknown's avatar
unknown committed
1330 1331 1332
    break;
  case INT_RESULT:
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
1333 1334 1335 1336 1337
  {
    int precision= args[0]->decimal_precision()*2 + prec_increment;
    decimals= min(args[0]->decimals + prec_increment, DECIMAL_MAX_SCALE);
    max_length= my_decimal_precision_to_length(precision, decimals,
                                               unsigned_flag);
1338

unknown's avatar
unknown committed
1339
    break;
unknown's avatar
unknown committed
1340
  }
unknown's avatar
unknown committed
1341 1342 1343 1344
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
1345
  DBUG_PRINT("info", ("Type: REAL_RESULT (%d, %d)", max_length, (int)decimals));
unknown's avatar
unknown committed
1346 1347 1348 1349
  DBUG_VOID_RETURN;
}


1350 1351
Item *Item_sum_variance::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1352
  return new (thd->mem_root) Item_sum_variance(thd, this);
1353 1354 1355
}


1356 1357 1358 1359 1360
/**
  Create a new field to match the type of value we're expected to yield.
  If we're grouping, then we need some space to serialize variables into, to
  pass around.
*/
unknown's avatar
unknown committed
1361 1362 1363
Field *Item_sum_variance::create_tmp_field(bool group, TABLE *table,
                                           uint convert_blob_len)
{
unknown's avatar
unknown committed
1364
  Field *field;
1365
  if (group)
unknown's avatar
unknown committed
1366
  {
1367 1368
    /*
      We must store both value and counter in the temporary table in one field.
1369
      The easiest way is to do this is to store both value in a string
1370 1371
      and unpack on access.
    */
1372
    field= new Field_string(sizeof(double)*2 + sizeof(longlong), 0, name, &my_charset_bin);
unknown's avatar
unknown committed
1373
  }
unknown's avatar
unknown committed
1374
  else
1375
    field= new Field_double(max_length, maybe_null, name, decimals, TRUE);
1376 1377

  if (field != NULL)
unknown's avatar
unknown committed
1378
    field->init(table);
1379

unknown's avatar
unknown committed
1380
  return field;
unknown's avatar
unknown committed
1381 1382 1383
}


1384
void Item_sum_variance::clear()
unknown's avatar
unknown committed
1385
{
1386
  count= 0; 
unknown's avatar
unknown committed
1387 1388 1389
}

bool Item_sum_variance::add()
unknown's avatar
unknown committed
1390
{
1391 1392 1393 1394 1395 1396 1397 1398
  /* 
    Why use a temporary variable?  We don't know if it is null until we
    evaluate it, which has the side-effect of setting null_value .
  */
  double nr= args[0]->val_real();
  
  if (!args[0]->null_value)
    variance_fp_recurrence_next(&recurrence_m, &recurrence_s, &count, nr);
unknown's avatar
unknown committed
1399 1400 1401
  return 0;
}

1402
double Item_sum_variance::val_real()
unknown's avatar
unknown committed
1403
{
1404
  DBUG_ASSERT(fixed == 1);
1405

1406 1407 1408 1409 1410 1411 1412 1413 1414 1415
  /*
    'sample' is a 1/0 boolean value.  If it is 1/true, id est this is a sample
    variance call, then we should set nullness when the count of the items
    is one or zero.  If it's zero, i.e. a population variance, then we only
    set nullness when the count is zero.

    Another way to read it is that 'sample' is the numerical threshhold, at and
    below which a 'count' number of items is called NULL.
  */
  DBUG_ASSERT((sample == 0) || (sample == 1));
1416
  if (count <= sample)
unknown's avatar
unknown committed
1417 1418 1419 1420
  {
    null_value=1;
    return 0.0;
  }
1421

unknown's avatar
unknown committed
1422
  null_value=0;
1423
  return variance_fp_recurrence_result(recurrence_s, count, sample);
unknown's avatar
unknown committed
1424 1425 1426 1427 1428
}


my_decimal *Item_sum_variance::val_decimal(my_decimal *dec_buf)
{
1429 1430
  DBUG_ASSERT(fixed == 1);
  return val_decimal_from_real(dec_buf);
unknown's avatar
unknown committed
1431 1432
}

1433

unknown's avatar
unknown committed
1434 1435
void Item_sum_variance::reset_field()
{
1436
  double nr;
1437
  uchar *res= result_field->ptr;
1438

1439
  nr= args[0]->val_real();              /* sets null_value as side-effect */
unknown's avatar
unknown committed
1440 1441 1442 1443 1444

  if (args[0]->null_value)
    bzero(res,sizeof(double)*2+sizeof(longlong));
  else
  {
1445 1446 1447 1448 1449 1450 1451 1452
    /* Serialize format is (double)m, (double)s, (longlong)count */
    ulonglong tmp_count;
    double tmp_s;
    float8store(res, nr);               /* recurrence variable m */
    tmp_s= 0.0;
    float8store(res + sizeof(double), tmp_s);
    tmp_count= 1;
    int8store(res + sizeof(double)*2, tmp_count);
unknown's avatar
unknown committed
1453 1454 1455
  }
}

1456

unknown's avatar
unknown committed
1457
void Item_sum_variance::update_field()
unknown's avatar
unknown committed
1458
{
1459
  ulonglong field_count;
1460
  uchar *res=result_field->ptr;
1461 1462 1463 1464

  double nr= args[0]->val_real();       /* sets null_value as side-effect */

  if (args[0]->null_value)
unknown's avatar
unknown committed
1465
    return;
unknown's avatar
unknown committed
1466

1467 1468 1469 1470
  /* Serialize format is (double)m, (double)s, (longlong)count */
  double field_recurrence_m, field_recurrence_s;
  float8get(field_recurrence_m, res);
  float8get(field_recurrence_s, res + sizeof(double));
unknown's avatar
unknown committed
1471
  field_count=sint8korr(res+sizeof(double)*2);
unknown's avatar
unknown committed
1472

1473 1474 1475 1476
  variance_fp_recurrence_next(&field_recurrence_m, &field_recurrence_s, &field_count, nr);

  float8store(res, field_recurrence_m);
  float8store(res + sizeof(double), field_recurrence_s);
1477 1478
  res+= sizeof(double)*2;
  int8store(res,field_count);
unknown's avatar
unknown committed
1479 1480
}

1481

unknown's avatar
unknown committed
1482 1483
/* min & max */

1484 1485
void Item_sum_hybrid::clear()
{
1486
  switch (hybrid_type) {
unknown's avatar
unknown committed
1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498
  case INT_RESULT:
    sum_int= 0;
    break;
  case DECIMAL_RESULT:
    my_decimal_set_zero(&sum_dec);
    break;
  case REAL_RESULT:
    sum= 0.0;
    break;
  default:
    value.length(0);
  }
1499 1500 1501
  null_value= 1;
}

1502
double Item_sum_hybrid::val_real()
unknown's avatar
unknown committed
1503
{
1504
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1505 1506
  if (null_value)
    return 0.0;
1507 1508
  switch (hybrid_type) {
  case STRING_RESULT:
1509 1510 1511
  {
    char *end_not_used;
    int err_not_used;
unknown's avatar
unknown committed
1512
    String *res;  res=val_str(&str_value);
1513 1514 1515
    return (res ? my_strntod(res->charset(), (char*) res->ptr(), res->length(),
			     &end_not_used, &err_not_used) : 0.0);
  }
1516 1517 1518 1519
  case INT_RESULT:
    if (unsigned_flag)
      return ulonglong2double(sum_int);
    return (double) sum_int;
unknown's avatar
unknown committed
1520 1521 1522
  case DECIMAL_RESULT:
    my_decimal2double(E_DEC_FATAL_ERROR, &sum_dec, &sum);
    return sum;
1523 1524
  case REAL_RESULT:
    return sum;
1525
  case ROW_RESULT:
unknown's avatar
unknown committed
1526
  default:
unknown's avatar
unknown committed
1527 1528 1529
    // This case should never be choosen
    DBUG_ASSERT(0);
    return 0;
unknown's avatar
unknown committed
1530
  }
1531 1532 1533 1534
}

longlong Item_sum_hybrid::val_int()
{
1535
  DBUG_ASSERT(fixed == 1);
1536 1537
  if (null_value)
    return 0;
1538
  switch (hybrid_type) {
unknown's avatar
unknown committed
1539 1540 1541 1542 1543 1544
  case INT_RESULT:
    return sum_int;
  case DECIMAL_RESULT:
  {
    longlong result;
    my_decimal2int(E_DEC_FATAL_ERROR, &sum_dec, unsigned_flag, &result);
1545
    return sum_int;
unknown's avatar
unknown committed
1546 1547
  }
  default:
1548
    return (longlong) rint(Item_sum_hybrid::val_real());
unknown's avatar
unknown committed
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
  }
}


my_decimal *Item_sum_hybrid::val_decimal(my_decimal *val)
{
  DBUG_ASSERT(fixed == 1);
  if (null_value)
    return 0;
  switch (hybrid_type) {
  case STRING_RESULT:
    string2my_decimal(E_DEC_FATAL_ERROR, &value, val);
    break;
  case REAL_RESULT:
    double2my_decimal(E_DEC_FATAL_ERROR, sum, val);
    break;
  case DECIMAL_RESULT:
    val= &sum_dec;
    break;
  case INT_RESULT:
    int2my_decimal(E_DEC_FATAL_ERROR, sum_int, unsigned_flag, val);
    break;
  case ROW_RESULT:
  default:
    // This case should never be choosen
    DBUG_ASSERT(0);
    break;
  }
  return val;					// Keep compiler happy
unknown's avatar
unknown committed
1578 1579 1580 1581 1582 1583
}


String *
Item_sum_hybrid::val_str(String *str)
{
1584
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1585 1586
  if (null_value)
    return 0;
1587 1588
  switch (hybrid_type) {
  case STRING_RESULT:
unknown's avatar
unknown committed
1589
    return &value;
1590
  case REAL_RESULT:
1591
    str->set_real(sum,decimals, &my_charset_bin);
1592
    break;
unknown's avatar
unknown committed
1593 1594 1595
  case DECIMAL_RESULT:
    my_decimal2string(E_DEC_FATAL_ERROR, &sum_dec, 0, 0, 0, str);
    return str;
1596
  case INT_RESULT:
1597
    str->set_int(sum_int, unsigned_flag, &my_charset_bin);
1598
    break;
1599
  case ROW_RESULT:
unknown's avatar
unknown committed
1600
  default:
unknown's avatar
unknown committed
1601 1602 1603
    // This case should never be choosen
    DBUG_ASSERT(0);
    break;
1604 1605
  }
  return str;					// Keep compiler happy
unknown's avatar
unknown committed
1606 1607
}

1608

1609 1610 1611 1612
void Item_sum_hybrid::cleanup()
{
  DBUG_ENTER("Item_sum_hybrid::cleanup");
  Item_sum::cleanup();
unknown's avatar
unknown committed
1613
  forced_const= FALSE;
unknown's avatar
unknown committed
1614

1615
  /*
unknown's avatar
unknown committed
1616
    by default it is TRUE to avoid TRUE reporting by
1617 1618 1619
    Item_func_not_all/Item_func_nop_all if this item was never called.

    no_rows_in_result() set it to FALSE if was not results found.
unknown's avatar
unknown committed
1620
    If some results found it will be left unchanged.
1621 1622
  */
  was_values= TRUE;
1623 1624 1625
  DBUG_VOID_RETURN;
}

1626 1627 1628
void Item_sum_hybrid::no_rows_in_result()
{
  was_values= FALSE;
1629
  clear();
1630 1631
}

1632

1633 1634
Item *Item_sum_min::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1635
  return new (thd->mem_root) Item_sum_min(thd, this);
1636 1637 1638
}


unknown's avatar
unknown committed
1639 1640
bool Item_sum_min::add()
{
1641 1642
  switch (hybrid_type) {
  case STRING_RESULT:
unknown's avatar
unknown committed
1643 1644 1645
  {
    String *result=args[0]->val_str(&tmp_value);
    if (!args[0]->null_value &&
1646
	(null_value || sortcmp(&value,result,collation.collation) > 0))
unknown's avatar
unknown committed
1647 1648 1649 1650 1651
    {
      value.copy(*result);
      null_value=0;
    }
  }
1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665
  break;
  case INT_RESULT:
  {
    longlong nr=args[0]->val_int();
    if (!args[0]->null_value && (null_value ||
				 (unsigned_flag && 
				  (ulonglong) nr < (ulonglong) sum_int) ||
				 (!unsigned_flag && nr < sum_int)))
    {
      sum_int=nr;
      null_value=0;
    }
  }
  break;
unknown's avatar
unknown committed
1666 1667
  case DECIMAL_RESULT:
  {
1668
    my_decimal value_buff, *val= args[0]->val_decimal(&value_buff);
unknown's avatar
unknown committed
1669 1670 1671 1672 1673 1674 1675 1676
    if (!args[0]->null_value &&
        (null_value || (my_decimal_cmp(&sum_dec, val) > 0)))
    {
      my_decimal2decimal(val, &sum_dec);
      null_value= 0;
    }
  }
  break;
1677
  case REAL_RESULT:
unknown's avatar
unknown committed
1678
  {
1679
    double nr= args[0]->val_real();
1680
    if (!args[0]->null_value && (null_value || nr < sum))
unknown's avatar
unknown committed
1681 1682 1683 1684 1685
    {
      sum=nr;
      null_value=0;
    }
  }
1686
  break;
1687
  case ROW_RESULT:
unknown's avatar
unknown committed
1688
  default:
unknown's avatar
unknown committed
1689 1690 1691
    // This case should never be choosen
    DBUG_ASSERT(0);
    break;
1692 1693 1694 1695 1696
  }
  return 0;
}


1697 1698
Item *Item_sum_max::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1699
  return new (thd->mem_root) Item_sum_max(thd, this);
1700 1701 1702
}


1703 1704 1705 1706
bool Item_sum_max::add()
{
  switch (hybrid_type) {
  case STRING_RESULT:
unknown's avatar
unknown committed
1707 1708 1709
  {
    String *result=args[0]->val_str(&tmp_value);
    if (!args[0]->null_value &&
1710
	(null_value || sortcmp(&value,result,collation.collation) < 0))
unknown's avatar
unknown committed
1711 1712 1713 1714 1715
    {
      value.copy(*result);
      null_value=0;
    }
  }
1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729
  break;
  case INT_RESULT:
  {
    longlong nr=args[0]->val_int();
    if (!args[0]->null_value && (null_value ||
				 (unsigned_flag && 
				  (ulonglong) nr > (ulonglong) sum_int) ||
				 (!unsigned_flag && nr > sum_int)))
    {
      sum_int=nr;
      null_value=0;
    }
  }
  break;
unknown's avatar
unknown committed
1730 1731
  case DECIMAL_RESULT:
  {
1732
    my_decimal value_buff, *val= args[0]->val_decimal(&value_buff);
unknown's avatar
unknown committed
1733 1734 1735 1736 1737 1738 1739 1740
    if (!args[0]->null_value &&
        (null_value || (my_decimal_cmp(val, &sum_dec) > 0)))
    {
      my_decimal2decimal(val, &sum_dec);
      null_value= 0;
    }
  }
  break;
1741 1742
  case REAL_RESULT:
  {
1743
    double nr= args[0]->val_real();
1744 1745 1746 1747 1748 1749 1750
    if (!args[0]->null_value && (null_value || nr > sum))
    {
      sum=nr;
      null_value=0;
    }
  }
  break;
1751
  case ROW_RESULT:
unknown's avatar
unknown committed
1752
  default:
unknown's avatar
unknown committed
1753 1754 1755
    // This case should never be choosen
    DBUG_ASSERT(0);
    break;
1756
  }
unknown's avatar
unknown committed
1757 1758 1759 1760 1761 1762 1763 1764
  return 0;
}


/* bit_or and bit_and */

longlong Item_sum_bit::val_int()
{
1765
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1766 1767 1768
  return (longlong) bits;
}

1769

1770
void Item_sum_bit::clear()
1771
{
1772
  bits= reset_bits;
1773 1774 1775
}

Item *Item_sum_or::copy_or_same(THD* thd)
unknown's avatar
unknown committed
1776
{
unknown's avatar
unknown committed
1777
  return new (thd->mem_root) Item_sum_or(thd, this);
unknown's avatar
unknown committed
1778 1779
}

1780

unknown's avatar
unknown committed
1781 1782 1783 1784 1785 1786 1787 1788
bool Item_sum_or::add()
{
  ulonglong value= (ulonglong) args[0]->val_int();
  if (!args[0]->null_value)
    bits|=value;
  return 0;
}

1789 1790
Item *Item_sum_xor::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1791
  return new (thd->mem_root) Item_sum_xor(thd, this);
1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802
}


bool Item_sum_xor::add()
{
  ulonglong value= (ulonglong) args[0]->val_int();
  if (!args[0]->null_value)
    bits^=value;
  return 0;
}

1803 1804
Item *Item_sum_and::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
1805
  return new (thd->mem_root) Item_sum_and(thd, this);
1806 1807 1808
}


unknown's avatar
unknown committed
1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822
bool Item_sum_and::add()
{
  ulonglong value= (ulonglong) args[0]->val_int();
  if (!args[0]->null_value)
    bits&=value;
  return 0;
}

/************************************************************************
** reset result of a Item_sum with is saved in a tmp_table
*************************************************************************/

void Item_sum_num::reset_field()
{
1823
  double nr= args[0]->val_real();
1824
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841

  if (maybe_null)
  {
    if (args[0]->null_value)
    {
      nr=0.0;
      result_field->set_null();
    }
    else
      result_field->set_notnull();
  }
  float8store(res,nr);
}


void Item_sum_hybrid::reset_field()
{
1842
  switch(hybrid_type) {
unknown's avatar
unknown committed
1843
  case STRING_RESULT:
unknown's avatar
unknown committed
1844 1845
  {
    char buff[MAX_FIELD_WIDTH];
1846
    String tmp(buff,sizeof(buff),result_field->charset()),*res;
unknown's avatar
unknown committed
1847 1848 1849 1850 1851 1852 1853 1854 1855 1856

    res=args[0]->val_str(&tmp);
    if (args[0]->null_value)
    {
      result_field->set_null();
      result_field->reset();
    }
    else
    {
      result_field->set_notnull();
1857
      result_field->store(res->ptr(),res->length(),tmp.charset());
unknown's avatar
unknown committed
1858
    }
unknown's avatar
unknown committed
1859
    break;
unknown's avatar
unknown committed
1860
  }
unknown's avatar
unknown committed
1861
  case INT_RESULT:
unknown's avatar
unknown committed
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874
  {
    longlong nr=args[0]->val_int();

    if (maybe_null)
    {
      if (args[0]->null_value)
      {
	nr=0;
	result_field->set_null();
      }
      else
	result_field->set_notnull();
    }
1875
    result_field->store(nr, unsigned_flag);
unknown's avatar
unknown committed
1876
    break;
unknown's avatar
unknown committed
1877
  }
unknown's avatar
unknown committed
1878
  case REAL_RESULT:
unknown's avatar
unknown committed
1879
  {
1880
    double nr= args[0]->val_real();
unknown's avatar
unknown committed
1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892

    if (maybe_null)
    {
      if (args[0]->null_value)
      {
	nr=0.0;
	result_field->set_null();
      }
      else
	result_field->set_notnull();
    }
    result_field->store(nr);
unknown's avatar
unknown committed
1893 1894 1895 1896
    break;
  }
  case DECIMAL_RESULT:
  {
1897
    my_decimal value_buff, *arg_dec= args[0]->val_decimal(&value_buff);
unknown's avatar
unknown committed
1898 1899 1900 1901 1902 1903 1904 1905

    if (maybe_null)
    {
      if (args[0]->null_value)
        result_field->set_null();
      else
        result_field->set_notnull();
    }
1906 1907 1908 1909 1910 1911 1912
    /*
      We must store zero in the field as we will use the field value in
      add()
    */
    if (!arg_dec)                               // Null
      arg_dec= &decimal_zero;
    result_field->store_decimal(arg_dec);
unknown's avatar
unknown committed
1913 1914 1915 1916 1917
    break;
  }
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
unknown's avatar
unknown committed
1918 1919 1920 1921 1922 1923
  }
}


void Item_sum_sum::reset_field()
{
unknown's avatar
unknown committed
1924 1925 1926
  if (hybrid_type == DECIMAL_RESULT)
  {
    my_decimal value, *arg_val= args[0]->val_decimal(&value);
1927 1928 1929
    if (!arg_val)                               // Null
      arg_val= &decimal_zero;
    result_field->store_decimal(arg_val);
unknown's avatar
unknown committed
1930 1931 1932 1933 1934 1935 1936
  }
  else
  {
    DBUG_ASSERT(hybrid_type == REAL_RESULT);
    double nr= args[0]->val_real();			// Nulls also return 0
    float8store(result_field->ptr, nr);
  }
unknown's avatar
unknown committed
1937 1938 1939 1940
  if (args[0]->null_value)
    result_field->set_null();
  else
    result_field->set_notnull();
unknown's avatar
unknown committed
1941 1942 1943 1944 1945
}


void Item_sum_count::reset_field()
{
1946
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
1947 1948
  longlong nr=0;

1949
  if (!args[0]->maybe_null || !args[0]->is_null())
unknown's avatar
unknown committed
1950 1951 1952 1953 1954 1955 1956
    nr=1;
  int8store(res,nr);
}


void Item_sum_avg::reset_field()
{
1957
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
1958 1959
  if (hybrid_type == DECIMAL_RESULT)
  {
1960
    longlong tmp;
unknown's avatar
unknown committed
1961 1962 1963
    my_decimal value, *arg_dec= args[0]->val_decimal(&value);
    if (args[0]->null_value)
    {
1964 1965
      arg_dec= &decimal_zero;
      tmp= 0;
unknown's avatar
unknown committed
1966 1967
    }
    else
1968 1969 1970 1971
      tmp= 1;
    my_decimal2binary(E_DEC_FATAL_ERROR, arg_dec, res, f_precision, f_scale);
    res+= dec_bin_size;
    int8store(res, tmp);
unknown's avatar
unknown committed
1972
  }
unknown's avatar
unknown committed
1973 1974
  else
  {
unknown's avatar
unknown committed
1975 1976 1977 1978 1979 1980
    double nr= args[0]->val_real();

    if (args[0]->null_value)
      bzero(res,sizeof(double)+sizeof(longlong));
    else
    {
1981
      longlong tmp= 1;
unknown's avatar
unknown committed
1982 1983 1984 1985
      float8store(res,nr);
      res+=sizeof(double);
      int8store(res,tmp);
    }
unknown's avatar
unknown committed
1986 1987 1988
  }
}

1989

unknown's avatar
unknown committed
1990
void Item_sum_bit::reset_field()
unknown's avatar
unknown committed
1991
{
1992 1993
  reset();
  int8store(result_field->ptr, bits);
unknown's avatar
unknown committed
1994 1995 1996
}

void Item_sum_bit::update_field()
unknown's avatar
unknown committed
1997
{
1998
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
1999 2000 2001
  bits= uint8korr(res);
  add();
  int8store(res, bits);
unknown's avatar
unknown committed
2002 2003
}

2004

unknown's avatar
unknown committed
2005 2006 2007 2008
/*
** calc next value and merge it with field_value
*/

unknown's avatar
unknown committed
2009
void Item_sum_sum::update_field()
unknown's avatar
unknown committed
2010
{
unknown's avatar
unknown committed
2011
  if (hybrid_type == DECIMAL_RESULT)
unknown's avatar
unknown committed
2012
  {
unknown's avatar
unknown committed
2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032
    my_decimal value, *arg_val= args[0]->val_decimal(&value);
    if (!args[0]->null_value)
    {
      if (!result_field->is_null())
      {
        my_decimal field_value,
                   *field_val= result_field->val_decimal(&field_value);
        my_decimal_add(E_DEC_FATAL_ERROR, dec_buffs, arg_val, field_val);
        result_field->store_decimal(dec_buffs);
      }
      else
      {
        result_field->store_decimal(arg_val);
        result_field->set_notnull();
      }
    }
  }
  else
  {
    double old_nr,nr;
2033
    uchar *res=result_field->ptr;
unknown's avatar
unknown committed
2034 2035 2036 2037 2038 2039 2040 2041 2042

    float8get(old_nr,res);
    nr= args[0]->val_real();
    if (!args[0]->null_value)
    {
      old_nr+=nr;
      result_field->set_notnull();
    }
    float8store(res,old_nr);
unknown's avatar
unknown committed
2043
  }
unknown's avatar
unknown committed
2044 2045 2046
}


unknown's avatar
unknown committed
2047
void Item_sum_count::update_field()
unknown's avatar
unknown committed
2048 2049
{
  longlong nr;
2050
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
2051

unknown's avatar
unknown committed
2052
  nr=sint8korr(res);
2053
  if (!args[0]->maybe_null || !args[0]->is_null())
unknown's avatar
unknown committed
2054 2055 2056 2057 2058
    nr++;
  int8store(res,nr);
}


unknown's avatar
unknown committed
2059
void Item_sum_avg::update_field()
unknown's avatar
unknown committed
2060 2061
{
  longlong field_count;
2062
  uchar *res=result_field->ptr;
unknown's avatar
unknown committed
2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074
  if (hybrid_type == DECIMAL_RESULT)
  {
    my_decimal value, *arg_val= args[0]->val_decimal(&value);
    if (!args[0]->null_value)
    {
      binary2my_decimal(E_DEC_FATAL_ERROR, res,
                        dec_buffs + 1, f_precision, f_scale);
      field_count= sint8korr(res + dec_bin_size);
      my_decimal_add(E_DEC_FATAL_ERROR, dec_buffs, arg_val, dec_buffs + 1);
      my_decimal2binary(E_DEC_FATAL_ERROR, dec_buffs,
                        res, f_precision, f_scale);
      res+= dec_bin_size;
2075
      field_count++;
unknown's avatar
unknown committed
2076 2077 2078 2079 2080
      int8store(res, field_count);
    }
  }
  else
  {
2081
    double nr;
unknown's avatar
unknown committed
2082

unknown's avatar
unknown committed
2083 2084 2085
    nr= args[0]->val_real();
    if (!args[0]->null_value)
    {
2086 2087 2088
      double old_nr;
      float8get(old_nr, res);
      field_count= sint8korr(res + sizeof(double));
unknown's avatar
unknown committed
2089
      old_nr+= nr;
2090 2091
      float8store(res,old_nr);
      res+= sizeof(double);
unknown's avatar
unknown committed
2092
      field_count++;
2093
      int8store(res, field_count);
unknown's avatar
unknown committed
2094
    }
unknown's avatar
unknown committed
2095 2096 2097
  }
}

2098

unknown's avatar
unknown committed
2099
void Item_sum_hybrid::update_field()
unknown's avatar
unknown committed
2100
{
2101
  switch (hybrid_type) {
unknown's avatar
unknown committed
2102
  case STRING_RESULT:
unknown's avatar
unknown committed
2103
    min_max_update_str_field();
unknown's avatar
unknown committed
2104 2105
    break;
  case INT_RESULT:
unknown's avatar
unknown committed
2106
    min_max_update_int_field();
unknown's avatar
unknown committed
2107 2108 2109 2110 2111
    break;
  case DECIMAL_RESULT:
    min_max_update_decimal_field();
    break;
  default:
unknown's avatar
unknown committed
2112
    min_max_update_real_field();
unknown's avatar
unknown committed
2113
  }
unknown's avatar
unknown committed
2114 2115 2116 2117
}


void
unknown's avatar
unknown committed
2118
Item_sum_hybrid::min_max_update_str_field()
unknown's avatar
unknown committed
2119 2120 2121
{
  String *res_str=args[0]->val_str(&value);

unknown's avatar
unknown committed
2122
  if (!args[0]->null_value)
unknown's avatar
unknown committed
2123
  {
2124
    result_field->val_str(&tmp_value);
unknown's avatar
unknown committed
2125 2126

    if (result_field->is_null() ||
2127
	(cmp_sign * sortcmp(res_str,&tmp_value,collation.collation)) < 0)
2128
      result_field->store(res_str->ptr(),res_str->length(),res_str->charset());
unknown's avatar
unknown committed
2129 2130 2131 2132 2133 2134
    result_field->set_notnull();
  }
}


void
unknown's avatar
unknown committed
2135
Item_sum_hybrid::min_max_update_real_field()
unknown's avatar
unknown committed
2136 2137 2138 2139
{
  double nr,old_nr;

  old_nr=result_field->val_real();
2140
  nr= args[0]->val_real();
unknown's avatar
unknown committed
2141 2142
  if (!args[0]->null_value)
  {
unknown's avatar
unknown committed
2143
    if (result_field->is_null(0) ||
unknown's avatar
unknown committed
2144 2145 2146 2147
	(cmp_sign > 0 ? old_nr > nr : old_nr < nr))
      old_nr=nr;
    result_field->set_notnull();
  }
unknown's avatar
unknown committed
2148
  else if (result_field->is_null(0))
unknown's avatar
unknown committed
2149 2150 2151 2152 2153 2154
    result_field->set_null();
  result_field->store(old_nr);
}


void
unknown's avatar
unknown committed
2155
Item_sum_hybrid::min_max_update_int_field()
unknown's avatar
unknown committed
2156 2157 2158 2159 2160 2161 2162
{
  longlong nr,old_nr;

  old_nr=result_field->val_int();
  nr=args[0]->val_int();
  if (!args[0]->null_value)
  {
unknown's avatar
unknown committed
2163
    if (result_field->is_null(0))
unknown's avatar
unknown committed
2164
      old_nr=nr;
2165 2166 2167 2168 2169 2170
    else
    {
      bool res=(unsigned_flag ?
		(ulonglong) old_nr > (ulonglong) nr :
		old_nr > nr);
      /* (cmp_sign > 0 && res) || (!(cmp_sign > 0) && !res) */
unknown's avatar
unknown committed
2171
      if ((cmp_sign > 0) ^ (!res))
2172 2173
	old_nr=nr;
    }
unknown's avatar
unknown committed
2174 2175
    result_field->set_notnull();
  }
unknown's avatar
unknown committed
2176
  else if (result_field->is_null(0))
unknown's avatar
unknown committed
2177
    result_field->set_null();
2178
  result_field->store(old_nr, unsigned_flag);
unknown's avatar
unknown committed
2179 2180 2181
}


unknown's avatar
unknown committed
2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208
void
Item_sum_hybrid::min_max_update_decimal_field()
{
  /* TODO: optimize: do not get result_field in case of args[0] is NULL */
  my_decimal old_val, nr_val;
  const my_decimal *old_nr= result_field->val_decimal(&old_val);
  const my_decimal *nr= args[0]->val_decimal(&nr_val);
  if (!args[0]->null_value)
  {
    if (result_field->is_null(0))
      old_nr=nr;
    else
    {
      bool res= my_decimal_cmp(old_nr, nr) > 0;
      /* (cmp_sign > 0 && res) || (!(cmp_sign > 0) && !res) */
      if ((cmp_sign > 0) ^ (!res))
        old_nr=nr;
    }
    result_field->set_notnull();
  }
  else if (result_field->is_null(0))
    result_field->set_null();
  result_field->store_decimal(old_nr);
}


Item_avg_field::Item_avg_field(Item_result res_type, Item_sum_avg *item)
unknown's avatar
unknown committed
2209 2210 2211
{
  name=item->name;
  decimals=item->decimals;
unknown's avatar
unknown committed
2212 2213
  max_length= item->max_length;
  unsigned_flag= item->unsigned_flag;
unknown's avatar
unknown committed
2214 2215
  field=item->result_field;
  maybe_null=1;
unknown's avatar
unknown committed
2216
  hybrid_type= res_type;
unknown's avatar
unknown committed
2217
  prec_increment= item->prec_increment;
unknown's avatar
unknown committed
2218 2219 2220 2221 2222 2223
  if (hybrid_type == DECIMAL_RESULT)
  {
    f_scale= item->f_scale;
    f_precision= item->f_precision;
    dec_bin_size= item->dec_bin_size;
  }
unknown's avatar
unknown committed
2224 2225
}

2226
double Item_avg_field::val_real()
unknown's avatar
unknown committed
2227
{
2228
  // fix_fields() never calls for this Item
2229 2230
  double nr;
  longlong count;
2231
  uchar *res;
2232

unknown's avatar
unknown committed
2233
  if (hybrid_type == DECIMAL_RESULT)
2234
    return val_real_from_decimal();
unknown's avatar
unknown committed
2235

2236 2237 2238 2239 2240 2241 2242
  float8get(nr,field->ptr);
  res= (field->ptr+sizeof(double));
  count= sint8korr(res);

  if ((null_value= !count))
    return 0.0;
  return nr/(double) count;
unknown's avatar
unknown committed
2243 2244
}

2245

unknown's avatar
unknown committed
2246 2247
longlong Item_avg_field::val_int()
{
2248
  return (longlong) rint(val_real());
unknown's avatar
unknown committed
2249 2250 2251
}


2252
my_decimal *Item_avg_field::val_decimal(my_decimal *dec_buf)
unknown's avatar
unknown committed
2253 2254
{
  // fix_fields() never calls for this Item
2255 2256 2257
  if (hybrid_type == REAL_RESULT)
    return val_decimal_from_real(dec_buf);

unknown's avatar
unknown committed
2258 2259
  longlong count= sint8korr(field->ptr + dec_bin_size);
  if ((null_value= !count))
2260
    return 0;
unknown's avatar
unknown committed
2261 2262 2263 2264 2265

  my_decimal dec_count, dec_field;
  binary2my_decimal(E_DEC_FATAL_ERROR,
                    field->ptr, &dec_field, f_precision, f_scale);
  int2my_decimal(E_DEC_FATAL_ERROR, count, 0, &dec_count);
unknown's avatar
unknown committed
2266 2267
  my_decimal_div(E_DEC_FATAL_ERROR, dec_buf,
                 &dec_field, &dec_count, prec_increment);
2268
  return dec_buf;
unknown's avatar
unknown committed
2269 2270
}

unknown's avatar
unknown committed
2271

unknown's avatar
unknown committed
2272 2273
String *Item_avg_field::val_str(String *str)
{
2274
  // fix_fields() never calls for this Item
unknown's avatar
unknown committed
2275
  if (hybrid_type == DECIMAL_RESULT)
2276 2277
    return val_string_from_decimal(str);
  return val_string_from_real(str);
unknown's avatar
unknown committed
2278 2279
}

2280

unknown's avatar
unknown committed
2281
Item_std_field::Item_std_field(Item_sum_std *item)
unknown's avatar
unknown committed
2282 2283 2284 2285
  : Item_variance_field(item)
{
}

2286

2287
double Item_std_field::val_real()
unknown's avatar
unknown committed
2288
{
2289
  double nr;
2290
  // fix_fields() never calls for this Item
2291 2292 2293
  nr= Item_variance_field::val_real();
  DBUG_ASSERT(nr >= 0.0);
  return sqrt(nr);
unknown's avatar
unknown committed
2294 2295
}

2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306

my_decimal *Item_std_field::val_decimal(my_decimal *dec_buf)
{
  /*
    We can't call val_decimal_from_real() for DECIMAL_RESULT as
    Item_variance_field::val_real() would cause an infinite loop
  */
  my_decimal tmp_dec, *dec;
  double nr;
  if (hybrid_type == REAL_RESULT)
    return val_decimal_from_real(dec_buf);
2307

2308 2309 2310 2311
  dec= Item_variance_field::val_decimal(dec_buf);
  if (!dec)
    return 0;
  my_decimal2double(E_DEC_FATAL_ERROR, dec, &nr);
2312 2313
  DBUG_ASSERT(nr >= 0.0);
  nr= sqrt(nr);
2314 2315 2316 2317 2318 2319
  double2my_decimal(E_DEC_FATAL_ERROR, nr, &tmp_dec);
  my_decimal_round(E_DEC_FATAL_ERROR, &tmp_dec, decimals, FALSE, dec_buf);
  return dec_buf;
}


unknown's avatar
unknown committed
2320
Item_variance_field::Item_variance_field(Item_sum_variance *item)
unknown's avatar
unknown committed
2321 2322 2323 2324
{
  name=item->name;
  decimals=item->decimals;
  max_length=item->max_length;
unknown's avatar
unknown committed
2325
  unsigned_flag= item->unsigned_flag;
unknown's avatar
unknown committed
2326 2327
  field=item->result_field;
  maybe_null=1;
2328
  sample= item->sample;
unknown's avatar
unknown committed
2329
  prec_increment= item->prec_increment;
unknown's avatar
unknown committed
2330 2331 2332 2333 2334 2335 2336 2337 2338
  if ((hybrid_type= item->hybrid_type) == DECIMAL_RESULT)
  {
    f_scale0= item->f_scale0;
    f_precision0= item->f_precision0;
    dec_bin_size0= item->dec_bin_size0;
    f_scale1= item->f_scale1;
    f_precision1= item->f_precision1;
    dec_bin_size1= item->dec_bin_size1;
  }
unknown's avatar
unknown committed
2339 2340
}

2341

2342
double Item_variance_field::val_real()
unknown's avatar
unknown committed
2343
{
2344
  // fix_fields() never calls for this Item
unknown's avatar
unknown committed
2345
  if (hybrid_type == DECIMAL_RESULT)
2346 2347
    return val_real_from_decimal();

2348 2349 2350
  double recurrence_s;
  ulonglong count;
  float8get(recurrence_s, (field->ptr + sizeof(double)));
unknown's avatar
unknown committed
2351 2352
  count=sint8korr(field->ptr+sizeof(double)*2);

2353
  if ((null_value= (count <= sample)))
unknown's avatar
unknown committed
2354
    return 0.0;
2355

2356
  return variance_fp_recurrence_result(recurrence_s, count, sample);
unknown's avatar
unknown committed
2357 2358 2359
}


unknown's avatar
unknown committed
2360 2361 2362 2363
/****************************************************************************
** COUNT(DISTINCT ...)
****************************************************************************/

2364
int simple_str_key_cmp(void* arg, uchar* key1, uchar* key2)
2365
{
2366
  Field *f= (Field*) arg;
2367
  return f->cmp(key1, key2);
2368 2369
}

unknown's avatar
unknown committed
2370 2371 2372 2373 2374 2375 2376
/*
  Did not make this one static - at least gcc gets confused when
  I try to declare a static function as a friend. If you can figure
  out the syntax to make a static function a friend, make this one
  static
*/

2377
int composite_key_cmp(void* arg, uchar* key1, uchar* key2)
2378 2379
{
  Item_sum_count_distinct* item = (Item_sum_count_distinct*)arg;
unknown's avatar
unknown committed
2380
  Field **field    = item->table->field;
2381
  Field **field_end= field + item->table->s->fields;
unknown's avatar
unknown committed
2382 2383 2384 2385 2386
  uint32 *lengths=item->field_lengths;
  for (; field < field_end; ++field)
  {
    Field* f = *field;
    int len = *lengths++;
2387
    int res = f->cmp(key1, key2);
unknown's avatar
unknown committed
2388 2389 2390 2391 2392
    if (res)
      return res;
    key1 += len;
    key2 += len;
  }
2393 2394 2395 2396
  return 0;
}


2397 2398
C_MODE_START

2399
static int count_distinct_walk(void *elem, element_count count, void *arg)
2400
{
2401
  (*((ulonglong*)arg))++;
2402 2403
  return 0;
}
2404

2405 2406
C_MODE_END

unknown's avatar
unknown committed
2407

unknown's avatar
unknown committed
2408
void Item_sum_count_distinct::cleanup()
unknown's avatar
unknown committed
2409
{
unknown's avatar
unknown committed
2410
  DBUG_ENTER("Item_sum_count_distinct::cleanup");
unknown's avatar
unknown committed
2411
  Item_sum_int::cleanup();
2412 2413

  /* Free objects only if we own them. */
2414 2415
  if (!original)
  {
2416 2417 2418 2419 2420 2421 2422
    /*
      We need to delete the table and the tree in cleanup() as
      they were allocated in the runtime memroot. Using the runtime
      memroot reduces memory footprint for PS/SP and simplifies setup().
    */
    delete tree;
    tree= 0;
2423
    is_evaluated= FALSE;
2424
    if (table)
unknown's avatar
unknown committed
2425
    {
2426
      free_tmp_table(table->in_use, table);
unknown's avatar
unknown committed
2427 2428
      table= 0;
    }
2429
    delete tmp_table_param;
unknown's avatar
unknown committed
2430
    tmp_table_param= 0;
2431
  }
2432
  always_null= FALSE;
unknown's avatar
unknown committed
2433
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
2434 2435
}

unknown's avatar
unknown committed
2436

2437 2438 2439 2440 2441 2442
/* This is used by rollup to create a separate usable copy of the function */

void Item_sum_count_distinct::make_unique()
{
  table=0;
  original= 0;
2443
  force_copy_fields= 1;
2444
  tree= 0;
2445
  is_evaluated= FALSE;
2446 2447 2448 2449 2450 2451 2452 2453
  tmp_table_param= 0;
  always_null= FALSE;
}


Item_sum_count_distinct::~Item_sum_count_distinct()
{
  cleanup();
2454 2455 2456
}


unknown's avatar
unknown committed
2457 2458 2459
bool Item_sum_count_distinct::setup(THD *thd)
{
  List<Item> list;
unknown's avatar
unknown committed
2460
  SELECT_LEX *select_lex= thd->lex->current_select;
2461 2462 2463 2464 2465 2466 2467 2468

  /*
    Setup can be called twice for ROLLUP items. This is a bug.
    Please add DBUG_ASSERT(tree == 0) here when it's fixed.
  */
  if (tree || table || tmp_table_param)
    return FALSE;

2469
  if (!(tmp_table_param= new TMP_TABLE_PARAM))
unknown's avatar
unknown committed
2470
    return TRUE;
2471

unknown's avatar
unknown committed
2472 2473
  /* Create a table with an unique key over all parameters */
  for (uint i=0; i < arg_count ; i++)
2474 2475 2476
  {
    Item *item=args[i];
    if (list.push_back(item))
unknown's avatar
unknown committed
2477
      return TRUE;                              // End of memory
2478 2479
    if (item->const_item() && item->is_null())
      always_null= 1;
2480 2481
  }
  if (always_null)
unknown's avatar
unknown committed
2482
    return FALSE;
2483
  count_field_types(select_lex, tmp_table_param, list, 0);
2484
  tmp_table_param->force_copy_fields= force_copy_fields;
2485
  DBUG_ASSERT(table == 0);
2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502
  /*
    Make create_tmp_table() convert BIT columns to BIGINT.
    This is needed because BIT fields store parts of their data in table's
    null bits, and we don't have methods to compare two table records, which
    is needed by Unique which is used when HEAP table is used.
  */
  {
    List_iterator_fast<Item> li(list);
    Item *item;
    while ((item= li++))
    {
      if (item->type() == Item::FIELD_ITEM &&
          ((Item_field*)item)->field->type() == FIELD_TYPE_BIT)
        item->marker=4;
    }
  }

2503
  if (!(table= create_tmp_table(thd, tmp_table_param, list, (ORDER*) 0, 1,
unknown's avatar
unknown committed
2504
				0,
2505
				(select_lex->options | thd->options),
2506
				HA_POS_ERROR, (char*)"")))
unknown's avatar
unknown committed
2507
    return TRUE;
unknown's avatar
unknown committed
2508
  table->file->extra(HA_EXTRA_NO_ROWS);		// Don't update rows
2509
  table->no_rows=1;
2510

unknown's avatar
unknown committed
2511
  if (table->s->db_type() == heap_hton)
unknown's avatar
unknown committed
2512
  {
2513 2514 2515 2516
    /*
      No blobs, otherwise it would have been MyISAM: set up a compare
      function and its arguments to use with Unique.
    */
unknown's avatar
unknown committed
2517 2518
    qsort_cmp2 compare_key;
    void* cmp_arg;
2519 2520 2521
    Field **field= table->field;
    Field **field_end= field + table->s->fields;
    bool all_binary= TRUE;
unknown's avatar
unknown committed
2522

2523
    for (tree_key_length= 0; field < field_end; ++field)
unknown's avatar
unknown committed
2524
    {
2525
      Field *f= *field;
2526
      enum enum_field_types f_type= f->type();
2527
      tree_key_length+= f->pack_length();
2528 2529 2530
      if ((f_type == MYSQL_TYPE_VARCHAR) ||
          !f->binary() && (f_type == MYSQL_TYPE_STRING ||
                           f_type == MYSQL_TYPE_VAR_STRING))
2531 2532 2533
      {
        all_binary= FALSE;
        break;
unknown's avatar
unknown committed
2534
      }
2535
    }
2536
    if (all_binary)
unknown's avatar
unknown committed
2537
    {
2538 2539 2540 2541 2542 2543
      cmp_arg= (void*) &tree_key_length;
      compare_key= (qsort_cmp2) simple_raw_key_cmp;
    }
    else
    {
      if (table->s->fields == 1)
unknown's avatar
unknown committed
2544
      {
2545 2546 2547 2548 2549 2550 2551 2552 2553
        /*
          If we have only one field, which is the most common use of
          count(distinct), it is much faster to use a simpler key
          compare method that can take advantage of not having to worry
          about other fields.
        */
        compare_key= (qsort_cmp2) simple_str_key_cmp;
        cmp_arg= (void*) table->field[0];
        /* tree_key_length has been set already */
unknown's avatar
unknown committed
2554 2555 2556
      }
      else
      {
2557 2558 2559 2560 2561 2562 2563 2564 2565 2566
        uint32 *length;
        compare_key= (qsort_cmp2) composite_key_cmp;
        cmp_arg= (void*) this;
        field_lengths= (uint32*) thd->alloc(table->s->fields * sizeof(uint32));
        for (tree_key_length= 0, length= field_lengths, field= table->field;
             field < field_end; ++field, ++length)
        {
          *length= (*field)->pack_length();
          tree_key_length+= *length;
        }
unknown's avatar
unknown committed
2567 2568
      }
    }
2569 2570 2571
    DBUG_ASSERT(tree == 0);
    tree= new Unique(compare_key, cmp_arg, tree_key_length,
                     thd->variables.max_heap_table_size);
unknown's avatar
unknown committed
2572
    /*
2573
      The only time tree_key_length could be 0 is if someone does
unknown's avatar
unknown committed
2574 2575 2576 2577
      count(distinct) on a char(0) field - stupid thing to do,
      but this has to be handled - otherwise someone can crash
      the server with a DoS attack
    */
2578
    is_evaluated= FALSE;
2579 2580
    if (! tree)
      return TRUE;
unknown's avatar
unknown committed
2581
  }
2582
  return FALSE;
2583
}
unknown's avatar
unknown committed
2584

2585 2586 2587

Item *Item_sum_count_distinct::copy_or_same(THD* thd) 
{
unknown's avatar
unknown committed
2588
  return new (thd->mem_root) Item_sum_count_distinct(thd, this);
2589 2590 2591
}


2592
void Item_sum_count_distinct::clear()
unknown's avatar
unknown committed
2593
{
2594
  /* tree and table can be both null only if always_null */
2595
  is_evaluated= FALSE;
2596
  if (tree)
2597
  {
2598
    tree->reset();
2599
  }
unknown's avatar
merge  
unknown committed
2600
  else if (table)
2601 2602 2603 2604 2605
  {
    table->file->extra(HA_EXTRA_NO_CACHE);
    table->file->delete_all_rows();
    table->file->extra(HA_EXTRA_WRITE_CACHE);
  }
unknown's avatar
unknown committed
2606 2607 2608 2609 2610
}

bool Item_sum_count_distinct::add()
{
  int error;
2611 2612
  if (always_null)
    return 0;
unknown's avatar
unknown committed
2613
  copy_fields(tmp_table_param);
unknown's avatar
unknown committed
2614
  copy_funcs(tmp_table_param->items_to_copy);
unknown's avatar
unknown committed
2615

2616 2617 2618 2619
  for (Field **field=table->field ; *field ; field++)
    if ((*field)->is_real_null(0))
      return 0;					// Don't count NULL

2620
  is_evaluated= FALSE;
2621
  if (tree)
unknown's avatar
unknown committed
2622 2623
  {
    /*
2624 2625 2626 2627
      The first few bytes of record (at least one) are just markers
      for deleted and NULLs. We want to skip them since they will
      bloat the tree without providing any valuable info. Besides,
      key_length used to initialize the tree didn't include space for them.
unknown's avatar
unknown committed
2628
    */
2629
    return tree->unique_add(table->record[0] + table->s->null_bytes);
unknown's avatar
unknown committed
2630
  }
2631
  if ((error= table->file->ha_write_row(table->record[0])) &&
2632
      table->file->is_fatal_error(error, HA_CHECK_DUP))
2633 2634
    return TRUE;
  return FALSE;
unknown's avatar
unknown committed
2635 2636
}

unknown's avatar
unknown committed
2637

unknown's avatar
unknown committed
2638 2639
longlong Item_sum_count_distinct::val_int()
{
2640
  int error;
2641
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2642 2643
  if (!table)					// Empty query
    return LL(0);
2644 2645
  if (tree)
  {
2646 2647
    if (is_evaluated)
      return count;
2648 2649 2650 2651 2652

    if (tree->elements == 0)
      return (longlong) tree->elements_in_tree(); // everything fits in memory
    count= 0;
    tree->walk(count_distinct_walk, (void*) &count);
2653
    is_evaluated= TRUE;
2654 2655
    return (longlong) count;
  }
2656 2657 2658 2659 2660 2661 2662 2663

  error= table->file->info(HA_STATUS_VARIABLE | HA_STATUS_NO_LOCK);

  if(error)
  {
    table->file->print_error(error, MYF(0));
  }

2664
  return table->file->stats.records;
unknown's avatar
unknown committed
2665 2666
}

2667

unknown's avatar
unknown committed
2668 2669 2670 2671 2672 2673 2674 2675 2676
/****************************************************************************
** Functions to handle dynamic loadable aggregates
** Original source by: Alexis Mikhailov <root@medinf.chuvashia.su>
** Adapted for UDAs by: Andreas F. Bobak <bobak@relog.ch>.
** Rewritten by: Monty.
****************************************************************************/

#ifdef HAVE_DLOPEN

2677
void Item_udf_sum::clear()
unknown's avatar
unknown committed
2678
{
2679
  DBUG_ENTER("Item_udf_sum::clear");
2680
  udf.clear();
2681
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
2682 2683 2684 2685
}

bool Item_udf_sum::add()
{
2686
  DBUG_ENTER("Item_udf_sum::add");
unknown's avatar
unknown committed
2687 2688 2689 2690
  udf.add(&null_value);
  DBUG_RETURN(0);
}

2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701
void Item_udf_sum::cleanup()
{
  /*
    udf_handler::cleanup() nicely handles case when we have not
    original item but one created by copy_or_same() method.
  */
  udf.cleanup();
  Item_sum::cleanup();
}


2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715
void Item_udf_sum::print(String *str)
{
  str->append(func_name());
  str->append('(');
  for (uint i=0 ; i < arg_count ; i++)
  {
    if (i)
      str->append(',');
    args[i]->print(str);
  }
  str->append(')');
}


2716 2717
Item *Item_sum_udf_float::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
2718
  return new (thd->mem_root) Item_sum_udf_float(thd, this);
2719 2720
}

2721
double Item_sum_udf_float::val_real()
unknown's avatar
unknown committed
2722
{
2723
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2724 2725 2726 2727 2728 2729
  DBUG_ENTER("Item_sum_udf_float::val");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
		     args[0]->result_type(), arg_count));
  DBUG_RETURN(udf.val(&null_value));
}

2730

unknown's avatar
unknown committed
2731 2732
String *Item_sum_udf_float::val_str(String *str)
{
2733
  return val_string_from_real(str);
unknown's avatar
unknown committed
2734 2735 2736
}


2737
my_decimal *Item_sum_udf_float::val_decimal(my_decimal *dec)
2738
{
2739
  return val_decimal_from_real(dec);
2740 2741 2742
}


unknown's avatar
unknown committed
2743 2744
String *Item_sum_udf_decimal::val_str(String *str)
{
2745
  return val_string_from_decimal(str);
unknown's avatar
unknown committed
2746 2747 2748 2749 2750
}


double Item_sum_udf_decimal::val_real()
{
2751
  return val_real_from_decimal();
unknown's avatar
unknown committed
2752 2753 2754 2755 2756
}


longlong Item_sum_udf_decimal::val_int()
{
2757
  return val_int_from_decimal();
unknown's avatar
unknown committed
2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777
}


my_decimal *Item_sum_udf_decimal::val_decimal(my_decimal *dec_buf)
{
  DBUG_ASSERT(fixed == 1);
  DBUG_ENTER("Item_func_udf_decimal::val_decimal");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
                     args[0]->result_type(), arg_count));

  DBUG_RETURN(udf.val_decimal(&null_value, dec_buf));
}


Item *Item_sum_udf_decimal::copy_or_same(THD* thd)
{
  return new (thd->mem_root) Item_sum_udf_decimal(thd, this);
}


2778 2779 2780 2781 2782
Item *Item_sum_udf_int::copy_or_same(THD* thd)
{
  return new (thd->mem_root) Item_sum_udf_int(thd, this);
}

unknown's avatar
unknown committed
2783 2784
longlong Item_sum_udf_int::val_int()
{
2785
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2786 2787 2788 2789 2790 2791
  DBUG_ENTER("Item_sum_udf_int::val_int");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
		     args[0]->result_type(), arg_count));
  DBUG_RETURN(udf.val_int(&null_value));
}

2792

unknown's avatar
unknown committed
2793 2794
String *Item_sum_udf_int::val_str(String *str)
{
2795
  return val_string_from_int(str);
unknown's avatar
unknown committed
2796 2797
}

2798 2799 2800 2801 2802 2803
my_decimal *Item_sum_udf_int::val_decimal(my_decimal *dec)
{
  return val_decimal_from_int(dec);
}


unknown's avatar
unknown committed
2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814
/* Default max_length is max argument length */

void Item_sum_udf_str::fix_length_and_dec()
{
  DBUG_ENTER("Item_sum_udf_str::fix_length_and_dec");
  max_length=0;
  for (uint i = 0; i < arg_count; i++)
    set_if_bigger(max_length,args[i]->max_length);
  DBUG_VOID_RETURN;
}

2815 2816 2817

Item *Item_sum_udf_str::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
2818
  return new (thd->mem_root) Item_sum_udf_str(thd, this);
2819 2820 2821
}


2822 2823 2824 2825 2826
my_decimal *Item_sum_udf_str::val_decimal(my_decimal *dec)
{
  return val_decimal_from_string(dec);
}

unknown's avatar
unknown committed
2827 2828
String *Item_sum_udf_str::val_str(String *str)
{
2829
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2830 2831 2832 2833 2834 2835 2836
  DBUG_ENTER("Item_sum_udf_str::str");
  String *res=udf.val_str(str,&str_value);
  null_value = !res;
  DBUG_RETURN(res);
}

#endif /* HAVE_DLOPEN */
2837 2838 2839 2840


/*****************************************************************************
 GROUP_CONCAT function
unknown's avatar
unknown committed
2841 2842

 SQL SYNTAX:
unknown's avatar
unknown committed
2843
  GROUP_CONCAT([DISTINCT] expr,... [ORDER BY col [ASC|DESC],...]
unknown's avatar
unknown committed
2844 2845
    [SEPARATOR str_const])

2846
 concat of values from "group by" operation
unknown's avatar
unknown committed
2847 2848 2849 2850 2851

 BUGS
   DISTINCT and ORDER BY only works if ORDER BY uses all fields and only fields
   in expression list
   Blobs doesn't work with DISTINCT or ORDER BY
2852 2853 2854
*****************************************************************************/

/*
unknown's avatar
SCRUM  
unknown committed
2855 2856
  function of sort for syntax:
  GROUP_CONCAT(DISTINCT expr,...)
2857 2858
*/

2859 2860
int group_concat_key_cmp_with_distinct(void* arg, uchar* key1,
				       uchar* key2)
2861
{
unknown's avatar
unknown committed
2862
  Item_func_group_concat* grp_item= (Item_func_group_concat*)arg;
unknown's avatar
unknown committed
2863
  TABLE *table= grp_item->table;
unknown's avatar
unknown committed
2864
  Item **field_item, **end;
unknown's avatar
unknown committed
2865

unknown's avatar
unknown committed
2866
  for (field_item= grp_item->args, end= field_item + grp_item->arg_count_field;
unknown's avatar
unknown committed
2867 2868
       field_item < end;
       field_item++)
2869
  {
unknown's avatar
unknown committed
2870 2871 2872 2873 2874 2875
    /*
      We have to use get_tmp_table_field() instead of
      real_item()->get_tmp_table_field() because we want the field in
      the temporary table, not the original field
    */
    Field *field= (*field_item)->get_tmp_table_field();
unknown's avatar
unknown committed
2876
    /* 
2877
      If field_item is a const item then either get_tmp_table_field returns 0
unknown's avatar
unknown committed
2878 2879 2880
      or it is an item over a const table. 
    */
    if (field && !(*field_item)->const_item())
2881
    {
unknown's avatar
unknown committed
2882
      int res;
2883 2884
      uint offset= (field->offset(field->table->record[0]) - 
                    table->s->null_bytes);
2885
      if ((res= field->cmp(key1 + offset, key2 + offset)))
unknown's avatar
unknown committed
2886
	return res;
2887
    }
unknown's avatar
unknown committed
2888
  }
2889 2890 2891
  return 0;
}

2892

2893
/*
unknown's avatar
SCRUM  
unknown committed
2894 2895
  function of sort for syntax:
  GROUP_CONCAT(expr,... ORDER BY col,... )
2896 2897
*/

2898
int group_concat_key_cmp_with_order(void* arg, uchar* key1, uchar* key2)
2899
{
unknown's avatar
unknown committed
2900
  Item_func_group_concat* grp_item= (Item_func_group_concat*) arg;
unknown's avatar
unknown committed
2901
  ORDER **order_item, **end;
unknown's avatar
unknown committed
2902
  TABLE *table= grp_item->table;
unknown's avatar
unknown committed
2903

unknown's avatar
unknown committed
2904
  for (order_item= grp_item->order, end=order_item+ grp_item->arg_count_order;
unknown's avatar
unknown committed
2905 2906
       order_item < end;
       order_item++)
2907
  {
unknown's avatar
unknown committed
2908
    Item *item= *(*order_item)->item;
unknown's avatar
unknown committed
2909 2910 2911 2912 2913 2914
    /*
      We have to use get_tmp_table_field() instead of
      real_item()->get_tmp_table_field() because we want the field in
      the temporary table, not the original field
    */
    Field *field= item->get_tmp_table_field();
unknown's avatar
unknown committed
2915 2916 2917 2918 2919
    /* 
      If item is a const item then either get_tp_table_field returns 0
      or it is an item over a const table. 
    */
    if (field && !item->const_item())
2920
    {
unknown's avatar
unknown committed
2921
      int res;
2922 2923
      uint offset= (field->offset(field->table->record[0]) -
                    table->s->null_bytes);
2924
      if ((res= field->cmp(key1 + offset, key2 + offset)))
unknown's avatar
unknown committed
2925
        return (*order_item)->asc ? res : -res;
2926
    }
unknown's avatar
unknown committed
2927
  }
unknown's avatar
SCRUM  
unknown committed
2928
  /*
unknown's avatar
unknown committed
2929 2930
    We can't return 0 because in that case the tree class would remove this
    item as double value. This would cause problems for case-changes and
unknown's avatar
unknown committed
2931
    if the returned values are not the same we do the sort on.
unknown's avatar
unknown committed
2932
  */
2933 2934 2935
  return 1;
}

2936

2937
/*
unknown's avatar
SCRUM  
unknown committed
2938 2939
  function of sort for syntax:
  GROUP_CONCAT(DISTINCT expr,... ORDER BY col,... )
unknown's avatar
unknown committed
2940 2941 2942 2943 2944

  BUG:
    This doesn't work in the case when the order by contains data that
    is not part of the field list because tree-insert will not notice
    the duplicated values when inserting things sorted by ORDER BY
2945
*/
unknown's avatar
SCRUM  
unknown committed
2946

2947 2948
int group_concat_key_cmp_with_distinct_and_order(void* arg,uchar* key1,
						 uchar* key2)
2949 2950 2951 2952 2953 2954
{
  if (!group_concat_key_cmp_with_distinct(arg,key1,key2))
    return 0;
  return(group_concat_key_cmp_with_order(arg,key1,key2));
}

2955

2956
/*
unknown's avatar
unknown committed
2957
  Append data from current leaf to item->result
2958
*/
unknown's avatar
SCRUM  
unknown committed
2959

2960
int dump_leaf_key(uchar* key, element_count count __attribute__((unused)),
unknown's avatar
unknown committed
2961
                  Item_func_group_concat *item)
2962
{
unknown's avatar
unknown committed
2963
  TABLE *table= item->table;
unknown's avatar
unknown committed
2964 2965 2966
  String tmp((char *)table->record[1], table->s->reclength,
             default_charset_info);
  String tmp2;
unknown's avatar
unknown committed
2967
  String *result= &item->result;
unknown's avatar
unknown committed
2968
  Item **arg= item->args, **arg_end= item->args + item->arg_count_field;
2969
  uint old_length= result->length();
unknown's avatar
unknown committed
2970

2971 2972 2973
  if (item->no_appended)
    item->no_appended= FALSE;
  else
unknown's avatar
unknown committed
2974
    result->append(*item->separator);
unknown's avatar
a fix  
unknown committed
2975

2976
  tmp.length(0);
unknown's avatar
unknown committed
2977 2978

  for (; arg < arg_end; arg++)
2979
  {
unknown's avatar
unknown committed
2980 2981
    String *res;
    if (! (*arg)->const_item())
2982
    {
unknown's avatar
unknown committed
2983 2984 2985 2986
      /*
	We have to use get_tmp_table_field() instead of
	real_item()->get_tmp_table_field() because we want the field in
	the temporary table, not the original field
unknown's avatar
unknown committed
2987 2988
        We also can't use table->field array to access the fields
        because it contains both order and arg list fields.
unknown's avatar
unknown committed
2989
      */
unknown's avatar
unknown committed
2990
      Field *field= (*arg)->get_tmp_table_field();
2991 2992
      uint offset= (field->offset(field->table->record[0]) -
                    table->s->null_bytes);
unknown's avatar
unknown committed
2993
      DBUG_ASSERT(offset < table->s->reclength);
2994
      res= field->val_str(&tmp, key + offset);
2995
    }
unknown's avatar
unknown committed
2996 2997 2998
    else
      res= (*arg)->val_str(&tmp);
    if (res)
unknown's avatar
unknown committed
2999
      result->append(*res);
3000
  }
unknown's avatar
unknown committed
3001

unknown's avatar
unknown committed
3002
  /* stop if length of result more than max_length */
unknown's avatar
unknown committed
3003
  if (result->length() > item->max_length)
3004
  {
3005 3006
    int well_formed_error;
    CHARSET_INFO *cs= item->collation.collation;
unknown's avatar
unknown committed
3007
    const char *ptr= result->ptr();
3008 3009 3010 3011 3012 3013 3014 3015
    uint add_length;
    /*
      It's ok to use item->result.length() as the fourth argument
      as this is never used to limit the length of the data.
      Cut is done with the third argument.
    */
    add_length= cs->cset->well_formed_len(cs,
                                          ptr + old_length,
unknown's avatar
unknown committed
3016 3017
                                          ptr + item->max_length,
                                          result->length(),
3018
                                          &well_formed_error);
unknown's avatar
unknown committed
3019
    result->length(old_length + add_length);
unknown's avatar
unknown committed
3020 3021
    item->count_cut_values++;
    item->warning_for_row= TRUE;
3022 3023 3024 3025 3026
    return 1;
  }
  return 0;
}

3027

3028
/*
unknown's avatar
SCRUM  
unknown committed
3029
  Constructor of Item_func_group_concat
unknown's avatar
unknown committed
3030 3031 3032 3033
  distinct_arg - distinct
  select_list - list of expression for show values
  order_list - list of sort columns
  separator_arg - string value of separator
3034
*/
unknown's avatar
SCRUM  
unknown committed
3035

unknown's avatar
unknown committed
3036
Item_func_group_concat::
3037 3038
Item_func_group_concat(Name_resolution_context *context_arg,
                       bool distinct_arg, List<Item> *select_list,
unknown's avatar
unknown committed
3039 3040 3041
                       SQL_LIST *order_list, String *separator_arg)
  :tmp_table_param(0), warning(0),
   separator(separator_arg), tree(0), table(0),
3042
   order(0), context(context_arg),
unknown's avatar
unknown committed
3043 3044 3045 3046 3047
   arg_count_order(order_list ? order_list->elements : 0),
   arg_count_field(select_list->elements),
   count_cut_values(0),
   distinct(distinct_arg),
   warning_for_row(FALSE),
unknown's avatar
unknown committed
3048
   force_copy_fields(0), original(0)
3049
{
unknown's avatar
unknown committed
3050 3051 3052
  Item *item_select;
  Item **arg_ptr;

unknown's avatar
unknown committed
3053
  quick_group= FALSE;
unknown's avatar
unknown committed
3054
  arg_count= arg_count_field + arg_count_order;
unknown's avatar
unknown committed
3055

unknown's avatar
SCRUM  
unknown committed
3056 3057
  /*
    We need to allocate:
unknown's avatar
unknown committed
3058 3059
    args - arg_count_field+arg_count_order
           (for possible order items in temporare tables)
unknown's avatar
SCRUM  
unknown committed
3060 3061
    order - arg_count_order
  */
unknown's avatar
unknown committed
3062
  if (!(args= (Item**) sql_alloc(sizeof(Item*) * arg_count +
unknown's avatar
unknown committed
3063
                                 sizeof(ORDER*)*arg_count_order)))
unknown's avatar
unknown committed
3064
    return;
unknown's avatar
unknown committed
3065

unknown's avatar
unknown committed
3066 3067
  order= (ORDER**)(args + arg_count);

unknown's avatar
unknown committed
3068
  /* fill args items of show and sort */
unknown's avatar
unknown committed
3069
  List_iterator_fast<Item> li(*select_list);
3070

unknown's avatar
unknown committed
3071 3072
  for (arg_ptr=args ; (item_select= li++) ; arg_ptr++)
    *arg_ptr= item_select;
unknown's avatar
unknown committed
3073

unknown's avatar
unknown committed
3074
  if (arg_count_order)
unknown's avatar
SCRUM  
unknown committed
3075
  {
unknown's avatar
unknown committed
3076
    ORDER **order_ptr= order;
unknown's avatar
unknown committed
3077 3078 3079
    for (ORDER *order_item= (ORDER*) order_list->first;
         order_item != NULL;
         order_item= order_item->next)
3080
    {
unknown's avatar
unknown committed
3081 3082 3083
      (*order_ptr++)= order_item;
      *arg_ptr= *order_item->item;
      order_item->item= arg_ptr++;
3084 3085 3086
    }
  }
}
unknown's avatar
unknown committed
3087

unknown's avatar
unknown committed
3088 3089

Item_func_group_concat::Item_func_group_concat(THD *thd,
unknown's avatar
unknown committed
3090 3091
                                               Item_func_group_concat *item)
  :Item_sum(thd, item),
unknown's avatar
unknown committed
3092 3093 3094 3095 3096 3097
  tmp_table_param(item->tmp_table_param),
  warning(item->warning),
  separator(item->separator),
  tree(item->tree),
  table(item->table),
  order(item->order),
3098
  context(item->context),
unknown's avatar
unknown committed
3099 3100 3101
  arg_count_order(item->arg_count_order),
  arg_count_field(item->arg_count_field),
  count_cut_values(item->count_cut_values),
unknown's avatar
unknown committed
3102 3103 3104
  distinct(item->distinct),
  warning_for_row(item->warning_for_row),
  always_null(item->always_null),
unknown's avatar
unknown committed
3105
  force_copy_fields(item->force_copy_fields),
unknown's avatar
unknown committed
3106 3107 3108
  original(item)
{
  quick_group= item->quick_group;
3109
  result.set_charset(collation.collation);
unknown's avatar
unknown committed
3110 3111
}

3112

unknown's avatar
SCRUM  
unknown committed
3113

unknown's avatar
unknown committed
3114 3115
void Item_func_group_concat::cleanup()
{
unknown's avatar
unknown committed
3116
  DBUG_ENTER("Item_func_group_concat::cleanup");
3117
  Item_sum::cleanup();
unknown's avatar
unknown committed
3118

3119 3120 3121 3122 3123
  /* Adjust warning message to include total number of cut values */
  if (warning)
  {
    char warn_buff[MYSQL_ERRMSG_SIZE];
    sprintf(warn_buff, ER(ER_CUT_VALUE_GROUP_CONCAT), count_cut_values);
3124
    warning->set_msg(current_thd, warn_buff);
3125 3126 3127
    warning= 0;
  }

unknown's avatar
unknown committed
3128 3129 3130 3131 3132 3133
  /*
    Free table and tree if they belong to this item (if item have not pointer
    to original item from which was made copy => it own its objects )
  */
  if (!original)
  {
unknown's avatar
unknown committed
3134 3135
    delete tmp_table_param;
    tmp_table_param= 0;
unknown's avatar
unknown committed
3136
    if (table)
unknown's avatar
unknown committed
3137
    {
unknown's avatar
unknown committed
3138
      THD *thd= table->in_use;
unknown's avatar
unknown committed
3139
      free_tmp_table(thd, table);
unknown's avatar
unknown committed
3140
      table= 0;
unknown's avatar
unknown committed
3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152
      if (tree)
      {
        delete_tree(tree);
        tree= 0;
      }
      if (warning)
      {
        char warn_buff[MYSQL_ERRMSG_SIZE];
        sprintf(warn_buff, ER(ER_CUT_VALUE_GROUP_CONCAT), count_cut_values);
        warning->set_msg(thd, warn_buff);
        warning= 0;
      }
unknown's avatar
unknown committed
3153
    }
3154
    DBUG_ASSERT(tree == 0 && warning == 0);
unknown's avatar
unknown committed
3155
  }
unknown's avatar
unknown committed
3156
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
3157 3158
}

unknown's avatar
unknown committed
3159

3160 3161
Item *Item_func_group_concat::copy_or_same(THD* thd)
{
unknown's avatar
unknown committed
3162
  return new (thd->mem_root) Item_func_group_concat(thd, this);
3163 3164 3165
}


3166
void Item_func_group_concat::clear()
3167 3168 3169
{
  result.length(0);
  result.copy();
unknown's avatar
SCRUM  
unknown committed
3170
  null_value= TRUE;
unknown's avatar
unknown committed
3171
  warning_for_row= FALSE;
3172
  no_appended= TRUE;
unknown's avatar
unknown committed
3173
  if (tree)
3174
    reset_tree(tree);
unknown's avatar
unknown committed
3175
  /* No need to reset the table as we never call write_row */
3176 3177
}

unknown's avatar
SCRUM  
unknown committed
3178

3179 3180
bool Item_func_group_concat::add()
{
unknown's avatar
BUG  
unknown committed
3181 3182
  if (always_null)
    return 0;
3183 3184 3185
  copy_fields(tmp_table_param);
  copy_funcs(tmp_table_param->items_to_copy);

unknown's avatar
unknown committed
3186
  for (uint i= 0; i < arg_count_field; i++)
3187
  {
unknown's avatar
unknown committed
3188
    Item *show_item= args[i];
unknown's avatar
SCRUM  
unknown committed
3189
    if (!show_item->const_item())
3190
    {
unknown's avatar
unknown committed
3191 3192
      Field *f= show_item->get_tmp_table_field();
      if (f->is_null_in_record((const uchar*) table->record[0]))
unknown's avatar
unknown committed
3193
        return 0;                               // Skip row if it contains null
3194 3195
    }
  }
unknown's avatar
unknown committed
3196

unknown's avatar
SCRUM  
unknown committed
3197
  null_value= FALSE;
3198 3199

  TREE_ELEMENT *el= 0;                          // Only for safety
unknown's avatar
unknown committed
3200 3201 3202
  if (tree)
    el= tree_insert(tree, table->record[0] + table->s->null_bytes, 0,
                    tree->custom_arg);
3203 3204 3205 3206 3207
  /*
    If the row is not a duplicate (el->count == 1)
    we can dump the row here in case of GROUP_CONCAT(DISTINCT...)
    instead of doing tree traverse later.
  */
3208
  if (!warning_for_row &&
unknown's avatar
unknown committed
3209 3210
      (!tree || (el->count == 1 && distinct && !arg_count_order)))
    dump_leaf_key(table->record[0] + table->s->null_bytes, 1, this);
3211

3212 3213 3214
  return 0;
}

unknown's avatar
SCRUM  
unknown committed
3215

3216
bool
3217
Item_func_group_concat::fix_fields(THD *thd, Item **ref)
3218
{
unknown's avatar
unknown committed
3219
  uint i;                       /* for loop variable */
3220
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
3221

unknown's avatar
unknown committed
3222
  if (init_sum_func_check(thd))
unknown's avatar
unknown committed
3223
    return TRUE;
unknown's avatar
unknown committed
3224

3225
  maybe_null= 1;
unknown's avatar
unknown committed
3226

unknown's avatar
SCRUM  
unknown committed
3227
  /*
unknown's avatar
unknown committed
3228
    Fix fields for select list and ORDER clause
unknown's avatar
SCRUM  
unknown committed
3229
  */
unknown's avatar
unknown committed
3230

unknown's avatar
unknown committed
3231
  for (i=0 ; i < arg_count ; i++)
3232
  {
unknown's avatar
unknown committed
3233
    if ((!args[i]->fixed &&
3234
         args[i]->fix_fields(thd, args + i)) ||
unknown's avatar
unknown committed
3235
        args[i]->check_cols(1))
unknown's avatar
unknown committed
3236
      return TRUE;
3237
  }
unknown's avatar
unknown committed
3238

3239
  if (agg_item_charsets(collation, func_name(),
3240 3241 3242
                        args,
			/* skip charset aggregation for order columns */
			arg_count - arg_count_order,
3243
			MY_COLL_ALLOW_CONV, 1))
3244 3245
    return 1;

unknown's avatar
unknown committed
3246
  result.set_charset(collation.collation);
unknown's avatar
SCRUM  
unknown committed
3247 3248
  result_field= 0;
  null_value= 1;
unknown's avatar
unknown committed
3249
  max_length= thd->variables.group_concat_max_len;
unknown's avatar
unknown committed
3250

3251 3252 3253 3254 3255 3256 3257 3258 3259
  uint32 offset;
  if (separator->needs_conversion(separator->length(), separator->charset(),
                                  collation.collation, &offset))
  {
    uint32 buflen= collation.collation->mbmaxlen * separator->length();
    uint errors, conv_length;
    char *buf;
    String *new_separator;

unknown's avatar
unknown committed
3260
    if (!(buf= (char*) thd->stmt_arena->alloc(buflen)) ||
3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271
        !(new_separator= new(thd->stmt_arena->mem_root)
                           String(buf, buflen, collation.collation)))
      return TRUE;
    
    conv_length= copy_and_convert(buf, buflen, collation.collation,
                                  separator->ptr(), separator->length(),
                                  separator->charset(), &errors);
    new_separator->length(conv_length);
    separator= new_separator;
  }

unknown's avatar
unknown committed
3272 3273 3274
  if (check_sum_func(thd, ref))
    return TRUE;

3275
  fixed= 1;
unknown's avatar
unknown committed
3276
  return FALSE;
3277 3278
}

unknown's avatar
SCRUM  
unknown committed
3279

3280 3281 3282
bool Item_func_group_concat::setup(THD *thd)
{
  List<Item> list;
unknown's avatar
unknown committed
3283
  SELECT_LEX *select_lex= thd->lex->current_select;
unknown's avatar
unknown committed
3284 3285
  qsort_cmp2 compare_key;
  DBUG_ENTER("Item_func_group_concat::setup");
3286

unknown's avatar
SCRUM  
unknown committed
3287
  /*
unknown's avatar
unknown committed
3288 3289
    Currently setup() can be called twice. Please add
    assertion here when this is fixed.
unknown's avatar
unknown committed
3290
  */
unknown's avatar
unknown committed
3291 3292 3293 3294 3295 3296 3297
  if (table || tree)
    DBUG_RETURN(FALSE);

  if (!(tmp_table_param= new TMP_TABLE_PARAM))
    DBUG_RETURN(TRUE);

  /* We'll convert all blobs to varchar fields in the temporary table */
unknown's avatar
unknown committed
3298 3299
  tmp_table_param->convert_blob_length= max_length *
                                        collation.collation->mbmaxlen;
unknown's avatar
unknown committed
3300
  /* Push all not constant fields to the list and create a temp table */
unknown's avatar
BUG  
unknown committed
3301
  always_null= 0;
unknown's avatar
unknown committed
3302
  for (uint i= 0; i < arg_count_field; i++)
3303
  {
unknown's avatar
SCRUM  
unknown committed
3304
    Item *item= args[i];
3305
    if (list.push_back(item))
unknown's avatar
unknown committed
3306
      DBUG_RETURN(TRUE);
3307 3308
    if (item->const_item())
    {
3309
      if (item->is_null())
unknown's avatar
unknown committed
3310 3311
      {
        always_null= 1;
unknown's avatar
unknown committed
3312
        DBUG_RETURN(FALSE);
unknown's avatar
unknown committed
3313
      }
3314 3315
    }
  }
unknown's avatar
unknown committed
3316

3317
  List<Item> all_fields(list);
unknown's avatar
unknown committed
3318 3319 3320 3321 3322 3323 3324
  /*
    Try to find every ORDER expression in the list of GROUP_CONCAT
    arguments. If an expression is not found, prepend it to
    "all_fields". The resulting field list is used as input to create
    tmp table columns.
  */
  if (arg_count_order &&
3325
      setup_order(thd, args, context->table_list, list, all_fields, *order))
unknown's avatar
unknown committed
3326
    DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
3327

3328
  count_field_types(select_lex, tmp_table_param, all_fields, 0);
3329
  tmp_table_param->force_copy_fields= force_copy_fields;
unknown's avatar
unknown committed
3330
  DBUG_ASSERT(table == 0);
unknown's avatar
unknown committed
3331 3332 3333 3334 3335 3336 3337 3338 3339
  /*
    Currently we have to force conversion of BLOB values to VARCHAR's
    if we are to store them in TREE objects used for ORDER BY and
    DISTINCT. This leads to truncation if the BLOB's size exceeds
    Field_varstring::MAX_SIZE.
  */
  if (arg_count_order > 0 || distinct)
    set_if_smaller(tmp_table_param->convert_blob_length, 
                   Field_varstring::MAX_SIZE);
unknown's avatar
SCRUM  
unknown committed
3340
  /*
unknown's avatar
unknown committed
3341
    We have to create a temporary table to get descriptions of fields
unknown's avatar
SCRUM  
unknown committed
3342
    (types, sizes and so on).
unknown's avatar
unknown committed
3343 3344 3345

    Note that in the table, we first have the ORDER BY fields, then the
    field list.
unknown's avatar
SCRUM  
unknown committed
3346
  */
unknown's avatar
unknown committed
3347 3348
  if (!(table= create_tmp_table(thd, tmp_table_param, all_fields,
                                (ORDER*) 0, 0, TRUE,
3349
                                (select_lex->options | thd->options),
unknown's avatar
unknown committed
3350 3351
                                HA_POS_ERROR, (char*) "")))
    DBUG_RETURN(TRUE);
3352
  table->file->extra(HA_EXTRA_NO_ROWS);
unknown's avatar
SCRUM  
unknown committed
3353
  table->no_rows= 1;
unknown's avatar
unknown committed
3354 3355


unknown's avatar
unknown committed
3356
  if (distinct || arg_count_order)
3357
  {
unknown's avatar
unknown committed
3358 3359 3360 3361 3362 3363 3364 3365
    /*
      Need sorting: init tree and choose a function to sort.
      Don't reserve space for NULLs: if any of gconcat arguments is NULL,
      the row is not added to the result.
    */
    uint tree_key_length= table->s->reclength - table->s->null_bytes;

    tree= &tree_base;
3366 3367 3368
    if (arg_count_order)
    {
      if (distinct)
unknown's avatar
SCRUM  
unknown committed
3369
        compare_key= (qsort_cmp2) group_concat_key_cmp_with_distinct_and_order;
3370
      else
unknown's avatar
SCRUM  
unknown committed
3371
        compare_key= (qsort_cmp2) group_concat_key_cmp_with_order;
3372 3373 3374
    }
    else
    {
3375
      compare_key= (qsort_cmp2) group_concat_key_cmp_with_distinct;
3376
    }
unknown's avatar
SCRUM  
unknown committed
3377
    /*
unknown's avatar
unknown committed
3378 3379 3380
      Create a tree for sorting. The tree is used to sort and to remove
      duplicate values (according to the syntax of this function). If there
      is no DISTINCT or ORDER BY clauses, we don't create this tree.
unknown's avatar
SCRUM  
unknown committed
3381
    */
3382 3383
    init_tree(tree, (uint) min(thd->variables.max_heap_table_size,
                               thd->variables.sortbuff_size/16), 0,
unknown's avatar
unknown committed
3384
              tree_key_length, compare_key, 0, NULL, (void*) this);
3385
  }
unknown's avatar
unknown committed
3386 3387

  DBUG_RETURN(FALSE);
3388 3389
}

unknown's avatar
unknown committed
3390

3391 3392 3393 3394
/* This is used by rollup to create a separate usable copy of the function */

void Item_func_group_concat::make_unique()
{
unknown's avatar
unknown committed
3395
  tmp_table_param= 0;
3396 3397
  table=0;
  original= 0;
3398
  force_copy_fields= 1;
unknown's avatar
unknown committed
3399
  tree= 0;
3400 3401 3402
}


3403 3404
String* Item_func_group_concat::val_str(String* str)
{
3405
  DBUG_ASSERT(fixed == 1);
unknown's avatar
SCRUM  
unknown committed
3406 3407
  if (null_value)
    return 0;
3408 3409 3410 3411
  if (!result.length() && tree)
    /* Tree is used for sorting as in ORDER BY */
    tree_walk(tree, (tree_walk_action)&dump_leaf_key, (void*)this,
              left_root_right);
3412
  if (count_cut_values && !warning)
unknown's avatar
unknown committed
3413
  {
3414 3415 3416 3417
    /*
      ER_CUT_VALUE_GROUP_CONCAT needs an argument, but this gets set in
      Item_func_group_concat::cleanup().
    */
unknown's avatar
unknown committed
3418 3419
    DBUG_ASSERT(table);
    warning= push_warning(table->in_use, MYSQL_ERROR::WARN_LEVEL_WARN,
3420 3421
                          ER_CUT_VALUE_GROUP_CONCAT,
                          ER(ER_CUT_VALUE_GROUP_CONCAT));
unknown's avatar
unknown committed
3422
  }
3423 3424
  return &result;
}
3425

unknown's avatar
unknown committed
3426

3427 3428
void Item_func_group_concat::print(String *str)
{
3429
  str->append(STRING_WITH_LEN("group_concat("));
3430
  if (distinct)
3431
    str->append(STRING_WITH_LEN("distinct "));
unknown's avatar
unknown committed
3432
  for (uint i= 0; i < arg_count_field; i++)
3433 3434 3435 3436 3437 3438 3439
  {
    if (i)
      str->append(',');
    args[i]->print(str);
  }
  if (arg_count_order)
  {
3440
    str->append(STRING_WITH_LEN(" order by "));
3441 3442 3443
    for (uint i= 0 ; i < arg_count_order ; i++)
    {
      if (i)
unknown's avatar
unknown committed
3444
        str->append(',');
3445
      (*order[i]->item)->print(str);
3446
      if (order[i]->asc)
3447
        str->append(STRING_WITH_LEN(" ASC"));
3448
      else
3449
        str->append(STRING_WITH_LEN(" DESC"));
3450 3451
    }
  }
3452
  str->append(STRING_WITH_LEN(" separator \'"));
3453
  str->append(*separator);
3454
  str->append(STRING_WITH_LEN("\')"));
3455
}