使用 Postgres 函数作为 ActiveRecord 模型

Posted 2023-03-25

【中文标题】使用 Postgres 函数作为 ActiveRecord 模型

【英文标题】：Use Postgres Function as ActiveRecord Model

【发布时间】：2014-03-10 17:32:02

【问题描述】：

我创建了一个 Postgres 函数，我正在使用它来执行一个复杂的查询，该查询连接了许多必须由动态日期字段过滤的表。

该函数完美运行，允许我执行"SELECT * FROM trail_for_date('2014-01-01')" 之类的查询并返回一个表。

Postgres 文档中关于函数的简化示例：

CREATE FUNCTION sum_n_product_with_tab (x int)
RETURNS TABLE(sum int, product int) AS $$
    SELECT $1 + tab.y, $1 * tab.y FROM tab;
$$ LANGUAGE SQL;

如何将此返回表用作 Rails/Ruby 模型，其中函数的参数是动态的？

类似下面的东西（显然行不通）：

class SimplifiedExample < ActiveRecord::Base
  self.table_name = 'sum_n_product_with_tab(:dynamic_input)'
end

【问题讨论】：

这是我原来的路线，但在我的查询中，我必须加入大约五个表，它们都在一个日期进行过滤并根据该日期进行计算......所以我需要能够传递某种变量或使用函数来完成此操作（我认为）。

我想我可以，尽管它有 100 多行查询，而且我必须使我的查询与数据库人员实现的数据库功能保持同步。

或者你可以使用 arel Model.select(Arel::Nodes::NamedFunction.new('sum_n_product_with_tab', 6)).to_sql

【参考方案1】：

创建一个包含您需要的数据的视图，然后您可以轻松地创建一个 ActiveRecord 模型来访问它。

您没有提供数据的具体细节，但作为一个简单的例子，在 Postgres 中创建您的视图以收集您的数据；

  create or replace view data_trails as
    select t.*, td.trail_date from trails t
    join trail_dates td on (td.trail_id = t.id)

然后创建你的模型

  class DataTrail < ActiveRecord::Base
    scope :on_date, -> (date)  where(trail_date: date) 
  end

  DataTrail.on_date(Date.today)

您可以在 Enterprise Rails 手册中找到更多信息。现在有点过时了，但原则是合理的。

http://enterpriserails.chak.org/full-text/chapter-11-view-backed-models

【讨论】：

视图不适合我，因为函数需要函数内部子查询的参数。如果我放弃子查询并在函数执行后进行过滤（如此处建议），那么函数将太慢，因为它必须计算表的所有行。

【参考方案2】：

创建一个与函数输出具有相同列的虚拟表：

CREATE TABLE report.compliance_year (
year TIMESTAMP,
compliance NUMERIC(20,2),
fund_id INT);

创建你的模型：

class Visualization::ComplianceByYear < ActiveRecord::Base
    self.table_name = 'report.compliance_year'
    def compliance_by_year(fund_id)
        Visualization::ComplianceByYear.find_by_sql(["
            SELECT year, compliance, fund_id
              FROM report.usp_compliance_year(ARRAY[?])", fund_id])
    end
end 

在您的控制器中引用它并使用函数调用结果填充它：

def visualizations
  @compliancebyyear = Visualization::ComplianceByYear.new()
  @compliancefunds = @compliancebyyear.compliance_by_year(current_group.id)
end