Python 写的一个MongoDB压力测试

$ pip install pymongo

#!/usr/bin/env python

#coding=utf-8

#Author: Ca0Gu0

from pymongo import MongoClient

import datetime,time

class MongCli(object):

def __init__(self, host="127.0.0.1", port=27017, user='pytest', passwd ='pytest123', database="pytest"):

self.host = host

self.port = port

self.user = user

self.passwd = passwd

self.database = database

client = MongoClient(self.host, self.port)

client.the_database.authenticate(self.user, self.passwd, source=self.database)

self.db=client[self.database]

self.posts = self.db.posts

def t(self, args1=None, args2=None):

c=datetime.datetime.now()

print "%s|%s... |%s|Row:%s| %s" %(self.host, args1, args2, self.posts.count(), c.strftime("%Y-%m-%d %H:%M:%S"))

return c

def write(self,number=100):

start = self.t(args1="start", args2="write")

for i in range(number):

post = {"author": "Mike"+str(i),

"text": "My first blog post!"+str(i),

"tags": ["mongodb", "python", "pymongo"],

"date": datetime.datetime.utcnow()}

post_id = self.posts.insert_one(post).inserted_id

end = self.t(args1="end", args2="write")

print "Total write runtime: %ss" %str((end-start).seconds)

def read(self):

start = self.t(args1="start", args2="read")

output = open("output.txt", 'w')

for post in self.posts.find():

try:

output.write(str(post)+"\n")

except Exception,e:

print e

output.close()

end = self.t(args1="end", args2="read")

print "Total read runtime: %ss" %str((end-start).seconds)

print "----------Split--------"

if __name__ == "__main__":

f = MongCli(host="127.0.0.1", port=27017, user='admin', passwd ='admin', database="admin")

f.write(20000)

f.read()

《MongoDB 权威指南》(MongoDB: The Definitive Guide)英文文字版[PDF] http://www.linuxidc.com/Linux/2012-07/66735.htm

MongoDB 的详细介绍：请点这里

MongoDB 的下载地址：请点这里