java面試題-多線程數(shù)組求和(CyclicBarrier實(shí)現(xiàn))

CyclicBarrier的作用是攔截多個(gè)線程,當(dāng)所有線程到達(dá)指定位置時(shí),在開始往下面執(zhí)行,否則線程等待。
兩種構(gòu)造函數(shù):

public CyclicBarrier(int parties)
public CyclicBarrier(int parties, Runnable barrierAction)

其中parties是線程個(gè)數(shù),barrierAction是所有線程到達(dá)后執(zhí)行的任務(wù)。
多線程數(shù)組求和,首先對(duì)數(shù)組進(jìn)行分段,每段一個(gè)線程進(jìn)行求和,當(dāng)所有線程都完成計(jì)算后,在barrierAction中進(jìn)行整體求和。

public class CyclicBarrierTest implements Runnable {

    CyclicBarrierTest(int count) {
        this.count = count;
        cyclicBarrier = new CyclicBarrier(count, this);
        executorService = Executors.newFixedThreadPool(count);
    }

    private Map<String, Integer> sumMap = new HashMap<>();
    private int count = 4;
    private CyclicBarrier cyclicBarrier;
    private ExecutorService executorService;

    public void sum(int[] arr) {
        int size = arr.length;
        int batch = size / count + 1;
        for (int i = 0; i < count; i++) {
            int end;
            int start;
            start = i * batch;
            end = Math.min(start + batch, size);

            System.out.println(start + " " + end);
            executorService.execute(new Calculator(arr, start, end));
        }
        executorService.shutdown();
    }

    public static void main(String[] args) {

        int[] arr = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11};

        CyclicBarrierTest cyclicBarrierTest = new CyclicBarrierTest(5);
        cyclicBarrierTest.sum(arr);

    }

    class Calculator implements Runnable {

        private int[] arr;
        private int start;
        private int end;

        Calculator(int[] arr, int start, int end) {
            this.arr = arr;
            this.start = start;
            this.end = end;
        }

        @Override
        public void run() {
            int sum = 0;
            for (int i = start; i < end; i++) {
                sum += arr[i];
            }
            sumMap.put(Thread.currentThread().getName(), sum);
            try {
              //  線程等待,當(dāng)所有線程都到此處時(shí)在往下面執(zhí)行
                cyclicBarrier.await();
            } catch (InterruptedException | BrokenBarrierException e) {
                e.printStackTrace();
            }
        }
    }

    @Override
    public void run() {
        long sum = sumMap.values().stream().reduce(Integer::sum).get();
        System.out.println(sum);
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容