MongoEngine是否有类似Django的AutoField()？求合适替代方案

1. 使用默认的ObjectIdField（推荐） #

MongoDB文档默认会自动生成_id字段，类型为ObjectId，这是MongoDB原生的主键设计。在MongoEngine中，模型无需手动定义，会自动添加该字段；若要显式声明，代码如下：

from mongoengine import Document, ObjectIdField
from bson.objectid import ObjectId

class YourModel(Document):
    id = ObjectIdField(primary_key=True, default=ObjectId)
    # 其他业务字段...

该方案完全贴合MongoDB的设计逻辑，无需额外维护，性能优异，是绝大多数场景的首选。

2. 实现自增整数主键（模拟AutoField） #

如果业务场景必须依赖自增整数ID，可通过以下两种方式实现：

方法一：使用SequenceField（简化实现） #

MongoEngine内置的SequenceField会自动维护一个计数器集合，通过原子操作生成唯一自增ID：

from mongoengine import Document, SequenceField

class YourModel(Document):
    auto_id = SequenceField(primary_key=True)
    # 其他业务字段...

注意：该字段依赖MongoDB的原子操作保证唯一性，但在分布式高并发场景下，可能存在轻微的锁竞争问题。

方法二：手动实现计数器逻辑（自定义控制） #

若需要更精细的控制（比如自定义初始值、分业务线计数），可手动创建计数器集合实现：

1. 定义计数器模型：

from mongoengine import Document, IntField, StringField

class Counter(Document):
    name = StringField(primary_key=True)
    value = IntField(default=0)

2. 在业务模型中重写save方法，生成自增ID：

from mongoengine import Document, IntField, DoesNotExist

class YourModel(Document):
    auto_id = IntField(primary_key=True)
    # 其他业务字段...

    def save(self, *args, **kwargs):
        if not self.auto_id:
            # 获取或初始化计数器
            try:
                counter = Counter.objects.get(name="your_model_counter")
            except DoesNotExist:
                counter = Counter(name="your_model_counter", value=0)
            # 原子递增计数器
            counter.value += 1
            counter.save()
            self.auto_id = counter.value
        super().save(*args, **kwargs)

注意事项 #

• 优先选择ObjectIdField，避免自增ID在分布式环境下的锁竞争和性能损耗。
• 自增ID方案必须保证计数器操作的原子性，否则会出现重复ID的风险。

内容的提问来源于stack exchange，提问作者kool