X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/5449cf4adcc1fad4b9f43426e6d3e4a8f65e6fbb..20992870ab1d58123712eafb4c9f8128bf471ed2:/lib/migrate.rb?ds=inline diff --git a/lib/migrate.rb b/lib/migrate.rb index 392060b60..05b3c90f2 100644 --- a/lib/migrate.rb +++ b/lib/migrate.rb @@ -158,6 +158,21 @@ module ActiveRecord def interval_constant(interval) "'#{interval}'::interval" end + + def add_index(table_name, column_name, options = {}) + column_names = Array(column_name) + index_name = index_name(table_name, :column => column_names) + + if Hash === options # legacy support, since this param was a string + index_type = options[:unique] ? "UNIQUE" : "" + index_name = options[:name] || index_name + index_method = options[:method] || "BTREE" + else + index_type = options + end + quoted_column_names = column_names.map { |e| quote_column_name(e) }.join(", ") + execute "CREATE #{index_type} INDEX #{quote_column_name(index_name)} ON #{quote_table_name(table_name)} USING #{index_method} (#{quoted_column_names})" + end end end end