MongoDB 测试 Python 代码

最新推荐文章于 2024-05-30 14:25:09 发布

原创最新推荐文章于 2024-05-30 14:25:09 发布 · 720 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#测试 #python #mongodb #数据库

Python 同时被 2 个专栏收录

22 篇文章

订阅专栏

MongoDB

16 篇文章

订阅专栏

本文提供了一组Python脚本来测试MongoDB的插入、查询及删除操作性能。通过循环进行大量数据的写入、随机查询及批量删除，评估MongoDB在高负载场景下的表现。

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

随机数查询测试 MongoDB 的 Python 代码

写库程序

#!/usr/bin/env python

from pymongo import Connection
import time
import datetime

connection = Connection('127.0.0.1', 27017)
db = connection['hawaii']

#时间记录器
def func_time(func):

    def _wrapper(*args,**kwargs):

        start = time.time()
        func(*args,**kwargs)
        print func.name,'run:',time.time()-start
    return _wrapper

@func_time
def insert(num):

    posts = db.userinfo

    for x in range(num):

        post = {"_id" : str(x),
                "author": str(x)+"Mike",
                "text": "My first blog post!",
                "tags": ["mongodb", "python", "pymongo"],
                "date": datetime.datetime.utcnow()}

        posts.insert(post)


if name == "main":

    #设定循环500万次
    num = 5000000
    insert(num)

查询程序

#!/usr/bin/env python

from pymongo import Connection
import time
import datetime
import random

connection = Connection('127.0.0.1', 27017)
db = connection['hawaii']

def func_time(func):

    def _wrapper(*args,**kwargs):
        start = time.time()
        func(*args,**kwargs)
        print func.name,'run:',time.time()-start
    return _wrapper

#@func_time
def randy():

    rand = random.randint(1,5000000)
    return rand

@func_time
def mread(num):

    find = db.userinfo
    for i in range(num):
        rand = randy()
        #随机数查询
        find.find( {"author": str(rand)+"Mike"} )


if name == "main":

    #设定循环100万次
    num = 1000000
    mread(num)

删除程序

#!/usr/bin/env python

from pymongo import Connection
import time
import datetime

connection = Connection('127.0.0.1', 27017)
db = connection['hawaii']

def func_time(func):

    def _wrapper(*args,**kwargs):

        start = time.time()
        func(*args,**kwargs)
        print func.name,'run:',time.time()-start

    return _wrapper

@func_time
def remove():

    posts = db.userinfo
    print 'count before remove:',posts.count();
    posts.remove({});
    print 'count after remove:',posts.count();

if name == "main":
    remove()