postgresql 将所有表的id列设置为自增主键,自增起始数值为该表的最大id

首先,使用`information_schema.columns`系统视图来获取所有具有 'id' 列的表的表名和列名。这个视图提供了有关数据库中所有表和列的元数据信息。

然后,使用循环来遍历每个满足条件的表,并执行以下操作:

  1. 使用动态SQL语句来获取当前表的最大 id 值,并将结果存储在 max_id 变量中。`COALESCE`函数用于处理表中没有数据时的情况。

  2. 创建一个序列,其名称是通过连接表名、列名和后缀 '_seq' 构成的。序列是用于生成自增值的。

  3. 获取序列名称,并将其存储在 sequence_name 变量中。

  4. 使用动态SQL语句来修改表的列,使用 `ALTER COLUMN` 子句设置默认值为 `nextval(sequence_name)`,即使用序列来自动生成值作为默认值。

  5. 使用动态SQL语句来设置当前列为主键约束,使用 `ALTER TABLE` 和 `ADD PRIMARY KEY` 子句。

通过以上步骤,代码会遍历所有具有 'id' 列的表,并设置自增起始值和主键。

sql 复制代码
DO
$$
DECLARE
  table_info RECORD;
  max_id BIGINT;
  sequence_name TEXT;
BEGIN
  FOR table_info IN (
    SELECT table_name, column_name
    FROM information_schema.columns
    WHERE column_name = 'id'
  )
  LOOP
    -- 获取当前表的最大id
    EXECUTE FORMAT('
      SELECT COALESCE(MAX(%I), 0)
      FROM %I',
      table_info.column_name,
      table_info.table_name
    ) INTO max_id;
    
    -- 创建序列并设置自增起始值为当前表的最大id + 1
    EXECUTE FORMAT('
      CREATE SEQUENCE %I START %s',
      table_info.table_name || '_' || table_info.column_name || '_seq',
      max_id + 1
    );
   
    -- 获取序列名称
    EXECUTE FORMAT('
      SELECT %L',
      table_info.table_name || '_' || table_info.column_name || '_seq'
    ) INTO sequence_name;

    -- 执行ALTER TABLE来设置自增起始值
    EXECUTE FORMAT('
      ALTER TABLE %I 
      ALTER COLUMN %I SET DEFAULT nextval(%L)',
      table_info.table_name,
      table_info.column_name,
      sequence_name
    );
   
    -- 执行ALTER TABLE来设置主键约束
    EXECUTE FORMAT('
      ALTER TABLE %I 
      ADD PRIMARY KEY (%I)',
      table_info.table_name,
      table_info.column_name
    );
  END LOOP;
END
$$
相关推荐
萧曵 丶10 分钟前
Spring @TransactionalEventListener
java·数据库·spring·事务·transactional·异步
坤坤不爱吃鱼12 分钟前
【MySQL\Oracle\PostgreSQL】迁移到openGauss数据出现的问题解决方案
mysql·postgresql·oracle
胡斌附体13 分钟前
mobaxterm终端sqlplus乱码问题解决
数据库·乱码·sqlplus·字符集设置
moon66sun23 分钟前
开源项目XYZ.ESB:数据库到数据库(DB->DB)集成
数据库·esb
TDengine (老段)41 分钟前
使用 StatsD 向 TDengine 写入
java·大数据·数据库·时序数据库·iot·tdengine·涛思数据
DarkAthena1 小时前
【GaussDB】深度解析:创建存储过程卡死且无法Kill会话的疑难排查
数据库·gaussdb
Gauss松鼠会1 小时前
GaussDB权限管理:从RBAC到精细化控制的企业级安全实践
大数据·数据库·安全·database·gaussdb
时序数据说1 小时前
时序数据库IoTDB用户自定义函数(UDF)使用指南
大数据·数据库·物联网·开源·时序数据库·iotdb
Java初学者小白2 小时前
秋招Day15 - Redis - 缓存设计
java·数据库·redis·缓存
绅士玖2 小时前
前端数据存储总结:Cookie、localStorage、sessionStorage与IndexedDB的使用与区别
前端·javascript·数据库