JMeter学习（十九）JMeter测试MongoDB

xuke123

　　JMeter测试MongoDB性能有两种方式，一种是利用JMeter直接进行测试MongoDB，还有一种是写Java代码方式测试MongoDB性能。
　　
　　第一种方法
　　1.编写Java代码，内容如下：



package com.test.mongodb;
import java.io.File;  
import java.io.FileOutputStream;  
import java.io.IOException;  
import java.util.UUID;
import java.util.concurrent.atomic.AtomicLong;  
import org.apache.jmeter.config.Arguments;  
import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;  
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;  
import org.apache.jmeter.samplers.SampleResult;  
import com.mongodb.BasicDBObject;  
import com.mongodb.DB;  
import com.mongodb.DBCollection;  
import com.mongodb.Mongo;  
import com.mongodb.MongoOptions;  
import com.mongodb.ServerAddress;  
import com.mongodb.WriteResult;  
/**
*  
* @author  2014-10-14
* @remark  一次性插入多条数据测试MongoDB的性能   
*/  
public class TestMongodb extends AbstractJavaSamplerClient {  
private static Mongo            m;  
private static DB               db;  
private static AtomicLong       read_key = new AtomicLong(0);  
private static FileOutputStream fos;  
static {  
try {  
fos = new FileOutputStream(new File("jmeter_error.log"));  
MongoOptions options = new MongoOptions();  
options.autoConnectRetry = true;  
options.connectionsPerHost = 1000;  
options.maxWaitTime = 5000;  
options.socketTimeout = 0;  
options.connectTimeout = 15000;  
options.threadsAllowedToBlockForConnectionMultiplier = 5000;  
m = new Mongo(new ServerAddress("127.0.0.1", 27017), options);  
db = m.getDB("iflashbuy-log");  
} catch (Exception e) {  
throw new RuntimeException(e);  
}  
}  
public Arguments getDefaultParameters() {
Arguments params = new Arguments();
params.addArgument("rw", "w");
return params;
}
public SampleResult runTest(JavaSamplerContext context) {  
// System.out.println(read_key.getAndIncrement());  
SampleResult results = new SampleResult();  
//System.out.println(context.getParameter("rw"));  
        results.sampleStart();  
DBCollection coll = db.getCollection("area");  
if ("w".equalsIgnoreCase(context.getParameter("rw"))) {  
long key = 1;  
BasicDBObject doc = new BasicDBObject();               
doc.put("_id", UUID.randomUUID());  
doc.put("_class", "com.iflashbuy.bi.source.pojo.AreaDataEntity");  
doc.put("province", "广东");  
doc.put("city", "广州");  
doc.put("pv", 35522924);  
doc.put("uv", 52556);  
doc.put("orderCount", 963);  
doc.put("orderTotal", 1548563);  
doc.put("entpriseCode", "00540001");  
db.requestStart();  
WriteResult rs = coll.insert(doc);  
try {  
if (rs.getError() != null) fos.write(rs.getError().getBytes());  
} catch (IOException e) {  
// TODO Auto-generated catch block  
                e.printStackTrace();  
}  
//System.out.println("w error->" + rs.getError());  
            db.requestDone();  
}  
results.setSuccessful(true);  
results.sampleEnd();  
return results;  
// return null;  
    }  
/*
public static void main(String args[]) throws Exception {  
Arguments a = new Arguments();  
a.addArgument("rw", "w");  
final JavaSamplerContext c = new JavaSamplerContext(a);  
long start = System.currentTimeMillis();  
for (int i = 0; i < 5; i++) {  
new Thread() {  
public void run() {  
TestMongodb t = new TestMongodb();  
for (int j = 0; j < 2; j++) {  
t.runTest(c);  
}  
}  
}.start();  
}  
}  
*/
}  
　　2.生成jar包，放在Jmeter_HOME/lib/ext文件夹下
　　3.启动JMeter，新建线程组，Java请求，察看结果树，聚合报告，图形结果

　　4.执行测试
　　检查MongoDB已添加成功数据。
　　
　　第二种方法
　　1.启动JMeter，新建线程组，设置线程组属性
　　2.右键添加－MongoDB Source Config
　　设置属性Server Address List：127.0.0.1
　　MongoDB Source：jmeterdbsource，如下图设置：

　　3.右键添加－Sampler－MongoDB Script
　　设置属性MongoDB Source：引用MongoDB source config中设置的好的名称，jmeterdbsource
　　Database Name：数据库名称，iflashbuy-log
　　Script：脚本

　　4.右键－监听器－察看结果树

　　
　　以上两种方式，都可用于测试MongoDB的性能。
　　小结：学习Jmeter有一阵子了，越来越体会到Jmeter对于Java的支持太强大了.
　　
　　附：
　　Mongodb亿级数据量的性能测试　http://www.iyunv.com/lovecindywang/archive/2011/03/02/1969324.html