CountDownLatch使用小Demo

2019独角兽企业重金招聘Python工程师标准>>>

假设我们又一个文件中有如下内容，对他们做求和操作

大概思路就是读取文件，有多少行开启多少个线程来计算，然后再汇总，在汇总之前让汇总的主线程处于阻塞状态，等待所有子线程计算完成再进行汇总

package cn.itchao.countdownlatch;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;

public class CountDownLatchDemo {
	
	private Integer[] nums;//总行数
	
	public CountDownLatchDemo(Integer line){
		nums = new Integer[line];
	}
	
	/**
	 * 
	 * @param lineText 行内容
	 * @param index 行数
	 */
	public void calc(String lineText,Integer index){
		String[] numsArray = lineText.split(",");
		Integer total = 0;
		for(String num : numsArray){
			total += Integer.parseInt(num);
		}
		nums[index] = total;
		System.out.println(Thread.currentThread().getName()+"  
                  线程开始计算...."+lineText+",计算结果为："+total);
	}
	
	public void sum(){
		System.out.println("汇总计算结果开始执行.....");
		int total = 0;
		for (int i = 0; i < nums.length; i++) {
			total +=nums[i];
		}
		System.out.println("最终的计算结果:"+total);
	}
	
	public static void main(String[] args) {
		List<String> contents = readFile();
		Integer lineNum = contents.size();
		CountDownLatch count = new CountDownLatch(lineNum);//定义一个计数器
		CountDownLatchDemo demo = new CountDownLatchDemo(lineNum);
		
		for (int i = 0; i < lineNum; i++) {
			final int j = i;
			new Thread(new Runnable() {
				@Override
				public void run() {
					demo.calc(contents.get(j), j);
					count.countDown();
				}
			}).start();
		}
		
		//汇总
		try {
			count.await();
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		demo.sum(); 
		
	}


	private static List<String> readFile() {
		List<String> contents = new ArrayList<>();
		String lineText = null;
		BufferedReader bufferedReader = null;
		try {
			String filePath = "/itchao-thread/src/main/resources/nums.txt";
			bufferedReader = new BufferedReader(new FileReader(filePath));
			while((lineText=bufferedReader.readLine())!=null){
				 contents.add(lineText);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}finally {
			if(bufferedReader!=null){
				try {
					bufferedReader.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		return contents;
	}
	
	

}

打印结果：

同样的道理，在真实企业级开发中对表格的每个sheel进行分别读取，最终进行汇总计算，即可实现高效的运算。

转载于:https://my.oschina.net/shxjinchao/blog/1936658