Skip to content

Commit b101837

Browse files
committed
Update the #remove_column to work with the #remove_columns alias for removing multiple columns at one time.
1 parent cb89240 commit b101837

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

lib/active_record/connection_adapters/sqlserver_adapter.rb

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -518,11 +518,13 @@ def add_column(table_name, column_name, type, options = {})
518518
remove_sqlserver_columns_cache_for(table_name)
519519
end
520520

521-
def remove_column(table_name, column_name)
522-
remove_check_constraints(table_name, column_name)
523-
remove_default_constraint(table_name, column_name)
524-
remove_indexes(table_name, column_name)
525-
execute "ALTER TABLE #{quote_table_name(table_name)} DROP COLUMN #{quote_column_name(column_name)}"
521+
def remove_column(table_name, *column_names)
522+
column_names.flatten.each do |column_name|
523+
remove_check_constraints(table_name, column_name)
524+
remove_default_constraint(table_name, column_name)
525+
remove_indexes(table_name, column_name)
526+
execute "ALTER TABLE #{quote_table_name(table_name)} DROP COLUMN #{quote_column_name(column_name)}"
527+
end
526528
remove_sqlserver_columns_cache_for(table_name)
527529
end
528530

0 commit comments

Comments
 (0)