具有 table 值参数的参数化查询

Question

如何在 PostgreSQL 中构建一个参数化的 query/view/stored 过程（或者不情愿地在 Java 中构建一个函数）以支持 table-values 参数？

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

一些上下文。

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

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;

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

我正在考虑将某种存储过程作为我的第一个选择，尽管据我了解 PostgreSQL 不支持 table-values 参数，因此这可能是不可能的。

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

如有任何提示或线索，我们将不胜感激。

Answer 1

您不能在视图中执行此操作，但您可以创建一个 PL/pgSQL 函数，该函数采用 table 或视图名称和 returns 符合查询定义的记录。由于事先不知道 table 名称，您必须使用 RETURN QUERY EXECUTE:

对传入的 table 名称动态执行查询

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;

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

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

Answer 2

看来你的主要问题是你的存储函数有一个 table 值的参数，这在 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

具有 table 值参数的参数化查询

Parameterised queries with table-valued parameters

sql

postgresql

jooq

以上使用 jOOQ