79 lines
2.3 KiB
Java
79 lines
2.3 KiB
Java
// concurrency/Tester.java
|
|
// (c)2016 MindView LLC: see Copyright.txt
|
|
// We make no guarantees that this code is fit for any purpose.
|
|
// Visit http://mindviewinc.com/Books/OnJava/ for more book information.
|
|
// Framework to test performance of concurrency collections.
|
|
import java.util.concurrent.*;
|
|
import onjava.*;
|
|
|
|
public abstract class Tester<C> {
|
|
static int testReps = 10;
|
|
static int testCycles = 1000;
|
|
static int collectionSize = 1000;
|
|
abstract C collectionInitializer();
|
|
abstract void startReadersAndWriters();
|
|
C testCollection;
|
|
String testId;
|
|
int nReaders;
|
|
int nWriters;
|
|
volatile long readResult = 0;
|
|
volatile long readTime = 0;
|
|
volatile long writeTime = 0;
|
|
CountDownLatch endLatch;
|
|
static ExecutorService exec =
|
|
Executors.newCachedThreadPool();
|
|
Integer[] writeData;
|
|
Tester(String testId, int nReaders, int nWriters) {
|
|
this.testId = testId + " " +
|
|
nReaders + "r " + nWriters + "w";
|
|
this.nReaders = nReaders;
|
|
this.nWriters = nWriters;
|
|
writeData = Generated.array(Integer.class,
|
|
new RandomSupplier.Integer(), collectionSize);
|
|
for(int i = 0; i < testReps; i++) {
|
|
runTest();
|
|
readTime = 0;
|
|
writeTime = 0;
|
|
}
|
|
}
|
|
void runTest() {
|
|
endLatch = new CountDownLatch(nReaders + nWriters);
|
|
testCollection = collectionInitializer();
|
|
startReadersAndWriters();
|
|
try {
|
|
endLatch.await();
|
|
} catch(InterruptedException ex) {
|
|
System.out.println("endLatch interrupted");
|
|
}
|
|
System.out.printf("%-27s %14d %14d\n",
|
|
testId, readTime, writeTime);
|
|
if(readTime != 0 && writeTime != 0)
|
|
System.out.printf("%-27s %14d\n",
|
|
"readTime + writeTime =", readTime + writeTime);
|
|
}
|
|
abstract class TestTask implements Runnable {
|
|
abstract void test();
|
|
abstract void putResults();
|
|
long duration;
|
|
public void run() {
|
|
long startTime = System.nanoTime();
|
|
test();
|
|
duration = System.nanoTime() - startTime;
|
|
synchronized(Tester.this) {
|
|
putResults();
|
|
}
|
|
endLatch.countDown();
|
|
}
|
|
}
|
|
public static void initMain(String[] args) {
|
|
if(args.length > 0)
|
|
testReps = new Integer(args[0]);
|
|
if(args.length > 1)
|
|
testCycles = new Integer(args[1]);
|
|
if(args.length > 2)
|
|
collectionSize = new Integer(args[2]);
|
|
System.out.printf("%-27s %14s %14s\n",
|
|
"Type", "Read time", "Write time");
|
|
}
|
|
}
|