在活动记录中添加自动递增列

Question

我正在尝试在我的 postgresql 表中添加一个基于整数的自动递增列，然后通过活动记录访问它。它不断被活动记录设置为 NULL。

明显的方法：

add_column :table_name, :id_name, :primary_key

这不起作用，因为表已经有一个主键。

接下来我尝试将其添加到 Active Record 迁移的上游部分：

execute("ALTER TABLE table_name ADD id_name SERIAL;")

这在创建表时非常有效，如果我通过数据库直接创建一个新条目，id_name 会按照人们的希望设置。但是，如果我使用 .new & .save 或 .create 通过 ActiveRecord 创建一个新条目，它会在尝试将 id_name 设置为 null 时爆炸。

我也试过了：

execute("CREATE SEQUENCE table_name_id_name_seq;")
execute("ALTER TABLE table_name ADD id_name INTEGER DEFAULT NEXTVAL('table_name_id_name_seq');")
execute("ALTER SEQUENCE table_name_id_name_seq OWNED BY table_name.id_name;")

这允许 .new & .save 以及 .create 进行条目，但结果的 id_name 为 null。

有什么方法可以说服 Rails 不要用 null 覆盖默认的 db 值？

Answer 1

这有点 hacky，但这里有一个可行的解决方案：

def up   
  add_column :table_name, :id_name, :integer
  n = 1
  TableName.all.each_with_index do |name|
    name.id_name = n
    name.save
    n += 1
  end
  execute "CREATE SEQUENCE table_name_id_name_seq START #{n} OWNED BY table_name.id_name;"
end


def next_seq_value(sequence)
  Article.connection.execute("SELECT nextval('#{sequence}')")[0]['nextval']
end


before_validation(:on => :create) do
  unless attribute_present?("id_name")
    self.id_name = next_seq_value('table_name_id_name_seq')
  end
 end

Answer 2

使用类型 serial 让数据库自动增加列的问题在于 schema.db 不支持它，因此例如，您将无法让该功能填充测试数据库。

相反，这是一种支持递增列的与数据库无关的方式：

  before_create :increment_identifier

  def increment_identifier
    sql = "SELECT COALESCE(MAX(identifier), 0) + 1 FROM claims"
    self.identifier = ActiveRecord::Base.connection.execute(sql).values[0][0]
  end

COALESCE 只是为空表返回 0 而不是 nil。而values[0][0] 是访问 ActiveRecord.Base 返回的时髦对象的必要方法。