java jmeter_使用Jmeter中的Java Request进行性能测试

在使用jmeter进行性能测试的时候，有可能会需要通过一些脚本去测试性能，(比如通过sokeeper的api去测试sokeeper的读写性能)。这时，我们可以通过Java Request实现，以下是操作步骤。

首先创建一个maven project。

因Jmeter的Java Request脚本依赖Jmeter的一些核心jar包，所以我们要maven项目的pom文件中添加Jmeter的一些依赖包：ApacheJMeter_core、ApacheJMeter_java：

编写Java Sample，继承AbstractJavaSample，并重写以下方法：

public Arguments getDefaultParameters()；可选，定义可用参数及默认值；

·  public void setupTest(JavaSamplerContext arg0)：可选，测试前执行，做一些初始化工作；

·  public SampleResult runTest(JavaSamplerContext arg0)；必选，实现自定义请求；

·  public void teardownTest(JavaSamplerContext arg0)：可选，测试结束时调用；

示例代码：

import org.apache.curator.RetryPolicy;

import org.apache.curator.framework.CuratorFramework;

import org.apache.curator.framework.CuratorFrameworkFactory;

import org.apache.curator.retry.ExponentialBackoffRetry;

import org.apache.curator.utils.EnsurePath;

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;

public class CuratorRequestSample extends AbstractJavaSamplerClient {

protected CuratorFramework client = null;

protected long subCount = 1l;

protected long TOTAL_SUB_COUNT = 100000l;

/**

* 定义可用参数及默认值

*/

public Arguments getDefaultParameters() {

Arguments params = new Arguments();

params.addArgument("connect_str", "192.168.103.136:2181,192.168.103.154:2181");

// 设置参数，并赋予默认值0

return params;

}

/**

* 可选，测试前执行，做一些初始化工作，在Jmeter执行的时候，一个Thread只执行一次setupTest方法

*/

public void setupTest(JavaSamplerContext context) {

String connect_str = context.getParameter("connect_str");

RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 4);

client =

CuratorFrameworkFactory.builder().connectString(connect_str)

.connectionTimeoutMs(3000).namespace("curator_sample")

.sessionTimeoutMs(5000).retryPolicy(retryPolicy).build();

client.start();

EnsurePath ensure = new EnsurePath("/curator_sample/setData");

try {

ensure.ensure(client.getZookeeperClient());

} catch (Exception e) {

e.printStackTrace();

}

}

/**

* 可选，测试结束时调用,用于资源释放，一个Thread只执行一次

*/

public void teardownTest(JavaSamplerContext context) {

client.close();

}

/**

* 必选，实现自定义请求

*/

public SampleResult runTest(JavaSamplerContext arg0) {

SampleResult resuts = new SampleResult();

resuts.sampleStart();

try {

byte[] bytes = client.getData().forPath("/setData");

resuts.setSuccessful(true);

} catch (Exception e) {

resuts.setSuccessful(false);

}

resuts.sampleEnd();

return resuts;

}

}

然后，把项目打成jar包，并将包放到${jmeter_home}/lib/ext/目录下。如还需要依赖第三方jar，需要将第三方的jar也放到${jmeter_home}/lib/ext/目录下。

启动Jmeter。

创建线程组：

创建Java Request Sample：

此时，会在Java请求页面的“类名称”下拉列表中看到我们之前写的Java Request脚本：

选中其中一个要测试的脚本，可以看到配置的参数：

然后再添加Summary Report即可，保存测试计划。设置线程组属性，执行测试计划。即可。

优化技巧一

很多情况下我们在编写Java Sample时，需要编写多个Java Sample，且这些Java Sample之间，拥有共同的资源初始化操作、资源释放操作，只是在核心业务不同而已，这时，我们可以封装一个基类。例：

import org.apache.curator.RetryPolicy;

import org.apache.curator.framework.CuratorFramework;

import org.apache.curator.framework.CuratorFrameworkFactory;

import org.apache.curator.retry.ExponentialBackoffRetry;

import org.apache.curator.utils.EnsurePath;

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;

/**

* Curator的Java请求测试基类

*

* @author majian

* @version 1.0, 2016-2-22

* @since 1.0, 2016-2-22

*/

public abstract class CuratorBaseRequest extends AbstractJavaSamplerClient {

protected CuratorFramework client = null;

protected long subCount = 1l;

protected long TOTAL_SUB_COUNT = 100000l;

/**

* 定义可用参数及默认值

*/

public Arguments getDefaultParameters() {

Arguments params = new Arguments();

params.addArgument("connect_str", "192.168.103.136:2181,192.168.103.154:2181");// 设置参数，并赋予默认值0

return params;

}

/**

* 可选，测试前执行，做一些初始化工作，在Jmeter执行的时候，一个Thread只执行一次setupTest方法

*/

public void setupTest(JavaSamplerContext context) {

String connect_str = context.getParameter("connect_str");

RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 4);

client =

CuratorFrameworkFactory.builder().connectString(connect_str)

.connectionTimeoutMs(3000).namespace("curator_sample")

.sessionTimeoutMs(5000).retryPolicy(retryPolicy).build();

client.start();

EnsurePath ensure = new EnsurePath("/curator_sample/setData");

try {

ensure.ensure(client.getZookeeperClient());

} catch (Exception e) {

e.printStackTrace();

}

}

/**

* 可选，测试结束时调用,用于资源释放，一个Thread只执行一次

*/

public void teardownTest(JavaSamplerContext context) {

client.close();

}

/**

* 必选，实现自定义请求

*/

public SampleResult runTest(JavaSamplerContext arg0) {

SampleResult resuts = new SampleResult();

resuts.sampleStart();

try {

testHandler();

resuts.setSuccessful(true);

} catch (Exception e) {

resuts.setSuccessful(false);

}

resuts.sampleEnd();

return resuts;

}

/**

* 核心业务执行功能实现，只需实现此接口即可

* @throws Exception

*/

public abstract void testHandler() throws Exception;

}

然后我们可以创建不同的Java请求，继承这个基类，只需实现testHandler()方法即可。如：

curator获取数据的Java Request:

public class GetDataRequest extends CuratorBaseRequest {

public void testHandler() throws Exception {

byte[] bytes = client.getData().forPath("/setData");

}

}

Curaotr设置数据的Java Request：

import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;

public class SetDataRequest extends CuratorBaseRequest {

private byte[] a100 = new byte[100];

public void setupTest(JavaSamplerContext context) {

for (int i = 0; i 

a100[i] = 'A';

}

super.setupTest(context);

}

public void testHandler() throws Exception {

client.setData().forPath("/setData", a100);

}

}

这样，可以避免大量的冗余代码。

优化技巧二

在编写完Java Sample后，每次都需要我们打成jar包，然后再复制jar包到${jmeter_home}\lib\ext下，然后再重起jmeter，这样操作比较繁琐。这时，我们可以通过在maven项目中添加一个maven插件来帮助我们简化操作。

org.apache.maven.plugins

maven-antrun-plugin

copy-files

package

overwrite="true" />

run

如此，我们便可在执行mvn package命令的时候，自动将项目打包，并将包复制到指定的目录下了，我们只需再重启jmeter就可以了。