提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  Javascript

当前栏目

Django自定义ID自增

2023-02-19 12:23:53 时间

ID自增

Django models里面ID自增通常使用models.AutoField(auto_created=True, primary_key=True)来定义主键，但是如何自定义ID自增怎么做呢？类似于在ID前面添加固定字符串如：AE8600000001、AE8600000002、AE8600000003……

案例分析

很显然上面的AutoField字段不能使用，因为AutoField字段是IntegerField字段，所以不能将pk用来AE860000001，从目标字段来看，很显然是字符串类型，所以只能将字段类型改为CharField，但是呢，如果数据量巨大，字符串作为主键显然是不理智的选择，这样会严重影响到查询性能问题，但是也不是不可使用。这样必须重写models里面的save方法，达到字段自增的目的（其实也可以用另外一张表记录ID游标的方式也可，但是个人觉得比较麻烦，不知道有没有大神知道更为简便的方法）。

获取最大值ID

class TestNew(models.Model):
  sequence_ID = models.CharField(primary_key=True, editable=False, max_length=50)
  def save(self, *args, **kwargs):
      if not self.sequence_ID:
          max_id = TestNew.objects.aggregate(id_max=Max('sequence_ID'))['id_max']
          self.sequence_ID = "{}{:08d}".format("AE86", max_id + 1 if max_id is not None else 1)
      super().save(*args, **kwargs)

弊端

重写save方法固然好用，但是有一种创建数据的方式不行，那就是批量创建的方法不适用，因为它的底层不调用save方法，objects.bulk_create这个方法，咱们可以看一下底层源码：

def bulk_create(self, objs, batch_size=None):
    """
    Inserts each of the instances into the database. This does *not* call
    save() on each of the instances, does not send any pre/post save
    signals, and does not set the primary key attribute if it is an
    autoincrement field (except if features.can_return_ids_from_bulk_insert=True).
    Multi-table models are not supported.
    """pass

写到这里，估计你们会产生疑问，这样写会不会出现问题，在生产环境能用吗？如果多个用户同时尝试，那么他就会发生IntegrityError错误。你们可以试试……

自定义方法

上面的方法存在一定的弊端，如果咱们每次重写save方法的时候去查询一下目前表里面的数量，创建临时字段ID，在此基础上加1是否可行？

class TestNew(models.Model):

  def ids():
    c_no = TestNew.objects.count()
    if c_no == None:
        return 1
    else:
        return c_no + 1
  emp_id = models.IntegerField(('Code'), default=ids, unique=True, editable=False)
  sequence_ID = models.CharField(primary_key=True, editable=False, max_length=50)
  def save(self, *args, **kwargs):
      if not self.sequence_ID:
          self.sequence_ID = "{}{:08d}".format("AE86", self.emp_id))
      super().save(*args, **kwargs)

这种发法也会和上面一样，如果多用户使用，也会产生IntegrityError错误。啊啊啊，有没有大神求教，目前只能这样实现了。

相关文章