从 postgresql 中的 table 推断行类型

Question

我的应用程序使用多个模式跨数据库对租户进行分区以提高性能。我正在尝试创建一个 plpgsql 函数，它将根据给定 table 的所有应用程序模式的联合给我一个任意结果集。这是我目前所拥有的（受 this blog post 启发）：

CREATE OR REPLACE FUNCTION app_union(tbl text) RETURNS SETOF RECORD AS $$
DECLARE
  schema RECORD;
  sql TEXT := '';
BEGIN
  FOR schema IN EXECUTE 'SELECT distinct schema FROM tenants' LOOP
    sql := sql || format('SELECT * FROM %I.%I %s UNION ALL ', schema.schema, tbl);
  END LOOP;

  RETURN QUERY EXECUTE left(sql, -11);
END
$$ LANGUAGE plpgsql;

这很好用，但必须在末尾使用行类型定义来调用：

select * from app_union('my_table') t(id uuid, name text, ...);

那么，如何在不提供行类型的情况下调用我的函数？

我知道我可以使用 information_schema.columns 自省我的 table，但我不确定如何在没有大量 case 语句的情况下动态生成类型声明 (columns不会像 pg_indexes 那样报告定义 sql。

即使我可以动态生成行声明，看来我还是必须将它作为动态 sql 附加到我以前的函数调用中，这 chicken/eggs 返回 a 的问题来自函数的任意类型的结果集。

Answer 1

您可以将 table 作为字符串提供，而不是将其提供为类型 anyelement 以指定返回数据的实际类型，然后推断 table 的名称使用 pg_typeof。您还可以使用 string_agg 而不是循环来构建您的 sql:

CREATE OR REPLACE FUNCTION app_union(tbl anyelement)
RETURNS setof anyelement AS $$
BEGIN
  return query execute string_agg(
    distinct format('select * from %I.%I', schema, pg_typeof(tbl)::text), 
    ' union all '
  ) from tenants;
END
$$ LANGUAGE plpgsql;

select * from app_union(null::my_table);

Simplified example

从 postgresql 中的 table 推断行类型

Infer row type from table in postgresql

sql

postgresql

dynamic