Ruby on Rails：如何判断ActiveRecord模型是否已被作用域限定？

问题核心在于：你只给模型类（比如User）添加了sample方法，但User.where(...)返回的是ActiveRecord::Relation实例，它调用的是Relation自身的sample方法，而非你定义的模型类方法。因此需要同时给模型类和Relation都添加自定义的sample方法，并在方法内区分调用上下文。

调整后的代码 #

module Sampler
  def self.included(base)
    if base == ActiveRecord::Base
      # 给所有ActiveRecord模型添加类方法
      base.extend(ClassMethods)
      # 给ActiveRecord::Relation添加实例方法（处理作用域后的调用）
      ActiveRecord::Relation.include(RelationMethods)
    end
  end

  module ClassMethods
    def sample(n=1)
      adapter = connection.adapter_name
      
      # 未作用域的模型类逻辑（比如User.sample）
      if adapter == 'PostgreSQL'
        # 示例：用PostgreSQL的TABLESAMPLE优化随机查询
        from("#{table_name} TABLESAMPLE SYSTEM(#{[100.0/n, 100].min}::float)").limit(n)
      else
        order_random(adapter).limit(n)
      end
    end

    private

    def order_random(adapter)
      case adapter
      when 'MySQL', 'Mysql2'
        order('RAND()')
      else
        order('RANDOM()')
      end
    end
  end

  module RelationMethods
    def sample(n=1)
      adapter = klass.connection.adapter_name
      
      # 已作用域的Relation逻辑（比如User.where(verified: true).sample）
      if adapter == 'PostgreSQL'
        # 带条件的作用域无法直接用TABLESAMPLE，回退到ORDER BY RANDOM()
        order('RANDOM()').limit(n)
      else
        order_random(adapter).limit(n)
      end
    end

    private

    def order_random(adapter)
      case adapter
      when 'MySQL', 'Mysql2'
        order('RAND()')
      else
        order('RANDOM()')
      end
    end
  end
end

# 在Rails初始化文件（如config/initializers/sampler.rb）中引入
ActiveRecord::Base.include(Sampler)

关键说明 #

• 区分调用上下文：
  • 调用User.sample时，执行ClassMethods中的sample，此时self是模型类（如User），属于未作用域场景。
  • 调用User.where(verified: true).sample时，执行RelationMethods中的sample，此时self是ActiveRecord::Relation实例，属于已作用域场景。
• PostgreSQL特殊处理：未作用域时用TABLESAMPLE提升性能，带条件的作用域无法使用该语法，因此回退到ORDER BY RANDOM()。
• 代码复用：把不同数据库的随机排序逻辑抽成私有方法，避免重复代码。

替代合并逻辑（可选） #

如果想把逻辑合并到单个方法中，可通过对象类型判断区分场景：

def sample(n=1)
  adapter = ::ActiveRecord::Base.connection.adapter_name
  
  if self.is_a?(ActiveRecord::Relation)
    # 已作用域逻辑
    adapter == 'PostgreSQL' ? order('RANDOM()').limit(n) : order_random(adapter).limit(n)
  else
    # 未作用域模型类逻辑
    adapter == 'PostgreSQL' ? from("#{table_name} TABLESAMPLE SYSTEM(#{100.0/n}::float)").limit(n) : order_random(adapter).limit(n)
  end
end

注意：这种方式需要确保该方法同时存在于模型类和Relation中，否则会出现方法未定义的问题。

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