Java并发测试工具类

java代码

package cn.pconline.common.utils;

import java.util.concurrent.CountDownLatch;

import java.util.concurrent.Executors;

import java.util.concurrent.ThreadFactory;

import java.util.concurrent.atomic.AtomicLong;

/**

* 并发测试工具类

* @author jie.zhao

* @date 创建时间：2017年10月17日

* @version v2.0

*/

public class TlUtil {

/**

* 测试耗时

*

* @param nThreads 线程数

* @param task 执行任务

* @param singleNum 单个线程执行个数

* @return

* @throws InterruptedException

*/

public static long timeTasks(int nThreads, int singleNum,

final Runnable task) {

final CountDownLatch startGate = new CountDownLatch(1);

final CountDownLatch endGate = new CountDownLatch(nThreads);

ThreadFactory tf = Executors.defaultThreadFactory();

final int singleExeNum = singleNum == 0 ? 1 : singleNum;

final AtomicLong sum = new AtomicLong();

final AtomicLong min = new AtomicLong(10000);

final AtomicLong max = new AtomicLong(0);

for (int i = 0; i < nThreads; i++) {

tf.newThread(new Thread() {

@Override

public void run() {

try {

startGate.await();

for (int j = 0; j < singleExeNum; j++) {

long start = System.nanoTime();

try {

task.run();

} finally {

long end = System.nanoTime();

long at = ((end - start) / 1000 / 1000);

sum.addAndGet(at);

if (min.get() > at) {

min.getAndSet(at);

}

if (max.get() < at) {

max.getAndSet(at);

}

}

}

endGate.countDown();

} catch (InterruptedException e) {

e.printStackTrace();

}

}

}).start();

}

long start = System.nanoTime();

startGate.countDown();

try {

endGate.await();

} catch (InterruptedException e) {

e.printStackTrace();

}

long end = System.nanoTime();

long at = ((end - start) / 1000 / 1000);

int allCount = singleExeNum * nThreads;

System.out.println("执行任务数：" + allCount);

System.out.println("------------------------");

System.out.println("所有线程共耗时：" + transStr(sum.get()));

System.out.println("并发执行完耗时：" + transStr(at));

System.out

.println("单任务平均耗时：" + transStr((double) sum.get() / allCount));

System.out.println("单线程最小耗时：" + transStr(min.get()));

System.out.println("单线程最大耗时：" + transStr(max.get()));

return end - start;

}

public static String transStr(long ms) {

return transStr((double) ms);

}

public static String transStr(double ms) {

if (ms < 1000) {

return ms + " ms";

}

double s = ms / 1000;

if (s < 1000) {

return s + " s";

}

double m = s / 60;

if (m < 60) {

return m + " m";

}

double h = m / 60;

if (h < 24) {

return h + " h";

}

double d = h / 24;

if (d < 30) {

return d + " d";

}

return d + " d";

}

}