要想通过自定义java请求来压测ZooKeeper，那么我们就需要做两件事情，第一我们需要知道java如何操作ZooKeeper，第二就是怎么能将我们写的jar包让jmeter识别，首先我们先来干第一件事。
一、java操作ZooKeeper
以下的代码是网上百度找到的，经过了一点点修改（对于测试其它的可以找其它的测试代码）
package com.comtop.ZookApi; import java.util.concurrent.CountDownLatch; import org.apache.zookeeper.CreateMode; import org.apache.zookeeper.WatchedEvent; import org.apache.zookeeper.Watcher; import org.apache.zookeeper.Watcher.Event.EventType; import org.apache.zookeeper.Watcher.Event.KeeperState; import org.apache.zookeeper.ZooDefs.Ids; import org.apache.zookeeper.ZooKeeper; /** * Zookeeper base学习笔记 */ public class ZookeeperBase { /** zookeeper地址 */ static final String CONNECT_ADDR = "10.10.3.136:32372"; /** session超时时间 */ static final int SESSION_OUTTIME = 2000;// ms /** 信号量，阻塞程序执行，用于等待zookeeper连接成功，发送成功信号 */ static final CountDownLatch connectedSemaphore = new CountDownLatch(1); public static void main(String[] args) throws Exception { ZooKeeper zk = new ZooKeeper(CONNECT_ADDR, SESSION_OUTTIME, new Watcher() { public void process(WatchedEvent event) { // 获取事件的状态 KeeperState keeperState = event.getState(); EventType eventType = event.getType(); // 如果是建立连接 if (KeeperState.SyncConnected == keeperState) { if (EventType.None == eventType) { // 如果建立连接成功，则发送信号量，让后续阻塞程序向下执行 System.out.println("zk 建立连接"); connectedSemaphore.countDown(); } } } }); // 进行阻塞 connectedSemaphore.await(); System.out.println(".."); // 创建父节点 // zk.create("/testRoot", "testRoot".getBytes(), Ids.OPEN_ACL_UNSAFE, // CreateMode.PERSISTENT); // 创建子节点 // zk.create("/testRoot/children", "children data".getBytes(), // Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); // 获取节点洗信息 // byte[] data = zk.getData("/testRoot/children", false, null); // System.out.println(new String(data)); // System.out.println(zk.getChildren("/testRoot", false)); // 修改节点的值 // zk.setData("/testRoot/children", "modify data root".getBytes(), -1); // byte[] data = zk.getData("/testRoot/children", false, null); // System.out.println(new String(data)); // 如果存在节点则删除节点 if (null != zk.exists("/testRoot/children", false)) { // 删除节点 zk.delete("/testRoot/children", -1); System.out.println(zk.exists("/testRoot/children", false)); } zk.close(); } }
二、自定义java请求
一、环境准备
1、新建一个java工程
2、导入jar包：ApacheJMeter_core.jar ApacheJMeter_java.jar （该包在本地C:\apache-jmeter-3.0\lib\ext下，当然路径取决于本地环境）
这两个jar是使用jmeter最基础的jar，能够让你的代码在jmeter中运行起来，如果在写代码的过程中需要其他的jar，自行导入。
二、写代码前该知道的
1、如果想要让你的代码在jmeter中运行起来，在创建类的时候需要去继承AbstractJavaSamplerClient抽象类或者是实现JavaSamplerClient接口
AbstractJavaSamplerClient抽象类是JavaSamplerClient接口的子类，当你不需要复写所有的需要复写的方法时，那么你只需要去继承AbstractJavaSamplerClient抽象类即可。
如果你选择了实现JavaSamplerClient接口，那么你需要复写的方法有：
public SampleResult runTest(JavaSamplerContext context) {}
public void setupTest(JavaSamplerContext context) {}
public void teardownTest(JavaSamplerContext context) {}
public Arguments getDefaultParameters() {}
这4个方法就必须要复写，但是如果选择继承AbstractJavaSamplerClient这个抽象类，那么只需要复写你需要的方法即可。
2、方法说明：
public Arguments getDefaultParameters() {}
这个方法由Jmeter在进行添加javaRequest时第一个运行，它决定了你要在GUI中默认显示出哪些属性。
public void setupTest(JavaSamplerContext context) {}
这个方法相当于loadrunner中的init，我们可以用它来进行一些初始化的动作。
public SampleResult runTest(JavaSamplerContext context) {}
这个方法相当于loadrunner中的action，我们的核心测试代码就在这里了。
public void teardownTest(JavaSamplerContext context) {}
这个方法相当于loadrunner中的end，收尾的工作可以由它来做。
3、除了以上2点，我们一般还需要去实现Serializable，序列化标记接口，这样可以让我们的类去实现序列化。
整体代码如下：
package com.comtop.ZkApiJM; import java.io.IOException; import java.io.InputStream; import java.io.Serializable; import java.net.MalformedURLException; import java.net.URL; import java.net.URLConnection; 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 java.util.concurrent.CountDownLatch; import javax.management.RuntimeErrorException; import org.apache.zookeeper.CreateMode; import org.apache.zookeeper.WatchedEvent; import org.apache.zookeeper.Watcher; import org.apache.zookeeper.Watcher.Event.EventType; import org.apache.zookeeper.Watcher.Event.KeeperState; import org.apache.zookeeper.ZooDefs.Ids; import org.apache.zookeeper.ZooKeeper.States; import org.apache.zookeeper.ZooKeeper; //public Arguments getDefaultParameters();设置可用参数及的默认值； //public void setupTest(JavaSamplerContext arg0)：每个线程测试前执行一次，做一些初始化工作； //public SampleResult runTest(JavaSamplerContext arg0)：开始测试，从arg0参数可以获得参数值； //public void teardownTest(JavaSamplerContext arg0)：测试结束时调用； public class ZkjavaRequest extends AbstractJavaSamplerClient implements Serializable { /** zookeeper地址 */ private String conn_addr; //存储用户输入的zk地址 // 测试用的数据 // private static String conn_addr1; // private static String session_timeout1; // private static String zk_father1; // private static String zk_children1; // private static String zk_context1; // private static String resultData1; private static final String ConnAddrName="conn_addr" ; //设置GUI页面显示的变量名称 //设置GUI页面默认显示的变量值,默认值为空 private static final String ConnAddrValueDefault="10.10.3.136:32372"; /** session超时时间 */ private String session_timeout; //存储用户输入的session超时时间(单位ms) private static final String SessTimeName="session_timeout" ; //设置GUI页面显示的变量名称 //设置GUI页面默认显示的变量值,默认值为空 private static final String SessTimeValueDefault="5000"; //zk父节点 private String zk_father = "test"; //存储用户输入的zk父节点 private static final String ZkFatherName="zk_father" ; //设置GUI页面显示的变量名称 //设置GUI页面默认显示的变量值,默认值为空 private static final String ZkFatherValueDefault="/test"; /** 信号量，阻塞程序执行，用于等待zookeeper连接成功，发送成功信号 */ static final CountDownLatch connectedSemaphore = new CountDownLatch(1); //zk连接对象 private ZooKeeper zk; // resultData变量用来存储响应的数据，目的是显示到查看结果树中。 private String resultData; // 测试代码 // public static void main(String[] args) { // haha(); // } // // public static void haha(){ // conn_addr1 = "10.10.3.136:32372"; // session_timeout1 = "2000"; // zk_father1 = "/test1"; // zk_children1 = "/test2"; // zk_context1 = "test2"; // try { // ZooKeeper zk = new ZooKeeper(conn_addr1, Integer.valueOf(session_timeout1), // new Watcher() { // public void process(WatchedEvent event) { // // 获取事件的状态 // KeeperState keeperState = event.getState(); // EventType eventType = event.getType(); // // 如果是建立连接 // if (KeeperState.SyncConnected == keeperState) { // if (EventType.None == eventType) { // // 如果建立连接成功，则发送信号量，让后续阻塞程序向下执行 // System.out.println("zk 建立连接"); // connectedSemaphore.countDown(); // } // } // } // }); // connectedSemaphore.await(); // // 如果存在父节点则不创建父节点 // if (null == zk.exists(zk_father1, false)) { // // 创建父节点 // zk.create(zk_father1, zk_father1.getBytes(), Ids.OPEN_ACL_UNSAFE, // CreateMode.PERSISTENT); // } // // 创建子节点 // zk.create(zk_father1+zk_children1, zk_context1.getBytes(), // Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); // // 获取节点洗信息 // byte[] data = zk.getData(zk_father1+zk_children1, false, null); // resultData1 = new String(data); // System.out.println(new String(data)); // System.out.println(zk.getChildren("/testRoot", false)); // } catch (Exception e) { // throw new RuntimeException(e); // } // //} /* * 这个方法用来控制显示在GUI页面的属性，由用户来进行设置。 * 此方法不用调用，是一个与生命周期相关的方法，类加载则运行。 * (non-Javadoc) * @see org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient#getDefaultParameters() */ @Override public Arguments getDefaultParameters() { System.out.println("读取属性值"); Arguments params = new Arguments(); params.addArgument("conn_addr",String.valueOf(ConnAddrValueDefault)); params.addArgument("session_timeout", String.valueOf(SessTimeValueDefault)); params.addArgument("zk_father", String.valueOf(ZkFatherValueDefault)); params.addArgument("zk_children",""); params.addArgument("zk_context",""); return params; } /** * 初始化方法，初始化性能测试时的每个线程 * 实际运行时每个线程仅执行一次，在测试方法运行前执行，类似于LoadRunner中的init方法 */ @Override public void setupTest(JavaSamplerContext jsc) { conn_addr = jsc.getParameter(ConnAddrName, ConnAddrValueDefault); session_timeout = jsc.getParameter(SessTimeName, SessTimeValueDefault); zk_father = jsc.getParameter(ZkFatherName, ZkFatherValueDefault); try { zk = new ZooKeeper(conn_addr, Integer.valueOf(session_timeout), new Watcher() { public void process(WatchedEvent event) { // 获取事件的状态 KeeperState keeperState = event.getState(); EventType eventType = event.getType(); // 如果是建立连接 if (KeeperState.SyncConnected == keeperState) { if (EventType.None == eventType) { // 如果建立连接成功，则发送信号量，让后续阻塞程序向下执行 System.out.println("zk 建立连接"); connectedSemaphore.countDown(); } } } }); if (States.CONNECTING == zk.getState()) { connectedSemaphore.await(); } // 如果存在父节点则不创建父节点 if (null == zk.exists(zk_father, false)) { // 创建父节点 zk.create(zk_father, zk_father.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); } } catch (Exception e) { // TODO Auto-generated catch block // e.printStackTrace(); throw new RuntimeException(e); } // 进行阻塞 // System.out.println(".."); } @Override public SampleResult runTest(JavaSamplerContext arg0) { String zk_children = arg0.getParameter("zk_children"); //获取zk_children String zk_context = arg0.getParameter("zk_context"); //获取zk_context // System.out.println(zk_context+" "+zk_children); /* * SampleResult这个类是用来将测试结果输出到查看结果树中的。 * 并且也是用来控制事务的开始和结束的。 */ SampleResult results = new SampleResult(); results.setSampleLabel("zk节点测试:"+zk_father); try{ // 事务开始标记 results.sampleStart(); // 创建子节点 zk.create(zk_father+zk_children, zk_context.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); // 获取节点洗信息 byte[] data = zk.getData(zk_father+zk_children, false, null); // System.out.println(new String(data)); resultData = new String(data); if(null == resultData){ results.setSuccessful(false); results.setResponseData("zk result is null",null); results.setDataType(SampleResult.TEXT); } results.setSuccessful(true); }catch(Exception e){ results.setSuccessful(false); results.setResponseData(e.toString(),null); results.setDataType(SampleResult.TEXT); e.printStackTrace(); return results; }finally{ //标记事务结束 results.sampleEnd(); } // System.out.println("写入成功"); // results.setSuccessful(true); results.setResponseData(resultData,null);//将数据打印到查看结果树当中 results.setDataType(SampleResult.TEXT); return results; } /** * 测试结束方法，结束测试中的每个线程 * 实际运行时，每个线程仅执行一次，在测试方法运行结束后执行，类似于Loadrunner中的End方法 */ public void teardownTest(JavaSamplerContext arg0) { try { zk.close(); System.out.println("关闭"); } catch (InterruptedException e) { // TODO Auto-generated catch block e.printStackTrace(); } } }
代码及依赖包如下：
413
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
