-
Notifications
You must be signed in to change notification settings - Fork 0
/
CyclicBarrierTest2.java
54 lines (49 loc) · 1.65 KB
/
CyclicBarrierTest2.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package edu.maskleo.module1;
import java.util.Map;
import java.util.concurrent.*;
public class CyclicBarrierTest2 implements Runnable {
/**
* 创建4个屏障,处理完之后执行当前类的run方法
*/
private CyclicBarrier c = new CyclicBarrier(4, this);
/**
* 假设只有4个sheet,所以只启动4个线程
*/
private Executor executor = Executors.newFixedThreadPool(4);
/**
* 保存每个sheet计算出的银流结果
*/
private ConcurrentHashMap<String, Integer> sheetBankWaterCount = new
ConcurrentHashMap<>();
public static void main(String[] args) {
CyclicBarrierTest2 bankWaterCount = new CyclicBarrierTest2();
bankWaterCount.count();
}
private void count() {
for (int i = 0; i < 4; i++) {
executor.execute(() -> {
// 计算当前sheet的银流数据,计算代码省略
sheetBankWaterCount
.put(Thread.currentThread().getName(), 1);
// 银流计算完成,插入一个屏障
try {
c.await();
} catch (InterruptedException |
BrokenBarrierException e) {
e.printStackTrace();
}
});
}
}
@Override
public void run() {
int result = 0;
// 汇总每个sheet计算出的结果
for (Map.Entry<String, Integer> sheet : sheetBankWaterCount.entrySet()) {
result += sheet.getValue();
}
// 将结果输出
sheetBankWaterCount.put("result", result);
System.out.println(result);
}
}