带有表值参数的参数化查询答案

【问题标题】：Parameterised queries with table-valued parameters带有表值参数的参数化查询
【发布时间】：2015-05-07 02:21:29
【问题描述】：

如何在 PostgreSQL 中构建参数化查询/视图/存储过程（或者，不情愿地，Java 中的函数）以支持表值参数？

我当前的工具链是带有 jooq 库的 PostgreSQL、JDBC 和 Java。

一些上下文。

我有一个（实际上相当复杂的）查询，它定义了一些统计数据的聚合处理。为简单起见，我们假设它的形式为：

create view aggregated_data as
  select table1.c1, sum(table2.c2) 
  from table1 
  inner join table2 on table1.c1 = table2.c1
  group by table1.c1;

我想要做的是参数化这个定义，这样我就可以将“table2”作为参数传递，因为我希望传递的“table2”只包含基于其他条件的行。

更明确地说，我追求的是：

create view aggregated_data(some_table_valued_parameter) as
  select table1.c1, sum(table2.c2) 
  from table1 
  inner join ( 
     select c1, c2 from some_table_valued_parameter
  ) table2 on table1.c1 = table2.c1
  group by table1.c1;

如何定义一个 something 让我保留所有的脚手架并只传入 some_table_valued_parameter？

我正在考虑将某种存储过程作为我的第一个选择，尽管据我了解 PostgreSQL 不支持表值参数，所以它可能是不可能的。

第二个选项（我很确定它会起作用）是使用 JOOQ 和 Java 函数来动态构建查询。不过，这不太可取，因为该数据库旨在支持人们在控制台登录并输入 SQL，而且我希望他们也能够从脚手架中受益。

任何提示或线索将不胜感激。

【问题讨论】：

很有趣。就像您试图从可能的许多视图中考虑包含聚合的复杂查询一样。据我所知，使用 SQL92 的唯一可能性是省略聚合并创建一个仅包含复杂查询（加上连接列）的视图。对于每个可能的 table2，您都可以加入该视图并聚合。
我知道返回表的函数。但从未听说过他们将表格作为论据。
除非 'some_table_valued_parameter' 子查询比声明的更复杂，否则您可以简单地将 'some_table_valued_parameter' 连接到 'table1'。

标签： sql postgresql jooq

【解决方案1】：

看来你的主要问题是你的存储函数有一个表值参数，这在 PostgreSQL 中确实不支持。但是您可以使用记录数组，这与更多样板（可能还有一些性能影响）基本相同：

CREATE TYPE my_rec AS (
  c1 BIGINT,
  c2 BIGINT
);

然后：

CREATE FUNCTION my_func(almost_a_table my_rec[]) 
RETURNS TABLE (c1 BIGINT, c2 BIGINT)
AS $$
BEGIN
  RETURN QUERY
  SELECT table1.c1, sum(table2.c2)
  FROM table1
  INNER JOIN (
    SELECT u.c1, u.c2 FROM unnest(almost_a_table) u
  ) table2 ON table1.c1 = table2.c1
  GROUP BY table1.c1;
END
$$ LANGUAGE plpgsql;

如何使用此功能：

SELECT *
FROM my_func((
  SELECT array_agg(row(t.a, t.b)::my_rec)
  FROM some_table t
))

以上使用jOOQ

注意，由于您提到您的堆栈中有 jOOQ，因此在使用 jOOQ 时这种方法也可能是理想的，因为 jOOQ 的代码生成器将为您生成所有样板代码，以便您可以访问 my_func 从你的 jOOQ 查询是一样的。详情在这里：

http://www.jooq.org/doc/latest/manual/sql-building/table-expressions/table-valued-functions

【讨论】：

哇，卢卡斯。太棒了。非常感谢。
非常感谢。卢卡斯。我如何传递静态记录而不是从表中选择。像这样：SELECT array_agg((row(3,3),row(4,4))::my_rec) 在函数调用中。
@UdeetSolanki：array[row(3,3),row(4,4)]::my_rec[]。不要使用array_agg
酷。谢谢。像魅力一样工作

【解决方案2】：

您不能在视图中执行此操作，但您可以创建一个 PL/pgSQL 函数，该函数采用表或视图名称并返回符合查询定义的记录。由于事先不知道表名，所以必须对使用RETURN QUERY EXECUTE传入的表名动态执行查询：

CREATE FUNCTION aggregated_data(nm name) RETURNS TABLE (c1 integer, c2 float) AS $$
BEGIN
  RETURN QUERY EXECUTE 
   'SELECT table1.c1, sum(table2.c2) ' ||
   'FROM table1 ' ||
   'INNER JOIN ( ' ||
   '  SELECT c1, c2 FROM ' || quote_ident(nm) ||
   ') table2 ON table2.c1 = table1.c1 ' ||
   'GROUP BY table1.c1';
END; $$ LANGUAGE plpgsql STRICT;

由于此函数返回一个关系，因此您可以（实际上应该）在较大查询的 FROM 子句中使用此函数，就像使用视图一样：

SELECT t.*, ad.c2
FROM t
JOIN aggregated_data('relation_with_c1_and_c2_columns') ad ON ad.c1 = t.id;

【讨论】：