从 postgresql 中的表推断行类型

如何解决从 postgresql 中的表推断行类型

我的应用程序使用多个架构在整个数据库中对租户进行分区以提高性能。我正在尝试创建一个 plpgsql 函数，该函数将根据给定表的所有应用程序模式的联合为我提供任意结果集。以下是我目前所拥有的（灵感来自 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 内省我的表，但是我不确定如何在没有大量 case 语句的情况下动态生成类型声明（columns 不报告定义 sql例如，pg_indexes 的方式）。

即使我可以动态生成行声明，看来我还是必须将它作为动态 sql 附加到我以前的函数调用中，哪一种鸡/蛋是从一个任意类型的结果集返回的问题功能。

解决方法

您可以将表提供为类型 anyelement 以指定返回数据的实际类型，而不是将表作为字符串提供，然后使用 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