您的位置:首页 > 其它

写一个多线程做压力测试的例子

2018-05-28 19:15 267 查看
写一个多线程做压力测试的例子

public interface RequestHandler<T> {

public void handle(T result);
}


package com.arch;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;

public class ConcurrentTestUtil {

/**
* 多线程并发执行某项任务
*
* @param concurrentThreads    并发线程数,可以用来模拟并发访问用户数
* @param times                总共执行多少次
* @param task                 任务
* @param requestHandler        结果处理器
* @param executeTimeoutMillis 执行任务总超时
* @throws InterruptedException
* @throws ExecutionException
*/
public static <T> void concurrentTest(long concurrentThreads, int times, final Callable<T> task,
RequestHandler<T> requestHandler, long executeTimeoutMillis)
throws InterruptedException, ExecutionException {

ExecutorService executor = Executors.newFixedThreadPool((int) concurrentThreads);
List<Future<T>> results = new ArrayList<Future<T>>(times);

long startTimeMillis = System.currentTimeMillis();
for (int i = 0; i < times; i++) {
results.add(executor.submit(task));
}
executor.shutdown();

boolean executeCompleteWithinTimeout = executor.awaitTermination(executeTimeoutMillis,TimeUnit.MILLISECONDS);
if (!executeCompleteWithinTimeout) {
System.out.println("Execute tasks out of timeout [" + executeTimeoutMillis + "ms]");

/*
* 取消所有任务
*/
for (Future<T> r : results) {
r.cancel(true);
}
} else {
long totalCostTimeMillis = System.currentTimeMillis() - startTimeMillis;

// 线程池此时肯定已关闭,处理任务结果
for (Future<T> r : results) {
if (requestHandler != null) {
requestHandler.handle(r.get());
}
}

System.out.println("concurrent threads: " + concurrentThreads + ", times: "   + times);
System.out.println("total cost time(ms): " + totalCostTimeMillis  + "ms, avg time(ms): " + ((double) totalCostTimeMillis / times));
System.out.println("tps: " + (double) (times * 1000) / totalCostTimeMillis);
}
}

public static void main(String[] args) throws InterruptedException, ExecutionException {
ConcurrentTestUtil.concurrentTest(1000, 3000,
new Callable<String>() {
@Override
public String call() throws Exception {
try {
Thread.sleep(100);
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
}
return "ok";
}
},
new RequestHandler<String>() {
@Override
public void handle(String result) {
System.out.println("result: " + result);
}
}, 6000);
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息