Django如何使用snowflake自定义生成主键而不是自动生成主键？

Django应用Snowflake算法

最新推荐文章于 2024-09-26 08:22:42 发布

原创

最新推荐文章于 2024-09-26 08:22:42 发布 · 2.1k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#django

本文探讨了在Django框架中使用Snowflake算法生成主键的四种方案，包括手动添加主键、重写save()方法、利用DjangoSignals的pre_save()信号以及模仿UUIDField创建SnowflakeIDField。

之前ID都是用自增实现的，那现在想用Snowflake算法生成主键，要做什么改动呢？

背景介绍

目前工程框架如下

Django
django.db.backends.postgresql_psycopg2

目前models声明和save()方法如下

# models.py
# 使用的是默认主键
class User(models.Model):
    name = models.CharField(max_length=100, verbose_name="名字")

# user的新增操作，默认主键
user = User()
user.name = "123"
user.save()

在user.save()时执行的SQL语句是

INSERT INTO "polls_user" ("name") VALUES ('xiaoming') RETURNING "polls_user"."id"; args=('xiaoming',)

很明显，并未在语句中设置ID主键

查看表结构，发现id自动是使用序列自增来实现的

CREATE TABLE "public"."polls_user" (
  "id" int4 NOT NULL DEFAULT nextval('polls_user_id_seq'::regclass),
  "name" varchar(100) COLLATE "pg_catalog"."default" NOT NULL,
  CONSTRAINT "polls_user_pkey" PRIMARY KEY ("id")
)

那么基于目前的情况，如何改写使用snowflake算法生成ID，而不是自动生成主键呢？

实现方案

方案1 - 手动添加主键

实现方式如下

（1）首先需要声明id，声明为非AutoField类型

如果不声明，在使用postgresql的情况下，会默认为主键设置为serial类型，使用序列生成主键

# models.py

class User(models.Model):
    id = models.BigIntegerField(primary_key=True)
    name = models.CharField(max_length

最低0.47元/天解锁文章