sql/sql_insert.cc (1,828 lines of code) (raw):

/* Copyright (c) 2000, 2025, Oracle and/or its affiliates. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License, version 2.0, as published by the Free Software Foundation. This program is designed to work with certain software (including but not limited to OpenSSL) that is licensed under separate terms, as designated in a particular file or component or in included license documentation. The authors of MySQL hereby grant you an additional permission to link the program and your derivative works with the separately licensed software that they have either included with the program or referenced in the documentation. 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, version 2.0, 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 */ /* Insert of records */ #include "sql/sql_insert.h" #include <errno.h> #include <stdio.h> #include <string.h> #include <atomic> #include <iterator> #include <map> #include <utility> #include "field_types.h" #include "lex_string.h" #include "my_alloc.h" #include "my_base.h" #include "my_bitmap.h" #include "my_dbug.h" #include "my_psi_config.h" #include "my_sys.h" #include "my_table_map.h" #include "my_thread_local.h" #include "mysql/components/services/bits/psi_bits.h" #include "mysql/mysql_lex_string.h" #include "mysql/psi/mysql_table.h" // IWYU pragma: keep #include "mysql/service_mysql_alloc.h" #include "mysql/strings/m_ctype.h" #include "mysql/udf_registration_types.h" #include "mysql_com.h" #include "mysqld_error.h" #include "pfs_table_provider.h" #include "prealloced_array.h" #include "sql/auth/auth_acls.h" #include "sql/auth/auth_common.h" // check_grant_all_columns #include "sql/binlog.h" #include "sql/create_field.h" #include "sql/dd/cache/dictionary_client.h" #include "sql/dd/dd.h" // dd::get_dictionary #include "sql/dd/dictionary.h" // dd::Dictionary #include "sql/dd_sql_view.h" // update_referencing_views_metadata #include "sql/debug_sync.h" // DEBUG_SYNC #include "sql/derror.h" // ER_THD #include "sql/discrete_interval.h" #include "sql/field.h" #include "sql/handler.h" #include "sql/item.h" #include "sql/key.h" #include "sql/lock.h" // mysql_unlock_tables #include "sql/locked_tables_list.h" #include "sql/mdl.h" #include "sql/mysqld.h" // stage_update #include "sql/nested_join.h" #include "sql/opt_explain.h" // Modification_plan #include "sql/opt_explain_format.h" #include "sql/partition_info.h" // partition_info #include "sql/protocol.h" #include "sql/query_options.h" #include "sql/rpl_rli.h" // Relay_log_info #include "sql/select_lex_visitor.h" #include "sql/sql_alter.h" #include "sql/sql_array.h" #include "sql/sql_base.h" // setup_fields #include "sql/sql_class.h" #include "sql/sql_const.h" #include "sql/sql_error.h" #include "sql/sql_gipk.h" #include "sql/sql_lex.h" #include "sql/sql_list.h" #include "sql/sql_resolver.h" // validate_gc_assignment #include "sql/sql_select.h" // check_privileges_for_list #include "sql/sql_show.h" // store_create_info #include "sql/sql_table.h" // quick_rm_table #include "sql/sql_view.h" // check_key_in_view #include "sql/stateless_allocator.h" #include "sql/system_variables.h" #include "sql/table_trigger_dispatcher.h" // Table_trigger_dispatcher #include "sql/thd_raii.h" #include "sql/transaction.h" // trans_commit_stmt #include "sql/transaction_info.h" #include "sql/trigger_def.h" #include "sql/visible_fields.h" #include "sql_string.h" #include "string_with_len.h" #include "template_utils.h" #include "thr_lock.h" namespace dd { class Table; } // namespace dd static bool check_view_insertability(THD *thd, Table_ref *view, const Table_ref *insert_table_ref); static void prepare_for_positional_update(TABLE *table, Table_ref *tables); /** Check that insert fields are from a single table of a multi-table view. @param fields The insert fields to be checked. @param view The view for insert. @param [out] insert_table_ref Reference to table to insert into This function is called to check that the fields being inserted into are from a single base table. This must be checked when the table to be inserted into is a multi-table view. @return false if success, true if an error was raised. */ static bool check_single_table_insert(const mem_root_deque<Item *> &fields, Table_ref *view, Table_ref **insert_table_ref) { // It is join view => we need to find the table for insert *insert_table_ref = nullptr; // reset for call to check_single_table() table_map tables = 0; for (Item *item : fields) tables |= item->used_tables(); if (view->check_single_table(insert_table_ref, tables)) { my_error(ER_VIEW_MULTIUPDATE, MYF(0), view->db, view->table_name); return true; } assert(*insert_table_ref && (*insert_table_ref)->is_insertable()); return false; } /** Check insert fields. @param thd The current thread. @param table_list The table for insert. @param fields The insert fields. @return false if success, true if error Resolved reference to base table is returned in lex->insert_table_leaf. */ static bool check_insert_fields(THD *thd, Table_ref *table_list, mem_root_deque<Item *> *fields) { LEX *const lex = thd->lex; #ifndef NDEBUG Table_ref *const saved_insert_table_leaf = lex->insert_table_leaf; #endif TABLE *table = table_list->table; assert(table_list->is_insertable()); if (fields->empty()) { /* No field list supplied, but a value list has been supplied. Use field list of table being updated. */ assert(table); // This branch is not reached with a view: lex->insert_table_leaf = table_list; Field_iterator_table_ref it; it.set(table_list); if (check_grant_all_columns(thd, INSERT_ACL, &it)) return true; for (it.set(table_list); !it.end_of_fields(); it.next()) { if (it.field()->is_hidden()) continue; Item *item = it.create_item(thd); if (item == nullptr) return true; fields->push_back(item); } } else { // INSERT with explicit field list. Query_block *query_block = thd->lex->query_block; Name_resolution_context *context = &query_block->context; Name_resolution_context_state ctx_state; /* Save the state of the current name resolution context. */ ctx_state.save_state(context, table_list); /* Perform name resolution only in the first table - 'table_list', which is the table that is inserted into. */ table_list->next_local = nullptr; context->resolve_in_table_list_only(table_list); const bool res = setup_fields(thd, /*want_privilege=*/INSERT_ACL, /*allow_sum_func=*/false, /*split_sum_funcs=*/false, /*column_update=*/true, /*typed_items=*/nullptr, fields, Ref_item_array()); /* Restore the current context. */ ctx_state.restore_state(context, table_list); if (res) return true; if (table_list->is_merged()) { if (check_single_table_insert(*fields, table_list, &lex->insert_table_leaf)) return true; table = lex->insert_table_leaf->table; } else { lex->insert_table_leaf = table_list; } // We currently don't check for unique columns when inserting via a view. const bool check_unique = !table_list->is_view(); if (check_unique && bitmap_bits_set(table->write_set) < fields->size()) { for (auto i = fields->cbegin(); i != fields->cend(); ++i) { // Skipping views means that we only have FIELD_ITEM. const Item *item1 = *i; for (auto j = std::next(i); j != fields->cend(); ++j) { const Item *item2 = *j; if (item1->eq(item2)) { my_error(ER_FIELD_SPECIFIED_TWICE, MYF(0), item1->item_name.ptr()); return true; } } } // A duplicate column name should have been found by now. assert(false); } } /* Mark all generated columns for write*/ if (table->vfield) table->mark_generated_columns(false); if (check_key_in_view(thd, table_list, lex->insert_table_leaf) || (table_list->is_view() && check_view_insertability(thd, table_list, lex->insert_table_leaf))) { my_error(ER_NON_INSERTABLE_TABLE, MYF(0), table_list->alias, "INSERT"); return true; } assert(saved_insert_table_leaf == nullptr || lex->insert_table_leaf == saved_insert_table_leaf); return false; } /** Check that table references are restricted to the supplied table map. The check can be skipped if the supplied table is a base table. @param view Table being specified @param values Values whose used tables are to be matched against table map @param map Table map to match against @return false if success, true if error */ static bool check_valid_table_refs(const Table_ref *view, const mem_root_deque<Item *> &values, table_map map) { if (!view->is_view()) // Ignore check if called with base table. return false; map |= PSEUDO_TABLE_BITS; for (Item *item : values) { if (item->used_tables() & ~map) { my_error(ER_VIEW_MULTIUPDATE, MYF(0), view->db, view->table_name); return true; } } return false; } /** Validates default value of fields which are not specified in the column list of INSERT statement. @note table->record[0] should be be populated with default values before calling this function. @param thd thread context @param table table to which values are inserted. @returns false if success, true if error */ bool validate_default_values_of_unset_fields(THD *thd, TABLE *table) { MY_BITMAP *write_set = table->write_set; DBUG_TRACE; for (Field **field = table->field; *field; field++) { if (!bitmap_is_set(write_set, (*field)->field_index()) && !(*field)->is_flag_set(NO_DEFAULT_VALUE_FLAG)) { if ((*field)->validate_stored_val(thd) && thd->is_error()) return true; } } return false; } /** Prepare triggers for INSERT-like statement. @param thd Thread handler @param table Table to which insert will happen @note Prepare triggers for INSERT-like statement by marking fields used by triggers and inform handlers that batching of UPDATE/DELETE cannot be done if there are BEFORE UPDATE/DELETE triggers. */ void prepare_triggers_for_insert_stmt(THD *thd, TABLE *table) { if (table->triggers) { if (table->triggers->has_triggers(TRG_EVENT_DELETE, TRG_ACTION_AFTER)) { /* The table has AFTER DELETE triggers that might access to subject table and therefore might need delete to be done immediately. So we turn-off the batching. */ (void)table->file->ha_extra(HA_EXTRA_DELETE_CANNOT_BATCH); } if (table->triggers->has_triggers(TRG_EVENT_UPDATE, TRG_ACTION_AFTER)) { /* The table has AFTER UPDATE triggers that might access to subject table and therefore might need update to be done immediately. So we turn-off the batching. */ (void)table->file->ha_extra(HA_EXTRA_UPDATE_CANNOT_BATCH); } } table->mark_columns_needed_for_insert(thd); } /** Setup data for field BLOB/GEOMETRY field types for execution of "INSERT...UPDATE" statement. For a expression in 'UPDATE' clause like "a= VALUES(a)", let as call Field* referring 'a' as LHS_FIELD and Field* referring field 'a' in "VALUES(a)" as RHS_FIELD This function creates a separate copy of the blob value for RHS_FIELD, if the field is updated as well as accessed through VALUES() function in 'UPDATE' clause of "INSERT...UPDATE" statement. @param [in] thd Pointer to THD object. @param [in] fields List of fields representing LHS_FIELD of all expressions in 'UPDATE' clause. @param [in] mem_root MEM_ROOT for blob copy. @return - Can fail only when we are out of memory. @retval false Success @retval true Failure */ static bool mysql_prepare_blob_values(THD *thd, const mem_root_deque<Item *> &fields, MEM_ROOT *mem_root) { DBUG_TRACE; if (fields.size() <= 1) return false; // Collect LHS_FIELD's which are updated in a 'set'. // This 'set' helps decide if we need to make copy of BLOB value // or not. Prealloced_array<Field_blob *, 16> blob_update_field_set( PSI_NOT_INSTRUMENTED); if (blob_update_field_set.reserve(fields.size())) return true; for (Item *fld : fields) { Item_field *field = fld->field_for_view_update(); Field *lhs_field = field->field; if (lhs_field->type() == MYSQL_TYPE_BLOB || lhs_field->type() == MYSQL_TYPE_VECTOR || lhs_field->type() == MYSQL_TYPE_GEOMETRY) blob_update_field_set.insert_unique(down_cast<Field_blob *>(lhs_field)); } // Traverse through thd->lex->insert_update_values_map // and make copy of BLOB values in RHS_FIELD, if the same field is // modified (present in above 'set' prepared). if (thd->lex->has_values_map()) { std::map<Item_field *, Field *>::iterator iter; for (iter = thd->lex->begin_values_map(); iter != thd->lex->end_values_map(); ++iter) { // Retrieve the Field_blob pointers from the map. // and initialize newly declared variables immediately. Field_blob *lhs_field = down_cast<Field_blob *>(iter->first->field); Field_blob *rhs_field = down_cast<Field_blob *>(iter->second); // Check if the Field_blob object is updated before making a copy. if (blob_update_field_set.count_unique(lhs_field) == 0) continue; // Copy blob value if (rhs_field->copy_blob_value(mem_root)) return true; } } return false; } bool Sql_cmd_insert_base::precheck(THD *thd) { DBUG_TRACE; /* Check that we have modify privileges for the first table and select privileges for the rest */ Access_bitmask privilege = INSERT_ACL | (duplicates == DUP_REPLACE ? DELETE_ACL : 0) | (update_value_list.empty() ? 0 : UPDATE_ACL); if (check_one_table_access(thd, privilege, lex->query_tables)) return true; return false; } bool Sql_cmd_insert_base::check_privileges(THD *thd) { DBUG_TRACE; if (check_all_table_privileges(thd)) return (true); if (check_privileges_for_list(thd, insert_field_list, INSERT_ACL)) return true; if (values_need_privilege_check) { for (List_item *values : insert_many_values) { if (check_privileges_for_list(thd, *values, SELECT_ACL)) return true; } } if (duplicates == DUP_UPDATE) { if (check_privileges_for_list(thd, update_field_list, UPDATE_ACL)) return true; if (check_privileges_for_list(thd, update_value_list, SELECT_ACL)) return true; } for (Query_block *sl = lex->unit->first_query_block(); sl; sl = sl->next_query_block()) { if (sl->check_column_privileges(thd)) return true; } return false; } /** Insert one or more rows from a VALUES list into a table @param thd thread handler @returns false if success, true if error */ bool Sql_cmd_insert_values::execute_inner(THD *thd) { DBUG_TRACE; assert(thd->lex->sql_command == SQLCOM_REPLACE || thd->lex->sql_command == SQLCOM_INSERT); /* We have three alternative syntax rules for the INSERT statement: 1) "INSERT (columns) VALUES ...", so non-listed columns need a default 2) "INSERT VALUES (), ..." so all columns need a default; note that "VALUES (),(expr_1, ..., expr_n)" is not allowed, so checking emptiness of the first row is enough 3) "INSERT VALUES (expr_1, ...), ..." so no defaults are needed; even if expr_i is "DEFAULT" (in which case the column is set by Item_default_value::save_in_field_inner()). */ const bool manage_defaults = column_count > 0 || // 1) value_count == 0; // 2) COPY_INFO info(COPY_INFO::INSERT_OPERATION, &insert_field_list, manage_defaults, duplicates); COPY_INFO update(COPY_INFO::UPDATE_OPERATION, &update_field_list, &update_value_list); Query_block *const query_block = lex->query_block; Table_ref *const table_list = lex->insert_table; TABLE *const insert_table = lex->insert_table_leaf->table; if (duplicates == DUP_UPDATE || duplicates == DUP_REPLACE) prepare_for_positional_update(insert_table, table_list); /* Must be done before can_prune_insert, due to internal initialization. */ if (info.add_function_default_columns(insert_table, insert_table->write_set)) return true; /* purecov: inspected */ if (duplicates == DUP_UPDATE && update.add_function_default_columns( insert_table, insert_table->write_set)) return true; /* purecov: inspected */ // Current error state inside and after the insert loop bool has_error = false; const bool select_insert = insert_many_values.empty(); /* Prune after locking if pruning was not completed in prepare phase. */ if (!select_insert && insert_table->part_info != nullptr && !insert_table->part_info->is_pruning_completed) { MY_BITMAP used_partitions; bool prune_needs_default_values = false; enum partition_info::enum_can_prune can_prune_partitions = partition_info::PRUNE_NO; if (insert_table->part_info->can_prune_insert( thd, duplicates, update, update_field_list, insert_field_list, value_count == 0, &can_prune_partitions, &prune_needs_default_values, &used_partitions)) { return true; /* purecov: inspected */ } if (can_prune_partitions != partition_info::PRUNE_NO) { if (prune_partitions(thd, prune_needs_default_values, insert_field_list, &used_partitions, insert_table, info, &can_prune_partitions, /*tables_locked*/ true)) { return true; } } } { // Statement plan is available within these braces const Modification_plan plan( thd, (lex->sql_command == SQLCOM_INSERT) ? MT_INSERT : MT_REPLACE, insert_table, nullptr, false, 0); DEBUG_SYNC(thd, "planned_single_insert"); if (lex->is_explain()) { const bool err = explain_single_table_modification(thd, thd, &plan, query_block); return err; } insert_table->next_number_field = insert_table->found_next_number_field; THD_STAGE_INFO(thd, stage_update); if (duplicates == DUP_REPLACE && (!insert_table->triggers || !insert_table->triggers->has_delete_triggers())) insert_table->file->ha_extra(HA_EXTRA_WRITE_CAN_REPLACE); if (duplicates == DUP_UPDATE) insert_table->file->ha_extra(HA_EXTRA_INSERT_WITH_UPDATE); /* let's *try* to start bulk inserts. It won't necessary start them as insert_many_values.elements should be greater than some - handler dependent - threshold. We should not start bulk inserts if this statement uses functions or invokes triggers since they may access to the same table and therefore should not see its inconsistent state created by this optimization. So we call start_bulk_insert to perform nesessary checks on insert_many_values.elements, and - if nothing else - to initialize the code to make the call of end_bulk_insert() below safe. */ if (duplicates != DUP_ERROR || lex->is_ignore()) insert_table->file->ha_extra(HA_EXTRA_IGNORE_DUP_KEY); /* This is a simple check for the case when the table has a trigger that reads from it, or when the statement invokes a stored function that reads from the table being inserted to. Engines can't handle a bulk insert in parallel with a read form the same table in the same connection. */ if (thd->locked_tables_mode <= LTM_LOCK_TABLES) insert_table->file->ha_start_bulk_insert(insert_many_values.size()); prepare_triggers_for_insert_stmt(thd, insert_table); /* Count warnings for all inserts. For single row insert, generate an error if trying to set a NOT NULL field to NULL. Notice that policy must be reset before leaving this function. */ thd->check_for_truncated_fields = ((insert_many_values.size() == 1 && !lex->is_ignore()) ? CHECK_FIELD_ERROR_FOR_NULL : CHECK_FIELD_WARN); thd->num_truncated_fields = 0L; for (Field **next_field = insert_table->field; *next_field; ++next_field) { (*next_field)->reset_warnings(); } for (const List_item *values : insert_many_values) { Autoinc_field_has_explicit_non_null_value_reset_guard after_each_row( insert_table); restore_record(insert_table, s->default_values); // Get empty record /* Check whether default values of the insert_field_list not specified in column list are correct or not. */ if (validate_default_values_of_unset_fields(thd, insert_table)) { has_error = true; break; } if (fill_record_n_invoke_before_triggers( thd, &info, insert_field_list, *values, insert_table, TRG_EVENT_INSERT, insert_table->s->fields, true, nullptr)) { assert(thd->is_error()); /* TODO: Convert warnings to errors if values_list.elements == 1 and check that all items return warning in case of problem with storing field. */ has_error = true; break; } if (check_that_all_fields_are_given_values(thd, insert_table, table_list)) { assert(thd->is_error()); has_error = true; break; } const int check_result = table_list->view_check_option(thd); if (check_result == VIEW_CHECK_SKIP) continue; else if (check_result == VIEW_CHECK_ERROR) { has_error = true; break; } if (invoke_table_check_constraints(thd, insert_table)) { if (thd->is_error()) { has_error = true; break; } // continue when IGNORE clause is used. continue; } if (write_record(thd, insert_table, &info, &update)) { has_error = true; break; } thd->get_stmt_da()->inc_current_row_for_condition(); } } // Statement plan is available within these braces assert(has_error == thd->get_stmt_da()->is_error()); /* Now all rows are inserted. Time to update logs and sends response to user */ { /* TODO: Only call this if insert_table->found_next_number_field.*/ insert_table->file->ha_release_auto_increment(); /* Make sure 'end_bulk_insert()' is called regardless of current error */ int loc_error = 0; if (thd->locked_tables_mode <= LTM_LOCK_TABLES) loc_error = insert_table->file->ha_end_bulk_insert(); /* Report error if 'end_bulk_insert()' failed, and set 'has_error' */ if (loc_error && !has_error) { /* purecov: begin inspected */ myf error_flags = MYF(0); if (insert_table->file->is_fatal_error(loc_error)) error_flags |= ME_FATALERROR; insert_table->file->print_error(loc_error, error_flags); has_error = true; /* purecov: end */ } const bool transactional_table = insert_table->file->has_transactions(); const bool changed [[maybe_unused]] = info.stats.copied || info.stats.deleted || info.stats.updated; if (!has_error || thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT)) { if (mysql_bin_log.is_open()) { int errcode = 0; if (!has_error) { /* [Guilhem wrote] Temporary errors may have filled thd->net.last_error/errno. For example if there has been a disk full error when writing the row, and it was MyISAM, then thd->net.last_error/errno will be set to "disk full"... and the mysql_file_pwrite() will wait until free space appears, and so when it finishes then the write_row() was entirely successful */ /* todo: consider removing */ thd->clear_error(); } else errcode = query_error_code(thd, thd->killed == THD::NOT_KILLED); /* bug#22725: A query which per-row-loop can not be interrupted with KILLED, like INSERT, and that does not invoke stored routines can be binlogged with neglecting the KILLED error. If there was no error (has_error == false) until after the end of inserting loop the KILLED flag that appeared later can be disregarded since previously possible invocation of stored routines did not result in any error due to the KILLED. In such case the flag is ignored for constructing binlog event. */ if (thd->binlog_query(THD::ROW_QUERY_TYPE, thd->query().str, thd->query().length, transactional_table, false, false, errcode)) has_error = true; } } assert( transactional_table || !changed || thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT)); } /* We'll report to the client this id: - if the table contains an autoincrement column and we successfully inserted an autogenerated value, the autogenerated value. - if the table contains no autoincrement column and LAST_INSERT_ID(X) was called, X. - if the table contains an autoincrement column, and some rows were inserted, the id of the last "inserted" row (if IGNORE, that value may not have been really inserted but ignored). */ const ulonglong id = (thd->first_successful_insert_id_in_cur_stmt > 0) ? thd->first_successful_insert_id_in_cur_stmt : (thd->arg_of_last_insert_id_function ? thd->first_successful_insert_id_in_prev_stmt : ((insert_table->next_number_field && info.stats.copied) ? insert_table->next_number_field->val_int() : 0)); insert_table->next_number_field = nullptr; // Remember to restore warning handling before leaving thd->check_for_truncated_fields = CHECK_FIELD_IGNORE; assert(has_error == thd->get_stmt_da()->is_error()); if (has_error) return true; if (insert_many_values.size() == 1 && (!(thd->variables.option_bits & OPTION_WARNINGS) || !thd->num_truncated_fields)) { my_ok(thd, info.stats.copied + info.stats.deleted + (thd->get_protocol()->has_client_capability(CLIENT_FOUND_ROWS) ? info.stats.touched : info.stats.updated), id); } else { char buff[160]; const ha_rows updated = thd->get_protocol()->has_client_capability(CLIENT_FOUND_ROWS) ? info.stats.touched : info.stats.updated; if (lex->is_ignore()) snprintf(buff, sizeof(buff), ER_THD(thd, ER_INSERT_INFO), (long)info.stats.records, (long)(info.stats.records - info.stats.copied), (long)thd->get_stmt_da()->current_statement_cond_count()); else snprintf(buff, sizeof(buff), ER_THD(thd, ER_INSERT_INFO), (long)info.stats.records, (long)(info.stats.deleted + updated), (long)thd->get_stmt_da()->current_statement_cond_count()); my_ok(thd, info.stats.copied + info.stats.deleted + updated, id, buff); } /* If we have inserted into a VIEW, and the base table has AUTO_INCREMENT column, but this column is not accessible through a view, then we should restore LAST_INSERT_ID to the value it had before the statement. */ if (table_list->is_view() && !table_list->contain_auto_increment) thd->first_successful_insert_id_in_cur_stmt = thd->first_successful_insert_id_in_prev_stmt; DBUG_EXECUTE_IF("after_mysql_insert", { const char act[] = "now " "wait_for signal.continue"; assert(opt_debug_sync_timeout > 0); assert(!debug_sync_set_action(thd, STRING_WITH_LEN(act))); };); return false; } /** Additional check for insertability for VIEW A view is insertable if the following conditions are true: - All columns being inserted into are from a single table. - All not used columns in table have default values. - All columns in view are distinct (not referring to the same column). - All columns in view are insertable-into. @param thd thread handler @param[in,out] view reference to view being inserted into. view->contain_auto_increment is true if and only if the view contains an auto_increment field. @param insert_table_ref reference to underlying table being inserted into @retval false if success @retval true if table is not insertable-into (no error is reported) */ static bool check_view_insertability(THD *thd, Table_ref *view, const Table_ref *insert_table_ref) { DBUG_TRACE; const uint num = view->view_query()->query_block->num_visible_fields(); TABLE *const table = insert_table_ref->table; MY_BITMAP used_fields; const enum_mark_columns save_mark_used_columns = thd->mark_used_columns; const uint used_fields_buff_size = bitmap_buffer_size(table->s->fields); uint32 *const used_fields_buff = (uint32 *)thd->alloc(used_fields_buff_size); if (!used_fields_buff) return true; /* purecov: inspected */ assert(view->table == nullptr && table != nullptr && view->field_translation != nullptr); (void)bitmap_init(&used_fields, used_fields_buff, table->s->fields); bitmap_clear_all(&used_fields); view->contain_auto_increment = false; thd->mark_used_columns = MARK_COLUMNS_NONE; // No privilege checking is done for these columns const Column_privilege_tracker column_privilege(thd, 0); /* check simplicity and prepare unique test of view */ Field_translator *const trans_start = view->field_translation; Field_translator *const trans_end = trans_start + num; for (Field_translator *trans = trans_start; trans != trans_end; trans++) { if (trans->item == nullptr) continue; /* @todo This fix_fields() call is necessary for execution of prepared statements. When repeated preparation is eliminated the call can be deleted. */ if (!trans->item->fixed && trans->item->fix_fields(thd, &trans->item)) return true; /* purecov: inspected */ // Extract the underlying base table column, if there is one Item_field *const field = trans->item->field_for_view_update(); // No underlying base table column, view is not insertable-into if (field == nullptr) return true; if (field->field->auto_flags & Field::NEXT_NUMBER) view->contain_auto_increment = true; /* prepare unique test */ /* remove collation (or other transparent for update function) if we have it */ trans->item = field; } thd->mark_used_columns = save_mark_used_columns; /* unique test */ for (Field_translator *trans = trans_start; trans != trans_end; trans++) { if (trans->item == nullptr) continue; /* Thanks to test above, we know that all columns are of type Item_field */ Item_field *field = down_cast<Item_field *>(trans->item); /* check fields belong to table in which we are inserting */ if (field->field->table == table && bitmap_test_and_set(&used_fields, field->field->field_index())) return true; } return false; } /** Recursive helper function for resolving join conditions for insertion into view for prepared statements. @param thd Thread handler @param tr Table structure which is traversed recursively @return false if success, true if error */ static bool fix_join_cond_for_insert(THD *thd, Table_ref *tr) { if (tr->join_cond() && !tr->join_cond()->fixed) { const Column_privilege_tracker column_privilege(thd, SELECT_ACL); if (tr->join_cond()->fix_fields(thd, nullptr)) return true; /* purecov: inspected */ } if (tr->nested_join == nullptr) return false; for (Table_ref *ti : tr->nested_join->m_tables) { if (fix_join_cond_for_insert(thd, ti)) return true; /* purecov: inspected */ } return false; } /** Get extra info for tables we insert into @param table table(TABLE object) we insert into, might be NULL in case of view @param tables (Table_ref object) or view we insert into */ static void prepare_for_positional_update(TABLE *table, Table_ref *tables) { if (table) { table->prepare_for_position(); return; } assert(tables->is_view()); for (Table_ref *tbl : *tables->view_tables) { prepare_for_positional_update(tbl->table, tbl); } } static bool allocate_column_bitmap(THD *thd, TABLE *table, MY_BITMAP **bitmap) { DBUG_TRACE; const uint number_bits = table->s->fields; MY_BITMAP *the_struct; my_bitmap_map *the_bits; if (multi_alloc_root(thd->mem_root, &the_struct, sizeof(MY_BITMAP), &the_bits, bitmap_buffer_size(number_bits), nullptr) == nullptr) return true; if (bitmap_init(the_struct, the_bits, number_bits) != 0) return true; *bitmap = the_struct; return false; } bool Sql_cmd_insert_base::get_default_columns( THD *thd, TABLE *table, MY_BITMAP **m_function_default_columns) { if (allocate_column_bitmap(thd, table, m_function_default_columns)) { return true; } /* Find columns with function default on insert or update, mark them in bitmap. */ for (uint i = 0; i < table->s->fields; ++i) { Field *f = table->field[i]; // if it's a default expression if (f->has_insert_default_general_value_expression()) { bitmap_set_bit(*m_function_default_columns, f->field_index()); } } // Remove from map the fields that are explicitly specified bitmap_subtract(*m_function_default_columns, table->write_set); // If no bit left exit if (bitmap_is_clear_all(*m_function_default_columns)) return false; // For each default function that is used restore the flags for (uint i = 0; i < table->s->fields; ++i) { Field *f = table->field[i]; if (bitmap_is_set(*m_function_default_columns, i)) { assert(f->m_default_val_expr != nullptr); // restore binlog safety flags thd->lex->set_stmt_unsafe_flags( f->m_default_val_expr->get_stmt_unsafe_flags()); // Mark the columns the expression reads in the table's read_set for (uint j = 0; j < table->s->fields; j++) { if (bitmap_is_set(&f->m_default_val_expr->base_columns_map, j)) { bitmap_set_bit(table->read_set, j); } } } } return false; } /** Prepare items in INSERT statement @param thd Thread handler WARNING You MUST set table->insert_values to 0 after calling this function before releasing the table object. @return false if success, true if error */ bool Sql_cmd_insert_base::prepare_inner(THD *thd) { DBUG_TRACE; // Save original number of columns in insert list column_count = insert_field_list.size(); const bool select_insert = insert_many_values.empty(); // Number of update fields must match number of update values assert(update_field_list.size() == update_value_list.size()); Query_expression *const unit = lex->unit; Query_block *const select = lex->query_block; Name_resolution_context *const context = &select->context; Name_resolution_context_state ctx_state; Table_ref *const table_list = lex->query_tables; lex->insert_table = table_list; const bool insert_into_view = table_list->is_view(); /* Save the state of the current name resolution context. Should be done only when select_insert is true, but compiler does not realize that. */ ctx_state.save_state(context, table_list); DBUG_PRINT("enter", ("table_list %p, view %d", table_list, (int)insert_into_view)); // This flag is used only for INSERT, make sure it is clear lex->in_update_value_clause = false; // first_query_block_table is the first table after the table inserted into Table_ref *first_query_block_table = table_list->next_local; // Setup the insert table only table_list->next_local = nullptr; // The VALUES table should only be available from within the update // expressions (i.e. the rhs of ODKU updates). The context must be restored // before resolve_update_expressions for ODKU statements. Table_ref *next_name_resolution_table = table_list->next_name_resolution_table; table_list->next_name_resolution_table = nullptr; if (select->setup_tables(thd, table_list, select_insert)) return true; /* purecov: inspected */ if (insert_into_view) { // Allowing semi-join would transform this table into a "join view" if (table_list->resolve_derived(thd, false)) return true; if (select->merge_derived(thd, table_list)) return true; /* purecov: inspected */ /* Require proper privileges for all leaf tables of the view. @todo - Check for target table only. */ Access_bitmask privilege = INSERT_ACL | (duplicates == DUP_REPLACE ? DELETE_ACL : 0) | (update_value_list.empty() ? 0 : UPDATE_ACL); if (select->check_view_privileges(thd, privilege, privilege)) return true; /* On second preparation, we may need to resolve view condition generated when merging the view. */ if (!select->first_execution && table_list->is_merged() && fix_join_cond_for_insert(thd, table_list)) return true; /* purecov: inspected */ } /* Insertability test is spread across several places: - Target table or view must be insertable (checked below) - A view containing LIMIT has special key requirements (checked in check_insert_fields) - A view has special requirements with respect to columns being specified (checked in check_view_insertability) - All inserted columns must be from an insertable component of a view (checked in check_insert_fields) - For INSERT ... VALUES, target table must not be same as one selected from (checked in unique_table) */ if (!table_list->is_insertable()) { my_error(ER_NON_INSERTABLE_TABLE, MYF(0), table_list->alias, "INSERT"); return true; } if (insert_into_view && column_count == 0) { if (table_list->is_multiple_tables()) { my_error(ER_VIEW_NO_INSERT_FIELD_LIST, MYF(0), table_list->db, table_list->table_name); return true; } if (insert_view_fields(&insert_field_list, table_list)) return true; } // REPLACE for a JOIN view is not permitted. if (table_list->is_multiple_tables() && duplicates == DUP_REPLACE) { my_error(ER_VIEW_DELETE_MERGE_VIEW, MYF(0), table_list->db, table_list->table_name); return true; } if (duplicates == DUP_UPDATE) { // Must be allocated before Item::fix_fields() if (table_list->set_insert_values(thd->mem_root)) return true; /* purecov: inspected */ } /* INSERT ... VALUES () has no WHERE clause, HAVING clause, GROUP BY clause, ORDER BY clause nor LIMIT clause. Table list is empty, except when an alias name is given for VALUES, which is represented as a derived table. */ assert(select_insert || ((first_query_block_table == nullptr || first_query_block_table->is_derived()) && select->where_cond() == nullptr && select->group_list.elements == 0 && select->having_cond() == nullptr && !select->has_limit())); // Prepare the lists of columns and values in the statement. if (check_insert_fields(thd, table_list, &insert_field_list)) return true; lex->insert_table_leaf->set_inserted(); if (duplicates == DUP_REPLACE) lex->insert_table_leaf->set_deleted(); if (duplicates == DUP_UPDATE) lex->insert_table_leaf->set_updated(); TABLE *const insert_table = lex->insert_table_leaf->table; uint field_count = insert_field_list.size(); const table_map map = lex->insert_table_leaf->map(); uint value_list_counter = 0; for (const List_item *values : insert_many_values) { value_list_counter++; /* Values for all fields in table must be specified, unless there is no field list and no value list is supplied (means all default values). */ if (values->size() != field_count && !(values->empty() && column_count == 0)) { my_error(ER_WRONG_VALUE_COUNT_ON_ROW, MYF(0), value_list_counter); return true; } // Each set of values specified must have the same cardinality if (value_list_counter > 1 && value_count != values->size()) { my_error(ER_WRONG_VALUE_COUNT_ON_ROW, MYF(0), value_list_counter); return true; } // Assign value count in the Sql_cmd object value_count = values->size(); /* If only default values, insert column items are not needed. And fill_record() doesn't accept when the list of column items and of specified values to insert have different lengths. */ if (value_count == 0) insert_field_list.clear(); // The const_cast is fine, since we're sending split_sum_funcs = false. if (setup_fields(thd, /*want_privilege=*/SELECT_ACL, /*allow_sum_func=*/false, /*split_sum_funcs=*/false, /*column_update=*/false, &insert_field_list, const_cast<mem_root_deque<Item *> *>(values), Ref_item_array())) return true; for (Item *item : *values) { if (!item->const_for_execution()) values_need_privilege_check = true; } if (check_valid_table_refs(table_list, *values, map)) return true; /* purecov: inspected */ if ((insert_table->has_gcol() || insert_table->gen_def_fields_ptr != nullptr) && validate_gc_assignment(insert_field_list, *values, insert_table)) return true; } /* check_insert_fields() will usually mark all inserted columns in write_set, except when - an explicit column list is given, and a view is inserted into (fields from field_translation list have already been fixed in resolve_derived(), thus setup_fields() in check_insert_fields() will not process them), or - no columns where provided in field list (except when no values are given - this is a special case that implies that all columns are given default values, and default values are managed in a different manner, see COPY_INFO for details). */ if ((insert_into_view || column_count == 0) && (select_insert || value_count > 0)) bitmap_set_all(insert_table->write_set); MY_BITMAP *function_default_columns = nullptr; if (get_default_columns(thd, insert_table, &function_default_columns)) return true; // If an alias for VALUES is specified, prepare the derived values table. This // must be done after setting up the insert table to have access to insert // fields. If values_field_list is not empty, the table is already set up // (e.g. as part of a PREPARE statement). if (values_column_list != nullptr && values_field_list.empty() && prepare_values_table(thd)) return true; if (duplicates == DUP_UPDATE) { // Setup the columns to be updated if (setup_fields(thd, /*want_privilege=*/UPDATE_ACL, /*allow_sum_func=*/false, /*split_sum_funcs=*/false, /*column_update=*/true, /*typed_items=*/nullptr, &update_field_list, Ref_item_array())) return true; if (check_valid_table_refs(table_list, update_field_list, map)) return true; } if (table_list->is_merged()) { const Column_privilege_tracker column_privilege(thd, SELECT_ACL); if (table_list->prepare_check_option(thd)) return true; /* purecov: inspected */ if (duplicates == DUP_REPLACE && table_list->prepare_replace_filter(thd)) return true; /* purecov: inspected */ } /* In ON DUPLICATE KEY clause, it is possible to refer to fields from the selected tables also, if the query expression is not a VALUES clause, not a UNION and the query block is not explicitly grouped. This has implications if ON DUPLICATE KEY values contain subqueries, due to the way Query_block::apply_local_transforms() is called: it is usually triggered only on the outer-most query block. Such subqueries are attached to the last query block of the INSERT statement (relevant if this is an INSERT statement with a query expression containing UNION). If the query is INSERT VALUES, processing is quite simple: - resolve VALUES expressions (above) - resolve ON DUPLICATE KEY values with same name resolution context. - call apply_local_transforms() on outer query block. If the query is INSERT SELECT and the query expression contains UNION, processing is performed as follows: - resolve ON DUPLICATE KEY expressions with same name resolution context. In this case, it is OK to resolve any subqueries before the outer query block, because references from the expressions into the tables of the query expression are not allowed. - resolve the query expression with insert table excluded from name resolution context. This will implicitly call apply_local_transforms() on the outer query blocks and all subqueries in ON DUPLICATE KEY expressions, which are attached to the last query block of the UNION. If the query is INSERT SELECT and the query expression does not have a UNION, processing is performed as follows: - set skip_local_transforms for the outer query block to prevent apply_local_transforms() from being called. - resolve the query expression with insert table excluded from name resolution context. - if query block is not grouped, combine the name resolution context for the insert table and the query expression, so that ON DUPLICATE KEY expressions may refer to all those tables (otherwise restore name resolution context as insert table only). - resolve ON DUPLICATE KEY expressions. - call apply_local_transforms() on outer query block, which also contains references to any subqueries from ON DUPLICATE KEY expressions. */ if (!select_insert) { // Duplicate tables in subqueries in VALUES clause are not allowed. Table_ref *const duplicate = unique_table(lex->insert_table_leaf, table_list->next_global, true); if (duplicate != nullptr) { update_non_unique_table_error(table_list, "INSERT", duplicate); return true; } } else { ulong added_options = SELECT_NO_UNLOCK; // The result needs to be buffered if the target table is used somewhere // in other parts of query. // This is not an issue if a secondary engine is involved, as the target // table will always be in the primary engine, and the source table will // be in the secondary engine, so they are always different for this // particular case. if (unique_table(lex->insert_table_leaf, table_list->next_global, false) && thd->secondary_engine_optimization() != Secondary_engine_optimization::SECONDARY) { added_options |= OPTION_BUFFER_RESULT; } /* INSERT...SELECT...ON DUPLICATE KEY UPDATE/REPLACE SELECT/ INSERT...IGNORE...SELECT can be unsafe, unless ORDER BY PRIMARY KEY clause is used in SELECT statement. We therefore use row based logging if mixed or row based logging is available. TODO: Check if the order of the output of the select statement is deterministic. Waiting for BUG#42415 */ if (lex->sql_command == SQLCOM_INSERT_SELECT && duplicates == DUP_UPDATE) lex->set_stmt_unsafe(LEX::BINLOG_STMT_UNSAFE_INSERT_SELECT_UPDATE); if (lex->sql_command == SQLCOM_INSERT_SELECT && lex->is_ignore()) lex->set_stmt_unsafe(LEX::BINLOG_STMT_UNSAFE_INSERT_IGNORE_SELECT); if (lex->sql_command == SQLCOM_REPLACE_SELECT) lex->set_stmt_unsafe(LEX::BINLOG_STMT_UNSAFE_REPLACE_SELECT); result = new (thd->mem_root) Query_result_insert(table_list, &insert_field_list, &insert_field_list, &update_field_list, &update_value_list, duplicates); if (result == nullptr) return true; /* purecov: inspected */ if (unit->is_set_operation()) { /* Update values may not have references to SELECT tables, so it is safe to resolve them before the query expression. */ if (duplicates == DUP_UPDATE && resolve_update_expressions(thd)) return true; } else { /* Delay apply_local_transforms() call until query block and any attached subqueries have been resolved. */ select->skip_local_transforms = true; } // Remove the insert table from the first query block select->m_table_list.first = first_query_block_table; context->table_list = first_query_block_table; context->first_name_resolution_table = first_query_block_table; if (unit->prepare(thd, result, &insert_field_list, added_options, 0)) return true; /* Restore the insert table but not the name resolution context */ if (first_query_block_table != select->get_table_list()) { // If we have transformation of the top block table list // by Query_block::transform_grouped_to_derived, we must update: first_query_block_table = select->get_table_list(); ctx_state.update_next_local(first_query_block_table); } select->m_table_list.first = table_list; context->table_list = table_list; table_list->next_local = first_query_block_table; if (field_count != unit->num_visible_fields()) { my_error(ER_WRONG_VALUE_COUNT_ON_ROW, MYF(0), 1); return true; } if ((insert_table->has_gcol() || insert_table->gen_def_fields_ptr) && validate_gc_assignment(insert_field_list, *unit->get_unit_column_types(), insert_table)) return true; } // The insert table should be a separate name resolution context assert(table_list->next_name_resolution_table == nullptr); if (duplicates == DUP_UPDATE) { if (select_insert) { if (!unit->is_set_operation() && !select->is_grouped()) { /* Make one context out of the two separate name resolution contexts: the INSERT table and the tables in the SELECT part, by concatenating the two lists: */ table_list->next_name_resolution_table = context->first_name_resolution_table; context->first_name_resolution_table = table_list; } else { // Restore the original name resolution context (the insert table) ctx_state.restore_state(context, table_list); } } else { // Resolve Item_insert_values _after_ their corresponding fields have been // fixed, as their field arguments should be resolved as normal. if (values_table != nullptr) resolve_values_table_columns(thd); // Restore the derived reference table that points to VALUES. table_list->next_name_resolution_table = next_name_resolution_table; } if (!unit->is_set_operation() && resolve_update_expressions(thd)) return true; } if (insert_table->triggers) { /* We don't need to mark columns which are used by ON DELETE and ON UPDATE triggers, which may be invoked in case of REPLACE or INSERT ... ON DUPLICATE KEY UPDATE, since before doing actual row replacement or update write_record() will mark all table fields as used. */ if (insert_table->triggers->mark_fields(TRG_EVENT_INSERT)) return true; } if (!unit->is_set_operation() && select->apply_local_transforms(thd, false)) return true; /* purecov: inspected */ if (select_insert) { // Restore the insert table and the name resolution context select->m_table_list.first = table_list; context->table_list = table_list; table_list->next_local = first_query_block_table; ctx_state.restore_state(context, table_list); } if (!select_insert && insert_table->part_info) { enum partition_info::enum_can_prune can_prune_partitions = partition_info::PRUNE_NO; /* We have three alternative syntax rules for the INSERT statement: 1) "INSERT (columns) VALUES ...", so non-listed columns need a default 2) "INSERT VALUES (), ..." so all columns need a default; note that "VALUES (),(expr_1, ..., expr_n)" is not allowed, so checking emptiness of the first row is enough 3) "INSERT VALUES (expr_1, ...), ..." so no defaults are needed; even if expr_i is "DEFAULT" (in which case the column is set by Item_default_value::save_in_field_inner()). */ const bool manage_defaults = column_count > 0 || // 1) value_count == 0; // 2) COPY_INFO info(COPY_INFO::INSERT_OPERATION, &insert_field_list, manage_defaults, duplicates); COPY_INFO update(COPY_INFO::UPDATE_OPERATION, &update_field_list, &update_value_list); // @todo FIX THIS /* Must be done before can_prune_insert, due to internal initialization. */ if (info.add_function_default_columns(insert_table, insert_table->write_set)) return true; /* purecov: inspected */ if (duplicates == DUP_UPDATE && update.add_function_default_columns( insert_table, insert_table->write_set)) return true; /* purecov: inspected */ MY_BITMAP used_partitions; bool prune_needs_default_values = false; if (insert_table->part_info->can_prune_insert( thd, duplicates, update, update_field_list, insert_field_list, value_count == 0, &can_prune_partitions, &prune_needs_default_values, &used_partitions)) { return true; /* purecov: inspected */ } if (can_prune_partitions != partition_info::PRUNE_NO) { if (prune_partitions(thd, prune_needs_default_values, insert_field_list, &used_partitions, insert_table, info, &can_prune_partitions, /*tables_locked*/ false)) { return true; } } } if (!select_insert) { unit->set_prepared(); } else { if (!is_regular()) { for (Table_ref *tr = lex->insert_table->first_leaf_table(); tr != nullptr; tr = tr->next_leaf) if (tr->save_properties()) return true; } } assert(CountHiddenFields(insert_field_list) == 0); return false; } /** Prepare the derived table created as a VALUES alias. When an optional table alias is specified for the VALUES of an INSERT statement, we create a derived table to contain its values. The field translation of this derived table is pointed to the insert buffer of the table we are inserting into, by means of the Item_insert_value objects we create here. An Item_insert_value object will originally contain the corresponding Item_field of the insert table, which is then cloned during its fix_field implementation to move the underlying Field to insert_values instead. The derived table is initialized in Sql_cmd_insert_base::make_cmd, but we have to wait until after the insert table is resolved to create our field translation indirection. If the table alias is not given explicit column names in the query, it must take the column names of the insert table; this information is not available to us until the insert table is resolved. @param thd Thread context. @returns false if success, true if error. */ bool Sql_cmd_insert_base::prepare_values_table(THD *thd) { // Insert_item_value items will be allocated for the field_translation of the // VALUES table. They should be persistent for prepared statements. const Prepared_stmt_arena_holder ps_arena_holder(thd); if (insert_field_list.empty()) { Table_ref *insert_table = lex->query_block->get_table_list(); Field_iterator_table_ref it; it.set(insert_table); for (it.set(insert_table); !it.end_of_fields(); it.next()) { if (it.field() != nullptr && it.field()->is_hidden_by_system()) continue; Item *item = it.create_item(thd); if (item == nullptr) return true; values_field_list.push_back(down_cast<Item_field *>(item->real_item())); } } else { for (Item *item : insert_field_list) { Item_field *field = down_cast<Item_field *>(item->real_item()); values_field_list.push_back(field); } } // If no column names were specified for the values table, we fill in names // corresponding to the same column of the insert table. if (values_column_list->empty()) { for (Item *item : values_field_list) { Item_field *field = down_cast<Item_field *>(item); values_column_list->push_back(to_lex_cstring(field->field_name)); } } if (check_duplicate_names(values_column_list, values_field_list, false)) return true; values_table->set_derived_column_names(values_column_list); Field_translator *transl = pointer_cast<Field_translator *>(thd->stmt_arena->alloc( values_column_list->size() * sizeof(Field_translator))); if (transl == nullptr) return true; uint field_count = 0; for (Item *item : values_field_list) { Item_field *field_arg = down_cast<Item_field *>(item); Item_insert_value *insert_value = new Item_insert_value( &thd->lex->current_query_block()->context, field_arg); if (insert_value == nullptr) return true; insert_value->context = &lex->query_block->context; transl[field_count].name = values_column_list->at(field_count).str; transl[field_count++].item = insert_value; } values_table->field_translation = transl; values_table->field_translation_end = transl + field_count; return false; } /** Resolve the columns of the optional VALUES table to the insert_values of the table inserted into. Field_translation of this table contains Item_insert_value objects pointing to the Item_field objects of the insert table. Caller is responsible for fixing these fields before calling this function. @param thd Thread context. @returns false if success, true if error. */ bool Sql_cmd_insert_base::resolve_values_table_columns(THD *thd) { // If insert_field_list is empty, we fill it by iterating over the fields of // the insert table in prepare_values_table(). These new Item_field objects // must be fixed as normal insert fields _before_ the Item_insert_value // objects in the VALUES table alias are fixed. if (insert_field_list.empty() && check_insert_fields(thd, thd->lex->query_tables, &values_field_list)) return true; Field_translator *field_translation = values_table->field_translation; while (field_translation != values_table->field_translation_end) { Item_insert_value *item = down_cast<Item_insert_value *>(field_translation->item); // We are specifically fixing fields for references in update clauses. thd->lex->in_update_value_clause = true; if (item->fix_fields(thd, nullptr)) return true; thd->lex->in_update_value_clause = false; ++field_translation; } // Signal that field_translation is valid. if (!values_table->is_merged()) values_table->set_merged(); return false; } /** Resolve ON DUPLICATE KEY UPDATE expressions. Caller is responsible for setting up the columns to be updated before calling this function. @param thd Thread handler @returns false if success, true if error */ bool Sql_cmd_insert_base::resolve_update_expressions(THD *thd) { DBUG_TRACE; Table_ref *const insert_table_ref = lex->query_tables; Table_ref *const insert_table_leaf = lex->insert_table_leaf; const bool select_insert = insert_many_values.empty(); const table_map map = lex->insert_table_leaf->map(); lex->in_update_value_clause = true; if (setup_fields(thd, /*want_privilege=*/SELECT_ACL, /*allow_sum_func=*/false, /*split_sum_funcs=*/false, /*column_update=*/false, /*typed_items=*/&update_field_list, &update_value_list, Ref_item_array())) return true; if (check_valid_table_refs(insert_table_ref, update_value_list, map)) return true; if (insert_table_leaf->table->has_gcol() && validate_gc_assignment(update_field_list, update_value_list, insert_table_leaf->table)) return true; lex->in_update_value_clause = false; if (select_insert && !lex->using_hypergraph_optimizer()) { /* Traverse the update values list and substitute fields from the select for references (Item_ref objects) to them. This is done in order to get correct values from those fields when the select employs a temporary table. This is not necessary for the hypergraph optimizer, since it changes the Item_field objects to point directly to the fields in the temporary table when the temporary table is created. */ Query_block *const select = lex->query_block; for (Item *&it : update_value_list) { Item *new_item = it->transform(&Item::update_value_transformer, pointer_cast<uchar *>(select)); if (new_item == nullptr) return true; it = new_item; } } return false; } bool Sql_cmd_insert_base::restore_cmd_properties(THD *thd) { if (duplicates == DUP_UPDATE && thd->lex->insert_table->set_insert_values(thd->mem_root)) return true; if (insert_many_values.empty()) lex->restore_properties_for_insert(); return Sql_cmd_dml::restore_cmd_properties(thd); } /** Check if there are more unique keys after the current one @param table table that keys are checked for @param keynr current key number @returns true if there are unique keys after the specified one */ static bool last_uniq_key(TABLE *table, uint keynr) { /* When an underlying storage engine informs that the unique key conflicts are not reported in the ascending order by setting the HA_DUPLICATE_KEY_NOT_IN_ORDER flag, we cannot rely on this information to determine the last key conflict. The information about the last key conflict will be used to do a replace of the new row on the conflicting row, rather than doing a delete (of old row) + insert (of new row). Hence check for this flag and disable replacing the last row by returning 0 always. Returning 0 will result in doing a delete + insert always. */ if (table->file->ha_table_flags() & HA_DUPLICATE_KEY_NOT_IN_ORDER) return false; /* purecov: inspected */ while (++keynr < table->s->keys) if (table->key_info[keynr].flags & HA_NOSAME) return false; return true; } /** Write a record to table with optional deletion of conflicting records, invoke proper triggers if needed. @param thd thread context @param table table to which record should be written @param info COPY_INFO structure describing handling of duplicates and which is used for counting number of records inserted and deleted. @param update COPY_INFO structure describing the UPDATE part (only used for INSERT ON DUPLICATE KEY UPDATE) Once this record is written to the table buffer, any AFTER INSERT trigger will be invoked. If instead of inserting a new record we end up updating an old one, both ON UPDATE triggers will fire instead. Similarly both ON DELETE triggers will be invoked if are to delete conflicting records. Call thd->transaction.stmt.mark_modified_non_trans_table() if table is a non-transactional table. @note In ON DUPLICATE KEY UPDATE case this call may set TABLE::autoinc_field_has_explicit_non_null_value flag to true (even in case of failure) so its caller should make sure that it is reset appropriately (@sa fill_record()). @returns false if success, true if error */ bool write_record(THD *thd, TABLE *table, COPY_INFO *info, COPY_INFO *update) { int error, trg_error = 0; char *key = nullptr; MY_BITMAP *save_read_set, *save_write_set; ulonglong prev_insert_id = table->file->next_insert_id; ulonglong insert_id_for_cur_row = 0; DBUG_TRACE; /* Here we are using separate MEM_ROOT as this memory should be freed once we exit write_record() function. This is marked as not instumented as it is allocated for very short time in a very specific case. */ MEM_ROOT mem_root(PSI_NOT_INSTRUMENTED, 256); info->stats.records++; save_read_set = table->read_set; save_write_set = table->write_set; const enum_duplicates duplicate_handling = info->get_duplicate_handling(); if (duplicate_handling == DUP_REPLACE || duplicate_handling == DUP_UPDATE) { assert(duplicate_handling != DUP_UPDATE || update != nullptr); while ((error = table->file->ha_write_row(table->record[0]))) { uint key_nr; /* If we do more than one iteration of this loop, from the second one the row will have an explicit value in the autoinc field, which was set at the first call of handler::update_auto_increment(). So we must save the autogenerated value to avoid thd->insert_id_for_cur_row to become 0. */ if (table->file->insert_id_for_cur_row > 0) insert_id_for_cur_row = table->file->insert_id_for_cur_row; else table->file->insert_id_for_cur_row = insert_id_for_cur_row; bool is_duplicate_key_error; if (!table->file->is_ignorable_error(error)) goto err; is_duplicate_key_error = (error == HA_ERR_FOUND_DUPP_KEY || error == HA_ERR_FOUND_DUPP_UNIQUE); if (!is_duplicate_key_error) { /* We come here when we had an ignorable error which is not a duplicate key error. In this we ignore error if ignore flag is set, otherwise report error as usual. We will not do any duplicate key processing. */ info->last_errno = error; table->file->print_error(error, MYF(0)); /* If IGNORE option is used, handler errors will be downgraded to warnings and don't have to stop the iteration. */ if (thd->is_error()) goto before_trg_err; goto ok_or_after_trg_err; /* Ignoring a not fatal error, return 0 */ } if ((int)(key_nr = table->file->get_dup_key(error)) < 0) { error = HA_ERR_FOUND_DUPP_KEY; /* Database can't find key */ goto err; } /* key index value is either valid in the range [0-MAX_KEY) or has value MAX_KEY as a marker for the case when no information about key can be found. In the last case we have to require that storage engine has the flag HA_DUPLICATE_POS turned on. If this invariant is false then assert will crash the server built in debug mode. For the server that was built without DEBUG we have additional check for the value of key_nr in the code below in order to report about error in any case. */ assert(key_nr != MAX_KEY || (key_nr == MAX_KEY && (table->file->ha_table_flags() & HA_DUPLICATE_POS))); DEBUG_SYNC(thd, "write_row_replace"); /* Read all columns for the row we are going to replace */ table->use_all_columns(); /* Don't allow REPLACE to replace a row when a auto_increment column was used. This ensures that we don't get a problem when the whole range of the key has been used. */ if (duplicate_handling == DUP_REPLACE && table->next_number_field && key_nr == table->s->next_number_index && (insert_id_for_cur_row > 0)) goto err; if (table->file->ha_table_flags() & HA_DUPLICATE_POS) { if (table->file->ha_rnd_pos(table->record[1], table->file->dup_ref)) goto err; } /* If the key index is equal to MAX_KEY it's treated as unknown key case and we shouldn't try to locate key info. */ else if (key_nr < MAX_KEY) { if (!key) { if (!(key = (char *)my_safe_alloca(table->s->max_unique_length, MAX_KEY_LENGTH))) { error = ENOMEM; goto err; } } /* If we convert INSERT operation internally to an UPDATE. An INSERT operation may update table->vfield for BLOB fields, So here we recalculate data for generated columns. */ if (table->vfield) { // Dont save old value while re-calculating generated fields. // Before image will already be saved in the first calculation. table->blobs_need_not_keep_old_value(); update_generated_write_fields(table->write_set, table); } key_copy((uchar *)key, table->record[0], table->key_info + key_nr, 0); if ((error = (table->file->ha_index_read_idx_map( table->record[1], key_nr, (uchar *)key, HA_WHOLE_KEY, HA_READ_KEY_EXACT)))) goto err; } else { /* For the server built in non-debug mode returns error if handler::get_dup_key() returned MAX_KEY as the value of key index. */ error = HA_ERR_FOUND_DUPP_KEY; /* Database can't find key */ goto err; } if (duplicate_handling == DUP_UPDATE) { int res = 0; /* We don't check for other UNIQUE keys - the first row that matches, is updated. If update causes a conflict again, an error is returned */ assert(table->insert_values != nullptr); /* The insert has failed, store the insert_id generated for this row to be re-used for the next insert. */ if (insert_id_for_cur_row > 0) prev_insert_id = insert_id_for_cur_row; store_record(table, insert_values); /* Special check for BLOB/GEOMETRY field in statements with "ON DUPLICATE KEY UPDATE" clause. See mysql_prepare_blob_values() function for more details. */ if (mysql_prepare_blob_values(thd, *update->get_changed_columns(), &mem_root)) goto before_trg_err; restore_record(table, record[1]); assert(update->get_changed_columns()->size() == update->update_values->size()); /* Reset TABLE::autoinc_field_has_explicit_non_null_value so we can figure out if ON DUPLICATE KEY UPDATE clause specifies value for auto-increment field as a side-effect of fill_record(). There is no need to clean-up this flag afterwards as this is responsibility of the caller. */ table->autoinc_field_has_explicit_non_null_value = false; bool is_row_changed = false; if (fill_record_n_invoke_before_triggers( thd, update, *update->get_changed_columns(), *update->update_values, table, TRG_EVENT_UPDATE, 0, true, &is_row_changed)) goto before_trg_err; bool insert_id_consumed = false; if ( // UPDATE clause specifies a value for the auto increment field table->autoinc_field_has_explicit_non_null_value && // An auto increment value has been generated for this row (insert_id_for_cur_row > 0)) { // After-update value: const ulonglong auto_incr_val = table->next_number_field->val_int(); if (auto_incr_val == insert_id_for_cur_row) { // UPDATE wants to use the generated value insert_id_consumed = true; } else if (table->file->auto_inc_interval_for_cur_row.in_range( auto_incr_val)) { /* UPDATE wants to use one auto generated value which we have already reserved for another (previous or following) row. That may cause a duplicate key error if we later try to insert the reserved value. Such conflicts on auto generated values would be strange behavior, so we return a clear error now. */ my_error(ER_AUTO_INCREMENT_CONFLICT, MYF(0)); goto before_trg_err; } } if (!insert_id_consumed) table->file->restore_auto_increment(prev_insert_id); info->stats.touched++; if (is_row_changed) { /* CHECK OPTION for VIEW ... ON DUPLICATE KEY UPDATE ... It is safe to not invoke CHECK OPTION for VIEW if records are same. In this case the row is coming from the view and thus should satisfy the CHECK OPTION. */ { const Table_ref *inserted_view = table->pos_in_table_list->belong_to_view; if (inserted_view != nullptr) { res = inserted_view->view_check_option(thd); if (res == VIEW_CHECK_SKIP) goto ok_or_after_trg_err; if (res == VIEW_CHECK_ERROR) goto before_trg_err; } } /* Existing rows in table should normally satisfy CHECK constraints. So it should be safe to check constraints only for rows that has really changed (i.e. after compare_records()). In future, once addition/enabling of CHECK constraints without their validation is supported, we might encounter old rows which do not satisfy CHECK constraints currently enabled. However, rejecting no-op updates to such invalid pre-existing rows won't make them valid and is probably going to be confusing for users. So it makes sense to stick to current behavior. */ if (invoke_table_check_constraints(thd, table)) { if (thd->is_error()) goto before_trg_err; // return false when IGNORE clause is used. goto ok_or_after_trg_err; } if ((error = table->file->ha_update_row(table->record[1], table->record[0])) && error != HA_ERR_RECORD_IS_THE_SAME) { info->last_errno = error; myf error_flags = MYF(0); if (table->file->is_fatal_error(error)) error_flags |= ME_FATALERROR; table->file->print_error(error, error_flags); /* If IGNORE option is used, handler errors will be downgraded to warnings and don't have to stop the iteration. */ if (thd->is_error()) goto before_trg_err; goto ok_or_after_trg_err; /* Ignoring a not fatal error, return 0 */ } if (error != HA_ERR_RECORD_IS_THE_SAME) info->stats.updated++; else error = 0; /* If ON DUP KEY UPDATE updates a row instead of inserting one, it's like a regular UPDATE statement: it should not affect the value of a next SELECT LAST_INSERT_ID() or mysql_insert_id(). Except if LAST_INSERT_ID(#) was in the INSERT query, which is handled separately by THD::arg_of_last_insert_id_function. */ insert_id_for_cur_row = table->file->insert_id_for_cur_row = 0; info->stats.copied++; } // Execute the 'AFTER, ON UPDATE' trigger trg_error = (table->triggers && table->triggers->process_triggers(thd, TRG_EVENT_UPDATE, TRG_ACTION_AFTER, true)); goto ok_or_after_trg_err; } else /* DUP_REPLACE */ { Table_ref *view = table->pos_in_table_list->belong_to_view; if (view && view->replace_filter) { const size_t record_length = table->s->reclength; void *record0_saved = my_malloc(PSI_NOT_INSTRUMENTED, record_length, MYF(MY_WME)); if (!record0_saved) { error = ENOMEM; goto err; } // Save the record used for comparison. memcpy(record0_saved, table->record[0], record_length); // Preparing the record for comparison. memcpy(table->record[0], table->record[1], record_length); // Checking if the row being conflicted is visible by the view. const bool found_row_in_view = view->replace_filter->val_int(); // Restoring the record back. memcpy(table->record[0], record0_saved, record_length); my_free(record0_saved); if (!found_row_in_view) { my_error(ER_REPLACE_INACCESSIBLE_ROWS, MYF(0)); goto err; } } /* The manual defines the REPLACE semantics that it is either an INSERT or DELETE(s) + INSERT; FOREIGN KEY checks in InnoDB do not function in the defined way if we allow MySQL to convert the latter operation internally to an UPDATE. We also should not perform this conversion if we have timestamp field with ON UPDATE which is different from DEFAULT. Another case when conversion should not be performed is when we have ON DELETE trigger on table so user may notice that we cheat here. Note that it is ok to do such conversion for tables which have ON UPDATE but have no ON DELETE triggers, we just should not expose this fact to users by invoking ON UPDATE triggers. */ if (last_uniq_key(table, key_nr) && !table->s->is_referenced_by_foreign_key() && (!table->triggers || !table->triggers->has_delete_triggers())) { if ((error = table->file->ha_update_row(table->record[1], table->record[0])) && error != HA_ERR_RECORD_IS_THE_SAME) goto err; if (error != HA_ERR_RECORD_IS_THE_SAME) info->stats.deleted++; else error = 0; thd->record_first_successful_insert_id_in_cur_stmt( table->file->insert_id_for_cur_row); /* Since we pretend that we have done insert we should call its after triggers. */ goto after_trg_n_copied_inc; } else { if (table->triggers && table->triggers->process_triggers(thd, TRG_EVENT_DELETE, TRG_ACTION_BEFORE, true)) goto before_trg_err; if ((error = table->file->ha_delete_row(table->record[1]))) goto err; info->stats.deleted++; if (!table->file->has_transactions()) thd->get_transaction()->mark_modified_non_trans_table( Transaction_ctx::STMT); if (table->triggers && table->triggers->process_triggers(thd, TRG_EVENT_DELETE, TRG_ACTION_AFTER, true)) { trg_error = 1; goto ok_or_after_trg_err; } /* Let us attempt do write_row() once more */ } } } /* If more than one iteration of the above while loop is done, from the second one the row being inserted will have an explicit value in the autoinc field, which was set at the first call of handler::update_auto_increment(). This value is saved to avoid thd->insert_id_for_cur_row becoming 0. Use this saved autoinc value. */ if (table->file->insert_id_for_cur_row == 0) table->file->insert_id_for_cur_row = insert_id_for_cur_row; thd->record_first_successful_insert_id_in_cur_stmt( table->file->insert_id_for_cur_row); /* Restore column maps if they where replaced during an duplicate key problem. */ if (table->read_set != save_read_set || table->write_set != save_write_set) table->column_bitmaps_set(save_read_set, save_write_set); } else if ((error = table->file->ha_write_row(table->record[0]))) { DEBUG_SYNC(thd, "write_row_noreplace"); info->last_errno = error; myf error_flags = MYF(0); if (table->file->is_fatal_error(error)) error_flags |= ME_FATALERROR; table->file->print_error(error, error_flags); /* If IGNORE option is used, handler errors will be downgraded to warnings and don't have to stop the iteration. */ if (thd->is_error()) goto before_trg_err; table->file->restore_auto_increment(prev_insert_id); goto ok_or_after_trg_err; } after_trg_n_copied_inc: info->stats.copied++; thd->record_first_successful_insert_id_in_cur_stmt( table->file->insert_id_for_cur_row); trg_error = (table->triggers && table->triggers->process_triggers( thd, TRG_EVENT_INSERT, TRG_ACTION_AFTER, true)); ok_or_after_trg_err: if (key) my_safe_afree(key, table->s->max_unique_length, MAX_KEY_LENGTH); if (!table->file->has_transactions()) thd->get_transaction()->mark_modified_non_trans_table( Transaction_ctx::STMT); return trg_error; err : { myf error_flags = MYF(0); /**< Flag for fatal errors */ info->last_errno = error; if (table->file->is_fatal_error(error)) error_flags |= ME_FATALERROR; table->file->print_error(error, error_flags); } before_trg_err: table->file->restore_auto_increment(prev_insert_id); if (key) my_safe_afree(key, table->s->max_unique_length, MAX_KEY_LENGTH); table->column_bitmaps_set(save_read_set, save_write_set); return true; } /** Check that all fields with aren't null_fields are used @param thd thread handler @param entry table that's checked @param table_list top-level table or view, used for generating error or warning message @retval true if all fields are given values */ bool check_that_all_fields_are_given_values(THD *thd, TABLE *entry, Table_ref *table_list) { MY_BITMAP *write_set = entry->fields_set_during_insert; for (Field **field = entry->field; *field; field++) { if (!bitmap_is_set(write_set, (*field)->field_index()) && ((*field)->is_flag_set(NO_DEFAULT_VALUE_FLAG) && ((*field)->m_default_val_expr == nullptr)) && ((*field)->real_type() != MYSQL_TYPE_ENUM)) { bool view = false; if (table_list) { table_list = table_list->top_table(); view = table_list->is_view(); } if (view) { if ((*field)->type() == MYSQL_TYPE_GEOMETRY) { my_error(ER_NO_DEFAULT_FOR_VIEW_FIELD, MYF(0), table_list->db, table_list->table_name); } else { (*field)->set_warning(Sql_condition::SL_WARNING, ER_NO_DEFAULT_FOR_VIEW_FIELD, 1, table_list->db, table_list->table_name); } } else { if ((*field)->type() == MYSQL_TYPE_GEOMETRY) { my_error(ER_NO_DEFAULT_FOR_FIELD, MYF(0), (*field)->field_name); } else { (*field)->set_warning(Sql_condition::SL_WARNING, ER_NO_DEFAULT_FOR_FIELD, 1); } } } } bitmap_clear_all(write_set); return thd->is_error(); } bool Query_result_insert::prepare(THD *thd, const mem_root_deque<Item *> &, Query_expression *u) { DBUG_TRACE; LEX *const lex = thd->lex; const enum_duplicates duplicate_handling = info.get_duplicate_handling(); unit = u; table = lex->insert_table_leaf->table; if (info.add_function_default_columns(table, table->write_set)) return true; if ((duplicate_handling == DUP_UPDATE) && update.add_function_default_columns(table, table->write_set)) return true; return false; } /** Set up the target table for execution. If the target table is the same as one of the source tables (INSERT SELECT), the target table is not finally set up in the join optimization phase. Do the final setup now. @returns false always */ bool Query_result_insert::start_execution(THD *thd) { DBUG_TRACE; table = thd->lex->insert_table_leaf->table; restore_record(table, s->default_values); // Get empty record table->next_number_field = table->found_next_number_field; const enum_duplicates duplicate_handling = info.get_duplicate_handling(); if (info.add_function_default_columns(table, table->write_set)) { return true; } if ((duplicate_handling == DUP_UPDATE) && update.add_function_default_columns(table, table->write_set)) return true; thd->num_truncated_fields = 0; if (thd->lex->is_ignore() || duplicate_handling != DUP_ERROR) table->file->ha_extra(HA_EXTRA_IGNORE_DUP_KEY); if (duplicate_handling == DUP_REPLACE && (!table->triggers || !table->triggers->has_delete_triggers())) table->file->ha_extra(HA_EXTRA_WRITE_CAN_REPLACE); if (duplicate_handling == DUP_UPDATE) table->file->ha_extra(HA_EXTRA_INSERT_WITH_UPDATE); prepare_triggers_for_insert_stmt(thd, table); for (Field **next_field = table->field; *next_field; ++next_field) { (*next_field)->reset_warnings(); (*next_field)->reset_tmp_null(); } if (thd->locked_tables_mode <= LTM_LOCK_TABLES && !thd->lex->is_explain()) { assert(!bulk_insert_started); // TODO: Is there no better estimation than 0 == Unknown number of rows? table->file->ha_start_bulk_insert((ha_rows)0); bulk_insert_started = true; } info.reset_counters(); return false; } void Query_result_insert::cleanup() { DBUG_TRACE; // table_list and table may be out of synch: if (current_thd->lex->insert_table_leaf != nullptr && current_thd->lex->insert_table_leaf->table == nullptr) table = nullptr; if (table != nullptr) { table->next_number_field = nullptr; table->file->ha_reset(); table = nullptr; } info.cleanup(); update.cleanup(); current_thd->check_for_truncated_fields = CHECK_FIELD_IGNORE; } bool Query_result_insert::send_data(THD *thd, const mem_root_deque<Item *> &values) { DBUG_TRACE; bool error = false; const Autoinc_field_has_explicit_non_null_value_reset_guard after_each_row( table); thd->check_for_truncated_fields = CHECK_FIELD_WARN; store_values(thd, values); thd->check_for_truncated_fields = CHECK_FIELD_ERROR_FOR_NULL; if (thd->is_error()) { return true; } /* TODO: This method is supposed to be called only once per-statement. But currently it is called for each row inserted by INSERT SELECT. Which looks like unnessary and results in resource wastage. */ prepare_triggers_for_insert_stmt(thd, table); if (table_list) // Not CREATE ... SELECT { switch (table_list->view_check_option(thd)) { case VIEW_CHECK_SKIP: return false; case VIEW_CHECK_ERROR: return true; } /* Replication may require extra check of data change statements, i.e., for DML executed for CREATE ... SELECT. We check this only once before inserting the first row. */ } else if (info.stats.records == 0 && run_before_dml_hook(thd)) { return true; } if (invoke_table_check_constraints(thd, table)) { // return false when IGNORE clause is used. return thd->is_error(); } error = write_record(thd, table, &info, &update); DEBUG_SYNC(thd, "create_select_after_write_rows_event"); if (!error && (table->triggers || info.get_duplicate_handling() == DUP_UPDATE)) { /* Restore fields of the record since it is possible that they were changed by ON DUPLICATE KEY UPDATE clause. If triggers exist then whey can modify some fields which were not originally touched by INSERT ... SELECT, so we have to restore their original values for the next row. */ restore_record(table, s->default_values); } if (!error && table->next_number_field) { /* If no value has been autogenerated so far, we need to remember the value we just saw, we may need to send it to client in the end. */ if (thd->first_successful_insert_id_in_cur_stmt == 0) // optimization autoinc_value_of_last_inserted_row = table->next_number_field->val_int(); /* Clear auto-increment field for the next record, if triggers are used we will clear it twice, but this should be cheap. */ table->next_number_field->reset(); } return error; } void Query_result_insert::store_values(THD *thd, const mem_root_deque<Item *> &values) { if (CountVisibleFields(*fields) != 0) { restore_record(table, s->default_values); if (!validate_default_values_of_unset_fields(thd, table)) fill_record_n_invoke_before_triggers(thd, &info, *fields, values, table, TRG_EVENT_INSERT, table->s->fields, true, nullptr); } else fill_record_n_invoke_before_triggers(thd, table->field, values, table, TRG_EVENT_INSERT, table->s->fields); check_that_all_fields_are_given_values(thd, table, table_list); } bool Query_result_insert::stmt_binlog_is_trans() const { return table->file->has_transactions(); } bool Query_result_insert::send_eof(THD *thd) { ulonglong id, row_count; bool changed [[maybe_unused]]; THD::killed_state killed_status = thd->killed; DBUG_TRACE; DBUG_PRINT("enter", ("trans_table=%d, table_type='%s'", table->file->has_transactions(), table->file->table_type())); int error = 0; if (bulk_insert_started) { error = table->file->ha_end_bulk_insert(); if (!error && thd->is_error()) error = thd->get_stmt_da()->mysql_errno(); bulk_insert_started = false; } changed = (info.stats.copied || info.stats.deleted || info.stats.updated); /* INSERT ... SELECT on non-transactional table which changes any rows must be marked as unsafe to rollback. */ assert(table->file->has_transactions() || !changed || thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT)); /* Write to binlog before committing transaction. No statement will be written by the binlog_query() below in RBR mode. All the events are in the transaction cache and will be written when ha_autocommit_or_rollback() is issued below. */ if (mysql_bin_log.is_open() && (!error || thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT))) { int errcode = 0; if (!error) thd->clear_error(); else errcode = query_error_code(thd, killed_status == THD::NOT_KILLED); if (thd->binlog_query(THD::ROW_QUERY_TYPE, thd->query().str, thd->query().length, stmt_binlog_is_trans(), false, false, errcode)) { table->file->ha_release_auto_increment(); return true; } } table->file->ha_release_auto_increment(); if (error) { myf error_flags = MYF(0); if (table->file->is_fatal_error(my_errno())) error_flags |= ME_FATALERROR; table->file->print_error(my_errno(), error_flags); return true; } /* For the strict_mode call of push_warning above results to set error in Diagnostic_area. Therefore it is necessary to check whether the error was set and leave method if it is true. If we didn't do so we would failed later when my_ok is called. */ if (thd->get_stmt_da()->is_error()) return true; char buff[160]; if (thd->lex->is_ignore()) snprintf(buff, sizeof(buff), ER_THD(thd, ER_INSERT_INFO), (long)info.stats.records, (long)(info.stats.records - info.stats.copied), (long)thd->get_stmt_da()->current_statement_cond_count()); else snprintf(buff, sizeof(buff), ER_THD(thd, ER_INSERT_INFO), (long)info.stats.records, (long)(info.stats.deleted + info.stats.updated), (long)thd->get_stmt_da()->current_statement_cond_count()); row_count = info.stats.copied + info.stats.deleted + (thd->get_protocol()->has_client_capability(CLIENT_FOUND_ROWS) ? info.stats.touched : info.stats.updated); id = (thd->first_successful_insert_id_in_cur_stmt > 0) ? thd->first_successful_insert_id_in_cur_stmt : (thd->arg_of_last_insert_id_function ? thd->first_successful_insert_id_in_prev_stmt : (info.stats.copied ? autoinc_value_of_last_inserted_row : 0)); my_ok(thd, row_count, id, buff); /* If we have inserted into a VIEW, and the base table has AUTO_INCREMENT column, but this column is not accessible through a view, then we should restore LAST_INSERT_ID to the value it had before the statement. */ if (table_list != nullptr && table_list->is_view() && !table_list->contain_auto_increment) thd->first_successful_insert_id_in_cur_stmt = thd->first_successful_insert_id_in_prev_stmt; return false; } void Query_result_insert::abort_result_set(THD *thd) { DBUG_TRACE; // table_list and table may be out of synch: if (thd->lex->insert_table_leaf != nullptr && thd->lex->insert_table_leaf->table == nullptr) table = nullptr; /* If the creation of the table failed (due to a syntax error, for example), no table will have been opened and therefore 'table' will be NULL. In that case, we still need to execute the rollback and the end of the function. */ if (table != nullptr) { bool changed [[maybe_unused]]; bool transactional_table; /* Try to end the bulk insert which might have been started before. We don't need to do this if we are in prelocked mode (since we don't use bulk insert in this case). Also we should not do this if tables are not locked yet (bulk insert is not started yet in this case). */ if (bulk_insert_started) { table->file->ha_end_bulk_insert(); bulk_insert_started = false; } /* If at least one row has been inserted/modified and will stay in the table (the table doesn't have transactions) we must write to the binlog (and the error code will make the slave stop). For many errors (example: we got a duplicate key error while inserting into a MyISAM table), no row will be added to the table, so passing the error to the slave will not help since there will be an error code mismatch (the inserts will succeed on the slave with no error). If table creation failed, the number of rows modified will also be zero, so no check for that is made. */ changed = (info.stats.copied || info.stats.deleted || info.stats.updated); transactional_table = table->file->has_transactions(); if (thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT)) { if (mysql_bin_log.is_open()) { int errcode = query_error_code(thd, thd->killed == THD::NOT_KILLED); /* error of writing binary log is ignored */ (void)thd->binlog_query(THD::ROW_QUERY_TYPE, thd->query().str, thd->query().length, transactional_table, false, false, errcode); } } assert( transactional_table || !changed || thd->get_transaction()->cannot_safely_rollback(Transaction_ctx::STMT)); table->file->ha_release_auto_increment(); } } /*************************************************************************** CREATE TABLE (SELECT) ... ***************************************************************************/ /** Create table from lists of fields and items (or just return TABLE object for pre-opened existing table). Used by CREATE SELECT. Let "source table" be the table in the SELECT part. Let "source table columns" be the set of columns in the SELECT list. An interesting peculiarity in the syntax CREATE TABLE (@<columns@>) SELECT is that function defaults are stripped from the the source table columns, but not from the additional columns defined in the CREATE TABLE part. The first @c TIMESTAMP column there is also subject to promotion to @c TIMESTAMP @c DEFAULT @c CURRENT_TIMESTAMP @c ON @c UPDATE @c CURRENT_TIMESTAMP, as usual. @param [in] thd Thread object @param [in] create_info Create information (like MAX_ROWS, ENGINE or temporary table flag) @param [in] create_table Pointer to Table_ref object providing database and name for table to be created or to be open @param [in,out] alter_info Initial list of columns and indexes for the table to be created @param [in] items The source table columns. Corresponding column definitions (Create_field's) will be added to the end of alter_info->create_list. @param [out] post_ddl_ht Set to handlerton for table's SE, if this SE supports atomic DDL, so caller can call SE post DDL hook after committing transaction. @note This function assumes that either table exists and was pre-opened and locked at open_and_lock_tables() stage (and in this case we just emit error or warning and return pre-opened TABLE object) or an exclusive metadata lock was acquired on table so we can safely create, open and lock table in it (we don't acquire metadata lock if this create is for temporary table). @note Since this function contains some logic specific to CREATE TABLE ... SELECT it should be changed before it can be used in other contexts. @retval non-zero Pointer to TABLE object for table created or opened @retval 0 Error */ static TABLE *create_table_from_items(THD *thd, HA_CREATE_INFO *create_info, Table_ref *create_table, Alter_info *alter_info, const mem_root_deque<Item *> &items, handlerton **post_ddl_ht) { DBUG_TRACE; // Check that the specified ENGINE exists and is enabled. if (get_viable_handlerton_for_create(thd, create_table->table_name, *create_info) == nullptr) { return nullptr; } if (!thd->variables.explicit_defaults_for_timestamp) promote_first_timestamp_column(&alter_info->create_list); TABLE_SHARE share; init_tmp_table_share(thd, &share, "", 0, "", "", nullptr); share.db_create_options = 0; share.db_low_byte_first = (create_info->db_type == myisam_hton || create_info->db_type == heap_hton); TABLE tmp_table; tmp_table.s = &share; tmp_table.set_not_started(); /* Add selected items to field list */ for (Item *item : VisibleFields(items)) { Create_field *create_field = generate_create_field(thd, item, &tmp_table); if (create_field == nullptr) return nullptr; /* purecov: deadcode */ // Array columns may be returned if show_hidden_columns is enabled. Raise an // error instead of attempting to create array columns in the new table. DBUG_EXECUTE("show_hidden_columns", { if (create_field->is_array) { my_error(ER_NOT_SUPPORTED_YET, MYF(0), "Creating tables with array columns."); return nullptr; } }); assert(!create_field->is_array); alter_info->create_list.push_back(create_field); } /* Acquire SU metadata locks for the tables referenced in the FK constraints. */ if (!(create_info->options & HA_LEX_CREATE_TMP_TABLE) && (create_info->db_type->flags & HTON_SUPPORTS_FOREIGN_KEYS)) { /* CREATE TABLE SELECT fails under LOCK TABLES at open_tables() time if target table doesn't exist already. So we don't need to handle LOCK TABLES case here by checking that parent tables for new FKs are properly locked and there are no orphan child tables for which table being created will become parent. */ assert(thd->locked_tables_mode != LTM_LOCK_TABLES && thd->locked_tables_mode != LTM_PRELOCKED_UNDER_LOCK_TABLES); MDL_request_list mdl_requests; if (collect_fk_parents_for_new_fks( thd, create_table->db, create_table->table_name, alter_info, MDL_SHARED_UPGRADABLE, nullptr, &mdl_requests, nullptr) || collect_fk_names_for_new_fks(thd, create_table->db, create_table->table_name, alter_info, create_info->db_type, 0, // No pre-existing FKs &mdl_requests)) { return nullptr; } if (!mdl_requests.is_empty() && thd->mdl_context.acquire_locks(&mdl_requests, thd->variables.lock_wait_timeout)) { return nullptr; } } // Prepare check constraints. if (prepare_check_constraints_for_create( thd, create_table->db, create_table->table_name, alter_info)) { return nullptr; } /* If mode to generate invisible primary key is active then, generate primary key for the table. */ if (is_generate_invisible_primary_key_mode_active(thd) && is_candidate_table_for_invisible_primary_key_generation(create_info, alter_info)) { if (validate_and_generate_invisible_primary_key(thd, alter_info)) return nullptr; } DEBUG_SYNC(thd, "create_table_select_before_create"); /* Create and lock table. Note that we are either creating (or opening existing) temporary table or creating base table on which name we have exclusive lock. So code below should not cause deadlocks or races. We don't log the statement, it will be logged later. If this is a HEAP table, the automatic DELETE FROM which is written to the binlog when a HEAP table is opened for the first time since startup, must not be written: 1) it would be wrong (imagine we're in CREATE SELECT: we don't want to delete from it) 2) it would be written before the CREATE TABLE, which is a wrong order. So we keep binary logging disabled when we open_table(). */ const size_t select_field_count = CountVisibleFields(items); if (mysql_create_table_no_lock( thd, create_table->db, create_table->table_name, create_info, alter_info, select_field_count, true, nullptr, post_ddl_ht)) { return nullptr; } DEBUG_SYNC(thd, "create_table_select_before_open"); if (!(create_info->options & HA_LEX_CREATE_TMP_TABLE)) { Open_table_context ot_ctx(thd, MYSQL_OPEN_REOPEN); /* Here we open the destination table, on which we already have an exclusive metadata lock. */ if (open_table(thd, create_table, &ot_ctx)) { /* Play safe, remove table share for the table from the cache. */ tdc_remove_table(thd, TDC_RT_REMOVE_ALL, create_table->db, create_table->table_name, false); if (!(create_info->db_type->flags & HTON_SUPPORTS_ATOMIC_DDL)) { quick_rm_table(thd, create_info->db_type, create_table->db, create_table->table_name, 0); } return nullptr; } } else { if (open_temporary_table(thd, create_table)) { /* This shouldn't happen as creation of temporary table should make it preparable for open. Anyway we can't drop temporary table if we are unable to find it. */ assert(0); return nullptr; } } return create_table->table; } Query_result_create::Query_result_create(Table_ref *table_arg, mem_root_deque<Item *> *fields, enum_duplicates duplic, Table_ref *select_tables_arg) : Query_result_insert(nullptr, // table_list_par nullptr, // target_columns fields, nullptr, // update_fields nullptr, // update_values duplic), create_table(table_arg), select_tables(select_tables_arg) {} bool Query_result_create::prepare(THD *, const mem_root_deque<Item *> &, Query_expression *u) { DBUG_TRACE; unit = u; return false; } /** Create new table @param thd thread handler @returns false on success, true on error */ bool Query_result_create::create_table_for_query_block(THD *thd) { DBUG_TRACE; DEBUG_SYNC(thd, "create_table_select_before_lock"); mem_root_deque<Item *> *select_exprs = unit->get_unit_column_types(); table = create_table_from_items(thd, create_info, create_table, alter_info, *select_exprs, &m_post_ddl_ht); if (table == nullptr) return true; // abort() deletes table // Ignore hidden fields uint field_count = table->s->fields; for (uint i = 0; i < table->s->fields; ++i) { if (table->s->field[i]->is_field_for_functional_index()) field_count--; } const size_t visible_select_exprs = CountVisibleFields(*select_exprs); if (field_count < visible_select_exprs) { my_error(ER_WRONG_VALUE_COUNT_ON_ROW, MYF(0), 1L); return true; } // First field to copy table_fields = table->field + (field_count - visible_select_exprs); for (Field **f = table_fields; *f != nullptr; f++) { if ((*f)->gcol_info && !(*f)->is_field_for_functional_index()) { /* Generated columns are not allowed to be given a value for CREATE TABLE .. SELECT statement. */ my_error(ER_NON_DEFAULT_VALUE_FOR_GENERATED_COLUMN, MYF(0), (*f)->field_name, (*f)->table->s->table_name.str); return true; } } // Turn off function defaults for columns filled from SELECT list: if (info.ignore_last_columns(table, visible_select_exprs)) { return true; } return false; } /// Lock the newly created table and prepare it for insertion. bool Query_result_create::start_execution(THD *thd) { DBUG_TRACE; MYSQL_LOCK *extra_lock = nullptr; table->reginfo.lock_type = TL_WRITE; /* mysql_lock_tables() below should never fail with request to reopen table since it won't wait for the table lock (we have exclusive metadata lock on the table) and thus can't get aborted. */ if (!(extra_lock = mysql_lock_tables(thd, &table, 1, 0)) || binlog_show_create_table(thd)) { if (extra_lock) { mysql_unlock_tables(thd, extra_lock); extra_lock = nullptr; } return true; } if (extra_lock) { assert(m_plock == nullptr); if (create_info->options & HA_LEX_CREATE_TMP_TABLE) m_plock = &m_lock; else m_plock = &thd->extra_lock; *m_plock = extra_lock; } /* Mark all fields that are given values */ for (Field **f = table_fields; *f != nullptr; f++) { bitmap_set_bit(table->write_set, (*f)->field_index()); bitmap_set_bit(table->fields_set_during_insert, (*f)->field_index()); } // Set up an empty bitmap of function defaults if (info.add_function_default_columns(table, table->write_set)) return true; if (info.add_function_default_columns(table, table->fields_set_during_insert)) return true; table->next_number_field = table->found_next_number_field; restore_record(table, s->default_values); // Get empty record thd->num_truncated_fields = 0; const enum_duplicates duplicate_handling = info.get_duplicate_handling(); if (thd->lex->is_ignore() || duplicate_handling != DUP_ERROR) table->file->ha_extra(HA_EXTRA_IGNORE_DUP_KEY); if (duplicate_handling == DUP_REPLACE && (!table->triggers || !table->triggers->has_delete_triggers())) table->file->ha_extra(HA_EXTRA_WRITE_CAN_REPLACE); if (duplicate_handling == DUP_UPDATE) table->file->ha_extra(HA_EXTRA_INSERT_WITH_UPDATE); if (thd->locked_tables_mode <= LTM_LOCK_TABLES) { table->file->ha_start_bulk_insert((ha_rows)0); bulk_insert_started = true; } const enum_check_fields save_check_for_truncated_fields = thd->check_for_truncated_fields; thd->check_for_truncated_fields = CHECK_FIELD_WARN; if (check_that_all_fields_are_given_values(thd, table, table_list)) return true; thd->check_for_truncated_fields = save_check_for_truncated_fields; table->mark_columns_needed_for_insert(thd); return false; } /* For row-based replication, the CREATE-SELECT statement is written in two pieces: the first one contain the CREATE TABLE statement necessary to create the table and the second part contain the rows that should go into the table. For non-temporary tables, the start of the CREATE-SELECT implicitly commits the previous transaction, and all events forming the statement will be stored the transaction cache. At end of the statement, the entire statement is committed as a transaction, and all events are written to the binary log. On the master, the table is locked for the duration of the statement, but since the CREATE part is replicated as a simple statement, there is no way to lock the table for accesses on the slave. Hence, we have to hold on to the CREATE part of the statement until the statement has finished. */ int Query_result_create::binlog_show_create_table(THD *thd) { DBUG_TRACE; Table_ref *save_next_global = create_table->next_global; create_table->next_global = select_tables; const int error = thd->decide_logging_format(create_table); create_table->next_global = save_next_global; if (error) return error; create_table->table->set_binlog_drop_if_temp( !thd->is_current_stmt_binlog_disabled() && !thd->is_current_stmt_binlog_format_row()); if (!thd->is_current_stmt_binlog_format_row() || table->s->tmp_table) return 0; /* Note 1: In RBR mode, we generate a CREATE TABLE statement for the created table by calling store_create_info() (behaves as SHOW CREATE TABLE). The 'CREATE TABLE' event will be put in the binlog statement cache with an Anonymous_gtid_log_event, and any subsequent events (e.g., table-map events and rows event) will be put in the binlog transaction cache with an Anonymous_gtid_log_event. So that the 'CREATE...SELECT' statement is logged as: Anonymous_gtid_log_event CREATE TABLE event Anonymous_gtid_log_event BEGIN rows event COMMIT We write the CREATE TABLE statement here and not in prepare() since there potentially are sub-selects or accesses to information schema that will do a close_thread_tables(), destroying the statement transaction cache. */ char buf[2048]; String query(buf, sizeof(buf), system_charset_info); int result; Table_ref tmp_table_list(table); query.length(0); // Have to zero it since constructor doesn't result = store_create_info(thd, &tmp_table_list, &query, create_info, /* show_database */ true, /* SHOW CREATE TABLE */ false); assert(result == 0); /* store_create_info() always return 0 */ if (mysql_bin_log.is_open()) { DEBUG_SYNC(thd, "create_select_before_write_create_event"); /* Binary log layer has special code to handle rollback of CREATE TABLE SELECT in RBR mode - it truncates statement cache in this case. If SE is transactional and supports atomic DDL, we log the Query_log event into transactional cache and do not flush it immediately. */ int errcode = query_error_code(thd, thd->killed == THD::NOT_KILLED); bool is_trans = false; bool direct = true; if (get_default_handlerton(thd, thd->lex->create_info->db_type)->flags & HTON_SUPPORTS_ATOMIC_DDL) { is_trans = true; direct = false; } result = thd->binlog_query(THD::STMT_QUERY_TYPE, query.ptr(), query.length(), is_trans, direct, /* suppress_use */ false, errcode); DEBUG_SYNC(thd, "create_select_after_write_create_event"); } return result; } void Query_result_create::store_values(THD *thd, const mem_root_deque<Item *> &values) { /* Evaluate function defaults and default expressions for the columns defined in the CREATE TABLE SELECT. fill_record_n_invoke_before_triggers version which does *not* set function default and default expression automagically is called from here. As a result function default and default expression is not evaluated for the columns defined in CREATE TABLE SELECT. Hence calling set_function_defaults explicitly. */ if (info.function_defaults_apply_on_columns(table->write_set)) { if (info.set_function_defaults(table)) return; } fill_record_n_invoke_before_triggers(thd, table_fields, values, table, TRG_EVENT_INSERT, table->s->fields); } bool Query_result_create::stmt_binlog_is_trans() const { /* Binary logging code assumes that CREATE TABLE statements are written to transactional cache iff they support atomic DDL. */ return (table->s->db_type()->flags & HTON_SUPPORTS_ATOMIC_DDL); } bool Query_result_create::send_eof(THD *thd) { /* The routine that writes the statement in the binary log is in Query_result_insert::send_eof(). For that reason, we mark the flag at this point. */ if (create_info->options & HA_LEX_CREATE_TMP_TABLE) thd->get_transaction()->mark_created_temp_table(Transaction_ctx::STMT); bool error = false; /* For non-temporary tables, we update the unique_constraint_name for the FKs of referencing tables, after acquiring exclusive metadata locks. We also need to upgrade the SU locks on referenced tables to be exclusive before invalidating the referenced tables. */ Foreign_key_parents_invalidator fk_invalidator; if (!(create_info->options & HA_LEX_CREATE_TMP_TABLE) && (create_info->db_type->flags & HTON_SUPPORTS_FOREIGN_KEYS)) { MDL_request_list mdl_requests; if ((!dd::get_dictionary()->is_dd_table_name(create_table->db, create_table->table_name) && collect_fk_children(thd, create_table->db, create_table->table_name, create_info->db_type, MDL_EXCLUSIVE, &mdl_requests)) || collect_fk_parents_for_new_fks(thd, create_table->db, create_table->table_name, alter_info, MDL_EXCLUSIVE, create_info->db_type, &mdl_requests, &fk_invalidator) || (!mdl_requests.is_empty() && thd->mdl_context.acquire_locks(&mdl_requests, thd->variables.lock_wait_timeout))) error = true; else { const dd::cache::Dictionary_client::Auto_releaser releaser( thd->dd_client()); const dd::Table *new_table = nullptr; if (thd->dd_client()->acquire(create_table->db, create_table->table_name, &new_table)) error = true; else { assert(new_table != nullptr); /* If we are to support FKs for storage engines which don't support atomic DDL we need to decide what to do for such SEs in case of failure to update children definitions and adjust code accordingly. */ assert(create_info->db_type->flags & HTON_SUPPORTS_ATOMIC_DDL); if (adjust_fk_children_after_parent_def_change( thd, create_table->db, create_table->table_name, create_info->db_type, new_table, nullptr) || adjust_fk_parents(thd, create_table->db, create_table->table_name, true, nullptr)) error = true; } } } { Uncommitted_tables_guard uncommitted_tables(thd); /* We can rollback target table creation by dropping it even for SEs which don't support atomic DDL. So there is no need to commit changes to metadata of dependent views below. Moreover, doing these intermediate commits can be harmful as in RBR mode they will flush CREATE TABLE event and row events to the binary log which, in case of later error, will create discrepancy with rollback of statement by target table removal. Such intermediate commits also wipe out transaction's unsafe-to-rollback flags which leads to broken assertions in Query_result_insert::send_eof(). */ if (!error) error = update_referencing_views_metadata(thd, create_table, false, &uncommitted_tables); } DBUG_EXECUTE_IF("crash_before_create_select_insert", DBUG_SUICIDE();); if (!error) error = Query_result_insert::send_eof(thd); if (error) abort_result_set(thd); else { DBUG_EXECUTE_IF("crash_after_create_select_insert", DBUG_SUICIDE();); /* Do an implicit commit at end of statement for non-temporary tables. This can fail in which case rollback will be done automatically. For storage engines supporting atomic DDL this will revert table creation in SE, data-dictionary and binlog changes. For other storage engines we might end-up with partially consistent state between data-dictionary, SE, data in table and binary log. However this should be extremely rare. */ if (!table->s->tmp_table) { thd->get_stmt_da()->set_overwrite_status(true); error = trans_commit_stmt(thd) || trans_commit_implicit(thd); thd->get_stmt_da()->set_overwrite_status(false); } if (!error && m_plock) { mysql_unlock_tables(thd, *m_plock); *m_plock = nullptr; m_plock = nullptr; } if (!error && m_post_ddl_ht) { m_post_ddl_ht->post_ddl(thd); } // The fk_invalidator.invalidate operation will close tables // in its parent map: here we tell the fk_invalidator about // tables that it should NOT close, as they will be closed // elsewhere. for (auto query_table = select_tables; query_table != nullptr; query_table = query_table->next_global) { fk_invalidator.mark_for_reopen_if_added(query_table->db, query_table->table_name); } fk_invalidator.invalidate(thd); } return error; } /** Close and drop just created table in CREATE TABLE ... SELECT in case of error. @note Here we assume that the table to be closed is open only by the calling thread, so we needn't wait until other threads close the table. We also assume that the table is first in thd->open_ables and a data lock on it, if any, has been released. */ void Query_result_create::drop_open_table(THD *thd) { DBUG_TRACE; if (table->s->tmp_table) { /* Call reset here since SE may depend on this to reset its state properly. Normally this is done when calling mark_tmp_table_for_reuse(table); at the end of a statement using temporary tables. In a Query_result_set_insert object it is done by the cleanup() member function. For a non-temporary table this is done by close_thread_table(). Calling ha_reset() from close_temporary_table() is not an options since this function gets called at times (boot) when is data structures needed by handler::reset() have not yet been initialized. */ table->file->ha_reset(); close_temporary_table(thd, table, true, true); } else { assert(table == thd->open_tables); handlerton *table_type = table->s->db_type(); table->file->ha_extra(HA_EXTRA_PREPARE_FOR_DROP); close_thread_table(thd, &thd->open_tables); /* Remove TABLE and TABLE_SHARE objects for the table we have failed to create from the caches. This also nicely covers the case when addition of table to data-dictionary was not even committed. */ tdc_remove_table(thd, TDC_RT_REMOVE_ALL, create_table->db, create_table->table_name, false); if (!(table_type->flags & HTON_SUPPORTS_ATOMIC_DDL)) { /* Removal of table by quick_rm_table() below commits statement transaction as a side-effect. If the statement is not rolled back here then binlog cache (containing log(s) of new table and inserts) will be written to the binlog file. We are not allowed to rollback a statement transactions inside stored function or trigger. OTOH in such contexts only creation of temporary tables is allowed. */ trans_rollback_stmt(thd); if (thd->transaction_rollback_request) trans_rollback_implicit(thd); quick_rm_table(thd, table_type, create_table->db, create_table->table_name, 0); } #ifdef HAVE_PSI_TABLE_INTERFACE else { /* quick_rm_table() was not called, so remove the P_S table share here. */ PSI_TABLE_CALL(drop_table_share) (false, create_table->db, strlen(create_table->db), create_table->table_name, strlen(create_table->table_name)); } #endif } } void Query_result_create::abort_result_set(THD *thd) { DBUG_TRACE; /* In Query_result_insert::abort_result_set() we roll back the statement, including truncating the transaction cache of the binary log. To do this, we pretend that the statement is transactional, even though it might be the case that it was not. We roll back the statement prior to deleting the table and prior to releasing the lock on the table, since there might be potential for failure if the rollback is executed after the drop or after unlocking the table. We also roll back the statement regardless of whether the creation of the table succeeded or not, since we need to reset the binary log state. */ { const Disable_binlog_guard binlog_guard(thd); Query_result_insert::abort_result_set(thd); thd->get_transaction()->reset_unsafe_rollback_flags(Transaction_ctx::STMT); } /* possible error of writing binary log is ignored deliberately */ (void)thd->binlog_flush_pending_rows_event(true, true); if (m_plock) { mysql_unlock_tables(thd, *m_plock); *m_plock = nullptr; m_plock = nullptr; } if (table) { drop_open_table(thd); table = nullptr; // Safety } if (!(create_info->options & HA_LEX_CREATE_TMP_TABLE)) { trans_rollback_stmt(thd); /* Rollback transaction both to clear THD::transaction_rollback_request (if it is set) and to synchronize DD state in cache and on disk (as statement rollback doesn't clear DD cache of modified uncommitted objects). */ trans_rollback_implicit(thd); if (m_post_ddl_ht) m_post_ddl_ht->post_ddl(thd); } } bool Sql_cmd_insert_base::accept(THD *thd, Select_lex_visitor *visitor) { // Columns for (Item *field : insert_field_list) { if (walk_item(field, visitor)) return true; } if (!insert_many_values.empty()) { // INSERT...VALUES statement for (const List_item *row : insert_many_values) { for (Item *item : *row) { if (walk_item(item, visitor)) return true; } } } else { // INSERT...SELECT statement if (thd->lex->query_block->accept(visitor)) return true; } // Update list (on duplicate update) auto it_value = update_value_list.begin(); auto it_column = update_field_list.begin(); while (it_value != update_value_list.end() && it_column != update_field_list.end()) { Item *value = *it_value++; Item *column = *it_column++; if (walk_item(column, visitor) || walk_item(value, visitor)) return true; } return visitor->visit(thd->lex->query_block); } const MYSQL_LEX_CSTRING * Sql_cmd_insert_select::eligible_secondary_storage_engine(THD *thd) const { // ON DUPLICATE KEY UPDATE cannot be offloaded if (!update_field_list.empty()) return nullptr; // Don't use secondary storage engines for REPLACE INTO SELECT statements if (is_replace) return nullptr; return get_eligible_secondary_engine(thd); } /** Perform partition pruning for INSERT query. */ bool Sql_cmd_insert_base::prune_partitions( THD *thd, bool prune_needs_default_values, const mem_root_deque<Item *> &insert_field_list, MY_BITMAP *used_partitions, TABLE *const insert_table, COPY_INFO &info, partition_info::enum_can_prune *can_prune_partitions, bool tables_locked) { auto its = insert_many_values.begin(); uint num_partitions = insert_table->part_info->lock_partitions.n_bits; uint counter = 1; /* Pruning probably possible, all partitions are unmarked for read/lock, and we must now add them on row by row basis. Check the first INSERT value. PRUNE_DEFAULTS means the partitioning fields are only set to DEFAULT values, so we only need to check the first INSERT value, since all the rest will be in the same partition. */ if (insert_table->part_info->set_used_partition( thd, insert_field_list, /*values=*/*(*its++), info, prune_needs_default_values, used_partitions, tables_locked)) { *can_prune_partitions = partition_info::PRUNE_NO; // set_used_partition may fail. if (thd->is_error()) { return true; } } while (its != insert_many_values.end()) { const mem_root_deque<Item *> *values = *its++; counter++; /* We check pruning for each row until we will use all partitions, Even if the number of rows is much higher than the number of partitions. TODO: Cache the calculated part_id and reuse in ha_partition::write_row() if possible. */ if (*can_prune_partitions == partition_info::PRUNE_YES) { if (insert_table->part_info->set_used_partition( thd, insert_field_list, *values, info, prune_needs_default_values, used_partitions, tables_locked)) { *can_prune_partitions = partition_info::PRUNE_NO; // set_used_partition may fail. if (thd->is_error()) { return true; } } if (!(counter % num_partitions)) { /* Check if we using all partitions in table after adding partition for current row to the set of used partitions. Do it only from time to time to avoid overhead from bitmap_is_set_all() call. */ if (bitmap_is_set_all(used_partitions)) { *can_prune_partitions = partition_info::PRUNE_NO; insert_table->part_info->is_pruning_completed = true; } } } } if (*can_prune_partitions != partition_info::PRUNE_NO) { /* Only lock the partitions we will insert into. And also only read from those partitions (duplicates etc.). If explicit partition selection 'INSERT INTO t PARTITION (p1)' is used, the new set of read/lock partitions is the intersection of read/lock partitions and used partitions, i.e only the partitions that exists in both sets will be marked for read/lock. It is also safe for REPLACE, since all potentially conflicting records always belong to the same partition as the one which we try to insert a row. This is because ALL unique/primary keys must include ALL partitioning columns. */ bitmap_intersect(&insert_table->part_info->read_partitions, used_partitions); bitmap_intersect(&insert_table->part_info->lock_partitions, used_partitions); insert_table->part_info->is_pruning_completed = true; } return false; }