多线程分配线程的实现方案：CountDownLatch类

需求：假如我们本地有4个文件需要解析，每个文件的内容为20万行。为了提高效率我们要创建4个线程进行处理。等4个线程处理完，要在文件日志表中记录处理状态。

一般的的解决方法是使用join，join用于让当前执行线程等待join线程执行结束。其实现原理是不停检查join线程是否存活，如果join线程存活则让当前线程永远wait。直到join线程中止后，线程的this.notifyAll会被调用。

但是DK1.5之后的并发包中提供的CountDownLatch也可以实现join的这个功能，并且比join的功能更多。

 public class CountDownLatchTest {

     static CountDownLatch c = new CountDownLatch(4);
 /**线程记录数，当线程开始调用时主线程阻塞，每当一个线程结束时调用countDown（）方法，线程记录数减一，当线程记录数为0时，主线程恢复调用。***/

     public static void main(String[] args) throws InterruptedException {
         new Thread(new Runnable() {
             @Override
             public void run() {
                 System.out.println(1);
                 c.countDown();//记录数减一，3
                 System.out.println(2);
                 c.countDown();//记录数减一,2
                                 System.out.println(3);
                 c.countDown();//记录数减一,1
                 System.out.println(4);
                 c.countDown();//记录数减一,0
             }
         }).start();

         c.await();
         System.out.println("主线程调用");//开始调用日志记录
     }

 }                    

CountDownLatch的构造函数接收一个int类型的参数作为计数器，如果你想等待N个点完成，这里就传入N。

当我们调用一次CountDownLatch的countDown方法时，N就会减1，CountDownLatch的await会阻塞当前线程，直到N变成零。由于countDown方法可以用在任何地方，所以这里说的N个点，可以是N个线程，也可以是1个线程里的N个执行步骤。用在多个线程时，你只需要把这个CountDownLatch的引用传递到线程里。

     /**
      * 单个文件处理入库
      * @author yanjp
      *
      */
     class Worker extends Thread{
         private DyFile dyFile;
         private CountDownLatch ct;
         private AnalyzeFileService analyzeFileService;
         public Worker(AnalyzeFileService analyzeFileService ,DyFile dyFile,CountDownLatch ct){
             this.dyFile=dyFile;
             this.ct=ct;
             this.analyzeFileService=analyzeFileService;
         }
         @Override
         public void run() {
             // TODO Auto-generated method stub
             try {
                 logger.info("begin入库处理文件："+dyFile.getFileName());
                 //合适的处理器,处理文件.
                 analyzeFileService.analyze(dyFile);
                 logger.info("end入库处理文件："+dyFile.getFileName());
             } catch (Exception e) {
                 // TODO Auto-generated catch block
                 logger.error("单个文件入库处理异常",e);
             }finally{
                 //完成一件任务
                 ct.countDown();
             }
         }

     }

如果有某个解析sheet的线程处理的比较慢，我们不可能让主线程一直等待，所以我们可以使用另外一个带指定时间的await方法，await(long time, TimeUnit unit): 这个方法等待特定时间后，就会不再阻塞当前线程。join也有类似的方法。

注意：计数器必须大于等于0，只是等于0时候，计数器就是零，调用await方法时不会阻塞当前线程。CountDownLatch不可能重新初始化或者修改CountDownLatch对象的内部计数器的值。一个线程调用countDown方法 happen-before 另外一个线程调用await方法。

CountDownLatch的一个非常典型的应用场景是：有一个任务想要往下执行，但必须要等到其他的任务执行完毕后才可以继续往下执行。假如我们这个想要继续往下执行的任务调用一个CountDownLatch对象的await()方法，其他的任务执行完自己的任务后调用同一个CountDownLatch对象上的countDown()方法，这个调用await()方法的任务将一直阻塞等待，直到这个CountDownLatch对象的计数值减到0为止。

参考：

http://ifeve.com/talk-concurrency-countdownlatch/