Created
April 28, 2011 12:29
-
-
Save dodecaphonic/946254 to your computer and use it in GitHub Desktop.
Function set sequences to max in Postgres (instead of writing the same piece of code in ruby again and again)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Taken from http://stackoverflow.com/questions/244243/how-to-reset-postgres-primary-key-sequence-when-it-falls-out-of-sync. | |
--drop function IF EXISTS reset_sequence (text,text) RESTRICT; | |
CREATE OR REPLACE FUNCTION "reset_sequence" (tablename text,columnname text) RETURNS bigint --"pg_catalog"."void" | |
AS | |
$body$ | |
DECLARE seqname character varying; | |
c integer; | |
BEGIN | |
select tablename || '_' || columnname || '_seq' into seqname; | |
EXECUTE 'SELECT max("' || columnname || '") FROM "' || tablename || '"' into c; | |
if c is null then c = 0; end if; | |
c = c+1; --because of substitution of setval with "alter sequence" | |
--EXECUTE 'SELECT setval( "' || seqname || '", ' || cast(c as character varying) || ', false)'; DOES NOT WORK!!! | |
EXECUTE 'alter sequence ' || seqname ||' restart with ' || cast(c as character varying); | |
RETURN nextval(seqname)-1; | |
END; | |
$body$ LANGUAGE 'plpgsql'; | |
select sequence_name, PG_CLASS.relname, PG_ATTRIBUTE.attname, | |
reset_sequence(PG_CLASS.relname,PG_ATTRIBUTE.attname) | |
from PG_CLASS | |
join PG_ATTRIBUTE on PG_ATTRIBUTE.attrelid = PG_CLASS.oid | |
join information_schema.sequences | |
on information_schema.sequences.sequence_name = PG_CLASS.relname || '_' || PG_ATTRIBUTE.attname || '_seq' | |
where sequence_schema='public'; |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment