Rails模型生成唯一数据库存储值的实现方法咨询

我懂你的需求——你不是要在保存时检查现有值是否唯一，而是要自动生成一个值，确保它在数据库表中没有重复，要是重复了就重新生成，直到拿到一个唯一的。你当前的after_initialize方法只是随机生成了个字符串，但完全没做唯一性校验，对吧？

下面给你一个可靠的实现方案：

核心实现代码 #

我们可以写一个专门的方法来生成并校验唯一值，然后在初始化时调用它：

after_initialize :init

def init
  # 只有当字段还没有值的时候才生成（避免更新时覆盖已有值）
  generate_unique_some_field unless self.some_field.present?
end

private

def generate_unique_some_field
  # 循环生成值，直到找到数据库中不存在的
  loop do
    candidate_value = rand(1..99_999_999).to_s
    # 检查当前模型类对应的表中是否存在这个值
    break self.some_field = candidate_value unless self.class.exists?(some_field: candidate_value)
  end
end

代码解释 #

• loop会持续运行，直到我们找到一个不重复的候选值
• self.class.exists?(some_field: candidate_value)直接查询数据库，确认这个值没有被使用过
• unless self.some_field.present?保证只有创建新记录时才生成值，更新记录时不会改动已有的some_field

高并发场景的额外保障 #

如果你的应用有较高的并发创建需求，可能会出现两个请求同时生成同一个值的竞态问题（比如A和B同时生成了"12345"，A先保存，B后保存时就会冲突）。这时候建议给some_field字段加数据库唯一索引，然后再加上保存重试机制：

第一步：添加数据库唯一索引 #

在迁移文件里加上：

add_index :your_table_name, :some_field, unique: true

第二步：添加保存重试方法 #

在模型里新增：

def save_with_unique_retry(retries = 3)
  save
rescue ActiveRecord::RecordNotUnique
  if retries > 0
    # 重新生成一个值，再尝试保存
    generate_unique_some_field
    save_with_unique_retry(retries - 1)
  else
    # 重试次数用完还是失败，抛出异常让上层处理
    raise
  end
end

之后创建记录时，调用save_with_unique_retry代替save就可以了。

内容的提问来源于stack exchange，提问作者Vitalina