如何急切加载与 current_user 的关联？

Question

我在 Rails 应用程序中使用 Devise 进行身份验证。我想在我的一些控制器中加载一些与用户相关的模型。像这样的事情：

class TeamsController < ApplicationController

  def show
    @team = Team.includes(:members).find params[:id]
    current_user.includes(:saved_listings)

    # normal controller stuff
  end
end

我怎样才能实现这个目标？

Answer 1

我遇到了同样的问题，虽然每个人都一直说没有必要这样做，但我发现有必要这样做，就像你一样。所以这对我有用：

# in application_controller.rb:
def current_user
  @current_user ||= super && User.includes(:saved_listings).find(@current_user.id)
end

请注意，这将加载所有控制器中的关联。对于我的用例，这正是我所需要的。如果您确实只希望在某些控制器中使用它，则必须对此进行更多调整。

这还将调用 User.find 两次，但对于查询缓存来说这应该不是问题，并且由于它可以防止大量额外的数据库命中，因此仍然可以提高性能。

Answer 2

覆盖 serialize_from_session 在您的 User 模型中。

class User
  devise :database_authenticatable

  def self.serialize_from_session key, salt
    record = where(id: key).eager_load(:saved_listings, roles: :accounts).first
    record if record && record.authenticatable_salt == salt
  end
end

然而，这将立即加载所有请求。

Answer 3

我想添加我认为更好的解决方案。正如评论中所述，现有的解决方案可能会通过 find 请求两次访问您的数据库。相反，我们可以使用 ActiveRecord::Associations::Preloader 来利用 Rails 加载关联的工作：

def current_user
  @current_user ||= super.tap do |user|
    ::ActiveRecord::Associations::Preloader.new.preload(user, :saved_listings)
  end
end

这将重用内存中的现有模型，而不是再次连接和查询整个表。

Answer 4

为什么不在模型上使用 default_scope 来实现呢？

像这样：

Class User  < ActiveRecord::Base
  ...
  default_scope includes(:saved_listings)
  ...
end