sql_update.cc 24.5 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
		 ha_rows limit,
53
		 enum enum_duplicates handle_duplicates)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
54
{
55 56
  bool 		using_limit=limit != HA_POS_ERROR;
  bool		safe_update= thd->options & OPTION_SAFE_UPDATES;
57
  bool		used_key_is_modified, transactional_table, log_delayed;
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,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
  if (setup_tables(table_list) || setup_conds(thd,table_list,&conds)
81
                               || setup_ftfuncs(&thd->lex.select_lex))
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(safe_update, limit)) || !limit)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
121 122 123 124 125 126 127
  {
    delete select;
    table->time_stamp=save_time_stamp;		// Restore timestamp pointer
    if (error)
    {
      DBUG_RETURN(-1);				// Error in where
    }
128
    send_ok(thd);				// No matching records
bk@work.mysql.com's avatar
bk@work.mysql.com committed
129 130 131
    DBUG_RETURN(0);
  }
  /* If running in safe sql mode, don't allow updates without keys */
132
  if (!table->quick_keys)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
133
  {
134
    thd->lex.select_lex.options|=QUERY_NO_INDEX_USED;
135
    if (safe_update && !using_limit)
136 137 138
    {
      delete select;
      table->time_stamp=save_time_stamp;
139
      send_error(thd,ER_UPDATE_WITHOUT_KEY_IN_SAFE_MODE);
140 141
      DBUG_RETURN(1);
    }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
142
  }
143
  init_ftfuncs(thd, &thd->lex.select_lex, 1);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
144 145 146 147 148 149 150 151 152 153
  /* 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;
154
  if (used_key_is_modified || order)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
155 156 157 158 159
  {
    /*
    ** We can't update table directly;  We must first search after all
    ** matching rows before updating the table!
    */
160
    table->file->extra(HA_EXTRA_DONT_USE_CURSOR_TO_UPDATE);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
161 162 163 164
    IO_CACHE tempfile;
    if (open_cached_file(&tempfile, mysql_tmpdir,TEMP_PREFIX,
			  DISK_BUFFER_SIZE, MYF(MY_WME)))
    {
165 166
      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
167 168 169 170 171 172 173
      DBUG_RETURN(-1);
    }
    if (old_used_keys & ((key_map) 1 << used_index))
    {
      table->key_read=1;
      table->file->extra(HA_EXTRA_KEYREAD);
    }
174 175 176 177 178 179 180 181

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

      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)) ||
191 192
          (table->found_records = filesort(thd, table, sortorder, length,
                                           (SQL_SELECT *) 0,
193
					   HA_POS_ERROR, &examined_rows))
194 195 196 197 198 199 200 201
          == 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
202
    init_read_record(&info,thd,table,select,0,1);
203
    thd->proc_info="Searching rows for update";
bk@work.mysql.com's avatar
bk@work.mysql.com committed
204 205 206 207 208 209 210 211 212

    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))
	{
213 214
	  error=1; /* purecov: inspected */
	  break; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
215 216 217 218
	}
      }
      else
      {
219
	if (!(test_flags & 512))		/* For debugging */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
	{
	  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
241
    {
bk@work.mysql.com's avatar
bk@work.mysql.com committed
242 243 244 245
      select= new SQL_SELECT;
      select->head=table;
    }
    if (reinit_io_cache(&tempfile,READ_CACHE,0L,0,0))
246
      error=1; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
247 248 249 250 251
    select->file=tempfile;			// Read row ptrs from this file
    if (error >= 0)
    {
      delete select;
      table->time_stamp=save_time_stamp;	// Restore timestamp pointer
252
      DBUG_RETURN(-1);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
253 254 255
    }
  }

monty@donna.mysql.com's avatar
monty@donna.mysql.com committed
256 257
  if (handle_duplicates == DUP_IGNORE)
    table->file->extra(HA_EXTRA_IGNORE_DUP_KEY);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
258 259 260 261 262
  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;
263
  thd->proc_info="Updating";
264
  query_id=thd->query_id;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
265 266 267 268 269 270 271

  while (!(error=info.read_record(&info)) && !thd->killed)
  {
    if (!(select && select->skipp_record()))
    {
      store_record(table,1);
      if (fill_record(fields,values))
272
	break; /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
273
      found++;
274
      if (compare_record(table, query_id))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
      {
	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
295 296
    else
      table->file->unlock_row();
bk@work.mysql.com's avatar
bk@work.mysql.com committed
297 298 299
  }
  end_read_record(&info);
  thd->proc_info="end";
monty@donna.mysql.com's avatar
monty@donna.mysql.com committed
300
  VOID(table->file->extra(HA_EXTRA_NO_IGNORE_DUP_KEY));
bk@work.mysql.com's avatar
bk@work.mysql.com committed
301
  table->time_stamp=save_time_stamp;	// Restore auto timestamp pointer
302 303 304
  transactional_table= table->file->has_transactions();
  log_delayed= (transactional_table || table->tmp_table);
  if (updated && (error <= 0 || !transactional_table))
bk@work.mysql.com's avatar
bk@work.mysql.com committed
305
  {
306 307 308
    mysql_update_log.write(thd,thd->query,thd->query_length);
    if (mysql_bin_log.is_open())
    {
309
      Query_log_event qinfo(thd, thd->query, thd->query_length,
310 311 312
			    log_delayed);
      if (mysql_bin_log.write(&qinfo) && transactional_table)
	error=1;				// Rollback update
313
    }
314
    if (!log_delayed)
315
      thd->options|=OPTION_STATUS_NO_TRANS_UPDATE;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
316
  }
317 318 319 320 321 322 323 324 325 326 327 328 329
  if (transactional_table)
  {
    if (ha_autocommit_or_rollback(thd, error >= 0))
      error=1;
  }
  /*
    Only invalidate the query cache if something changed or if we
    didn't commit the transacion (query cache is automaticly
    invalidated on commit)
  */
  if (updated &&
      (!transactional_table ||
       thd->options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN)))
bell@sanja.is.com.ua's avatar
merge  
bell@sanja.is.com.ua committed
330
  {
331
    query_cache_invalidate3(thd, table_list, 1);
bell@sanja.is.com.ua's avatar
merge  
bell@sanja.is.com.ua committed
332
  }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
333 334 335 336 337
  if (thd->lock)
  {
    mysql_unlock_tables(thd, thd->lock);
    thd->lock=0;
  }
338

bk@work.mysql.com's avatar
bk@work.mysql.com committed
339 340
  delete select;
  if (error >= 0)
341
    send_error(thd,thd->killed ? ER_SERVER_SHUTDOWN : 0); /* purecov: inspected */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
342 343 344 345 346
  else
  {
    char buff[80];
    sprintf(buff,ER(ER_UPDATE_INFO), (long) found, (long) updated,
	    (long) thd->cuted_fields);
347
    send_ok(thd,
bk@work.mysql.com's avatar
bk@work.mysql.com committed
348 349 350 351 352
	    (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
353
  free_io_cache(table);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
354 355
  DBUG_RETURN(0);
}
356 357

/***************************************************************************
358
  Update multiple tables from join 
359 360
***************************************************************************/

361 362 363 364 365 366
multi_update::multi_update(THD *thd_arg, TABLE_LIST *ut, List<Item> &fs, 
			   enum enum_duplicates handle_duplicates, 
			   uint num)
  : update_tables (ut), thd(thd_arg), updated(0), found(0), fields(fs),
    dupl(handle_duplicates), num_of_tables(num), num_fields(0), num_updated(0),
    error(0),  do_update(false)
367 368 369 370 371
{
  save_time_stamps = (uint *) sql_calloc (sizeof(uint) * num_of_tables);
  tmp_tables = (TABLE **)NULL;
  int counter=0;
  ulong timestamp_query_id;
372
  not_trans_safe=false;
373 374 375
  for (TABLE_LIST *dt=ut ; dt ; dt=dt->next,counter++)
  {
    TABLE *table=ut->table;
376
    // (void) ut->table->file->extra(HA_EXTRA_NO_KEYREAD);
377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
    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
394
multi_update::prepare(List<Item> &values, SELECT_LEX_UNIT *u)
395 396
{
  DBUG_ENTER("multi_update::prepare");
397
  unit= u;
398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415
  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);
      }
    }
  }
416 417 418 419 420 421 422
  /*
    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
  */
423 424 425 426 427 428 429 430 431

  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;
432 433
    for (table_ref=update_tables;  table_ref;
	 table_ref=table_ref->next, counter++)
434
    {
435
      if (table_ref->table == item->field->table)
436
      {
437 438 439 440 441 442 443 444 445 446 447
	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;
	}
448 449 450 451 452
	break;
      }
    }
    if (!table_ref)
    {
453
      net_printf(thd, ER_NOT_SUPPORTED_YET, "JOIN SYNTAX WITH MULTI-TABLE UPDATES");
454 455 456 457 458
      DBUG_RETURN(1);
    }
    else
      *int_ptr++=counter;
  }
459
  if (!num_updated--)
460
  {
461
    net_printf(thd, ER_NOT_SUPPORTED_YET, "SET CLAUSE MUST CONTAIN TABLE.FIELD REFERENCE");
462 463 464
    DBUG_RETURN(1);
  }

465 466 467 468
  /*
    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.
  */
469
  if (num_updated)
470
  {
471 472 473
    tmp_tables = (TABLE **) sql_calloc(sizeof(TABLE *) * num_updated);
    infos = (COPY_INFO *) sql_calloc(sizeof(COPY_INFO) * num_updated);
    fields_by_tables = (List_item **)sql_calloc(sizeof(List_item *) * (num_updated + 1));
474 475 476 477 478 479
    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;
480
      // Here we have to add row offset as an additional field ...
481 482 483 484 485 486 487 488 489 490 491 492 493 494
      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)
      {
495
	Field_string offset(table_ref->table->file->ref_length, false,
496
                            "offset", table_ref->table, my_charset_bin);
497
	temp_fields->push_front(new Item_field(((Field *)&offset)));
monty@narttu.mysql.fi's avatar
monty@narttu.mysql.fi committed
498 499

	// Make a temporary table
500 501 502 503
	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;
504 505 506
	if (!(tmp_tables[cnt]=create_tmp_table(thd, &tmp_table_param,
					       *temp_fields,
					       (ORDER*) 0, 1, 0, 0,
507 508
					       TMP_TABLE_ALL_COLUMNS,
					       unit)))
509 510 511 512 513 514 515 516 517 518 519 520 521
	{
	  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++;
    }
  }
522
  init_ftfuncs(thd, thd->lex.current_select->select_lex(), 1);
523 524 525 526 527 528 529 530
  error = 0; // Timestamps do not need to be restored, so far ...
  DBUG_RETURN(0);
}


void
multi_update::initialize_tables(JOIN *join)
{
531
#ifdef NOT_YET
532
   We skip it as it only makes a mess ...........
533 534 535 536 537 538 539 540 541 542 543 544
  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)
    {
545
//       We are going to update from this table 
546
       TABLE *tbl=walk->table=tab->table;
547
       /* Don't use KEYREAD optimization on this table */
548 549
       tbl->no_keyread=1;
       walk=walk->next;
550 551
    }
  }
552
#endif
553 554 555 556 557 558 559 560 561 562 563
}


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;
564
    table->no_keyread=0;
565 566 567 568
    if (error) 
      table->time_stamp=save_time_stamps[counter];
  }
  if (tmp_tables)
569
    for (uint counter = 0; counter < num_updated; counter++)
570 571 572 573 574 575 576 577 578 579
      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();
580
  // We have skipped fields ....
581
  if (!num_updated)
582 583 584 585 586 587 588 589 590 591 592 593
  {
    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]);
594
      // Only one table being updated receives a completely different treatment
595 596 597 598 599 600 601 602
      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++;
603
      table->file->extra(HA_EXTRA_NO_CACHE);
604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630
      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);
      }
631
      // Here I am breaking values as per each table    
632 633 634 635 636 637 638 639 640
      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])))
641
	{
642
	  updated++;
643 644
	  table->file->extra(HA_EXTRA_NO_CACHE);
	}
645 646 647 648 649 650 651 652 653
	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
654 655
	// Here we insert into each temporary table
	values_by_table.push_front(new Item_string((char*) table->file->ref,
656 657
						   table->file->ref_length,
						   system_charset_info));
658
	fill_record(tmp_tables[secure_counter]->field,values_by_table);
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
659 660
	error= write_record(tmp_tables[secure_counter],
			    &(infos[secure_counter]));
661 662 663 664 665 666 667 668 669 670 671 672 673
	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
674 675

  //TODO error should be sent at the query processing end
676
  /* First send error what ever it is ... */
677
  ::send_error(thd,errcode,err);
678 679

  /* reset used flags */
680
  //  update_tables->table->no_keyread=0;
681 682 683 684

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

686
  /* Something already updated so we have to invalidate cache */
687 688
  query_cache_invalidate3(thd, update_tables, 1);

689 690 691 692 693 694 695 696 697 698 699
  /* 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() &&
700
       table_being_updated == update_tables) || !not_trans_safe)
701
    ha_rollback_stmt(thd);
702
  else if (do_update && num_updated)
703 704 705 706 707 708
    VOID(do_updates(true));
}


int multi_update::do_updates (bool from_send_error)
{
709
  int local_error= 0, counter= 0;
710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733

  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))
    {
734
      local_error=1;
735 736 737 738
      break;
    }
    List<Item> list;
    Field **ptr=tmp_table->field,*field;
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
739
    // This is supposed to be something like insert_fields
740 741 742 743 744 745 746 747 748 749
    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;
750 751 752 753 754 755
    local_error=0;
    list.pop();			// we get position some other way ...
    local_error = tmp_table->file->rnd_init(1);
    if (local_error) 
      return local_error;
    while (!(local_error=tmp_table->file->rnd_next(tmp_table->record[0])) &&
756 757 758
	   (!thd->killed ||  from_send_error || not_trans_safe))
    {
      found++; 
759 760 761 762
      local_error= table->file->rnd_pos(table->record[0],
					(byte*) (*(tmp_table->field))->ptr);
      if (local_error)
	return local_error;
763 764
      table->status|= STATUS_UPDATED;
      store_record(table,1); 
765 766 767 768
      local_error= (fill_record(*fields_by_tables[counter + 1],list) ||
		    /* compare_record(table, query_id) || */
		    table->file->update_row(table->record[1],table->record[0]));
      if (local_error)
769
      {
770
	table->file->print_error(local_error,MYF(0));
771 772 773 774 775
	break;
      }
      else
	updated++;
    }
776 777
    if (local_error == HA_ERR_END_OF_FILE)
      local_error = 0;
778
  }
779
  return local_error;
780 781 782
}


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

785 786
bool multi_update::send_eof()
{
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
787
  thd->proc_info="updating the  reference tables";
788 789

  /* Does updates for the last n - 1 tables, returns 0 if ok */
790
  int local_error = (num_updated) ? do_updates(false) : 0;
791 792

  /* reset used flags */
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
793 794 795
#ifndef NOT_USED
  update_tables->table->no_keyread=0;
#endif
796 797 798
  if (local_error == -1)
    local_error= 0;
  thd->proc_info= "end";
monty@mashka.mysql.fi's avatar
monty@mashka.mysql.fi committed
799
  // TODO:  Error should be sent at the query processing end
800 801
  if (local_error)
    send_error(local_error, "An error occured in multi-table update");
802

monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
803 804 805 806 807 808
  /*
    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.
  */
809

810
  if (updated || not_trans_safe)
811 812
  {
    mysql_update_log.write(thd,thd->query,thd->query_length);
813
    Query_log_event qinfo(thd, thd->query, thd->query_length, 0);
814

monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
815 816 817 818
    /*
      mysql_bin_log is not open if binlogging or replication
      is not used
    */
819 820

    if (mysql_bin_log.is_open() &&  mysql_bin_log.write(&qinfo) &&
821
	!not_trans_safe)
822
      local_error=1;  /* Log write failed: roll back the SQL statement */
823 824

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