使用Postgres函数作为ActiveRecord模型

时间:2014-03-10 17:32:02

标签: ruby-on-rails ruby-on-rails-3 postgresql

我创建了一个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

2 个答案:

答案 0 :(得分: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

答案 1 :(得分:1)

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

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