未验证 提交 8baca31d 编写于 作者: R Ryuta Kamizono 提交者: GitHub

Bring back ability to insert zero value on primary key for fixtures (#31795)

Since #29504, mysql2 adapter lost ability to insert zero value on
primary key due to enforce `NO_AUTO_VALUE_ON_ZERO` disabled.
That is for using `DEFAULT` on auto increment column, but we can use
`NULL` instead in that case.
上级 301cd91c
......@@ -414,6 +414,9 @@ def join_to_update(update, select, key) # :nodoc:
alias join_to_delete join_to_update
private
def default_insert_value(column)
Arel.sql("DEFAULT")
end
def build_fixture_sql(fixtures, table_name)
columns = schema_cache.columns_hash(table_name)
......@@ -432,7 +435,7 @@ def build_fixture_sql(fixtures, table_name)
bind = Relation::QueryAttribute.new(name, fixture[name], type)
with_yaml_fallback(bind.value_for_database)
else
Arel.sql("DEFAULT")
default_insert_value(column)
end
end
end
......
......@@ -526,22 +526,15 @@ def default_index_type?(index) # :nodoc:
index.using == :btree || super
end
def insert_fixtures(*)
without_sql_mode("NO_AUTO_VALUE_ON_ZERO") { super }
end
def insert_fixtures_set(fixture_set, tables_to_delete = [])
iterate_over_results = -> { while raw_connection.next_result; end; }
with_multi_statements do
without_sql_mode("NO_AUTO_VALUE_ON_ZERO") do
super(fixture_set, tables_to_delete, &iterate_over_results)
end
super(fixture_set, tables_to_delete, &iterate_over_results)
end
end
private
def combine_multi_statements(total_sql)
total_sql.each_with_object([]) do |sql, total_sql_chunks|
previous_packet = total_sql_chunks.last
......@@ -580,19 +573,6 @@ def with_multi_statements
reconnect!
end
def without_sql_mode(mode)
result = execute("SELECT @@SESSION.sql_mode")
current_mode = result.first[0]
return yield unless current_mode.include?(mode)
sql_mode = "REPLACE(@@sql_mode, '#{mode}', '')"
execute("SET @@SESSION.sql_mode = #{sql_mode}")
yield
ensure
sql_mode = "CONCAT(@@sql_mode, ',#{mode}')"
execute("SET @@SESSION.sql_mode = #{sql_mode}")
end
def initialize_type_map(m = type_map)
super
......
......@@ -50,6 +50,9 @@ def exec_delete(sql, name = nil, binds = [])
alias :exec_update :exec_delete
private
def default_insert_value(column)
Arel.sql("DEFAULT") unless column.auto_increment?
end
def last_inserted_id(result)
@connection.last_id
......
......@@ -121,7 +121,7 @@ def test_insert_fixtures_set_raises_an_error_when_max_allowed_packet_is_smaller_
conn = ActiveRecord::Base.connection
mysql_margin = 2
packet_size = 1024
bytes_needed_to_have_a_1024_bytes_fixture = 855
bytes_needed_to_have_a_1024_bytes_fixture = 858
fixtures = {
"traffic_lights" => [
{ "location" => "US", "state" => ["NY"], "long_state" => ["a" * bytes_needed_to_have_a_1024_bytes_fixture] },
......@@ -199,26 +199,17 @@ def test_insert_fixtures_set_concat_total_sql_into_a_single_packet_smaller_than_
end
end
def test_no_auto_value_on_zero_is_disabled
skip unless current_adapter?(:Mysql2Adapter)
begin
fixtures = [
{ "name" => "first", "wheels_count" => 2 },
{ "name" => "second", "wheels_count" => 3 }
]
subscriber = InsertQuerySubscriber.new
subscription = ActiveSupport::Notifications.subscribe("sql.active_record", subscriber)
assert_nothing_raised do
ActiveRecord::Base.connection.insert_fixtures_set("aircraft" => fixtures)
end
expected_sql = "INSERT INTO `aircraft` (`id`, `name`, `wheels_count`) VALUES (DEFAULT, 'first', 2), (DEFAULT, 'second', 3);\n"
assert_equal expected_sql, subscriber.events.first
ensure
ActiveSupport::Notifications.unsubscribe(subscription)
def test_auto_value_on_primary_key
fixtures = [
{ "name" => "first", "wheels_count" => 2 },
{ "name" => "second", "wheels_count" => 3 }
]
conn = ActiveRecord::Base.connection
assert_nothing_raised do
conn.insert_fixtures_set({ "aircraft" => fixtures }, ["aircraft"])
end
result = conn.select_all("SELECT name, wheels_count FROM aircraft ORDER BY id")
assert_equal fixtures, result.to_a
end
def test_deprecated_insert_fixtures
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册