package com.yzu.zhang.thread.concurrent;

import java.util.Map.Entry;

import java.util.Random;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.CyclicBarrier;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;

/**

 * 银行流水处理服务类

 * @date 2016年11月21日

 */

public class BankCountService implements Runnable{

    private int threadCount = 4;

    private Random random = new Random();

    /**

     * 创建4个屏障类，都处理完之后执行当前类的run方法

     */

    private CyclicBarrier c = new CyclicBarrier(threadCount, this);

    private ExecutorService executor = Executors.newFixedThreadPool(threadCount);

    private ConcurrentHashMap<String, Integer> countMap = new ConcurrentHashMap<String, Integer>();

    /**

     * 开启线程池进行计算

     */

    private void count() {

        System.out.println(">>>>>开始计算>>>>>");

        for (int i = 0; i < threadCount; i++) {

            executor.execute(new Runnable() {

                @Override

                public void run() {

                    //计算当前sheet的银行流水，模拟计算

                    int value = random.nextInt(10000);

                    try {

                        Thread.sleep(value);

                    } catch (InterruptedException e1) {

                        e1.printStackTrace();

                    }

                    String threadName = Thread.currentThread().getName();

                    countMap.put(threadName, value);

                    System.out.println("["+threadName+"]计算完成:"+value+", 等待汇总...");

                    //银行流水计算完成,插入一个屏蔽，等待其他线程的计算

                    try {

                        c.await();

                    } catch (Exception e) {

                        e.printStackTrace();

                    }

                }

            });

        }

    }

    @Override

    public void run() {

        int total = 0;

        System.out.println("开始汇总...");

        //汇总结果

        for( String key : countMap.keySet() ) {

            total += countMap.get(key);

        }

        for(Entry<String,Integer> entry : countMap.entrySet()) {

            //total += entry.getValue().intValue();

        }

        //将结果输出

        System.out.println("银行总流水==="+total);

        //关闭线程池

        if (executor != null) {

            executor.shutdown();

            System.out.println(">>>>>计算结束>>>>>");

        }

    }

    public static void main(String[] args) {

        BankCountService service = new BankCountService();

        service.count();

    }

}

>>>>>开始计算>>>>>

[pool-1-thread-1]计算完成:750, 等待汇总...

[pool-1-thread-2]计算完成:6426, 等待汇总...

[pool-1-thread-4]计算完成:6538, 等待汇总...

[pool-1-thread-3]计算完成:9430, 等待汇总...

开始汇总...

银行总流水===23144

>>>>>计算结束>>>>>