Rails 迁移：可以使用动态代码 update_all 吗？

Question

我想在表中添加一个新字段。

我的用户模型中的新“secret_code”字段应等于 Digest::SHA1.hexdigest([Time.now, rand].join)[1..12]。

我想做的是生成一个迁移，将字段添加到表中，并使用（某种）唯一的“secret_code”填充我的现有用户。

class AddSecretCodeToUsers < ActiveRecord::Migration
  def self.up
    add_column :users, :secret_code, :string
    User.update_all ["secret_code =?", Digest::SHA1.hexdigest([Time.now, rand].join)[1..12]]
  end

  def self.down
    remove_column :users, :secret_code
  end
end

问题是此迁移使用相同的密码填充所有现有用户！

一种解决方案是不使用 update_all 并运行循环来获取每个用户并向每个用户发送更新，但在这种情况下，我的迁移将非常慢。

有没有办法将“唯一”随机值发送到 update_all 方法？

谢谢， 奥古斯托

Answer 1

尝试将其更改为 Digest::SHA1.hexdigest([Time.now, rand].to_s) 但我个人会为上述内容创建一个 rake 任务，因为它并不是真正的迁移。

您的 rake 任务可以

User.all.each do |u|
  u.update_attribute(:secret_code, Digest::SHA1.hexdigest([Time.now, rand].to_s))
end

但是，对于您的迁移，我还会添加 t.string :secret_code, :default =>摘要::SHA1.hexdigest([Time.now, rand].to_s)
添加到属性中，以便将其添加到新创建的记录上。

Answer 2

对于 MySQL，您可以将其直接放入 self.up 中：

connection.execute(%Q{
    update users
    set secret_code = substring(sha1(rand()) from 1 for 12)
})

PostgreSQL 默认情况下不支持 SHA1，但它确实具有 MD5，这可能已经足够了：

connection.execute(%Q{
    update users
    set secret_code = substring(md5(random()::varchar) from 1 for 12)
})

如果您有 pgcrypto 软件包后，您就可以对两者使用 SHA1。

这两者都将让数据库完成所有工作，并避免往返整个表的所有开销。如果你想混合时间，你也可以尝试一下你的哈希值：

md5(random()::varchar || now()::varchar) -- PostgreSQL
sha(rand()            || now()         ) -- MySQL