Chris@1464: class AddUniqueIndexOnCustomFieldsProjects < ActiveRecord::Migration
Chris@1464:   def up
Chris@1464:     table_name = "#{CustomField.table_name_prefix}custom_fields_projects#{CustomField.table_name_suffix}"
Chris@1464:     duplicates = CustomField.connection.select_rows("SELECT custom_field_id, project_id FROM #{table_name} GROUP BY custom_field_id, project_id HAVING COUNT(*) > 1")
Chris@1464:     duplicates.each do |custom_field_id, project_id|
Chris@1464:       # Removes duplicate rows
Chris@1464:       CustomField.connection.execute("DELETE FROM #{table_name} WHERE custom_field_id=#{custom_field_id} AND project_id=#{project_id}")
Chris@1464:       # And insert one
Chris@1464:       CustomField.connection.execute("INSERT INTO #{table_name} (custom_field_id, project_id) VALUES (#{custom_field_id}, #{project_id})")
Chris@1464:     end
Chris@1464: 
Chris@1464:     if index_exists? :custom_fields_projects, [:custom_field_id, :project_id]
Chris@1464:       remove_index :custom_fields_projects, [:custom_field_id, :project_id]
Chris@1464:     end
Chris@1464:     add_index :custom_fields_projects, [:custom_field_id, :project_id], :unique => true
Chris@1464:   end
Chris@1464: 
Chris@1464:   def down
Chris@1464:     if index_exists? :custom_fields_projects, [:custom_field_id, :project_id]
Chris@1464:       remove_index :custom_fields_projects, [:custom_field_id, :project_id]
Chris@1464:     end
Chris@1464:     add_index :custom_fields_projects, [:custom_field_id, :project_id]
Chris@1464:   end
Chris@1464: end