开发性能测试工具——自己动手实现迭代功能

在用Jmeter进行性能测试时也许有人遇到过这样的问题：

测试时TPS相当大，一台，二台负载机都找不到拐点，最后准备好多台负载机才解决问题，而且管理这些机器时比较麻烦。

我们能不能解决这些问题呢？当然可以，我们可以自动动手来实现一个迭代器。之所以能够高效，是因为我们去掉了测试工具的诸多辅助功能。

比如实时计算当前TPS、平均响应时间；记录日志等待工作，所以能够更快。

废话少说，下面实现一个简单的迭代器，以备不时之需。

1.定义一个接口，说明要做什么；参照Junit与LoadRunner  Java Vuser来定义。

package com.seling.thread;

public interface BaseTask extends Runnable {
	/**
	 * 初始化操作
	 */
	public void init();
	/**
	 * 具体迭代的内容，要执行的测试程序放在此方法中，可以有入参
	 * @param i
	 */
	public String action(int i);
	/**
	 * 收尾工作
	 */
	public void end();

}

2.实现接口程序，也就是具体任务内容。

</pre><pre name="code" class="java">package com.seling.thread;

import org.apache.log4j.Logger;
import com.seling.test.UserDaoI;
import com.seling.test.UserDaoImpl;

public abstract class Task implements BaseTask {

	private static Logger log = Logger.getLogger(Task.class);
	
	private String threadNo;//入参示例，这里传入线程号，实际可参与对象
	
	private UserDaoI ud;//此测待测试的接口
	/**
	 * 通过构造方法来初始化入参
	 * @param thread
	 */
	public Task(String thread){
		this.threadNo = thread;
	}
	
	@Override
	public abstract void run();

	@Override
	public void init() {
		// TODO Auto-generated method stub
		log.info("init "+this.threadNo);
		ud = new UserDaoImpl();
	}

	@Override
	public String action(int i) {
		// TODO Auto-generated method stub
		log.info("action "+this.threadNo+" iterator "+i);
		//System.out.println("action "+this.threadNo+" iterator "+i);
		//测试ud.getUserById()接口
		return ud.getUserById(this.threadNo+"-"+i);//这个是我们要测试的接口主法
		
	}

	@Override
	public void end() {
		// TODO Auto-generated method stub

	}

}

3.开启多个线程来执行任务

package com.seling.thread;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;

import org.apache.log4j.Logger;

public class TestTask extends Task {

	private static Logger log = Logger.getLogger(TestTask.class);

	private static AtomicLong totalfailRecords = new AtomicLong(0);

	private static AtomicLong totalPassRecords = new AtomicLong(0);
	
	private static int  runTime = 30000 ;//测试执行时长，单位毫秒

	public TestTask(String thread) {
		super(thread);
	}

	/**
	 * 开5个线程执行任务，平时测试中开启多少个线程自己决定
	 * 
	 * @param args
	 */
	public static void main(String args[]) {
		for (int i = 1; i <= 5; i++) {
			new Thread(new TestTask("t" + i)).start();//开启线程执行任务
		}
	}

	@Override
	public void run() {
		this.init();
		int i = 0;
		boolean ifRun = true;
		long start = System.currentTimeMillis();
		while (ifRun) {
			String user = this.action(i++);
			if (null != user)//验证事务是否成功
				totalPassRecords.incrementAndGet();
			else
				totalfailRecords.incrementAndGet();
			// 模拟ThinkTime，单位毫秒,对于TPS过万的接口建议不用模拟ThinkTime
			try {
				TimeUnit.MILLISECONDS.sleep(2000L);
			} catch (InterruptedException e) {
				e.printStackTrace();
			}
			long end = System.currentTimeMillis();
			long runInstance = end - start;
			if(runInstance >= runTime){
				ifRun = false;
				//执行完之后计算TPS
				System.out.println("Duration(ms): " + (runInstance));
				System.out.println("---------------------------------------------");
				System.out.println("pass trasactions : " + totalPassRecords.get());
				System.out.println("fail trasactions : " + totalfailRecords.get());
				System.out.println("TPS: " + totalPassRecords.get() / (runInstance / 1000));
				System.out.println("---------------------------------------------");
				Thread.currentThread().stop();
			}
			//log.info("Fail trasactions : " + totalfailRecords.get());
			//log.info("Pass trasactions ：  " + totalPassRecords.get());
			//System.out.println("Iterator : " + totalPassRecords.get());
			
		}
	}

}

代码贴完，上面只是简单的实现了一个迭代功能。

如果对于入参有更多要求，下面几个思路可以作参考：

1.下面入参改为传对象

public TestTask(String thread) {
		super(thread);
	}

2.可以在init()方法中从文件中获取参数

3.在TestTask定义新的方法专门用来生成参数

最后附上UserDaoI接口及实现。

UserDaoI：不能再简单了，纯为了演示而用

package com.seling.test;

public interface UserDaoI {
	
	public String getUserById(String id);
	
	public void inOrModifyUser(UserInfo userInfo);
	
	public void delUser(String id);
	
	
}

UserDaoImpl：

package com.seling.test;

public class UserDaoImpl implements UserDaoI {

	@Override
	public String getUserById(String id) {
		// TODO Auto-generated method stub
		return id;
	}

	@Override
	public void inOrModifyUser(UserInfo userInfo) {
		// TODO Auto-generated method stub
		System.out.println(userInfo.getUserName());
	}

	@Override
	public void delUser(String id) {
		// TODO Auto-generated method stub
		System.out.println("del "+id);
	}

}

UserInfo：

package com.seling.test;

public class UserInfo {
	
	private String userId;
	
	private String userName;
	
	private String address;
	
	private String tele;
	
	private boolean  male;

	public String getUserId() {
		return userId;
	}

	public void setUserId(String userId) {
		this.userId = userId;
	}

	public String getUserName() {
		return userName;
	}

	public void setUserName(String userName) {
		this.userName = userName;
	}

	public String getAddress() {
		return address;
	}

	public void setAddress(String address) {
		this.address = address;
	}

	public String getTele() {
		return tele;
	}

	public void setTele(String tele) {
		this.tele = tele;
	}

	public boolean isMale() {
		return male;
	}

	public void setMale(boolean male) {
		this.male = male;
	}
		
	
}

见笑