Rails 成语避免在 has_many 中重复：通过

Question

我的 Rails 应用程序中的用户和角色之间存在标准的多对多关系：

class User < ActiveRecord::Base
  has_many :user_roles
  has_many :roles, :through => :user_roles
end

我想确保一个用户只能被分配一次任何角色。任何插入重复项的尝试都应忽略该请求，而不是引发错误或导致验证失败。我真正想要表示的是一个“集合”，其中插入集合中已经存在的元素没有任何效果。 {1,2,3} U {1} = {1,2,3}，而不是 {1,1,2,3}。

我意识到我可以这样做：

user.roles << role unless user.roles.include?(role)

或者通过创建一个包装方法（例如add_to_roles(role)），但我希望有一些惯用的方式通过关联使其自动化，以便我可以编写：

user.roles << role  # automatically checks roles.include?

它只是为我工作。这样，我就不必记住检查重复或使用自定义方法。我缺少的框架中有什么东西吗？我首先认为 has_many 的 :uniq 选项可以做到这一点，但它基本上只是“选择不同的”。

有没有办法以声明的方式做到这一点？如果没有，也许通过使用关联扩展？

以下是默认行为如何失败的示例：

 >> u = User.create
      用户创建 (0.6ms) INSERT INTO "users" ("name") VALUES(NULL)
    => #
    >> u.roles 
      角色加载 (0.5ms) SELECT * FROM "roles" LIMIT 1
      UserRole Create (0.5ms) INSERT INTO "user_roles" ("role_id", "user_id") VALUES(1, 3)
      角色加载 (0.4ms) SELECT "roles".* FROM "roles" INNER JOIN "user_roles" ON "roles".id = "user_roles".role_id WHERE (("user_roles".user_id = 3))
    => [#]
    >> u.roles 
      角色加载 (0.4ms) SELECT * FROM "roles" LIMIT 1
      UserRole Create (0.5ms) INSERT INTO "user_roles" ("role_id", "user_id") VALUES(1, 3)
    => [#, #]

Answer 1

也许可以创建验证规则

validates_uniqueness_of :user_roles

然后捕获验证异常并优雅地进行。然而，如果可能的话，这感觉真的很hacky并且非常不雅。

Answer 2

只要附加的角色是 ActiveRecord 对象，你在做什么：

user.roles << role

应该为:has_many 关联自动重复数据删除。

对于has_many :through，请尝试：

class User
  has_many :roles, :through => :user_roles do
    def <<(new_item)
      super( Array(new_item) - proxy_association.owner.roles )
    end
  end
end

如果 super 不起作用，你可能需要设置一个 alias_method_chain。

Answer 3

我认为正确的验证规则在您的 users_roles 加入模型中：

validates_uniqueness_of :user_id, :scope => [:role_id]

Answer 4

您可以在主模型中使用 validates_uniqueness_of 和覆盖

validates_uniqueness_of :user_id, :scope => [:role_id]

class User
  has_many :roles, :through => :user_roles do
    def <<(*items)
      super(items) rescue ActiveRecord::RecordInvalid
    end
  end
end

Answer 5

我认为你想做这样的事情：

user.roles.find_or_create_by(role_id: role.id) # saves association to database
user.roles.find_or_initialize_by(role_id: role.id) # builds association to be saved later

Answer 6

我今天遇到了这个问题，最终使用了#replace，它“将执行差异并仅删除/添加已更改的记录”。

因此，您需要传递现有角色的联合（这样它们就不会被删除）和您的新角色：

new_roles = [role]
user.roles.replace(user.roles | new_roles)

重要的是要注意，这个答案和接受的答案都将关联的roles 对象加载到内存中，以执行 Array diff (-) 和 union (|)。如果您要处理大量相关记录，这可能会导致性能问题。

如果这是一个问题，您可能需要先查看通过查询检查是否存在的选项，或者使用INSERT ON DUPLICATE KEY UPDATE (mysql) 类型的查询进行插入。

Answer 7

使用数组的|= Join 方法。

您可以使用 Array 的 |= 连接方法将元素添加到 Array，除非它已经存在。只需确保将元素包装在一个数组中即可。

role                  #=> #<Role id: 1, name: "1">

user.roles            #=> []

user.roles |= [role]  #=> [#<Role id: 1, name: "1">]

user.roles |= [role]  #=> [#<Role id: 1, name: "1">]

也可用于添加可能已经存在或不存在的多个元素：

role1                         #=> #<Role id: 1, name: "1">
role2                         #=> #<Role id: 2, name: "2">

user.roles                    #=> [#<Role id: 1, name: "1">]

user.roles |= [role1, role2]  #=> [#<Role id: 1, name: "1">, #<Role id: 2, name: "2">]

user.roles |= [role1, role2]  #=> [#<Role id: 1, name: "1">, #<Role id: 2, name: "2">]

在this StackOverflow answer 上发现了这种技术。

Answer 8

即使多次调用Refer rails guide，这也只会在数据库中创建一个关联。

user.roles=[Role.first]