利用MongoDB技术开发中遇到的写入冲突问题的解决方案探究

# 更新数据时使用乐观锁
def update_data_with_optimistic_locking(collection, document_id, update_data):
    document = collection.find_one({'_id': document_id})
    if document:
        version = document.get('version')

        # 将版本号添加到更新数据中
        update_data['version'] = version + 1

        # 使用版本号进行更新
        result = collection.update_one({'_id': document_id, 'version': version}, {'$set': update_data})

        if result.modified_count == 1:
            print("数据更新成功")
        else:
            print("数据更新失败，可能存在并发冲突")
    else:
        print("未找到指定的数据")

# 使用悲观锁进行更新数据
def update_data_with_pessimistic_locking(collection, document_id, update_data):
    with collection.find_one_and_lock({'_id': document_id}) as doc:
        if doc:
            # 执行更新操作
            result = collection.update_one({'_id': document_id}, {'$set': update_data})

            if result.modified_count == 1:
                print("数据更新成功")
            else:
                print("数据更新失败，可能存在并发冲突")
        else:
            print("未找到指定的数据")