Commit 92642031 authored by Felipe Artur's avatar Felipe Artur

change add_concurrent_index function arguments

parent 8450fe30
...@@ -11,7 +11,7 @@ module Gitlab ...@@ -11,7 +11,7 @@ module Gitlab
# add_concurrent_index :users, :some_column # add_concurrent_index :users, :some_column
# #
# See Rails' `add_index` for more info on the available arguments. # See Rails' `add_index` for more info on the available arguments.
def add_concurrent_index(*args) def add_concurrent_index(table_name, column_name, options = {})
if transaction_open? if transaction_open?
raise 'add_concurrent_index can not be run inside a transaction, ' \ raise 'add_concurrent_index can not be run inside a transaction, ' \
'you can disable transactions by calling disable_ddl_transaction! ' \ 'you can disable transactions by calling disable_ddl_transaction! ' \
...@@ -19,14 +19,10 @@ module Gitlab ...@@ -19,14 +19,10 @@ module Gitlab
end end
if Database.postgresql? if Database.postgresql?
if args[2].present? options = options.merge({ algorithm: :concurrently })
args[2].merge!({ algorithm: :concurrently })
else
args << { algorithm: :concurrently }
end
end end
add_index(*args) add_index(table_name, column_name, options)
end end
# Updates the value of a column in batches. # Updates the value of a column in batches.
......
...@@ -16,14 +16,21 @@ describe Gitlab::Database::MigrationHelpers, lib: true do ...@@ -16,14 +16,21 @@ describe Gitlab::Database::MigrationHelpers, lib: true do
end end
context 'using PostgreSQL' do context 'using PostgreSQL' do
it 'creates the index concurrently' do before { expect(Gitlab::Database).to receive(:postgresql?).and_return(true) }
expect(Gitlab::Database).to receive(:postgresql?).and_return(true)
it 'creates the index concurrently' do
expect(model).to receive(:add_index). expect(model).to receive(:add_index).
with(:users, :foo, algorithm: :concurrently) with(:users, :foo, algorithm: :concurrently)
model.add_concurrent_index(:users, :foo) model.add_concurrent_index(:users, :foo)
end end
it 'creates unique index concurrently' do
expect(model).to receive(:add_index).
with(:users, :foo, { algorithm: :concurrently, unique: true })
model.add_concurrent_index(:users, :foo, unique: true)
end
end end
context 'using MySQL' do context 'using MySQL' do
...@@ -31,7 +38,7 @@ describe Gitlab::Database::MigrationHelpers, lib: true do ...@@ -31,7 +38,7 @@ describe Gitlab::Database::MigrationHelpers, lib: true do
expect(Gitlab::Database).to receive(:postgresql?).and_return(false) expect(Gitlab::Database).to receive(:postgresql?).and_return(false)
expect(model).to receive(:add_index). expect(model).to receive(:add_index).
with(:users, :foo) with(:users, :foo, {})
model.add_concurrent_index(:users, :foo) model.add_concurrent_index(:users, :foo)
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment