使用JMeter测试你的EJB

对EJB进行一些性能基准测试是非常有必要和有帮助的，测试的方法和工具有很多，不过我最近发现，Apache JMeter是进行基准测试的一个优秀工具。可惜的是，JMeter没有提供一个可测试任意EJB的通用取样器(sampler)，不过，我们可以自己动手来创建一个。

首先，让我们简单的了解一下Apache JMeter，它是一个100%的纯Java桌面应用，可用于压力测试和性能测量。它最初被设计用于Web应用测试但后来扩展到其他测试领域。

在本篇文章中，我使用JBoss应用服务器来运行我的EJB。使用其它容器的实现过程应该也非常类似。

1、首先为EJB创建一个工厂(factory)类

我们需要做的第一件事情是，创建一个简单的Singleton factory class，以为你的测试创建EJB客户端实例。参考代码如下：

public class MyServiceFactory {

private static final Log log = LogFactory.getLog(MyServiceFactory.class);

private static MyService service;

private static MyServiceFactory me;

private MyServiceFactory() { }

static {

MyServiceFactory.me = new MyServiceFactory();

}

public static MyServiceFactory getInstance() {

return MyServiceFactory.me;

}

public MyService getService() {

if (MyService.service == null) {

try {

log.info("Loading the service...");

Context ctx = new InitialContext();

service = (MyService)ctx.lookup("MyAction/remote");

if (service == null) {

log.error("Didn't get the service!");

}

} catch (NamingException e) {

log.error("Error looking up the remote service", e);

return null;

}

}

return service;

}

}

　　2、编写测试代码

接下来我们需要自己编写测试代码，为了实现这个目的，我们可以对JMeter的 org.apache.jmeter.protocol.java.sampler包中的AbstractJavaSamplerClient 类进行扩展。这个抽象类具有一个runTest方法，我们需要对其进行重写(override)，使用这个方法来实现实际的测试。另外，我们还将重写 getDefaultParameters方法，以使其提供某些合理的默认值，它们将在JMeter图形应用界面中显示。

package us.mikedesjardins.demo.jmeter;

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 DigitalContentServiceEJBTestSampler extends AbstractJavaSamplerClient {

public SampleResult runTest(JavaSamplerContext context) {

SampleResult results = new SampleResult();

MyService service = MyServiceFactory.getInstance().getService();

results.sampleStart();

Long param1 = context.getLongParameter("PARAM_1");

String param2 = context.getStringParameter("PARAM_2");

MyResult result = service.myMethod(param1, param2);

if (result != null) {

results.setSuccessful(true);

results.setResponseCodeOK();

results.setResponseMessage("'myResult:" + myResult);

} else {

results.setSuccessful(false);

}

results.sampleEnd();

return results;

}

@Override

public Arguments getDefaultParameters() {

Arguments args = new Arguments();

args.addArgument("PARAM_1", "4815162342");

args.addArgument("PARAM_2", "Iculus");

return args;

}

}

　　3、运行JMeter

JMeter的扩展lib目录是${JMETER_INSTALL_LIB}/lib/ext。你需要把所有EJB客户端所需要的jar文件拷贝到这个目录中。如果你使用的是JBoss，则需要把jbossall-client.jar拷贝到这个目录中;对于其它应用服务器，则拷贝类似的客户端jar文件到这个目录。

当你启动JMeter后，你的新取样器(sampler)将会出现在它的Sampler菜单中，这样你就可以使用它来测试你的EJB了。