filesort.cc

Go to the documentation of this file.

/* Copyright (C) 2000-2006 MySQL AB

   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
   the Free Software Foundation; version 2 of the License.

   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.

   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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */


#include <config.h>

#include <float.h>
#include <limits.h>

#include <queue>
#include <algorithm>
#include <iostream>

#include <drizzled/drizzled.h>
#include <drizzled/sql_sort.h>
#include <drizzled/filesort.h>
#include <drizzled/error.h>
#include <drizzled/probes.h>
#include <drizzled/session.h>
#include <drizzled/table.h>
#include <drizzled/table_list.h>
#include <drizzled/optimizer/range.h>
#include <drizzled/records.h>
#include <drizzled/internal/iocache.h>
#include <drizzled/internal/my_sys.h>
#include <plugin/myisam/myisam.h>
#include <drizzled/plugin/transactional_storage_engine.h>
#include <drizzled/atomics.h>
#include <drizzled/global_buffer.h>
#include <drizzled/sort_field.h>
#include <drizzled/item/subselect.h>
#include <drizzled/statistics_variables.h>
#include <drizzled/system_variables.h>

using namespace std;

namespace drizzled {

/* Defines used by filesort and uniques */
#define MERGEBUFF   7
#define MERGEBUFF2    15

class BufferCompareContext
{
public:
  qsort_cmp2 key_compare;
  void *key_compare_arg;

  BufferCompareContext() :
    key_compare(0),
    key_compare_arg(0)
  { }

};

class SortParam {
public:
  uint32_t rec_length;          /* Length of sorted records */
  uint32_t sort_length;     /* Length of sorted columns */
  uint32_t ref_length;      /* Length of record ref. */
  uint32_t addon_length;        /* Length of added packed fields */
  uint32_t res_length;          /* Length of records in final sorted file/buffer */
  uint32_t keys;        /* Max keys / buffer */
  ha_rows max_rows,examined_rows;
  Table *sort_form;     /* For quicker make_sortkey */
  SortField *local_sortorder;
  SortField *end;
  sort_addon_field *addon_field; /* Descriptors for companion fields */
  unsigned char *unique_buff;
  bool not_killable;
  char *tmp_buffer;
  /* The fields below are used only by Unique class */
  qsort2_cmp compare;
  BufferCompareContext cmp_context;

  SortParam() :
    rec_length(0),
    sort_length(0),
    ref_length(0),
    addon_length(0),
    res_length(0),
    keys(0),
    max_rows(0),
    examined_rows(0),
    sort_form(0),
    local_sortorder(0),
    end(0),
    addon_field(0),
    unique_buff(0),
    not_killable(0),
    tmp_buffer(0),
    compare(0)
  {
  }

  ~SortParam()
  {
    free(tmp_buffer);
  }

  int write_keys(unsigned char * *sort_keys,
                 uint32_t count,
                 internal::io_cache_st *buffer_file,
                 internal::io_cache_st *tempfile);

  void make_sortkey(unsigned char *to,
                    unsigned char *ref_pos);
  void register_used_fields();
  void save_index(unsigned char **sort_keys,
                  uint32_t count,
                  filesort_info *table_sort);

};

/* functions defined in this file */

static char **make_char_array(char **old_pos, uint32_t fields,
                              uint32_t length);

static unsigned char *read_buffpek_from_file(internal::io_cache_st *buffer_file,
                                             uint32_t count,
                                             unsigned char *buf);

static uint32_t suffix_length(uint32_t string_length);
static void unpack_addon_fields(sort_addon_field *addon_field,
                                unsigned char *buff);

FileSort::FileSort(Session &arg) :
  _session(arg)
{ 
}

ha_rows FileSort::run(Table *table, SortField *sortorder, uint32_t s_length,
                      optimizer::SqlSelect *select, ha_rows max_rows,
                      bool sort_positions, ha_rows &examined_rows)
{
  int error= 1;
  uint32_t memavl= 0, min_sort_memory;
  uint32_t maxbuffer;
  size_t allocated_sort_memory= 0;
  buffpek *buffpek_inst= 0;
  ha_rows records= HA_POS_ERROR;
  unsigned char **sort_keys= 0;
  internal::io_cache_st tempfile;
  internal::io_cache_st buffpek_pointers;
  internal::io_cache_st *selected_records_file;
  internal::io_cache_st *outfile;
  SortParam param;
  bool multi_byte_charset;

  /*
    Don't use table->sort in filesort as it is also used by
    QuickIndexMergeSelect. Work with a copy and put it back at the end
    when index_merge select has finished with it.
  */
  filesort_info table_sort(table->sort);
  table->sort.io_cache= NULL;

  TableList *tab= table->pos_in_table_list;
  Item_subselect *subselect= tab ? tab->containing_subselect() : 0;

  DRIZZLE_FILESORT_START(table->getShare()->getSchemaName(), table->getShare()->getTableName());

  /*
   Release InnoDB's adaptive hash index latch (if holding) before
   running a sort.
  */
  plugin::TransactionalStorageEngine::releaseTemporaryLatches(&getSession());


  outfile= table_sort.io_cache;
  assert(tempfile.buffer == 0);
  assert(buffpek_pointers.buffer == 0);

  param.sort_length= sortlength(sortorder, s_length, &multi_byte_charset);
  param.ref_length= table->cursor->ref_length;

  if (!(table->cursor->getEngine()->check_flag(HTON_BIT_FAST_KEY_READ)) && !sort_positions)
  {
    /*
      Get the descriptors of all fields whose values are appended
      to sorted fields and get its total length in param.spack_length.
    */
    param.addon_field= get_addon_fields(table->getFields(),
                                        param.sort_length,
                                        &param.addon_length);
  }

  table_sort.addon_buf= 0;
  table_sort.addon_length= param.addon_length;
  table_sort.addon_field= param.addon_field;
  table_sort.unpack= unpack_addon_fields;
  if (param.addon_field)
  {
    param.res_length= param.addon_length;
    table_sort.addon_buf= (unsigned char *) malloc(param.addon_length);
  }
  else
  {
    param.res_length= param.ref_length;
    /*
      The reference to the record is considered
      as an additional sorted field
    */
    param.sort_length+= param.ref_length;
  }
  param.rec_length= param.sort_length+param.addon_length;
  param.max_rows= max_rows;

  if (select && select->quick)
  {
    getSession().status_var.filesort_range_count++;
  }
  else
  {
    getSession().status_var.filesort_scan_count++;
  }
#ifdef CAN_TRUST_RANGE
  if (select && select->quick && select->quick->records > 0L)
  {
    records= min((ha_rows) (select->quick->records*2+EXTRA_RECORDS*2),
                 table->cursor->stats.records)+EXTRA_RECORDS;
    selected_records_file=0;
  }
  else
#endif
  {
    records= table->cursor->estimate_rows_upper_bound();
    /*
      If number of records is not known, use as much of sort buffer
      as possible.
    */
    if (records == HA_POS_ERROR)
      records--;  // we use 'records+1' below.
    selected_records_file= 0;
  }

  if (multi_byte_charset)
    param.tmp_buffer= (char*) malloc(param.sort_length);

  memavl= getSession().variables.sortbuff_size;
  min_sort_memory= max((uint32_t)MIN_SORT_MEMORY, param.sort_length*MERGEBUFF2);
  while (memavl >= min_sort_memory)
  {
    uint32_t old_memavl;
    uint32_t keys= memavl/(param.rec_length+sizeof(char*));
    param.keys= (uint32_t) min(records+1, (ha_rows)keys);

    allocated_sort_memory= param.keys * param.rec_length;
    if (not global_sort_buffer.add(allocated_sort_memory))
    {
      my_error(ER_OUT_OF_GLOBAL_SORTMEMORY, MYF(ME_ERROR+ME_WAITTANG));
      goto err;
    }

    if ((table_sort.sort_keys=
   (unsigned char **) make_char_array((char **) table_sort.sort_keys,
                                            param.keys, param.rec_length)))
      break;

    global_sort_buffer.sub(allocated_sort_memory);
    old_memavl= memavl;
    if ((memavl= memavl/4*3) < min_sort_memory && old_memavl > min_sort_memory)
      memavl= min_sort_memory;
  }
  sort_keys= table_sort.sort_keys;
  if (memavl < min_sort_memory)
  {
    my_error(ER_OUT_OF_SORTMEMORY,MYF(ME_ERROR+ME_WAITTANG));
    goto err;
  }

  if (buffpek_pointers.open_cached_file(drizzle_tmpdir.c_str(),TEMP_PREFIX, DISK_BUFFER_SIZE, MYF(MY_WME)))
  {
    goto err;
  }

  param.keys--;       /* TODO: check why we do this */
  param.sort_form= table;
  param.end=(param.local_sortorder=sortorder)+s_length;
  if ((records= find_all_keys(&param,select,sort_keys, &buffpek_pointers,
                              &tempfile, selected_records_file)) == HA_POS_ERROR)
  {
    goto err;
  }
  maxbuffer= (uint32_t)(buffpek_pointers.tell() / sizeof(*buffpek_inst));

  if (maxbuffer == 0)     // The whole set is in memory
  {
    param.save_index(sort_keys,(uint32_t) records, &table_sort);
  }
  else
  {
    if (table_sort.buffpek && table_sort.buffpek_len < maxbuffer)
    {
      free(table_sort.buffpek);
      table_sort.buffpek = 0;
    }
    if (!(table_sort.buffpek=
          (unsigned char *) read_buffpek_from_file(&buffpek_pointers, maxbuffer, table_sort.buffpek)))
    {
      goto err;
    }
    buffpek_inst= (buffpek *) table_sort.buffpek;
    table_sort.buffpek_len= maxbuffer;
    buffpek_pointers.close_cached_file();
  /* Open cached file if it isn't open */
    if (not outfile->inited() && outfile->open_cached_file(drizzle_tmpdir.c_str(),TEMP_PREFIX,READ_RECORD_BUFFER, MYF(MY_WME)))
    {
      goto err;
    }

    if (outfile->reinit_io_cache(internal::WRITE_CACHE,0L,0,0))
    {
      goto err;
    }

    /*
      Use also the space previously used by string pointers in sort_buffer
      for temporary key storage.
    */
    param.keys=((param.keys*(param.rec_length+sizeof(char*))) / param.rec_length-1);

    maxbuffer--;        // Offset from 0
    if (merge_many_buff(&param,(unsigned char*) sort_keys,buffpek_inst,&maxbuffer, &tempfile))
    {
      goto err;
    }

    if (tempfile.flush() || tempfile.reinit_io_cache(internal::READ_CACHE,0L,0,0))
    {
      goto err;
    }

    if (merge_index(&param,(unsigned char*) sort_keys,buffpek_inst,maxbuffer,&tempfile, outfile))
    {
      goto err;
    }
  }

  if (records > param.max_rows)
  {
    records= param.max_rows;
  }
  error =0;

 err:
  if (not subselect || not subselect->is_uncacheable())
  {
    free(sort_keys);
    table_sort.sort_keys= 0;
    free(buffpek_inst);
    table_sort.buffpek= 0;
    table_sort.buffpek_len= 0;
  }

  tempfile.close_cached_file();
  buffpek_pointers.close_cached_file();

  if (outfile->inited())
  {
    if (outfile->flush())
    {
      error= 1;
    }
    {
      internal::my_off_t save_pos= outfile->pos_in_file;
      /* For following reads */
      if (outfile->reinit_io_cache(internal::READ_CACHE,0L,0,0))
      {
        error=1;
      }
      outfile->end_of_file=save_pos;
    }
  }

  if (error)
  {
    my_message(ER_FILSORT_ABORT, ER(ER_FILSORT_ABORT),
               MYF(ME_ERROR+ME_WAITTANG));
  }
  else
  {
    getSession().status_var.filesort_rows+= (uint32_t) records;
  }
  examined_rows= param.examined_rows;
  global_sort_buffer.sub(allocated_sort_memory);
  table->sort= table_sort;
  DRIZZLE_FILESORT_DONE(error, records);
  return (error ? HA_POS_ERROR : records);
} /* filesort */

static char **make_char_array(char **old_pos, uint32_t fields,
                              uint32_t length)
{
  if (not old_pos)
    old_pos= (char**) malloc((uint32_t) fields * (length + sizeof(char*)));
  char** pos= old_pos; 
  char* char_pos= ((char*) (pos+fields)) - length;
  while (fields--) 
    *(pos++) = (char_pos+= length);

  return old_pos;
} /* make_char_array */


static unsigned char *read_buffpek_from_file(internal::io_cache_st *buffpek_pointers, uint32_t count, unsigned char *buf)
{
  uint32_t length= sizeof(buffpek)*count;
  unsigned char *tmp= buf;
  if (count > UINT_MAX/sizeof(buffpek))
    return 0; /* sizeof(buffpek)*count will overflow */
  if (!tmp)
    tmp= (unsigned char *)malloc(length);
  {
    if (buffpek_pointers->reinit_io_cache(internal::READ_CACHE,0L,0,0) ||
      buffpek_pointers->read(tmp, length))
    {
      free(tmp);
      tmp= 0;
    }
  }
  return tmp;
}


ha_rows FileSort::find_all_keys(SortParam *param, 
                                optimizer::SqlSelect *select,
                                unsigned char **sort_keys,
                                internal::io_cache_st *buffpek_pointers,
                                internal::io_cache_st *tempfile, internal::io_cache_st *indexfile)
{
  int error,flag,quick_select;
  uint32_t idx,indexpos,ref_length;
  unsigned char *ref_pos,*next_pos,ref_buff[MAX_REFLENGTH];
  internal::my_off_t record;
  Table *sort_form;
  volatile Session::killed_state_t *killed= getSession().getKilledPtr();
  Cursor *file;
  boost::dynamic_bitset<> *save_read_set= NULL;
  boost::dynamic_bitset<> *save_write_set= NULL;

  idx=indexpos=0;
  error=quick_select=0;
  sort_form=param->sort_form;
  file= sort_form->cursor;
  ref_length=param->ref_length;
  ref_pos= ref_buff;
  quick_select=select && select->quick;
  record=0;
  flag= ((!indexfile && ! file->isOrdered())
   || quick_select);
  if (indexfile || flag)
    ref_pos= &file->ref[0];
  next_pos=ref_pos;
  if (! indexfile && ! quick_select)
  {
    next_pos=(unsigned char*) 0;      /* Find records in sequence */
    if (file->startTableScan(1))
      return(HA_POS_ERROR);
    file->extra_opt(HA_EXTRA_CACHE, getSession().variables.read_buff_size);
  }

  ReadRecord read_record_info;
  if (quick_select)
  {
    if (select->quick->reset())
      return(HA_POS_ERROR);

    if (read_record_info.init_read_record(&getSession(), select->quick->head, select, 1, 1))
      return(HA_POS_ERROR);
  }

  /* Remember original bitmaps */
  save_read_set=  sort_form->read_set;
  save_write_set= sort_form->write_set;
  /* Set up temporary column read map for columns used by sort */
  sort_form->tmp_set.reset();
  /* Temporary set for register_used_fields and register_field_in_read_map */
  sort_form->read_set= &sort_form->tmp_set;
  param->register_used_fields();
  if (select && select->cond)
    select->cond->walk(&Item::register_field_in_read_map, 1,
                       (unsigned char*) sort_form);
  sort_form->column_bitmaps_set(sort_form->tmp_set, sort_form->tmp_set);

  for (;;)
  {
    if (quick_select)
    {
      if ((error= read_record_info.read_record(&read_record_info)))
      {
        error= HA_ERR_END_OF_FILE;
        break;
      }
      file->position(sort_form->getInsertRecord());
    }
    else          /* Not quick-select */
    {
      if (indexfile)
      {
  if (indexfile->read(ref_pos, ref_length))
  {
    error= errno ? errno : -1;    /* Abort */
    break;
  }
  error=file->rnd_pos(sort_form->getInsertRecord(),next_pos);
      }
      else
      {
  error=file->rnd_next(sort_form->getInsertRecord());

  if (!flag)
  {
    internal::my_store_ptr(ref_pos,ref_length,record); // Position to row
    record+= sort_form->getShare()->db_record_offset;
  }
  else if (!error)
    file->position(sort_form->getInsertRecord());
      }
      if (error && error != HA_ERR_RECORD_DELETED)
  break;
    }

    if (*killed)
    {
      if (!indexfile && !quick_select)
      {
        (void) file->extra(HA_EXTRA_NO_CACHE);
        file->endTableScan();
      }
      return(HA_POS_ERROR);
    }
    if (error == 0)
      param->examined_rows++;
    if (error == 0 && (!select || select->skip_record() == 0))
    {
      if (idx == param->keys)
      {
  if (param->write_keys(sort_keys, idx, buffpek_pointers, tempfile))
    return(HA_POS_ERROR);
  idx=0;
  indexpos++;
      }
      param->make_sortkey(sort_keys[idx++], ref_pos);
    }
    else
    {
      file->unlock_row();
    }

    /* It does not make sense to read more keys in case of a fatal error */
    if (getSession().is_error())
      break;
  }
  if (quick_select)
  {
    /*
      index_merge quick select uses table->sort when retrieving rows, so free
      resoures it has allocated.
    */
    read_record_info.end_read_record();
  }
  else
  {
    (void) file->extra(HA_EXTRA_NO_CACHE);  /* End cacheing of records */
    if (!next_pos)
      file->endTableScan();
  }

  if (getSession().is_error())
    return(HA_POS_ERROR);

  /* Signal we should use orignal column read and write maps */
  sort_form->column_bitmaps_set(*save_read_set, *save_write_set);

  if (error != HA_ERR_END_OF_FILE)
  {
    sort_form->print_error(error,MYF(ME_ERROR | ME_WAITTANG));
    return(HA_POS_ERROR);
  }

  if (indexpos && idx && param->write_keys(sort_keys,idx,buffpek_pointers,tempfile))
  {
    return(HA_POS_ERROR);
  }

  return tempfile->inited() ? (ha_rows) (tempfile->tell() / param->rec_length) : idx;
} /* find_all_keys */


int SortParam::write_keys(unsigned char **sort_keys, uint32_t count,
                          internal::io_cache_st *buffpek_pointers, internal::io_cache_st *tempfile)
{
  buffpek buffpek;

  internal::my_string_ptr_sort((unsigned char*) sort_keys, (uint32_t) count, sort_length);
  if (not tempfile->inited() &&
      tempfile->open_cached_file(drizzle_tmpdir.c_str(), TEMP_PREFIX, DISK_BUFFER_SIZE, MYF(MY_WME)))
  {
    return 1;
  }
  /* check we won't have more buffpeks than we can possibly keep in memory */
  if (buffpek_pointers->tell() + sizeof(buffpek) > UINT_MAX)
  {
    return 1;
  }

  buffpek.file_pos= tempfile->tell();
  if ((ha_rows) count > max_rows)
    count=(uint32_t) max_rows;

  buffpek.count=(ha_rows) count;

  for (unsigned char **ptr= sort_keys + count ; sort_keys != ptr ; sort_keys++)
  {
    if (tempfile->write(*sort_keys, rec_length))
    {
      return 1;
    }
  }

  if (buffpek_pointers->write(&buffpek, sizeof(buffpek)))
  {
    return 1;
  }

  return 0;
} /* write_keys */


static inline void store_length(unsigned char *to, uint32_t length, uint32_t pack_length)
{
  switch (pack_length) {
  case 1:
    *to= (unsigned char) length;
    break;
  case 2:
    mi_int2store(to, length);
    break;
  case 3:
    mi_int3store(to, length);
    break;
  default:
    mi_int4store(to, length);
    break;
  }
}


void SortParam::make_sortkey(unsigned char *to, unsigned char *ref_pos)
{
  Field *field;
  SortField *sort_field;
  size_t length;

  for (sort_field= local_sortorder ;
       sort_field != end ;
       sort_field++)
  {
    bool maybe_null=0;
    if ((field=sort_field->field))
    {           // Field
      if (field->maybe_null())
      {
  if (field->is_null())
  {
    if (sort_field->reverse)
      memset(to, 255, sort_field->length+1);
    else
      memset(to, 0, sort_field->length+1);
    to+= sort_field->length+1;
    continue;
  }
  else
    *to++=1;
      }
      field->sort_string(to, sort_field->length);
    }
    else
    {           // Item
      Item *item=sort_field->item;
      maybe_null= item->maybe_null;

      switch (sort_field->result_type) {
      case STRING_RESULT:
        {
          const charset_info_st * const cs=item->collation.collation;
          char fill_char= ((cs->state & MY_CS_BINSORT) ? (char) 0 : ' ');
          int diff;
          uint32_t sort_field_length;

          if (maybe_null)
            *to++=1;
          /* All item->str() to use some extra byte for end null.. */
          String tmp((char*) to,sort_field->length+4,cs);
          String *res= item->str_result(&tmp);
          if (!res)
          {
            if (maybe_null)
              memset(to-1, 0, sort_field->length+1);
            else
            {
              /*
                This should only happen during extreme conditions if we run out
                of memory or have an item marked not null when it can be null.
                This code is here mainly to avoid a hard crash in this case.
              */
              assert(0);
              memset(to, 0, sort_field->length);  // Avoid crash
            }
            break;
          }
          length= res->length();
          sort_field_length= sort_field->length - sort_field->suffix_length;
          diff=(int) (sort_field_length - length);
          if (diff < 0)
          {
            diff=0;
            length= sort_field_length;
          }
          if (sort_field->suffix_length)
          {
            /* Store length last in result_string */
            store_length(to + sort_field_length, length,
                         sort_field->suffix_length);
          }
          if (sort_field->need_strxnfrm)
          {
            char *from=(char*) res->ptr();
            uint32_t tmp_length;
            if ((unsigned char*) from == to)
            {
              set_if_smaller(length,sort_field->length);
              memcpy(tmp_buffer,from,length);
              from= tmp_buffer;
            }
            tmp_length= cs->strnxfrm(to,sort_field->length, (unsigned char*) from, length);
            assert(tmp_length == sort_field->length);
          }
          else
          {
            cs->strnxfrm((unsigned char*)to,length,(const unsigned char*)res->ptr(),length);
            cs->cset->fill(cs, (char *)to+length,diff,fill_char);
          }
          break;
        }
      case INT_RESULT:
        {
          int64_t value= item->val_int_result();
          if (maybe_null)
          {
            *to++=1;
            if (item->null_value)
            {
              if (maybe_null)
                memset(to-1, 0, sort_field->length+1);
              else
              {
                memset(to, 0, sort_field->length);
              }
              break;
            }
          }
          to[7]= (unsigned char) value;
          to[6]= (unsigned char) (value >> 8);
          to[5]= (unsigned char) (value >> 16);
          to[4]= (unsigned char) (value >> 24);
          to[3]= (unsigned char) (value >> 32);
          to[2]= (unsigned char) (value >> 40);
          to[1]= (unsigned char) (value >> 48);
          if (item->unsigned_flag)                    /* Fix sign */
            to[0]= (unsigned char) (value >> 56);
          else
            to[0]= (unsigned char) (value >> 56) ^ 128; /* Reverse signbit */
          break;
        }
      case DECIMAL_RESULT:
        {
          type::Decimal dec_buf, *dec_val= item->val_decimal_result(&dec_buf);
          if (maybe_null)
          {
            if (item->null_value)
            {
              memset(to, 0, sort_field->length+1);
              to++;
              break;
            }
            *to++=1;
          }
          dec_val->val_binary(E_DEC_FATAL_ERROR, to,
                              item->max_length - (item->decimals ? 1:0),
                              item->decimals);
          break;
        }
      case REAL_RESULT:
        {
          double value= item->val_result();
          if (maybe_null)
          {
            if (item->null_value)
            {
              memset(to, 0, sort_field->length+1);
              to++;
              break;
            }
            *to++=1;
          }
          change_double_for_sort(value,(unsigned char*) to);
          break;
        }
      case ROW_RESULT:
      default:
        // This case should never be choosen
        assert(0);
        break;
      }
    }

    if (sort_field->reverse)
    {             /* Revers key */
      if (maybe_null)
        to[-1]= ~to[-1];
      length=sort_field->length;
      while (length--)
      {
  *to = (unsigned char) (~ *to);
  to++;
      }
    }
    else
    {
      to+= sort_field->length;
    }
  }

  if (addon_field)
  {
    /*
      Save field values appended to sorted fields.
      First null bit indicators are appended then field values follow.
      In this implementation we use fixed layout for field values -
      the same for all records.
    */
    sort_addon_field *addonf= addon_field;
    unsigned char *nulls= to;
    assert(addonf != 0);
    memset(nulls, 0, addonf->offset);
    to+= addonf->offset;
    for ( ; (field= addonf->field) ; addonf++)
    {
      if (addonf->null_bit && field->is_null())
      {
        nulls[addonf->null_offset]|= addonf->null_bit;
#ifdef HAVE_VALGRIND
  memset(to, 0, addonf->length);
#endif
      }
      else
      {
#ifdef HAVE_VALGRIND
        unsigned char *end= field->pack(to, field->ptr);
  uint32_t local_length= (uint32_t) ((to + addonf->length) - end);
  assert((int) local_length >= 0);
  if (local_length)
    memset(end, 0, local_length);
#else
        (void) field->pack(to, field->ptr);
#endif
      }
      to+= addonf->length;
    }
  }
  else
  {
    /* Save filepos last */
    memcpy(to, ref_pos, (size_t) ref_length);
  }
}


/*
  fields used by sorting in the sorted table's read set
*/

void SortParam::register_used_fields()
{
  SortField *sort_field;
  Table *table= sort_form;

  for (sort_field= local_sortorder ;
       sort_field != end ;
       sort_field++)
  {
    Field *field;
    if ((field= sort_field->field))
    {
      if (field->getTable() == table)
        table->setReadSet(field->position());
    }
    else
    {           // Item
      sort_field->item->walk(&Item::register_field_in_read_map, 1,
                             (unsigned char *) table);
    }
  }

  if (addon_field)
  {
    sort_addon_field *addonf= addon_field;
    Field *field;
    for ( ; (field= addonf->field) ; addonf++)
      table->setReadSet(field->position());
  }
  else
  {
    /* Save filepos last */
    table->prepare_for_position();
  }
}


void SortParam::save_index(unsigned char **sort_keys, uint32_t count, filesort_info *table_sort)
{
  internal::my_string_ptr_sort((unsigned char*) sort_keys, (uint32_t) count, sort_length);
  uint32_t offset= rec_length - res_length;

  if ((ha_rows) count > max_rows)
    count=(uint32_t) max_rows;

  unsigned char* to= table_sort->record_pointers= (unsigned char*) malloc(res_length*count);

  for (unsigned char **end_ptr= sort_keys+count ; sort_keys != end_ptr ; sort_keys++)
  {
    memcpy(to, *sort_keys+offset, res_length);
    to+= res_length;
  }
}


int FileSort::merge_many_buff(SortParam *param, unsigned char *sort_buffer,
                              buffpek *buffpek_inst, uint32_t *maxbuffer, internal::io_cache_st *t_file)
{
  internal::io_cache_st t_file2,*from_file,*to_file,*temp;
  buffpek *lastbuff;

  if (*maxbuffer < MERGEBUFF2)
    return 0;
  if (t_file->flush() ||
      t_file2.open_cached_file(drizzle_tmpdir.c_str(),TEMP_PREFIX,DISK_BUFFER_SIZE, MYF(MY_WME)))
  {
    return 1;
  }

  from_file= t_file ; to_file= &t_file2;
  while (*maxbuffer >= MERGEBUFF2)
  {
    if (from_file->reinit_io_cache(internal::READ_CACHE, 0, 0, 0)
      || to_file->reinit_io_cache(internal::WRITE_CACHE, 0, 0, 0))
      break;

    uint32_t i= 0;
    lastbuff= buffpek_inst;
    for (; i <= *maxbuffer - MERGEBUFF * 3 / 2; i += MERGEBUFF)
    {
      if (merge_buffers(param, from_file, to_file, sort_buffer, lastbuff++, buffpek_inst + i, buffpek_inst + i + MERGEBUFF - 1, 0))
      {
        goto cleanup;
      }
    }

    if (merge_buffers(param, from_file, to_file, sort_buffer, lastbuff++, buffpek_inst + i, buffpek_inst + *maxbuffer, 0)
      || to_file->flush())
      break;

    temp=from_file; from_file=to_file; to_file=temp;
    from_file->setup_io_cache();
    to_file->setup_io_cache();
    *maxbuffer= (uint32_t) (lastbuff-buffpek_inst)-1;
  }

cleanup:
  to_file->close_cached_file();     // This holds old result
  if (to_file == t_file)
  {
    *t_file=t_file2;        // Copy result file
    t_file->setup_io_cache();
  }

  return(*maxbuffer >= MERGEBUFF2); /* Return 1 if interrupted */
} /* merge_many_buff */


uint32_t FileSort::read_to_buffer(internal::io_cache_st *fromfile, buffpek *buffpek_inst, uint32_t rec_length)
{
  uint32_t count;
  uint32_t length;

  if ((count= (uint32_t) min((ha_rows) buffpek_inst->max_keys,buffpek_inst->count)))
  {
    if (pread(fromfile->file,(unsigned char*) buffpek_inst->base, (length= rec_length*count),buffpek_inst->file_pos) == 0)
      return((uint32_t) -1);

    buffpek_inst->key= buffpek_inst->base;
    buffpek_inst->file_pos+= length;      /* New filepos */
    buffpek_inst->count-= count;
    buffpek_inst->mem_count= count;
  }
  return (count*rec_length);
} /* read_to_buffer */


class compare_functor
{
  qsort2_cmp key_compare;
  void *key_compare_arg;

  public:
  compare_functor(qsort2_cmp in_key_compare, void *in_compare_arg) :
    key_compare(in_key_compare),
    key_compare_arg(in_compare_arg)
  { }
  
  inline bool operator()(const buffpek *i, const buffpek *j) const
  {
    int val= key_compare(key_compare_arg, &i->key, &j->key);

    return (val >= 0);
  }
};


int FileSort::merge_buffers(SortParam *param, internal::io_cache_st *from_file,
                            internal::io_cache_st *to_file, unsigned char *sort_buffer,
                            buffpek *lastbuff, buffpek *Fb, buffpek *Tb,
                            int flag)
{
  int error;
  uint32_t rec_length,res_length,offset;
  size_t sort_length;
  uint32_t maxcount;
  ha_rows max_rows,org_max_rows;
  internal::my_off_t to_start_filepos;
  unsigned char *strpos;
  buffpek *buffpek_inst;
  qsort2_cmp cmp;
  void *first_cmp_arg;
  volatile Session::killed_state_t *killed= getSession().getKilledPtr();
  Session::killed_state_t not_killable;

  getSession().status_var.filesort_merge_passes++;
  if (param->not_killable)
  {
    killed= &not_killable;
    not_killable= Session::NOT_KILLED;
  }

  error=0;
  rec_length= param->rec_length;
  res_length= param->res_length;
  sort_length= param->sort_length;
  offset= rec_length-res_length;
  maxcount= (uint32_t) (param->keys/((uint32_t) (Tb-Fb) +1));
  to_start_filepos= to_file->tell();
  strpos= (unsigned char*) sort_buffer;
  org_max_rows=max_rows= param->max_rows;

  /* The following will fire if there is not enough space in sort_buffer */
  assert(maxcount!=0);

  if (param->unique_buff)
  {
    cmp= param->compare;
    first_cmp_arg= (void *) &param->cmp_context;
  }
  else
  {
    cmp= internal::get_ptr_compare(sort_length);
    first_cmp_arg= (void*) &sort_length;
  }
  priority_queue<buffpek *, vector<buffpek *>, compare_functor >
    queue(compare_functor(cmp, first_cmp_arg));
  for (buffpek_inst= Fb ; buffpek_inst <= Tb ; buffpek_inst++)
  {
    buffpek_inst->base= strpos;
    buffpek_inst->max_keys= maxcount;
    strpos+= (uint32_t) (error= (int) read_to_buffer(from_file, buffpek_inst,
                                                                         rec_length));
    if (error == -1)
      return -1;

    buffpek_inst->max_keys= buffpek_inst->mem_count;  // If less data in buffers than expected
    queue.push(buffpek_inst);
  }

  if (param->unique_buff)
  {
    /*
       Called by Unique::get()
       Copy the first argument to param->unique_buff for unique removal.
       Store it also in 'to_file'.

       This is safe as we know that there is always more than one element
       in each block to merge (This is guaranteed by the Unique:: algorithm
    */
    buffpek_inst= queue.top();
    memcpy(param->unique_buff, buffpek_inst->key, rec_length);
    if (to_file->write(buffpek_inst->key, rec_length))
    {
      return 1;
    }
    buffpek_inst->key+= rec_length;
    buffpek_inst->mem_count--;
    if (!--max_rows)
    {
      error= 0;
      goto end;
    }
    /* Top element has been used */
    queue.pop();
    queue.push(buffpek_inst);
  }
  else
  {
    cmp= 0;                                        // Not unique
  }

  while (queue.size() > 1)
  {
    if (*killed)
    {
      return 1;
    }
    for (;;)
    {
      buffpek_inst= queue.top();
      if (cmp)                                        // Remove duplicates
      {
        if (!(*cmp)(first_cmp_arg, &(param->unique_buff),
                    (unsigned char**) &buffpek_inst->key))
              goto skip_duplicate;
            memcpy(param->unique_buff, buffpek_inst->key, rec_length);
      }
      if (flag == 0)
      {
        if (to_file->write(buffpek_inst->key, rec_length))
        {
          return 1;
        }
      }
      else
      {
        if (to_file->write(buffpek_inst->key+offset, res_length))
        {
          return 1;
        }
      }
      if (!--max_rows)
      {
        error= 0;
        goto end;
      }

    skip_duplicate:
      buffpek_inst->key+= rec_length;
      if (! --buffpek_inst->mem_count)
      {
        if (!(error= (int) read_to_buffer(from_file,buffpek_inst,
                                          rec_length)))
        {
          queue.pop();
          break;                        /* One buffer have been removed */
        }
        else if (error == -1)
        {
          return -1;
        }
      }
      /* Top element has been replaced */
      queue.pop();
      queue.push(buffpek_inst);
    }
  }
  buffpek_inst= queue.top();
  buffpek_inst->base= sort_buffer;
  buffpek_inst->max_keys= param->keys;

  /*
    As we know all entries in the buffer are unique, we only have to
    check if the first one is the same as the last one we wrote
  */
  if (cmp)
  {
    if (!(*cmp)(first_cmp_arg, &(param->unique_buff), (unsigned char**) &buffpek_inst->key))
    {
      buffpek_inst->key+= rec_length;         // Remove duplicate
      --buffpek_inst->mem_count;
    }
  }

  do
  {
    if ((ha_rows) buffpek_inst->mem_count > max_rows)
    {                                        /* Don't write too many records */
      buffpek_inst->mem_count= (uint32_t) max_rows;
      buffpek_inst->count= 0;                        /* Don't read more */
    }
    max_rows-= buffpek_inst->mem_count;
    if (flag == 0)
    {
      if (to_file->write(buffpek_inst->key, (rec_length*buffpek_inst->mem_count)))
      {
        return 1;
      }
    }
    else
    {
      unsigned char *end;
      strpos= buffpek_inst->key+offset;
      for (end= strpos+buffpek_inst->mem_count*rec_length ;
           strpos != end ;
           strpos+= rec_length)
      {
        if (to_file->write(strpos, res_length))
        {
          return 1;
        }
      }
    }
  }

  while ((error=(int) read_to_buffer(from_file,buffpek_inst, rec_length))
         != -1 && error != 0);

end:
  lastbuff->count= min(org_max_rows-max_rows, param->max_rows);
  lastbuff->file_pos= to_start_filepos;

  return error;
} /* merge_buffers */


  /* Do a merge to output-file (save only positions) */

int FileSort::merge_index(SortParam *param, unsigned char *sort_buffer,
                          buffpek *buffpek_inst, uint32_t maxbuffer,
                          internal::io_cache_st *tempfile, internal::io_cache_st *outfile)
{
  if (merge_buffers(param,tempfile,outfile,sort_buffer,buffpek_inst,buffpek_inst,
        buffpek_inst+maxbuffer,1))
    return 1;

  return 0;
} /* merge_index */


static uint32_t suffix_length(uint32_t string_length)
{
  if (string_length < 256)
    return 1;
  if (string_length < 256L*256L)
    return 2;
  if (string_length < 256L*256L*256L)
    return 3;
  return 4;                                     // Can't sort longer than 4G
}



uint32_t FileSort::sortlength(SortField *sortorder, uint32_t s_length, bool *multi_byte_charset)
{
  uint32_t length;
  const charset_info_st *cs;
  *multi_byte_charset= 0;

  length=0;
  for (; s_length-- ; sortorder++)
  {
    sortorder->need_strxnfrm= 0;
    sortorder->suffix_length= 0;
    if (sortorder->field)
    {
      cs= sortorder->field->sort_charset();
      sortorder->length= sortorder->field->sort_length();
      cs= sortorder->field->sort_charset();
      if (cs->use_strnxfrm())
      {
        sortorder->need_strxnfrm= 1;
        *multi_byte_charset= 1;
        sortorder->length= cs->coll->strnxfrmlen(cs, sortorder->length);
      }
      if (sortorder->field->maybe_null())
  length++;       // Place for NULL marker
    }
    else
    {
      sortorder->result_type= sortorder->item->result_type();
      if (sortorder->item->result_as_int64_t())
        sortorder->result_type= INT_RESULT;

      switch (sortorder->result_type) {
      case STRING_RESULT:
        sortorder->length=sortorder->item->max_length;
        set_if_smaller(sortorder->length, getSession().variables.max_sort_length);
        cs= sortorder->item->collation.collation;
        if (cs->use_strnxfrm())
        {
          sortorder->length= cs->coll->strnxfrmlen(cs, sortorder->length);
          sortorder->need_strxnfrm= 1;
          *multi_byte_charset= 1;
        }
        else if (cs == &my_charset_bin)
        {
          /* Store length last to be able to sort blob/varbinary */
          sortorder->suffix_length= suffix_length(sortorder->length);
          sortorder->length+= sortorder->suffix_length;
        }
        break;
      case INT_RESULT:
        sortorder->length=8;      // Size of intern int64_t
        break;
      case DECIMAL_RESULT:
        sortorder->length=
          class_decimal_get_binary_size(sortorder->item->max_length -
                                     (sortorder->item->decimals ? 1 : 0),
                                     sortorder->item->decimals);
        break;
      case REAL_RESULT:
        sortorder->length=sizeof(double);
        break;
      case ROW_RESULT:
        // This case should never be choosen
        assert(0);
        break;
      }
      if (sortorder->item->maybe_null)
        length++;       // Place for NULL marker
    }
    set_if_smaller(sortorder->length, (size_t)getSession().variables.max_sort_length);
    length+=sortorder->length;
  }
  sortorder->field= (Field*) 0;     // end marker
  return length;
}


sort_addon_field *FileSort::get_addon_fields(Field **ptabfield, uint32_t sortlength_arg, uint32_t *plength)
{
  Field **pfield;
  Field *field;
  sort_addon_field *addonf;
  uint32_t length= 0;
  uint32_t fields= 0;
  uint32_t null_fields= 0;

  /*
    If there is a reference to a field in the query add it
    to the the set of appended fields.
    Note for future refinement:
    This this a too strong condition.
    Actually we need only the fields referred in the
    result set. And for some of them it makes sense to use
    the values directly from sorted fields.
  */
  *plength= 0;

  for (pfield= ptabfield; (field= *pfield) ; pfield++)
  {
    if (!(field->isReadSet()))
      continue;
    if (field->flags & BLOB_FLAG)
      return 0;
    length+= field->max_packed_col_length(field->pack_length());
    if (field->maybe_null())
      null_fields++;
    fields++;
  }
  if (!fields)
    return 0;
  length+= (null_fields+7)/8;

  if (length+sortlength_arg > getSession().variables.max_length_for_sort_data)
    return 0;
  addonf= (sort_addon_field *) malloc(sizeof(sort_addon_field) * (fields+1));

  *plength= length;
  length= (null_fields+7)/8;
  null_fields= 0;
  for (pfield= ptabfield; (field= *pfield) ; pfield++)
  {
    if (!(field->isReadSet()))
      continue;
    addonf->field= field;
    addonf->offset= length;
    if (field->maybe_null())
    {
      addonf->null_offset= null_fields/8;
      addonf->null_bit= 1<<(null_fields & 7);
      null_fields++;
    }
    else
    {
      addonf->null_offset= 0;
      addonf->null_bit= 0;
    }
    addonf->length= field->max_packed_col_length(field->pack_length());
    length+= addonf->length;
    addonf++;
  }
  addonf->field= 0;     // Put end marker

  return (addonf-fields);
}


static void
unpack_addon_fields(sort_addon_field *addon_field, unsigned char *buff)
{
  Field *field;
  sort_addon_field *addonf= addon_field;

  for ( ; (field= addonf->field) ; addonf++)
  {
    if (addonf->null_bit && (addonf->null_bit & buff[addonf->null_offset]))
    {
      field->set_null();
      continue;
    }
    field->set_notnull();
    field->unpack(field->ptr, buff + addonf->offset);
  }
}

/*
** functions to change a double or float to a sortable string
** The following should work for IEEE
*/

#define DBL_EXP_DIG (sizeof(double)*8-DBL_MANT_DIG)

void change_double_for_sort(double nr,unsigned char *to)
{
  unsigned char *tmp=(unsigned char*) to;
  if (nr == 0.0)
  {           /* Change to zero string */
    tmp[0]=(unsigned char) 128;
    memset(tmp+1, 0, sizeof(nr)-1);
  }
  else
  {
#ifdef WORDS_BIGENDIAN
    memcpy(tmp,&nr,sizeof(nr));
#else
    {
      unsigned char *ptr= (unsigned char*) &nr;
#if defined(__FLOAT_WORD_ORDER) && (__FLOAT_WORD_ORDER == __BIG_ENDIAN)
      tmp[0]= ptr[3]; tmp[1]=ptr[2]; tmp[2]= ptr[1]; tmp[3]=ptr[0];
      tmp[4]= ptr[7]; tmp[5]=ptr[6]; tmp[6]= ptr[5]; tmp[7]=ptr[4];
#else
      tmp[0]= ptr[7]; tmp[1]=ptr[6]; tmp[2]= ptr[5]; tmp[3]=ptr[4];
      tmp[4]= ptr[3]; tmp[5]=ptr[2]; tmp[6]= ptr[1]; tmp[7]=ptr[0];
#endif
    }
#endif
    if (tmp[0] & 128)       /* Negative */
    {           /* make complement */
      uint32_t i;
      for (i=0 ; i < sizeof(nr); i++)
  tmp[i]=tmp[i] ^ (unsigned char) 255;
    }
    else
    {         /* Set high and move exponent one up */
      uint16_t exp_part=(((uint16_t) tmp[0] << 8) | (uint16_t) tmp[1] |
           (uint16_t) 32768);
      exp_part+= (uint16_t) 1 << (16-1-DBL_EXP_DIG);
      tmp[0]= (unsigned char) (exp_part >> 8);
      tmp[1]= (unsigned char) exp_part;
    }
  }
}

} /* namespace drizzled */