巧用KingbaseES中的动态DDL

2022-10-15,,,

概述 :在DBA的日常工作中,经常遇到一些需要基于数据库当前状态的实用程序查询的实例。比如一个逻辑复制的目标表,主键ID列与生成数据的序列不同步,这将导致插入新行是,会有主键冲突。要纠正这个问题,需要设置序列可以生成的值,超过表中当前最大值。

解决此类问题可以使用动态DDL,虽然SQL 中的数据定义语言(DDL) 本身是非动态的,因为DDL需要有严格的解析规则,而且其中的子查询预定义的数据类型、表结构,以及基于已知关联列。

下面的案例,可以让 SQL语句随心所欲地执行动态 DDL,而不必每次都手动编写这些查询。

一、改变序列重启值的解决方案

希望通过查询显式地为序列提供RESTART值,重置一个序列,使其从它所关联的表的当前最大值之后开始。预想中,使用简单直接的DDL语句:

ALTER SEQUENCE tablename_id_seq RESTART (SELECT max(id)+1 FROM tablename);
ERROR: syntax error at or near "(", at character 38
STATEMENT: ALTER SEQUENCE tablename_id_seq RESTART (SELECT max(id) + 1 FROM tablename);

正如所看到的,KingbaseE语法不支持这种方法,因为这里需要的是实际值,而不是子查询。

1、使用ksql的变量替换

使用ksql变量,首先选择一个变量的值,然后将这个值替换到表达式中。

-- 使用\gset设置一个带有查询结果的变量

SELECT max(id) + 1 as id_max from tablename
\gset
-- substitute the variable in a new query
ALTER SEQUENCE tablename _id_seq RESTART :id_max ;

在实例中,使用\gset命令来得到max(id)+1的结果,并将其存储在会话中,供以后使用。然后使用:id_max语法将该变量插入到SQL语句中,将其直接传递给服务器。

2、使用ksql的 \gexec 命令

使用ksql进行动态SQL的另一种方法,是设计一个查询SELECT语句,返回希望运行的语句,然后使用\gexec命令执行。

SELECT 'ALTER SEQUENCE tablename_id_seq RESTART ' || max(id) + 1 as query FROM tablename;

\gexec query

ALTER SEQUENCE tablename_id_seq RESTART 100001

ALTER SEQUENCE

与变量替换相比,这种方法的一个优点是,它可以处理更复杂的语句和多个返回值,因此可以基于任意条件设计查询语句,并生成多个SQL语句;第一个方法实现仅限于每次返回单个行的查询。在使用\gexec对服务器执行SQL语句之前,这还可以预览将要运行的SQL语句,因此提供一定程度的安全性。

二、非ksql的动态SQL

ksql具有处理动态SQL的功能,但如何在第三方应用中,实现动态SQL的功能。现在使用基本类似的方法,编写sql语句块来生成查询,然后执行返回的SQL语句。这些方法与上面的ksql部分中的方法大致相关,因此适用于单个或多个动态语句。

1、匿名块

要使用服务器端动态SQL,我们需要使用format函数构造查询,并执行底层文本,就像我们自己发出底层查询一样。

BEGIN
EXECUTE format('ALTER SEQUENCE tablename_id_seq RESTART %s', (SELECT max(id) + 1 FROM tablename));
END;

2、创建函数 exec()

与匿名块的功能几乎相同,我们也可以创建一个简单的用户函数,简单地调用它的输入参数EXECUTE。

CREATE OR REPLACE FUNCTION exec(raw_query text) RETURNS text AS
BEGIN
EXECUTE raw_query;
RETURN raw_query;
END;
SELECT exec(format('ALTER SEQUENCE tablename_id_seq RESTART %s', (SELECT max(id) + 1 FROM tablename)));

函数方法基本上只是将查询移动到传入的参数中,但它带来更灵活的针对构造的查询列表,在SQL语句中调用此函数,可以选择依次运行每个 SQL。

限制

这些方法对动态SQL运行的限制,在显式事务块之外,运行的SQL命令受到限制。如果想在所有已知索引上运行REINDEX CONCURRENTLY,那么使用exec()函数为public模式中的所有索引执行REINDEX CONCURRENTLY语句:

SELECT exec(format('REINDEX INDEX CONCURRENTLY %I', relname))
FROM sys_class r
JOIN sys_namespace s ON r.relnamespace = s.oid
WHERE relkind = 'i' AND nspname = 'public' ; ERROR: REINDEX CONCURRENTLY cannot be executed from a function
CONTEXT: SQL statement "REINDEX INDEX CONCURRENTLY tablename_pkey"
PL/SQL function exec(text) line 3 at EXECUTE

从这里看到,由于REINDEX CONCURRENTLY需要管理自己的事务状态,因此它不能作为函数运行;在 KingbaseES 中,函数的内在事务运行时,是以函数的SQL语句完全成功或完全失败。

试一试命令 \gexec

SELECT format('REINDEX INDEX CONCURRENTLY %I', relname)
FROM sys_class r
JOIN sys_namespace s ON r.relnamespace = s.oid
WHERE relkind = 'i' AND nspname = 'public' \gexec

  

由于\gexec处理是由ksql完成的,因此生成的语句独立运行。

巧用KingbaseES中的动态DDL的相关教程结束。

《巧用KingbaseES中的动态DDL.doc》

下载本文的Word格式文档,以方便收藏与打印。