sql_update.cc 24.1 KB
Newer Older
bk@work.mysql.com's avatar
bk@work.mysql.com committed
1
/* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
2

bk@work.mysql.com's avatar
bk@work.mysql.com committed
3 4 5 6
   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.
7

bk@work.mysql.com's avatar
bk@work.mysql.com committed
8 9 10 11
   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.
12

bk@work.mysql.com's avatar
bk@work.mysql.com committed
13 14 15 16 17
   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 */


18 19 20
/* Update of records 
   Multi-table updates were introduced by Monty and Sinisa <sinisa@mysql.com>
*/
bk@work.mysql.com's avatar
bk@work.mysql.com committed
21 22 23

#include "mysql_priv.h"
#include "sql_acl.h"
24
#include "sql_select.h"
bk@work.mysql.com's avatar
bk@work.mysql.com committed
25 26 27

/* Return 0 if row hasn't changed */

28
static bool compare_record(TABLE *table, ulong query_id)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
29 30 31 32 33 34 35 36 37
{
  if (!table->blob_fields)
    return cmp_record(table,1);
  if (memcmp(table->null_flags,
	     table->null_flags+table->rec_buff_length,
	     table->null_bytes))
    return 1;					// Diff in NULL value
  for (Field **ptr=table->field ; *ptr ; ptr++)
  {
38
    if ((*ptr)->query_id == query_id &&
bk@work.mysql.com's avatar
bk@work.mysql.com committed
39 40 41 42 43 44 45
	(*ptr)->cmp_binary_offset(table->rec_buff_length))
      return 1;
  }
  return 0;
}


46 47 48 49 50 51
int mysql_update(THD *thd,
                 TABLE_LIST *table_list,
                 List<Item> &fields,
		 List<Item> &values,
                 COND *conds,
                 ORDER *order,
bk@work.mysql.com's avatar
bk@work.mysql.com committed
52 53 54 55 56
		 ha_rows limit,
		 enum enum_duplicates handle_duplicates,
		 thr_lock_type lock_type)
{
  bool 		using_limit=limit != HA_POS_ERROR;
57
  bool		used_key_is_modified, using_transactions;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
58
  int		error=0;
59 60
  uint		save_time_stamp, used_index, want_privilege;
  ulong		query_id=thd->query_id, timestamp_query_id;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
61 62 63 64 65 66
  key_map	old_used_keys;
  TABLE		*table;
  SQL_SELECT	*select;
  READ_RECORD	info;
  DBUG_ENTER("mysql_update");
  LINT_INIT(used_index);
67
  LINT_INIT(timestamp_query_id);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
68 69

  if (!(table = open_ltable(thd,table_list,lock_type)))
70
    DBUG_RETURN(-1); /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
71 72 73 74
  save_time_stamp=table->time_stamp;
  table->file->info(HA_STATUS_VARIABLE | HA_STATUS_NO_LOCK);
  thd->proc_info="init";

75 76 77 78 79
  /* Calculate "table->used_keys" based on the WHERE */
  table->used_keys=table->keys_in_use;
  table->quick_keys=0;
  want_privilege=table->grant.want_privilege;
  table->grant.want_privilege=(SELECT_ACL & ~table->grant.privilege);
80 81
  if (setup_tables(table_list) || setup_conds(thd,table_list,&conds)
                               || setup_ftfuncs(thd))
82 83 84
    DBUG_RETURN(-1);				/* purecov: inspected */
  old_used_keys=table->used_keys;		// Keys used in WHERE

bk@work.mysql.com's avatar
bk@work.mysql.com committed
85
  /*
86 87
    Change the query_id for the timestamp column so that we can
    check if this is modified directly
bk@work.mysql.com's avatar
bk@work.mysql.com committed
88
  */
89 90 91 92 93
  if (table->timestamp_field)
  {
    timestamp_query_id=table->timestamp_field->query_id;
    table->timestamp_field->query_id=thd->query_id-1;
  }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
94

95 96
  /* Check the fields we are going to modify */
  table->grant.want_privilege=want_privilege;
monty@tik.mysql.fi's avatar
monty@tik.mysql.fi committed
97
  if (setup_fields(thd,table_list,fields,1,0,0))
98 99 100 101 102 103 104 105 106
    DBUG_RETURN(-1);				/* purecov: inspected */
  if (table->timestamp_field)
  {
    // Don't set timestamp column if this is modified
    if (table->timestamp_field->query_id == thd->query_id)
      table->time_stamp=0;
    else
      table->timestamp_field->query_id=timestamp_query_id;
  }
107

108 109
  /* Check values */
  table->grant.want_privilege=(SELECT_ACL & ~table->grant.privilege);
monty@tik.mysql.fi's avatar
monty@tik.mysql.fi committed
110
  if (setup_fields(thd,table_list,values,0,0,0))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
111 112 113 114
  {
    table->time_stamp=save_time_stamp;		// Restore timestamp pointer
    DBUG_RETURN(-1);				/* purecov: inspected */
  }
115

116 117
  // Don't count on usage of 'only index' when calculating which key to use
  table->used_keys=0;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
118 119
  select=make_select(table,0,0,conds,&error);
  if (error ||
120
      (select && select->check_quick(test(thd->options & OPTION_SAFE_UPDATES),
monty@tik.mysql.fi's avatar
monty@tik.mysql.fi committed
121 122
				     limit)) ||
      !limit)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
123 124 125 126 127 128 129
  {
    delete select;
    table->time_stamp=save_time_stamp;		// Restore timestamp pointer
    if (error)
    {
      DBUG_RETURN(-1);				// Error in where
    }
130
    send_ok(thd);				// No matching records
bk@work.mysql.com's avatar
bk@work.mysql.com committed
131 132 133
    DBUG_RETURN(0);
  }
  /* If running in safe sql mode, don't allow updates without keys */
134
  if (!table->quick_keys)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
135
  {
136
    thd->lex.select_lex.options|=QUERY_NO_INDEX_USED;
137 138 139 140
    if ((thd->options & OPTION_SAFE_UPDATES) && limit == HA_POS_ERROR)
    {
      delete select;
      table->time_stamp=save_time_stamp;
141
      send_error(thd,ER_UPDATE_WITHOUT_KEY_IN_SAFE_MODE);
142 143
      DBUG_RETURN(1);
    }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
144
  }
145
  init_ftfuncs(thd,1);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
146 147 148 149 150 151 152 153 154 155
  /* Check if we are modifying a key that we are used to search with */
  if (select && select->quick)
    used_key_is_modified= (!select->quick->unique_key_range() &&
			   check_if_key_used(table,
					     (used_index=select->quick->index),
					     fields));
  else if ((used_index=table->file->key_used_on_scan) < MAX_KEY)
    used_key_is_modified=check_if_key_used(table, used_index, fields);
  else
    used_key_is_modified=0;
156
  if (used_key_is_modified || order)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
157 158 159 160 161
  {
    /*
    ** We can't update table directly;  We must first search after all
    ** matching rows before updating the table!
    */
162
    table->file->extra(HA_EXTRA_DONT_USE_CURSOR_TO_UPDATE);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
163 164 165 166
    IO_CACHE tempfile;
    if (open_cached_file(&tempfile, mysql_tmpdir,TEMP_PREFIX,
			  DISK_BUFFER_SIZE, MYF(MY_WME)))
    {
167 168
      delete select; /* purecov: inspected */
      table->time_stamp=save_time_stamp;	// Restore timestamp pointer /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
169 170 171 172 173 174 175
      DBUG_RETURN(-1);
    }
    if (old_used_keys & ((key_map) 1 << used_index))
    {
      table->key_read=1;
      table->file->extra(HA_EXTRA_KEYREAD);
    }
176 177 178 179 180 181 182 183

    if (order)
    {
      uint         length;
      SORT_FIELD  *sortorder;
      TABLE_LIST   tables;
      List<Item>   fields;
      List<Item>   all_fields;
184
      ha_rows examined_rows;
185 186 187 188 189 190 191 192

      bzero((char*) &tables,sizeof(tables));
      tables.table = table;

      table->io_cache = (IO_CACHE *) my_malloc(sizeof(IO_CACHE),
                                               MYF(MY_FAE | MY_ZEROFILL));
      if (setup_order(thd, &tables, fields, all_fields, order) ||
          !(sortorder=make_unireg_sortorder(order, &length)) ||
193 194
          (table->found_records = filesort(thd, table, sortorder, length,
                                           (SQL_SELECT *) 0,
195
					   HA_POS_ERROR, &examined_rows))
196 197 198 199 200 201 202 203
          == HA_POS_ERROR)
      {
	delete select;
	table->time_stamp=save_time_stamp;	// Restore timestamp pointer
	DBUG_RETURN(-1);
      }
    }

bk@work.mysql.com's avatar
bk@work.mysql.com committed
204
    init_read_record(&info,thd,table,select,0,1);
205
    thd->proc_info="Searching rows for update";
bk@work.mysql.com's avatar
bk@work.mysql.com committed
206 207 208 209 210 211 212 213 214

    while (!(error=info.read_record(&info)) && !thd->killed)
    {
      if (!(select && select->skipp_record()))
      {
	table->file->position(table->record[0]);
	if (my_b_write(&tempfile,table->file->ref,
		       table->file->ref_length))
	{
215 216
	  error=1; /* purecov: inspected */
	  break; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
217 218 219 220
	}
      }
      else
      {
221
	if (!(test_flags & 512))		/* For debugging */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
	{
	  DBUG_DUMP("record",(char*) table->record[0],table->reclength);
	}
      }
    }
    end_read_record(&info);
    if (table->key_read)
    {
      table->key_read=0;
      table->file->extra(HA_EXTRA_NO_KEYREAD);
    }
    /* Change select to use tempfile */
    if (select)
    {
      delete select->quick;
      if (select->free_cond)
	delete select->cond;
      select->quick=0;
      select->cond=0;
    }
    else
243
    {
bk@work.mysql.com's avatar
bk@work.mysql.com committed
244 245 246 247
      select= new SQL_SELECT;
      select->head=table;
    }
    if (reinit_io_cache(&tempfile,READ_CACHE,0L,0,0))
248
      error=1; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
249 250 251 252 253
    select->file=tempfile;			// Read row ptrs from this file
    if (error >= 0)
    {
      delete select;
      table->time_stamp=save_time_stamp;	// Restore timestamp pointer
254
      DBUG_RETURN(-1);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
255 256 257
    }
  }

monty@donna.mysql.com's avatar
monty@donna.mysql.com committed
258 259
  if (handle_duplicates == DUP_IGNORE)
    table->file->extra(HA_EXTRA_IGNORE_DUP_KEY);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
260 261 262 263 264
  init_read_record(&info,thd,table,select,0,1);

  ha_rows updated=0L,found=0L;
  thd->count_cuted_fields=1;			/* calc cuted fields */
  thd->cuted_fields=0L;
265
  thd->proc_info="Updating";
266
  query_id=thd->query_id;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
267 268 269 270 271 272 273

  while (!(error=info.read_record(&info)) && !thd->killed)
  {
    if (!(select && select->skipp_record()))
    {
      store_record(table,1);
      if (fill_record(fields,values))
274
	break; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
275
      found++;
276
      if (compare_record(table, query_id))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296
      {
	if (!(error=table->file->update_row((byte*) table->record[1],
					    (byte*) table->record[0])))
	{
	  updated++;
	  if (!--limit && using_limit)
	  {
	    error= -1;
	    break;
	  }
	}
	else if (handle_duplicates != DUP_IGNORE ||
		 error != HA_ERR_FOUND_DUPP_KEY)
	{
	  table->file->print_error(error,MYF(0));
	  error= 1;
	  break;
	}
      }
    }
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
297 298
    else
      table->file->unlock_row();
bk@work.mysql.com's avatar
bk@work.mysql.com committed
299 300 301
  }
  end_read_record(&info);
  thd->proc_info="end";
monty@donna.mysql.com's avatar
monty@donna.mysql.com committed
302
  VOID(table->file->extra(HA_EXTRA_NO_IGNORE_DUP_KEY));
bk@work.mysql.com's avatar
bk@work.mysql.com committed
303
  table->time_stamp=save_time_stamp;	// Restore auto timestamp pointer
304
  using_transactions=table->file->has_transactions();
305
  if (updated && (error <= 0 || !using_transactions))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
306
  {
307 308 309
    mysql_update_log.write(thd,thd->query,thd->query_length);
    if (mysql_bin_log.is_open())
    {
310 311
      Query_log_event qinfo(thd, thd->query, thd->query_length,
			    using_transactions);
312 313
      if (mysql_bin_log.write(&qinfo) && using_transactions)
	error=1;
314
    }
315
    if (!using_transactions)
316
      thd->options|=OPTION_STATUS_NO_TRANS_UPDATE;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
317
  }
318
  if (using_transactions && ha_autocommit_or_rollback(thd, error >= 0))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
319
    error=1;
320
  if (updated)
bell@sanja.is.com.ua's avatar
merge  
bell@sanja.is.com.ua committed
321
  {
322
    query_cache_invalidate3(thd, table_list, 1);
bell@sanja.is.com.ua's avatar
merge  
bell@sanja.is.com.ua committed
323
  }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
324 325 326 327 328
  if (thd->lock)
  {
    mysql_unlock_tables(thd, thd->lock);
    thd->lock=0;
  }
329

bk@work.mysql.com's avatar
bk@work.mysql.com committed
330 331
  delete select;
  if (error >= 0)
332
    send_error(thd,thd->killed ? ER_SERVER_SHUTDOWN : 0); /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
333 334 335 336 337
  else
  {
    char buff[80];
    sprintf(buff,ER(ER_UPDATE_INFO), (long) found, (long) updated,
	    (long) thd->cuted_fields);
338
    send_ok(thd,
bk@work.mysql.com's avatar
bk@work.mysql.com committed
339 340 341 342 343
	    (thd->client_capabilities & CLIENT_FOUND_ROWS) ? found : updated,
	    thd->insert_id_used ? thd->insert_id() : 0L,buff);
    DBUG_PRINT("info",("%d records updated",updated));
  }
  thd->count_cuted_fields=0;			/* calc cuted fields */
serg@serg.mysql.com's avatar
serg@serg.mysql.com committed
344
  free_io_cache(table);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
345 346
  DBUG_RETURN(0);
}
347 348

/***************************************************************************
349
  Update multiple tables from join 
350 351 352 353 354 355 356 357 358 359 360
***************************************************************************/

multi_update::multi_update(THD *thd_arg, TABLE_LIST *ut, List<Item> &fs, 		 
			   enum enum_duplicates handle_duplicates,  thr_lock_type lock_option_arg, uint num)
  : update_tables (ut), thd(thd_arg), updated(0), found(0), fields(fs), lock_option(lock_option_arg),
    dupl(handle_duplicates), num_of_tables(num), num_fields(0), num_updated(0) , error(0),  do_update(false)
{
  save_time_stamps = (uint *) sql_calloc (sizeof(uint) * num_of_tables);
  tmp_tables = (TABLE **)NULL;
  int counter=0;
  ulong timestamp_query_id;
361
  not_trans_safe=false;
362 363 364
  for (TABLE_LIST *dt=ut ; dt ; dt=dt->next,counter++)
  {
    TABLE *table=ut->table;
365
    // (void) ut->table->file->extra(HA_EXTRA_NO_KEYREAD);
366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382
    dt->table->used_keys=0;
    if (table->timestamp_field)
    {
      // Don't set timestamp column if this is modified
      timestamp_query_id=table->timestamp_field->query_id;
      table->timestamp_field->query_id=thd->query_id-1;
      if (table->timestamp_field->query_id == thd->query_id)
	table->time_stamp=0;
      else
	table->timestamp_field->query_id=timestamp_query_id;
    }
    save_time_stamps[counter]=table->time_stamp;
  }
  error = 1; // In case we do not reach prepare we have to reset timestamps
}

int
383
multi_update::prepare(List<Item> &values, SELECT_LEX_UNIT *u)
384 385
{
  DBUG_ENTER("multi_update::prepare");
386
  unit= u;
387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404
  do_update = true;   
  thd->count_cuted_fields=1;
  thd->cuted_fields=0L;
  thd->proc_info="updating the  main table";
  TABLE_LIST *table_ref;

  if (thd->options & OPTION_SAFE_UPDATES)
  {
    for (table_ref=update_tables;  table_ref; table_ref=table_ref->next)
    {
      TABLE *table=table_ref->table;
      if ((thd->options & OPTION_SAFE_UPDATES) && !table->quick_keys)
      {
	my_error(ER_UPDATE_WITHOUT_KEY_IN_SAFE_MODE,MYF(0));
	DBUG_RETURN(1);
      }
    }
  }
405 406 407 408 409 410 411
  /*
    Here I have to connect fields with tables and only update tables that
    need to be updated.
    I calculate num_updated and fill-up table_sequence
    Set table_list->shared  to true or false, depending on whether table is
    to be updated or not
  */
412 413 414 415 416 417 418 419 420

  Item_field *item;
  List_iterator<Item> it(fields);
  num_fields=fields.elements;
  field_sequence = (uint *) sql_alloc(sizeof(uint)*num_fields);
  uint *int_ptr=field_sequence;
  while ((item= (Item_field *)it++))
  {
    unsigned int counter=0;
421 422
    for (table_ref=update_tables;  table_ref;
	 table_ref=table_ref->next, counter++)
423
    {
424
      if (table_ref->table == item->field->table)
425
      {
426 427 428 429 430 431 432 433 434 435 436
	if (!table_ref->shared)
	{
	  TABLE *tbl=table_ref->table;
	  num_updated++;
	  table_ref->shared=1;
	  if (!not_trans_safe && !table_ref->table->file->has_transactions())
	    not_trans_safe=true;
	  // to be moved if initialize_tables has to be used
	  tbl->no_keyread=1;
	  tbl->used_keys=0;
	}
437 438 439 440 441
	break;
      }
    }
    if (!table_ref)
    {
442
      net_printf(thd, ER_NOT_SUPPORTED_YET, "JOIN SYNTAX WITH MULTI-TABLE UPDATES");
443 444 445 446 447 448 449
      DBUG_RETURN(1);
    }
    else
      *int_ptr++=counter;
  }
  if (!num_updated)
  {
450
    net_printf(thd, ER_NOT_SUPPORTED_YET, "SET CLAUSE MUST CONTAIN TABLE.FIELD REFERENCE");
451 452 453
    DBUG_RETURN(1);
  }

454 455 456 457
  /*
    Here, I have to allocate the array of temporary tables
    I have to treat a case of num_updated=1 differently in send_data() method.
  */
458 459 460 461 462 463 464 465 466 467 468
  if (num_updated > 1)
  {
    tmp_tables = (TABLE **) sql_calloc(sizeof(TABLE *) * (num_updated - 1));
    infos = (COPY_INFO *) sql_calloc(sizeof(COPY_INFO) * (num_updated - 1));
    fields_by_tables = (List_item **)sql_calloc(sizeof(List_item *) * num_updated);
    unsigned int counter;
    List<Item> *temp_fields;
    for (table_ref=update_tables, counter = 0;  table_ref; table_ref=table_ref->next)
    {
      if (!table_ref->shared) 
	continue;
469
      // Here we have to add row offset as an additional field ...
470 471 472 473 474 475 476 477 478 479 480 481 482 483
      if (!(temp_fields = (List_item *)sql_calloc(sizeof(List_item))))
      {
	error = 1; // A proper error message is due here 
	DBUG_RETURN(1);
      }
      temp_fields->empty();
      it.rewind(); int_ptr=field_sequence;
      while ((item= (Item_field *)it++))
      {
	if (*int_ptr++ == counter)
	  temp_fields->push_back(item);
      }
      if (counter)
      {
484
	Field_string offset(table_ref->table->file->ref_length, false,
bell@sanja.is.com.ua's avatar
merge  
bell@sanja.is.com.ua committed
485 486
                            "offset", table_ref->table, true,
                            default_charset_info);
487
	temp_fields->push_front(new Item_field(((Field *)&offset)));
monty@narttu.mysql.fi's avatar
monty@narttu.mysql.fi committed
488 489

	// Make a temporary table
490 491 492 493
	int cnt=counter-1;
	TMP_TABLE_PARAM tmp_table_param;
	bzero((char*) &tmp_table_param,sizeof(tmp_table_param));
	tmp_table_param.field_count=temp_fields->elements;
494 495 496
	if (!(tmp_tables[cnt]=create_tmp_table(thd, &tmp_table_param,
					       *temp_fields,
					       (ORDER*) 0, 1, 0, 0,
497 498
					       TMP_TABLE_ALL_COLUMNS,
					       unit)))
499 500 501 502 503 504 505 506 507 508 509 510 511
	{
	  error = 1; // A proper error message is due here 
	  DBUG_RETURN(1);
	}
	tmp_tables[cnt]->file->extra(HA_EXTRA_WRITE_CACHE);
	tmp_tables[cnt]->file->extra(HA_EXTRA_IGNORE_DUP_KEY);
	infos[cnt].handle_duplicates=DUP_IGNORE;
	temp_fields->pop(); // because we shall use those for values only ...
      }
      fields_by_tables[counter]=temp_fields;
      counter++;
    }
  }
512
  init_ftfuncs(thd,1);
513 514 515 516 517 518 519 520
  error = 0; // Timestamps do not need to be restored, so far ...
  DBUG_RETURN(0);
}


void
multi_update::initialize_tables(JOIN *join)
{
521
#ifdef NOT_YET
522
   We skip it as it only makes a mess ...........
523 524 525 526 527 528 529 530 531 532 533 534
  TABLE_LIST *walk;
  table_map tables_to_update_from=0;
  for (walk= update_tables ; walk ; walk=walk->next)
    tables_to_update_from|= walk->table->map;
  
  walk= update_tables;
  for (JOIN_TAB *tab=join->join_tab, *end=join->join_tab+join->tables;
       tab < end;
       tab++)
  {
    if (tab->table->map & tables_to_update_from)
    {
535
//       We are going to update from this table 
536
       TABLE *tbl=walk->table=tab->table;
537
       /* Don't use KEYREAD optimization on this table */
538 539
       tbl->no_keyread=1;
       walk=walk->next;
540 541
    }
  }
542
#endif
543 544 545 546 547 548 549 550 551 552 553
}


multi_update::~multi_update()
{
  int counter = 0;
  for (table_being_updated=update_tables ;
       table_being_updated ;
       counter++, table_being_updated=table_being_updated->next)
  {
    TABLE *table=table_being_updated->table;
554
    table->no_keyread=0;
555 556 557 558 559 560 561 562 563 564 565 566 567 568 569
    if (error) 
      table->time_stamp=save_time_stamps[counter];
  }
  if (tmp_tables)
    for (uint counter = 0; counter < num_updated-1; counter++)
      if (tmp_tables[counter])
	free_tmp_table(thd,tmp_tables[counter]);
}


bool multi_update::send_data(List<Item> &values)
{
  List<Item> real_values(values);
  for (uint counter = 0; counter < fields.elements; counter++)
    real_values.pop();
570
  // We have skipped fields ....
571 572 573 574 575 576 577 578 579 580 581 582 583
  if (num_updated == 1)
  {
    for (table_being_updated=update_tables ;
	 table_being_updated ;
	 table_being_updated=table_being_updated->next)
    {
      if (!table_being_updated->shared) 
	continue;
      TABLE *table=table_being_updated->table;
      /* Check if we are using outer join and we didn't find the row */
      if (table->status & (STATUS_NULL_ROW | STATUS_UPDATED))
	return 0;
      table->file->position(table->record[0]);
584
      // Only one table being updated receives a completely different treatment
585 586 587 588 589 590 591 592
      table->status|= STATUS_UPDATED;
      store_record(table,1); 
      if (fill_record(fields,real_values))
	return 1;
      found++;
      if (/* compare_record(table, query_id)  && */
	  !(error=table->file->update_row(table->record[1], table->record[0])))
	updated++;
593
      table->file->extra(HA_EXTRA_NO_CACHE);
594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
      return error;
    }
  }
  else
  {
    int secure_counter= -1;
    for (table_being_updated=update_tables ;
	 table_being_updated ;
	 table_being_updated=table_being_updated->next, secure_counter++)
    {
      if (!table_being_updated->shared) 
	continue;
      
      TABLE *table=table_being_updated->table;
      /* Check if we are using outer join and we didn't find the row */
      if (table->status & (STATUS_NULL_ROW | STATUS_UPDATED))
	continue;
      table->file->position(table->record[0]);
      Item *item;
      List_iterator<Item> it(real_values);
      List <Item> values_by_table;
      uint *int_ptr=field_sequence;
      while ((item= (Item *)it++))
      {
	if (*int_ptr++ == (uint) (secure_counter + 1))
	  values_by_table.push_back(item);
      }
621
      // Here I am breaking values as per each table    
622 623 624 625 626 627 628 629 630
      if (secure_counter < 0)
      {
	table->status|= STATUS_UPDATED;
	store_record(table,1); 
	if (fill_record(*fields_by_tables[0],values_by_table))
	  return 1;
	found++;
	if (/*compare_record(table, query_id)  && */
	    !(error=table->file->update_row(table->record[1], table->record[0])))
631
	{
632
	  updated++;
633 634
	  table->file->extra(HA_EXTRA_NO_CACHE);
	}
635 636 637 638 639 640 641 642 643
	else
	{
	  table->file->print_error(error,MYF(0));
	  if (!error) error=1;
	  return 1;
	}
      }
      else
      {
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
644 645
	// Here we insert into each temporary table
	values_by_table.push_front(new Item_string((char*) table->file->ref,
646 647
						   table->file->ref_length,
						   system_charset_info));
648
	fill_record(tmp_tables[secure_counter]->field,values_by_table);
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
649 650
	error= write_record(tmp_tables[secure_counter],
			    &(infos[secure_counter]));
651 652 653 654 655 656 657 658 659 660 661 662 663
	if (error)
	{
	  error=-1;
	  return 1;
	}
      }
    }
  }
  return 0;
}

void multi_update::send_error(uint errcode,const char *err)
{
bell@sanja.is.com.ua's avatar
bell@sanja.is.com.ua committed
664 665

  //TODO error should be sent at the query processing end
666
  /* First send error what ever it is ... */
667
  ::send_error(thd,errcode,err);
668 669

  /* reset used flags */
670
  //  update_tables->table->no_keyread=0;
671 672 673 674

  /* If nothing updated return */
  if (!updated)
    return;
675

676
  /* Something already updated so we have to invalidate cache */
677 678
  query_cache_invalidate3(thd, update_tables, 1);

679 680 681 682 683 684 685 686 687 688 689
  /* Below can happen when thread is killed early ... */
  if (!table_being_updated)
    table_being_updated=update_tables;

  /*
    If rows from the first table only has been updated and it is transactional,
    just do rollback.
    The same if all tables are transactional, regardless of where we are.
    In all other cases do attempt updates ...
  */
  if ((table_being_updated->table->file->has_transactions() &&
690
       table_being_updated == update_tables) || !not_trans_safe)
691
    ha_rollback_stmt(thd);
692
  else if (do_update && num_updated > 1)
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
    VOID(do_updates(true));
}


int multi_update::do_updates (bool from_send_error)
{
  int error = 0, counter = 0;

  if (from_send_error)
  {
    /* Found out table number for 'table_being_updated' */
    for (TABLE_LIST *aux=update_tables;
	 aux != table_being_updated;
	 aux=aux->next)
      counter++;
  }
  else
    table_being_updated = update_tables;

  do_update = false;
  for (table_being_updated=table_being_updated->next;
       table_being_updated ;
       table_being_updated=table_being_updated->next, counter++)
  { 
    if (!table_being_updated->shared) 
      continue;

    TABLE *table = table_being_updated->table;
    TABLE *tmp_table=tmp_tables[counter];
    if (tmp_table->file->extra(HA_EXTRA_NO_CACHE))
    {
      error=1;
      break;
    }
    List<Item> list;
    Field **ptr=tmp_table->field,*field;
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
729
    // This is supposed to be something like insert_fields
730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747
    thd->used_tables|=tmp_table->map;
    while ((field = *ptr++))
    {
      list.push_back((Item *)new Item_field(field));
      if (field->query_id == thd->query_id)
	thd->dupp_field=field;
      field->query_id=thd->query_id;
      tmp_table->used_keys&=field->part_of_key;
    }
    tmp_table->used_fields=tmp_table->fields;
    error=0; list.pop(); // we get position some other way ...
    error = tmp_table->file->rnd_init(1);
    if (error) 
      return error;
    while (!(error=tmp_table->file->rnd_next(tmp_table->record[0])) &&
	   (!thd->killed ||  from_send_error || not_trans_safe))
    {
      found++; 
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
748 749
      error= table->file->rnd_pos(table->record[0],
				  (byte*) (*(tmp_table->field))->ptr);
750 751 752 753
      if (error)
	return error;
      table->status|= STATUS_UPDATED;
      store_record(table,1); 
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
754 755
      error= fill_record(*fields_by_tables[counter + 1],list) ||
	/* compare_record(table, query_id) || */
756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771
	table->file->update_row(table->record[1],table->record[0]);
      if (error)
      {
	table->file->print_error(error,MYF(0));
	break;
      }
      else
	updated++;
    }
    if (error == HA_ERR_END_OF_FILE)
      error = 0;
  }
  return error;
}


monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
772 773
/* out: 1 if error, 0 if success */

774 775
bool multi_update::send_eof()
{
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
776
  thd->proc_info="updating the  reference tables";
777 778

  /* Does updates for the last n - 1 tables, returns 0 if ok */
Sinisa@sinisa.nasamreza.org's avatar
Sinisa@sinisa.nasamreza.org committed
779
  int error = (num_updated > 1) ? do_updates(false) : 0;   /* do_updates returns 0 if success */
780 781

  /* reset used flags */
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
782 783 784 785 786
#ifndef NOT_USED
  update_tables->table->no_keyread=0;
#endif
  if (error == -1)
    error = 0;
787
  thd->proc_info="end";
bell@sanja.is.com.ua's avatar
bell@sanja.is.com.ua committed
788
  //TODO error should be sent at the query processing end
789 790 791
  if (error)
    send_error(error,"An error occured in multi-table update");

monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
792 793 794 795 796 797
  /*
    Write the SQL statement to the binlog if we updated
    rows and we succeeded, or also in an error case when there
    was a non-transaction-safe table involved, since
    modifications in it cannot be rolled back.
  */
798

799
  if (updated || not_trans_safe)
800 801
  {
    mysql_update_log.write(thd,thd->query,thd->query_length);
802
    Query_log_event qinfo(thd, thd->query, thd->query_length);
803

monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
804 805 806 807
    /*
      mysql_bin_log is not open if binlogging or replication
      is not used
    */
808 809

    if (mysql_bin_log.is_open() &&  mysql_bin_log.write(&qinfo) &&
810
	!not_trans_safe)
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
811
      error=1;  /* Log write failed: roll back the SQL statement */
812 813 814 815 816 817 818 819 820 821 822

    /* Commit or rollback the current SQL statement */ 
    VOID(ha_autocommit_or_rollback(thd,error > 0));
  }
  else
    error=0; // this can happen only if it is end of file error
  if (!error) // if the above log write did not fail ...
  {
    char buff[80];
    sprintf(buff,ER(ER_UPDATE_INFO), (long) found, (long) updated,
	    (long) thd->cuted_fields);
823
    if (updated)
824
    {
825
      query_cache_invalidate3(thd, update_tables, 1);
826
    }
827
    ::send_ok(thd,
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
828 829
	      (thd->client_capabilities & CLIENT_FOUND_ROWS) ? found : updated,
	      thd->insert_id_used ? thd->insert_id() : 0L,buff);
830 831 832 833
  }
  thd->count_cuted_fields=0;
  return 0;
}