pymongo实现_id自增长

最新推荐文章于 2023-07-28 00:51:15 发布

①岁①枯荣

最新推荐文章于 2023-07-28 00:51:15 发布

阅读量473

点赞数 2

分类专栏： python mongo 文章标签： mongodb 数据库 nosql python

本文链接：https://blog.csdn.net/sinat_34343838/article/details/124783527

版权

本文介绍如何利用pymongo在MongoDB中实现_id的自增长，主要涉及初始化计数器、更新计数器以及在实际文档集合中的应用。通过get_next_sequence_value方法结合find_one_and_update操作，确保每个新插入的文档拥有唯一的_id。

摘要由CSDN通过智能技术生成

import pymongo

# 数据库
db = pymongo.MongoClient('mongodb://localhost:27017/')['abc']

# 计数器
counter = db['counter']

# book文档集合
books = db['books']


# 初始化计数器
def init_counter(sequence_name):
    count_doc = {
        '_id': sequence_name,
        'sequence_value': 0
    }
    counter.insert_one(count_doc)


# 获取计数器自增值
def get_next_sequence_value(sequence_name):
    updated_count_doc = counter.find_one_and_update({'_id': sequence_name},
                                                    {'$inc': {'sequence_value': 1}}, return_document=True)
    return updated_count_doc['sequence_value']


if __name__ == '__main__':
    init_counter('book_id')

init_counter在这里作为book的计数器,只需初始化执行一次就可以，sequence_name为id名称，如：book_id,

若是想作为其他文档集合计数器，直接改参数（即将'book_id'