【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的类是可能是多线程的。