Jmeter插件开发实现对mongo数据库的操作

背景：使用JMeter做接口自动化，用例执行前需要清除一些数据，操作完后校验mongo数据。因为Jmeter3没有对Mongo数据库的支持，所以自己开发java请求实现对mongo的操作。
1.eclipse新建java工程，依赖jar包如下



2.写一个类实现连接mongo数据库，返回要操作的collection对象public DBCollection ConnectionMongoClient(String host, int port, String dbname,String collection) {
         Mongo connection = new Mongo(host,port);
DB db = connection.getDB(dbname);
DBCollection coll = db.getCollection(collection);
return coll;

} 3.参看官网API，java请求需要继承AbstractJavaSamplerClient抽象类，下面以mongo的查询操作为例的实现



sampler核心代码



more details see 点击打开链接
getDefaultParameters:设置请求中的参数以及默认值
public Arguments getDefaultParameters(){
Arguments params=new Arguments();
params.addArgument("ip","127.0.0.1");
params.addArgument("port","27017");
params.addArgument("dbname","dummy");
params.addArgument("collection","users");
params.addArgument("queryjsonString","");
return params;
}setupTest:参数初始化public void setupTest(JavaSamplerContext context){
System.out.println("执行setup");
ip= context.getParameter("ip");
port=context.getIntParameter("port");
dbname= context.getParameter("dbname");
collection= context.getParameter("collection");
queryjsonString=context.getParameter("queryjsonString");

}runTest：操作核心代码：设置请求的名称，执行Mongo的查询，设置查看结果树的响应数据@Override
public SampleResult runTest(JavaSamplerContext context) {
System.out.println("执行runTest...");

SampleResult sampleresult = new SampleResult();
sampleresult.sampleStart();
//设置sampler请求的名称
sampleresult.setSampleLabel("MongodbSampler" + Thread.currentThread().getId());

//连接mongo
DBCollection coll=new TestMongoConection().ConnectionMongoClient(ip, port, dbname, collection);
DBObject query=new BasicDBObject();
if (queryjsonString!=""&&queryjsonString!=null){
query=(DBObject) com.mongodb.util.JSON.parse(queryjsonString);
}

//日志打印查询条件,查看查询条件是否传值错误
getLogger().debug(query.toString());

DBCursor cursor = coll.find(query);
while (cursor.hasNext()){
DBObject dbobj=cursor.next();
responseData =responseData+dbobj.toString()+ "\n";
}
if ( responseData!= "" || !responseData.equals("")) {
//设置响应数据为查询结果
sampleresult.setResponseData(responseData.getBytes());
}else {
sampleresult.setResponseData(("the return data is null !").getBytes());
}

sampleresult.setSuccessful(true);
sampleresult.sampleEnd();
return sampleresult;

}删除等操作，仅需修改对mongo操作的相应代码 WriteResult wr=coll.remove(query);
int deleteDocNum=wr.getN();
if (deleteDocNum>0){
responseData=deleteDocNum+" documents deleted successfully";
}else{
responseData="no document deleted";
}
4.界面操作
4.1 请求：



4.2 查看结果，设置的请求名称和响应数据。可对响应数据进行断言，确定用例结果



4.3 将一些关键信息打印入日志，方便调试问题