如何减少Rails应用中的SQL请求？控制器优化遇阻求助

嘿，我懂你现在的困扰——在Rails里揪出那些拖慢速度的SQL请求确实头疼，尤其是用循环遍历用户对象统计数据的时候，一不小心就触发N+1查询，性能唰唰往下掉。你朋友说的用哈希/数组存储数据来优化的思路是对的，但关键是要先把数据高效加载到内存再做统计，或者直接让数据库帮你完成计算（数据库在这类计数逻辑上比Ruby高效多了）。

先看看你当前代码的核心问题：

• 遍历x.each do |user|时，如果x是未预加载的ActiveRecord集合，每访问user.facebook_token都会触发一次单独的SQL查询，这就是典型的N+1问题
• Ruby层面循环累加统计，效率远不如直接让数据库处理这类简单计数逻辑

下面给你几个优化方案，按推荐程度排序：

方案1：直接用数据库查询统计（最优） #

把统计逻辑交给数据库，只需要一次SQL请求就能拿到结果，完全避免Ruby循环和N+1问题：

# 统计使用Facebook登录的用户数
def fb_connection(users)
  users.where.not(facebook_token: nil).count
end

# 统计使用邮箱登录的用户数（假设业务逻辑是：无facebook_token即为邮箱登录）
def email_connection(users)
  users.where(facebook_token: nil).count
end

如果你的count方法只是单纯调用var.count，其实完全可以删掉，直接用上面的写法就行。

方案2：加载数据到数组后再统计（适合复杂业务逻辑） #

如果你的统计逻辑比单纯计数复杂（比如需要多字段判断），可以先把需要的字段一次性加载到内存数组里，再在Ruby层面处理，这样也能避免N+1：

def fb_connection(users)
  # 用pluck一次性取出所有facebook_token，返回一个数组，只发一次SQL
  facebook_tokens = users.pluck(:facebook_token)
  # 统计非空值的数量（compact会去掉nil元素）
  facebook_tokens.compact.size
end

def email_connection(users)
  facebook_tokens = users.pluck(:facebook_token)
  # 统计空值的数量
  facebook_tokens.count(nil)
end

pluck只会查询你需要的字段，不会实例化整个User对象，比加载完整的ActiveRecord集合更省内存。

方案3：把逻辑放到模型里（符合Rails MVC原则） #

控制器里尽量别堆业务逻辑，把这些统计逻辑放到User模型里，代码更整洁，复用性也更高：

# app/models/user.rb
class User < ApplicationRecord
  # 定义scope，方便链式调用
  scope :with_facebook_login, -> { where.not(facebook_token: nil) }
  scope :with_email_login, -> { where(facebook_token: nil) }

  # 或者定义类方法
  def self.count_facebook_logins
    with_facebook_login.count
  end

  def self.count_email_logins
    with_email_login.count
  end
end

然后控制器里直接调用，代码会非常简洁：

def some_action
  @facebook_login_count = User.count_facebook_logins
  @email_login_count = User.count_email_logins
end

额外提醒：避免N+1的关键 #

不管用哪种方案，确保你获取用户集合的时候，不要加载不必要的字段。比如如果只需要统计facebook_token，就用User.select(:facebook_token)或者User.pluck(:facebook_token)，这样能减少数据库传输的数据量，进一步提升速度。

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