PostgreSQL 主键弃用自增ID使用雪花算法

使用PG创建带主键自增的表

CREATE TABLE "public"."temp" (
  "id" BIGSERIAL PRIMARY KEY,
  "name" varchar(255) COLLATE "pg_catalog"."default"
)

使用这种方式创建的表pg会自动生产一个seq自增与主键绑定。

DROP TABLE IF EXISTS "public"."temp";
CREATE TABLE "public"."temp" (
  "id" int8 NOT NULL DEFAULT nextval('temp_id_seq'::regclass),
  "name" varchar(255) COLLATE "pg_catalog"."default"
)
;

-- ----------------------------
-- Primary Key structure for table temp
-- ----------------------------
ALTER TABLE "public"."temp" ADD CONSTRAINT "temp_pkey" PRIMARY KEY ("id");

DEV环境向UAT环境同步过程中，因为自增序列的原因或者，在同步某一张表的过程中不能将表结构直接导出，再其他数据库执行，会报关系不存在问题

放弃pg原有的自增主键 

使用如下函数

CREATE SEQUENCE table_id_seq increment by 1 maxvalue 99999999 minvalue 1 start 1 cycle;


CREATE OR REPLACE FUNCTION snow_next_id(OUT result bigint) AS $$
DECLARE
our_epoch bigint := 1314226531721;
seq_id bigint;
now_millis bigint;
shard_id int := 5;
BEGIN
seq_id := nextval('table_id_seq') % 1024;
SELECT FLOOR(EXTRACT(EPOCH FROM clock_timestamp()) * 1000) INTO now_millis;
result := (now_millis - our_epoch) << 23;
result := result | (shard_id << 10);
result := result | (seq_id);
END;
$$ LANGUAGE PLPGSQL;

建表语句

CREATE TABLE "public"."temp2" (
  "id" int8 NOT NULL DEFAULT snow_next_id(),
  "name" varchar(255) COLLATE "pg_catalog"."default"
)

补充：

在后续业务中，向表中写入大量数据，会出现主键不唯一，此时将

seq_id := nextval('table_id_seq') % 4096;

修改，也可修改为更大的值（2的指数倍）