ruby-on-rails - 通过查询将 has_many 和 has_many 合并在一起

Question

这似乎是一个典型的问题，但很难搜索。

我想选择用户通过 has_many 拥有的项目和用户通过 has_many through 关联的项目。

考虑以下模型:

class User < ActiveRecord::Base
  has_many :projects,
           inverse_of: :owner

  has_many :project_associations,
           class_name: 'ProjectUser',
           inverse_of: :user

  has_many :associated_projects,
           through: :project_associations,
           source: :project
end

class Project < ActiveRecord::Base
  belongs_to :owner,
             class_name: 'User',
             foreign_key: :owner_id,
             inverse_of: :projects

  has_many :user_associations,
           class_name: 'ProjectUser',
           inverse_of: :project

  has_many :associated_users,
           through: :user_associations,
           source: :user
end

class ProjectUser < ActiveRecord::Base
  belongs_to :project,
             inverse_of: :user_associations

  belongs_to :user,
             inverse_of: :project_associations
end

使用多个查询执行此操作很简单:

user = User.find(1)
all_projects = user.projects + user.associated_projects

但我怀疑它可以使用 Arel 进行优化到单个查询中。

---

编辑:

我第一次尝试使用 find_by_sql 方法解决问题是:

Project.find_by_sql([
  'SELECT "projects".* FROM "projects" ' \
  'INNER JOIN "project_users" ' \
  'ON "projects"."id" = "project_users"."project_id" ' \
  'WHERE "project_users"."user_id" = :user_id ' \
  'OR "projects"."owner_id" = :user_id',
  { user_id: 1 }
])

这会产生我期望的结果，但我想避免使用 find_by_sql 而是让 Arel 构建 SQL。

Answer 1

这应该有效:

Project.joins(:user_associations)
       .where("projects.owner_id = ? OR project_users.user_id = ?", id, id)

您可以将其放入名为 User#all_projects 的方法中。

你也可以使用 UNION ，尽管在这里看起来有点矫枉过正。

只有一个警告:根据我的经验，这种连接模型的两条备用路径的结构(其中一条路径是连接表，另一条路径是直接外键)会造成很多麻烦。您将不得不一直处理这两种情况，并且在复杂查询中，OR 可能会使数据库查询计划器感到困惑。如果可以，您可能需要重新考虑。我认为，如果您删除 owner_id 列并改为使用 project_users 上的 role 列来标识所有者，我认为您会更少头疼。

编辑 或在 Arel 中:

Project.joins(:user_associations)
       .where(Project.arel_table[:owner_id].eq(id).or(
              ProjectUser.arel_table[:user_id].eq(id)))