如何在单个 Select 语句中执行动态 SQL 语句？

Question

我只是想知道如何使用一个 select 来评估动态 SQL 的内容； 这就是例子。 这只是一个例子。 但我想要动态函数，并使用单选进行管理。 （我知道 sqls 只适用于 SELECT 而不是修改......但在这个深层次的被问者中，我正在成为一个疯狂的开发者）

SELECT 'SELECT SETVAL(' || chr(39) || c.relname || chr(39)|| ' , 
(SELECT MAX(Id)+1 FROM '  || regexp_replace(c.relname, '_[a-zA-Z]+_[a-zA-Z]+(_[a-zA-Z0-9]+)?', '', 'g')   ||' ), true );' 
FROM pg_class c WHERE c.relkind = 'S';

原output为：

SELECT SETVAL('viewitem_id_seq' , (SELECT MAX(Id)+1 FROM viewitem ), true );
SELECT SETVAL('userform_id_seq' , (SELECT MAX(Id)+1 FROM userform ), true );

这是动态语句： (SELECT MAX(Id)+1 FROM ' || regexp_replace(c.relname, ' [a-zA-Z]+ [a-zA-Z]+(_[a-zA-Z0- 9]+)?', '', 'g')

是一个生成为 output 和 SQL 的字符串，如何在同一行中评估此语句？

所需的 output 是：

SELECT SETVAL('viewitem_id_seq' , 25, true );
SELECT SETVAL('userform_id_seq' , 85, true );

谢谢！

Answer 1

如果这些是serial或identity列，最好使用pg_get_serial_sequence()来获取表的列与其序列之间的链接。

您实际上可以使用query_to_xml()在 SQL 语句中运行动态 SQL

如果我需要将serial （或identity ）列的序列与其实际值同步，我使用以下脚本：

with sequences as (
  -- this query is only to identify all sequences that belong to a column
  -- it's essentially similar to your select * from pg_class where reltype = 'S'
  -- but returns the sequence name, table and column name to which the 
  -- sequence belongs
  select *
  from (
    select table_schema,
           table_name,
           column_name,
           pg_get_serial_sequence(format('%I.%I', table_schema, table_name), column_name) as col_sequence
    from information_schema.columns
    where table_schema not in ('pg_catalog', 'information_schema')
  ) t
  where col_sequence is not null
), maxvals as (
  select table_schema, table_name, column_name, col_sequence,
         --
         -- this is the "magic" that runs the SELECT MAX() query
         --
          (xpath('/row/max/text()',
             query_to_xml(format('select max(%I) from %I.%I', column_name, table_schema, table_name), true, true, ''))
          )[1]::text::bigint as max_val
  from sequences
) 
select table_schema, 
       table_name, 
       column_name, 
       col_sequence,
       coalesce(max_val, 0) as max_val,
       setval(col_sequence, coalesce(max_val, 1)) --<< this uses the value from the dynamic query
from maxvals;

这里的动态部分是对query_to_xml()的调用

首先，我使用format()来正确处理标识符。 它还使编写 SQL 更容易，因为不需要连接。 因此，对于第一个 CTE 返回的每个表，都会执行以下操作：

query_to_xml('select max(id) from public.some_table', true, true, '')

这会返回类似：

<row xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
  <max>42</max>
</row>

该值比使用 xpath() 从 XML 值中提取并转换为一个数字，然后在最终 SELECT 中使用以实际调用setval()

具有多个 CTE 的嵌套仅用于使每个部分更具可读性。

例如，可以使用相同的方法来查找所有表的行数

关于query_to_xml()如何工作的一些背景知识