写一个多线程做压力测试的例子
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 {
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);
}
}
