sql_union.cc 7.07 KB
Newer Older
unknown's avatar
unknown committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
/* Copyright (C) 2000 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; either version 2 of the License, or
   (at your option) any later version.

   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */


/*
  UNION  of select's
  UNION's  were introduced by Monty and Sinisa <sinisa@mysql.com>
*/


#include "mysql_priv.h"
#include "sql_select.h"


28
int mysql_union(THD *thd, LEX *lex,select_result *result)
unknown's avatar
unknown committed
29
{
unknown's avatar
unknown committed
30
  SELECT_LEX *sl, *last_sl, *lex_sl;
unknown's avatar
unknown committed
31 32 33
  ORDER *order;
  List<Item> item_list;
  TABLE *table;
34 35
  int describe=(lex->select_lex.options & SELECT_DESCRIBE) ? 1 : 0;
  int res;
36
  TABLE_LIST result_table_list;
unknown's avatar
unknown committed
37 38 39 40
  TMP_TABLE_PARAM tmp_table_param;
  select_union *union_result;
  DBUG_ENTER("mysql_union");

41
  /* Fix tables 'to-be-unioned-from' list to point at opened tables */
unknown's avatar
unknown committed
42 43 44 45
  last_sl= &lex->select_lex;
  for (sl= last_sl;
       sl && sl->linkage != NOT_A_SELECT;
       last_sl=sl, sl=sl->next)
46 47 48 49 50 51 52
  {
    for (TABLE_LIST *cursor= (TABLE_LIST *)sl->table_list.first;
	 cursor;
	 cursor=cursor->next)
      cursor->table= ((TABLE_LIST*) cursor->table)->table;
  }

unknown's avatar
unknown committed
53
  /* last_sel now points at the last select where the ORDER BY is stored */
54 55
  if (sl)
  {
unknown's avatar
unknown committed
56 57 58 59 60 61 62 63 64 65 66 67
    /*
      The found SL is an extra SELECT_LEX argument that contains
      the ORDER BY and LIMIT parameter for the whole UNION
    */
    lex_sl= sl;
    last_sl->next=0;				// Remove this extra element
    order=  (ORDER *) lex_sl->order_list.first;
  }
  else if (!last_sl->braces)
  {
    lex_sl= last_sl;				// ORDER BY is here
    order=  (ORDER *) lex_sl->order_list.first;
68 69
  }
  else
unknown's avatar
unknown committed
70 71 72 73
  {
    lex_sl=0;
    order=0;
  }
74 75
  
  if (describe)
unknown's avatar
unknown committed
76
  {
77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
    Item *item;
    item_list.push_back(new Item_empty_string("table",NAME_LEN));
    item_list.push_back(new Item_empty_string("type",10));
    item_list.push_back(item=new Item_empty_string("possible_keys",
						  NAME_LEN*MAX_KEY));
    item->maybe_null=1;
    item_list.push_back(item=new Item_empty_string("key",NAME_LEN));
    item->maybe_null=1;
    item_list.push_back(item=new Item_int("key_len",0,3));
    item->maybe_null=1;
    item_list.push_back(item=new Item_empty_string("ref",
						    NAME_LEN*MAX_REF_PARTS));
    item->maybe_null=1;
    item_list.push_back(new Item_real("rows",0.0,0,10));
    item_list.push_back(new Item_empty_string("Extra",255));
unknown's avatar
unknown committed
92
  }
93
  else
unknown's avatar
unknown committed
94 95 96
  {
    Item *item;
    List_iterator<Item> it(lex->select_lex.item_list);
97
    TABLE_LIST *first_table= (TABLE_LIST*) lex->select_lex.table_list.first;
unknown's avatar
unknown committed
98 99 100 101 102 103 104 105

    /* Create a list of items that will be in the result set */
    while ((item= it++))
      if (item_list.push_back(item))
	DBUG_RETURN(-1);
    if (setup_fields(thd,first_table,item_list,0,0,1))
      DBUG_RETURN(-1);
  }
unknown's avatar
unknown committed
106

unknown's avatar
unknown committed
107
  bzero((char*) &tmp_table_param,sizeof(tmp_table_param));
unknown's avatar
unknown committed
108
  tmp_table_param.field_count=item_list.elements;
unknown's avatar
unknown committed
109
  if (!(table=create_tmp_table(thd, &tmp_table_param, item_list,
110
			       (ORDER*) 0, !describe & !lex->union_option,
unknown's avatar
unknown committed
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
			       1, 0,
			       (lex->select_lex.options | thd->options |
				TMP_TABLE_ALL_COLUMNS))))
    DBUG_RETURN(-1);
  table->file->extra(HA_EXTRA_WRITE_CACHE);
  table->file->extra(HA_EXTRA_IGNORE_DUP_KEY);
  bzero((char*) &result_table_list,sizeof(result_table_list));
  result_table_list.db= (char*) "";
  result_table_list.real_name=result_table_list.name=(char*) "union";
  result_table_list.table=table;

  if (!(union_result=new select_union(table)))
  {
    res= -1;
    goto exit;
  }
127 128
  union_result->save_time_stamp=!describe;

129
  for (sl=lex->select=&lex->select_lex;sl;sl=lex->select=sl->next)
unknown's avatar
unknown committed
130 131 132 133 134 135 136 137
  {
    thd->offset_limit=sl->offset_limit;
    thd->select_limit=sl->select_limit+sl->offset_limit;
    if (thd->select_limit < sl->select_limit)
      thd->select_limit= HA_POS_ERROR;		// no limit
    if (thd->select_limit == HA_POS_ERROR)
      sl->options&= ~OPTION_FOUND_ROWS;

138
    res=mysql_select(thd, (TABLE_LIST*) sl->table_list.first,
unknown's avatar
unknown committed
139 140
		     sl->item_list,
		     sl->where,
141
		     (sl->braces) ? (ORDER *)sl->order_list.first : (ORDER *) 0,
unknown's avatar
unknown committed
142 143 144
		     (ORDER*) sl->group_list.first,
		     sl->having,
		     (ORDER*) NULL,
145
		     sl->options | thd->options | SELECT_NO_UNLOCK | ((describe) ? SELECT_DESCRIBE : 0),
unknown's avatar
unknown committed
146 147 148 149 150 151 152 153 154 155
		     union_result);
    if (res)
      goto exit;
  }
  if (union_result->flush())
  {
    res= 1;					// Error is already sent
    goto exit;
  }
  delete union_result;
156 157

  /* Send result to 'result' */
158
  lex->select = &lex->select_lex;
unknown's avatar
unknown committed
159 160 161 162 163
  res =-1;
  {
    /* Create a list of fields in the temporary table */
    List_iterator<Item> it(item_list);
    Field **field;
164
#if 0
unknown's avatar
unknown committed
165 166
    List<Item_func_match> ftfunc_list;
    ftfunc_list.empty();
167 168 169
#else
    thd->lex.select_lex.ftfunc_list.empty();
#endif
unknown's avatar
unknown committed
170 171 172 173 174 175 176

    for (field=table->field ; *field ; field++)
    {
      (void) it++;
      (void) it.replace(new Item_field(*field));
    }
    if (!thd->fatal_error)			// Check if EOM
177
    {
unknown's avatar
unknown committed
178
      if (lex_sl)
179
      {
unknown's avatar
unknown committed
180 181 182
	thd->offset_limit=lex_sl->offset_limit;
	thd->select_limit=lex_sl->select_limit+lex_sl->offset_limit;
	if (thd->select_limit < lex_sl->select_limit)
183 184 185 186
	  thd->select_limit= HA_POS_ERROR;		// no limit
	if (thd->select_limit == HA_POS_ERROR)
	  thd->options&= ~OPTION_FOUND_ROWS;
      }
187 188
      if (describe)
	thd->select_limit= HA_POS_ERROR;		// no limit
unknown's avatar
unknown committed
189
      res=mysql_select(thd,&result_table_list,
190
		       item_list, NULL, /*ftfunc_list,*/ order,
unknown's avatar
unknown committed
191 192
		       (ORDER*) NULL, NULL, (ORDER*) NULL,
		       thd->options, result);
193
    }
unknown's avatar
unknown committed
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
  }

exit:
  free_tmp_table(thd,table);
  DBUG_RETURN(res);
}


/***************************************************************************
** store records in temporary table for UNION
***************************************************************************/

select_union::select_union(TABLE *table_par)
    :table(table_par)
{
  bzero((char*) &info,sizeof(info));
  /*
    We can always use DUP_IGNORE because the temporary table will only
    contain a unique key if we are using not using UNION ALL
  */
  info.handle_duplicates=DUP_IGNORE;
}

select_union::~select_union()
{
}

unknown's avatar
unknown committed
221

unknown's avatar
unknown committed
222 223
int select_union::prepare(List<Item> &list)
{
224
  if (save_time_stamp && list.elements != table->fields)
unknown's avatar
unknown committed
225 226 227 228 229
  {
    my_message(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT,
	       ER(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT),MYF(0));
    return -1;
  }
unknown's avatar
unknown committed
230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
  return 0;
}

bool select_union::send_data(List<Item> &values)
{
  if (thd->offset_limit)
  {						// using limit offset,count
    thd->offset_limit--;
    return 0;
  }
  fill_record(table->field,values);
  return write_record(table,&info) ? 1 : 0;
}

bool select_union::send_eof()
{
  return 0;
}

bool select_union::flush()
{
unknown's avatar
unknown committed
251 252
  int error;
  if ((error=table->file->extra(HA_EXTRA_NO_CACHE)))
unknown's avatar
unknown committed
253 254 255 256 257 258 259
  {
    table->file->print_error(error,MYF(0));
    ::send_error(&thd->net);
    return 1;
  }
  return 0;
}