MongoDB 唯一复合索引允许重复答案

【问题标题】：MongoDB Unique Compound Index is Allowing DuplicatesMongoDB 唯一复合索引允许重复
【发布时间】：2020-05-01 20:48:56
【问题描述】：

我正在使用 PyMongo 和 PyMODM 来处理一个相当简单的文档结构。

这就是我的模型的样子：

class User(MongoModel):
    subscriber_uid: fields.CharField = fields.CharField(required=True)
    interface: fields.CharField = fields.CharField(required=True)
    ...other fields...

    class Meta:
        """Defines MongoDB options for this model"""
        cascade = True  # currently nothing to cascade deletes against
        indexes = [
            IndexModel(
                keys=[('subscriber_uid', pymongo.ASCENDING),
                      ('interface', pymongo.ASCENDING)],
                unique=True,
            ),
        ]

我的单元测试看起来像这样，注意 DB 是一个包装实际 PyMongo 命令的 API。例如：put_user 在一些异常处理逻辑中包装了user.save()，而read_user 包装了User.objects.get()。

...
    user = User(**TEST_USER)
    user.interface = '2/2/2/2'
    user.subscriber_uid = 'some other suid'
    DB.put_user(user)
    user_from_db = DB.read_user(subscriber_uid=user.subscriber_uid,
                                interface=user.interface)
    assert user_from_db.interface == user.interface
    assert user_from_db.subscriber_uid == user.subscriber_uid

    # attempt to create a new record with a non-unique suid+iface pair
    # ie: ensure this updates instead of creates
    user = User(**TEST_USER)
    user.someotherattr = 1023
    DB.put_user(user)
    user_from_db: User = DB.read_user(subscriber_uid=user.subscriber_uid,
                                      interface=user.interface)
    assert user_from_db.seed_index == user.seed_index
...

当我运行此测试时，read_user()/User.objects.get() 调用失败并返回 MultipleObjectsReturned 对象错误，我可以确认有两条记录具有相同的 interface 和 subscriber_uid 值。

根据我在其他 StackOverflow answers 和 comments 中看到的情况，上面的复合唯一索引应该可以防止这种情况发生。

每次测试运行时都会删除数据库，因此问题不在于陈旧数据挥之不去。我是不是对复合索引有误解？

【问题讨论】：

标签： python mongodb pymongo pymodm

【解决方案1】：

明显缺乏索引是一个红鲱鱼。

PyMODM docs state：

请注意，必须先调用 connect()（定义相应的连接别名，如果有），然后才能将任何 MongoModel 与该别名一起使用。如果在 Meta 上定义了索引，则必须在评估 MongoModel 类之前。

我正在使用一个函数来启动与 MongoDB 的连接，这当然是在我导入模型类之后发生的。对于其他数据库客户端，如果您不确定在导入模块时您的数据库是否可用（通常在容器化工作负载中，应用程序启动通常比数据库启动花费更少的时间），那么这通常是要走的路。 PyMongo client is special though：

从 3.0 版开始，MongoClient 构造函数在连接到一个或多个服务器时不再阻塞，如果它们不可用，它不再引发 ConnectionFailure，如果用户的凭据错误，它也不会引发 ConfigurationError。相反，构造函数立即返回并在后台线程上启动连接过程。

将connect() 移动到我的模型模块的顶部解决了这个问题。即使在调用connect() 时数据库不可用。一旦数据库可用，就会创建索引，并且一切正常。

【讨论】：