Question

我的数据库使用Postgres架构为用户提供独立的多租户环境。每个模式都有相同表的副本。

我有一个特定的查询，我需要在模式中加入，返回一个记录列表（在本例中为children）。我通过动态SQL查询工作，如下所示。但是，我想在每个结果中添加一列，指定行来自的模式的名称。

当前动态查询

（架构看起来像：OPERATOR_SCHEMA_my-great-company）

CREATE OR REPLACE FUNCTION all_children_dynamic() RETURNS SETOF children AS $$
DECLARE
  schema RECORD;
BEGIN
  FOR schema IN EXECUTE
    format(
      'SELECT schema_name FROM information_schema.schemata WHERE left(schema_name, 16) = %L',
      'OPERATOR_SCHEMA_'
    )
  LOOP
    RETURN QUERY EXECUTE
      format('SELECT * FROM %I.children', schema.schema_name);
  END LOOP;
END;
$$ LANGUAGE plpgsql;

--------

-- Usage: SELECT "id", "name" FROM all_children_dynamic();

返回类似于：

的内容

-------------
| id | name |
| 1  | Bob  |
| 2  | Joe  |
-------------

我想要它返回类似的内容：

-------------------------------
| id | name | schema_name     |
| 1  | Bob  | darcy's-store   |
| 2  | Joe  | bob's-4th-store |
-------------------------------

应该注意的是，模式名称是用户定义的，并且可以在其中包含引号。

如何为每个孩子添加相关的模式名称？

我尝试过以下几种变体：

LOOP
   RETURN QUERY EXECUTE
     format('SELECT %s AS schema_name, * FROM %1$I.children', schema.schema_name);
END LOOP;

但我在格式化方面存在一些问题。我可能会在这里使用一些quote_X功能。

我对Postgres（以及一般的数据库）不是很了解，所以请耐心等待！

更新

以下是我为一些变化而得到的确切错误。

Input:
format('SELECT %s AS schema_name, * FROM %1$I.children', schema.schema_name);

ERROR:  column "operator_schema_don" does not exist
LINE 1: SELECT OPERATOR_SCHEMA_don-t-display-data AS schema_name, * ...
QUERY:  SELECT OPERATOR_SCHEMA_don-t-display-data AS schema_name, * FROM "OPERATOR_SCHEMA_don-t-display-data".children

Input:
format('SELECT %s AS schema_name, * FROM %I.children', quote_literal(schema.schema_name), schema.schema_name);

ERROR:  structure of query does not match function result type
DETAIL:  Returned type unknown does not match expected type uuid in column 1.

更新2

我越来越近了，但还没到那里。

CREATE OR REPLACE FUNCTION all_children_dynamic() RETURNS TABLE (id uuid, schema_name varchar) AS $$
DECLARE
  schema RECORD;
BEGIN
  FOR schema IN EXECUTE
    format(
      'SELECT schema_name FROM information_schema.schemata WHERE left(schema_name, 16) = %L',
      'OPERATOR_SCHEMA_'
    )
  LOOP
    RETURN QUERY EXECUTE
      format('SELECT id, %L AS schema_name FROM %I.children', quote_literal(schema.schema_name), schema.schema_name);
  END LOOP;
END;
$$ LANGUAGE plpgsql;

ERROR:  structure of query does not match function result type
DETAIL:  Returned type unknown does not match expected type character varying in column 2.
CONTEXT:  PL/pgSQL function all_children_dynamic() line 11 at RETURN QUERY

为什么返回类型会以未知方式返回？我希望它会插入一个字符串并返回该类型。

Answer 1

由于要向查询添加新列（在本例中为模式名称），因此函数结果不是setof children。因此，需要返回一个表类型，其中包含其他列。语法可以看作here。

像这样......

CREATE OR REPLACE FUNCTION all_children_dynamic() RETURNS 
TABLE(col1 col1type,col2 col2type,...., schema_name varchar) AS $$

第二个错误是由于postgres的不正确转换，因为版本9是关于返回数据类型的真实特定。例如。如果要在函数返回类型中返回varchar（8），则必须返回相同长度的varchar。因此需要铸造。

将模式名称添加到动态Postgres查询

1 个答案: