motor -- 异步操作Mongo

原创已于 2022-07-11 17:46:43 修改 · 3.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2022-01-18 15:19:03 首次发布

MyCode 专栏收录该内容

32 篇文章

订阅专栏

本文展示了一种使用Python的异步IO库asyncio结合Motor库进行MongoDB操作的方法。通过对比两种不同的文档检索方式，一种是将所有文档转换为列表再遍历，另一种是直接在异步游标上进行迭代，演示了异步编程如何提高数据库操作效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码

# -*- coding: utf-8 -*-
# @Author   : zbz

import time
import asyncio

from motor.motor_asyncio import AsyncIOMotorClient


async def asy_mongo_io():
    mcli = AsyncIOMotorClient('mongodb://localhost:27017')
    table = mcli["ztasks"]["info"]
    l = await table.count_documents({})

    # 慢
    begin = time.time()
    for _ in range(l):
        docu = (await table.find({}).to_list(length=1))[0]
        print(docu)
    print("cons time {}".format(time.time() - begin))

    # 快
    begin = time.time()
    async for docu in table.find({}):
        print(docu)
    print("cons time {}".format(time.time() - begin))


if __name__ == '__main__':
    coro = asy_mongo_io()
    loop = asyncio.get_event_loop()
    loop.run_until_complete(coro)