benchmark
package org.springside.modules.test.benchmark;
import java.math.BigDecimal;
import java.util.concurrent.TimeUnit;
/**
* Benchmark中任务线程的基类.
*
* @see ConcurrentBenchmark
*/
public abstract class BenchmarkTask implements Runnable {
protected int taskSequence;
protected ConcurrentBenchmark parent;
protected long previousRequests = 0L;
protected long nextPrintTime;
@Override
public void run() {
setUp();
onThreadStart();
try {
for (int i = 1; i <= parent.loopCount; i++) {
execute(i);
printProgressMessage(i);
}
} finally {
tearDown();
onThreadFinish();
}
}
abstract protected void execute(final int requestSequence);
/**
* Override for thread local connection and data setup.
*/
protected void setUp() {
}
/**
* Override for thread local connection and data cleanup.
*/
protected void tearDown() {
}
/**
* Must be invoked in children class when each thread finish the setup.
*/
protected void onThreadStart() {
parent.startLock.countDown();
// wait for all other threads ready
try {
parent.startLock.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
nextPrintTime = System.currentTimeMillis() + parent.intervalMillis;
}
/**
* Must be invoked in children class when each thread finish loop, before the tearDown().
*/
protected void onThreadFinish() {
// notify test finish
parent.finishLock.countDown();
// print finish summary message
printThreadFinishMessage();
}
/**
* 间隔固定时间打印进度信息.
*/
protected void printProgressMessage(final int currentRequests) {
long currentTime = System.currentTimeMillis();
if (currentTime > nextPrintTime) {
long lastIntervalMillis = parent.intervalMillis + (currentTime - nextPrintTime);
nextPrintTime = currentTime + parent.intervalMillis;
long lastRequests = currentRequests - previousRequests;
long totalTimeMillis = currentTime - parent.startTime.getTime();
long totalTimeSeconds = TimeUnit.MILLISECONDS.toSeconds(totalTimeMillis);
long totalTps = (currentRequests * 1000) / totalTimeMillis;
long lastTps = (lastRequests * 1000) / lastIntervalMillis;
BigDecimal lastLatency = new BigDecimal(lastIntervalMillis).divide(new BigDecimal(lastRequests), 2,
BigDecimal.ROUND_HALF_UP);
BigDecimal totalLatency = new BigDecimal(totalTimeMillis).divide(new BigDecimal(currentRequests), 2,
BigDecimal.ROUND_HALF_UP);
System.out
.printf("Thread %02d process %,d requests after %s seconds. Last tps/latency is %,d/%sms. Total tps/latency is %,d/%sms.\n",
taskSequence, currentRequests, totalTimeSeconds, lastTps, lastLatency.toString(),
totalTps, totalLatency.toString());
previousRequests = currentRequests;
}
}
/**
* 打印线程结果信息.
*/
protected void printThreadFinishMessage() {
long totalTimeMillis = System.currentTimeMillis() - parent.startTime.getTime();
long totalRequest = parent.loopCount;
long totalTps = (totalRequest * 1000) / totalTimeMillis;
BigDecimal totalLatency = new BigDecimal(totalTimeMillis).divide(new BigDecimal(totalRequest), 2,
BigDecimal.ROUND_HALF_UP);
System.out.printf("Thread %02d finish.Total tps/latency is %,d/%sms\n", taskSequence, totalTps,
totalLatency.toString());
}
}