当前位置: 代码迷 >> 综合 >> pgsql 重置所有按规则命名的序列
  详细解决方案

pgsql 重置所有按规则命名的序列

热度:18   发布时间:2023-11-20 22:00:47.0

pgsql 重置所有按规则命名的序列

因为在项目中使用hibernate使用注解进行映射,使用序列获取id值,但是数据库迁移后序列并没有迁移,导致程序启动后,自动创建序列进行查询时,对于有多条记录的表,序列仍从1开始,向表中插入数据,仍会出现很多错误,因此参考了其它文章代码改动,进行序列重设

需要根据情况更改逻辑

do $body$
declare seq_name text;table_and_id_string text;table_name text;id_name text;sqltext text:='';delimiter_reverse_pos integer;
begin-- 遍历获取到所有的序列for seq_name in select relname from pg_class where relkind='S' order by relname loop-- 根据实际的规则,条件判断是否是要处理的序列(这里是找到以s_开头的序列进行处理)if seq_name like 's_%' thentable_name:=substr(seq_name,3,length(seq_name));-- 设置序列的currentValuesqltext:=sqltext||'select setval(''schema.'||seq_name|| ''',(select max("id") from schema.'||table_name||'));'||chr(10);-- 重置表的序列sqltext:=sqltext||'alter table schema.'||table_name||' alter column id set default nextval(''yth_dap.'||seq_name||''');'||chr(10);	end if;end loop;RAISE NOTICE '%',sqltext;  -- 也通过此语句打印出拼接的sql语句-- execute sqltext; -- 可以在此执行语句EXCEPTIONWHEN others THENRAISE NOTICE '%', '执行出错,请检查是否从序列名中匹配的对象不存在!';RAISE EXCEPTION '(%)', SQLERRM; 
end
$body$;
···