Question

我正在寻找在rails中编写迁移的方法，这些迁移可以对数据库多次执行而不会失败。

例如，假设我有这样的迁移：

class AddUrlToProfile < ActiveRecord::Migration
  def self.up
    add_column :profile, :url, :string
  end

  def self.down
    remove_column :profile, :url
  end
end

如果Profile表中已经存在url列（例如，如果schema.rb被意外修改），则我的迁移将失败，说明它是重复的！

那么，只有在迫不得已的情况下，如何执行这种迁移呢？

谢谢

Answer 1

你可以这样做：

class AddUrlToProfile < ActiveRecord::Migration
  def self.up
    Profile.reset_column_information
    add_column(:profile, :url, :string) unless Profile.column_names.include?( url )

  end

  def self.down
    Profile.reset_column_information
    remove_column(:profile, :url) if Profile.column_names.include?( url )
  end
end

这将在开始之前重置列信息，确保Profile模型具有实际表中的最新列信息。然后，它只会在列不存在的情况下添加该列。down函数也会发生同样的情况，但它只删除存在的列。

如果您有多个这样的用例，您可以将代码分解为一个函数，并在迁移中重用它。

Answer 2

对于Rails3.X，是否存在column_exists？（：table_name，：column_name）方法。

对于Rails2.X，您可以使用以下命令检查列的存在：

columns("<table name>").index {|col| col.name == "<column name>"}

…或者，如果您不在迁移文件中：

ActiveRecord::Base.connection.columns("<table name>").index {|col| col.name == "<column name>"}

如果返回nil，则不存在这样的列。如果它返回一个Fixnum，那么该列确实存在。当然，如果您希望不仅仅通过列的名称来标识列，您可以在{…}之间放置更多选择性参数，例如：

{ |col| col.name == "foo" and col.sql_type == "tinyint(1)" and col.primary == nil }

Answer 3

这应该有效

def self.table_exists?(name)
  ActiveRecord::Base.connection.tables.include?(name)
end

if table_exists?(:profile) && !Profile.column_names.include?("url")
  add_column :profile, :url, :string
end

Answer 4

Wrapping my migration in a conditional worked for me. Rails 4.X

class AddUrlToProfile < ActiveRecord::Migration
  unless Profile.column_names.include?("url")
    def self.up
      add_column :profile, :url, :string
    end

    def self.down
      remove_column :profile, :url
    end
  end
end

友情链接