【java】【开源代码分析】java并发编程与junit4

看完《java并发编程实战》这本书之后，我想在这里把书上的知识与一些开源代码浅显地联系起来，以巩固自己的知识。

junit4本身虽然还没有支持多线程的注解（我在网上没找到），要多线程的测试功能的话大多数都是用自己或第三方实现的runner，但junit4也提供了一个并行运行runner和suites的策略。

实现这个功能的主要类位于org.junit.experimental包中（感觉包名暗示这个功能是实验性的），它是ParallelComputer。代码如下：

public class ParallelComputer extends Computer {
private final boolean classes;

private final boolean methods;

public ParallelComputer(boolean classes, boolean methods) {
this.classes = classes;
this.methods = methods;
}

public static Computer classes() {
return new ParallelComputer(true, false);
}

public static Computer methods() {
return new ParallelComputer(false, true);
}

private static Runner parallelize(Runner runner) {
if (runner instanceof ParentRunner) {
((ParentRunner<?>) runner).setScheduler(new RunnerScheduler() {
private final ExecutorService fService = Executors.newCachedThreadPool();

public void schedule(Runnable childStatement) {
fService.submit(childStatement);
}

public void finished() {
try {
fService.shutdown();
fService.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
} catch (InterruptedException e) {
e.printStackTrace(System.err);
}
}
});
}
return runner;
}

@Override
public Runner getSuite(RunnerBuilder builder, java.lang.Class<?>[] classes)
throws InitializationError {
Runner suite = super.getSuite(builder, classes);
return this.classes ? parallelize(suite) : suite;
}

@Override
protected Runner getRunner(RunnerBuilder builder, Class<?> testClass)
throws Throwable {
Runner runner = super.getRunner(builder, testClass);
return methods ? parallelize(runner) : runner;
}
}

它和java并发编程的关系

我们先看下RunnerScheduler接口

public interface RunnerScheduler {
void schedule(Runnable childStatement);
void finished();
}

这个接口是封装了测试方法（被@Test修饰的方法）的调度执行策略，每一个测试方法的执行都会先封装到Runnable的匿名内部类中再传到这个接口的schedule方法去执行。（详细可以看ParentRunner类）

所以要并行执行测试方法当然要实现一个并行的RunnerScheduler。我们现在把注意力放到ParallelComputer的parallelize（）方法。

private static Runner parallelize(Runner runner) {
if (runner instanceof ParentRunner) {
((ParentRunner<?>) runner).setScheduler(new RunnerScheduler() {
private final ExecutorService fService = Executors.newCachedThreadPool();

public void schedule(Runnable childStatement) {
fService.submit(childStatement);
}

public void finished() {
try {
fService.shutdown();
fService.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
} catch (InterruptedException e) {
e.printStackTrace(System.err);
}
}
});
}
return runner;
}

代码中调用了runner的setScheduler方法更换了Runner的调度策略为匿名类实现的并行调度。匿名类把传给自己的任务委托ExecutorService执行。恩，非常简单。但也可以用《java并发编程实战》6，7，8章的内容来从各个角度去分析它。

Executors.newCachedThreadPool()可以联系第6章的Executor框架



finished方法的任务取消策略可以联系第7章的任务取消与关闭（下图为章节小结）



junit4另外一个与java并发相关的地方是ParentRunner类中的filter方法和sort方法还有getFilteredChildren方法，它通过ReentrantLock保护filteredChildren集合中的测试方法数据安全性和可见性。原因是第三方实现自ParentRunner的类是可能是多线程的。