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

2024-04-16 09:38

本文主要是介绍Django如何使用snowflake自定义生成主键而不是自动生成主键?,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

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

目录

    • 背景介绍
    • 实现方案
        • 方案1 - 手动添加主键
        • 方案2 - 重写save()方法
        • 方案3 - 使用 Django Signals 中的pre_save()
        • 方案4 - 仿照models.UUIDField,写一个models.SnowflakeIDField
    • 总结

背景介绍

目前工程框架如下

  • 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.pyclass User(models.Model):id = models.BigIntegerField(primary_key=True)name = models.CharField(max_length=100, verbose_name="名字")
  • (2)其次是save( )的时候,需要手动设置主键
# user的新增操作user = User()
user.id = snowflake.next_id()
user.name = "123"
# 可以不指定force_insert=True,如果不指定,默认会先update看是否有记录,如果没有,再执行insert
user.save(force_insert=True)

改动:(1)所有model增加id设置

(2)所有save()之前设置id=snowflake.next_id()

方案2 - 重写save()方法

实现方式如下

  • 单独改写某一个model的save( )方法
# models.pyclass User(models.Model):id = models.BigIntegerField(primary_key=True)name = models.CharField(max_length=100, verbose_name="名字")def save(self, *args, **kwargs):if not self.id:self.id = snowflake.next_id()super(User, self).save(*args, **kwargs)return self.id
  • 调用的地方不用修改,不需要设置 user.id = snowflake.next_id()

按照上面的方法的话,每个新的model都是需要重写save( )方法的,显然工作量仍然比较多。

改动:(1)所有model增加id设置

(2)所有model重写save()方法

方案3 - 使用 Django Signals 中的pre_save()

官方文档地址:https://docs.djangoproject.com/en/3.0/topics/signals/

实现方式如下

  • 修改models.py文件

使用pre_save( )方法,在调用models.save( )之后立即执行

注意

(1)如auth_开头的表的save()以及 其他django_开头的表都会走此逻辑,需要排除这些model的save

(2)save() 包括update和insert,过滤执行update的model

from django.db.models.signals import pre_save
from django.dispatch import receiver
import logging@receiver(pre_save)
def pre_save_set_snowflake_id(sender, instance, *args, **kwargs):"""Django Signals, pre_save适用所有的modelIf we dont include the sender argument in the decorator,like @receiver(pre_save, sender=MyModel), the callback will be called for all models."""# print(__name__)  # = polls.models# print(type(instance)) # = <class 'polls.models.Question'>if __name__ in str(type(instance)) and not instance.id:# 满足条件(1)在本models.py中声明的model (2)id不为空的model 才会使用snowflake生成id# 原因是如果不加条件(1)会有如auth_开头的表以及django_开头的表也会用snowflake生成的id,但是其id长度不够instance.id = snowflake.next_id()class User(models.Model):id = models.BigIntegerField(primary_key=True)name = models.CharField(max_length=100, verbose_name="名字")

如上,修改好models.py之后,就可以安心使用save()了,原先的业务逻辑也不用有其他修改

改动(1)所有model增加id设置,统一设置 id = models.BigIntegerField(primary_key=True)

(2)增加pre_save( ) 方法,统一设置 id = snowflake.next_id()

方案4 - 仿照models.UUIDField,写一个models.SnowflakeIDField

尝试中,但是还没有写出来。

思路就是新写一个field,比如类似models.UUIDField,能否写一个models.SnowflakeIDField

如下是UUIDField写法

# models.py
import uuidclass TestUUID(models.Model):# 一定要设置default=uuid.uuid4id = models.UUIDField(primary_key=True, default=uuid.uuid4)name = models.CharField(max_length=100, verbose_name="名字")
# 新增操作test_uuid = TestUUID()
test_uuid.name = '123123'
test_uuid.save()
-- 执行SQL
(0.005) INSERT INTO "polls_testuuid" ("id", "name") VALUES ('74f8e76c-b80a-4197-a469-81992c46e3c4'::uuid, '123123'); args=(UUID('74f8e76c-b80a-4197-a469-81992c46e3c4'), '123123')

TODO… 那么,SnowflakeIDField如何实现呢?

最终了解下来,发现完全没有必要自己写models.SnowflakeIDField

如下,即可完成…

# models.py
# 添加 default=snowflake_util.next_id 即可
id = models.BigIntegerField(primary_key=True, default=snowflake_util.next_id)

总结

  • 选择上面的方案3,使用 Django Signals 中的pre_save(),可以增加对于参数的校验,不过本次不需要
  • 选择上面的方案4,最简单

这篇关于Django如何使用snowflake自定义生成主键而不是自动生成主键?的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/908438

相关文章

微信公众号脚本-获取热搜自动新建草稿并发布文章

《微信公众号脚本-获取热搜自动新建草稿并发布文章》本来想写一个自动化发布微信公众号的小绿书的脚本,但是微信公众号官网没有小绿书的接口,那就写一个获取热搜微信普通文章的脚本吧,:本文主要介绍微信公众... 目录介绍思路前期准备环境要求获取接口token获取热搜获取热搜数据下载热搜图片给图片加上标题文字上传图片

vue使用docxtemplater导出word

《vue使用docxtemplater导出word》docxtemplater是一种邮件合并工具,以编程方式使用并处理条件、循环,并且可以扩展以插入任何内容,下面我们来看看如何使用docxtempl... 目录docxtemplatervue使用docxtemplater导出word安装常用语法 封装导出方

Linux换行符的使用方法详解

《Linux换行符的使用方法详解》本文介绍了Linux中常用的换行符LF及其在文件中的表示,展示了如何使用sed命令替换换行符,并列举了与换行符处理相关的Linux命令,通过代码讲解的非常详细,需要的... 目录简介检测文件中的换行符使用 cat -A 查看换行符使用 od -c 检查字符换行符格式转换将

Java编译生成多个.class文件的原理和作用

《Java编译生成多个.class文件的原理和作用》作为一名经验丰富的开发者,在Java项目中执行编译后,可能会发现一个.java源文件有时会产生多个.class文件,从技术实现层面详细剖析这一现象... 目录一、内部类机制与.class文件生成成员内部类(常规内部类)局部内部类(方法内部类)匿名内部类二、

使用Jackson进行JSON生成与解析的新手指南

《使用Jackson进行JSON生成与解析的新手指南》这篇文章主要为大家详细介绍了如何使用Jackson进行JSON生成与解析处理,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录1. 核心依赖2. 基础用法2.1 对象转 jsON(序列化)2.2 JSON 转对象(反序列化)3.

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

Elasticsearch 在 Java 中的使用教程

《Elasticsearch在Java中的使用教程》Elasticsearch是一个分布式搜索和分析引擎,基于ApacheLucene构建,能够实现实时数据的存储、搜索、和分析,它广泛应用于全文... 目录1. Elasticsearch 简介2. 环境准备2.1 安装 Elasticsearch2.2 J

使用C#代码在PDF文档中添加、删除和替换图片

《使用C#代码在PDF文档中添加、删除和替换图片》在当今数字化文档处理场景中,动态操作PDF文档中的图像已成为企业级应用开发的核心需求之一,本文将介绍如何在.NET平台使用C#代码在PDF文档中添加、... 目录引言用C#添加图片到PDF文档用C#删除PDF文档中的图片用C#替换PDF文档中的图片引言在当

Java中List的contains()方法的使用小结

《Java中List的contains()方法的使用小结》List的contains()方法用于检查列表中是否包含指定的元素,借助equals()方法进行判断,下面就来介绍Java中List的c... 目录详细展开1. 方法签名2. 工作原理3. 使用示例4. 注意事项总结结论:List 的 contain

C#使用SQLite进行大数据量高效处理的代码示例

《C#使用SQLite进行大数据量高效处理的代码示例》在软件开发中,高效处理大数据量是一个常见且具有挑战性的任务,SQLite因其零配置、嵌入式、跨平台的特性,成为许多开发者的首选数据库,本文将深入探... 目录前言准备工作数据实体核心技术批量插入:从乌龟到猎豹的蜕变分页查询:加载百万数据异步处理:拒绝界面