Java并发之CyclicBarria的使用（二）

public class CyclicBarriaTest {
    
    static class ProcessDir extends Thread{
        
        private CyclicBarrier cb;
        
        private CountDownLatch cdl;
        
        private String path;
        
        private int num;
        
        public ProcessDir(String path, CyclicBarrier cb, int num, CountDownLatch cdl){
            this.num = num;
            this.cb = cb;
            this.path = path;
            this.cdl = cdl;
        }
        
        @Override
        public void run() {
            try {
                cb.await();
                copyDir();
                cb.await();
                deleteDir();
            } catch (Exception e) {
                e.printStackTrace();
            }finally{
                cdl.countDown();
            } 
        }
        
        /**
         * 拷贝文件夹
         * @throws IOException
         */
        public void copyDir() throws IOException{
            if(num == 1){
                FileUtils.copyDirectory(new File("D:/eclipse"), new File(path));
            }
            
            if(num == 2){
                FileUtils.copyDirectory(new File("D:/maven"), new File(path));
            }
        }
        
        /**
         * 删除文件夹
         * @throws IOException
         */
        public void deleteDir() throws IOException{
            if(num == 1){
                FileUtils.deleteDirectory(new File("d:/test/eclipse"));
            }
            
            if(num == 2){
                FileUtils.deleteDirectory(new File("d:/test/maven"));
            }
        }
    }
    
    static class ShowInfo implements Runnable{
        
        private boolean flag;
        
        public ShowInfo(boolean flag){
            this.flag = flag;
        }
        
        @Override
        public void run() {
            if(flag){
                System.out.println("所有的线程已经 准备完毕，开始执行拷贝");
                flag = false;
            }else{
                System.out.println("数据拷贝完毕，开始执行删除");
            }
        }
    }
    
    public static void main(String[] args) throws InterruptedException {
        CyclicBarrier cb = new CyclicBarrier(2, new ShowInfo(true));
        CountDownLatch cdl = new CountDownLatch(2);
        
        for(int i = 1; i <= 2; i++){
            new ProcessDir("d:/test", cb, i, cdl).start();
        }
        
        cdl.await();
        
        File file = new File("d:/test");
        file.delete();
    }
}