如何让PostgreSQL函数支持Rails模型的链式查询？

这个问题我之前也碰到过！用find_by_sql返回实例数组确实没法继续用ActiveRecord的链式方法，改代码成本太高了，不过有几个很实用的解决方案，能让你继续享受ActiveRecord的便捷：

1. 直接用from方法包装函数调用 #

最直接的方式就是用ActiveRecord的from方法，把PostgreSQL函数的调用当成一个“虚拟表”来使用，这样返回的是ActiveRecord::Relation对象，完全支持链式调用：

# 直接调用，返回Relation，可继续链式操作
results = Model.from('my_pg_function(?, ?)', param1, param2)
               .where(status: 'active')
               .order(created_at: :desc)
               .limit(10)

# 甚至可以直接用find方法
specific_record = Model.from('my_pg_function(?, ?)', param1, param2).find(123)

这里的关键是from方法接受SQL片段和参数占位符，Rails会自动处理参数转义，避免SQL注入风险，而且返回的Relation可以像操作普通数据库表一样，组合各种查询条件。

2. 封装成模型类方法 #

如果这个函数调用在多处使用，建议把它封装成模型的类方法，让代码更整洁、可读性更高：

class Model < ApplicationRecord
  def self.from_pg_function(param1, param2)
    from('my_pg_function(?, ?)', param1, param2)
  end
end

# 调用起来更简洁
Model.from_pg_function(100, '2024-01-01')
     .where(user_id: current_user.id)
     .paginate(page: params[:page])

3. 用Scope封装带参数的查询 #

如果你习惯用Rails的Scope风格，也可以用lambda定义带参数的Scope，这样更符合Rails的最佳实践：

class Model < ApplicationRecord
  scope :using_pg_function, ->(param1, param2) { from('my_pg_function(?, ?)', param1, param2) }
end

# 调用方式
Model.using_pg_function(50, 'draft')
     .order(updated_at: :asc)
     .pluck(:name)

为什么这些方法有效？ #

find_by_sql会直接执行SQL并返回实例数组，而from方法是构建一个查询的“基础来源”，返回的是ActiveRecord::Relation——这是Rails查询的核心对象，支持延迟加载、链式组合各种查询条件，完全和操作普通表的体验一致。

备注：内容来源于stack exchange，提问作者Cristiano Mendonça