并发编程实现模型之（二）Master-Worker模式

不念不忘少年蓝@ 2022-06-06 13:51 208阅读 0赞

Master-Worker模式是常用的并行模式之一，它的核心思想是，系统有两个进程协作工作：Master进程，负责接收和分配任务；Worker进程，负责处理子任务。当Worker进程将子任务处理完成后，结果返回给Master进程，由Master进程做归纳汇总，最后得到最终的结果。

**一、什么是Master-Worker模式：**

该模式的结构图：

![SouthEast][]

结构图：

![SouthEast 1][]

Worker：用于实际处理一个任务；

Master：任务的分配和最终结果的合成；

Main：启动程序，调度开启Master。

**二、代码实现：**

下面的是一个简易的Master-Worker框架实现。

（1）Master部分：

**\[java\]** [view plain][] [copy][view plain]

1.  **package** MasterWorker;  
2.    
3.  **import** java.util.HashMap;  
4.  **import** java.util.Map;  
5.  **import** java.util.Queue;  
6.  **import** java.util.concurrent.ConcurrentHashMap;  
7.  **import** java.util.concurrent.ConcurrentLinkedQueue;  
8.    
9.  **public** **class** Master \{  
10.   
11.     //任务队列  
12.     **protected** Queue<Object> workQueue= **new** ConcurrentLinkedQueue<Object>();  
13.     //Worker进程队列  
14.     **protected** Map<String ,Thread> threadMap= **new** HashMap<String ,Thread>();  
15.     //子任务处理结果集  
16.     **protected** Map<String ,Object> resultMap= **new** ConcurrentHashMap<String, Object>();  
17.     //是否所有的子任务都结束了  
18.     **public** **boolean** isComplete()\{  
19.         **for**(Map.Entry<String , Thread> entry:threadMap.entrySet())\{  
20.             **if**(entry.getValue().getState()!=Thread.State.TERMINATED)\{  
21.                 **return** **false**;  
22.             \}  
23.                   
24.         \}  
25.         **return** **true** ;  
26.     \}  
27.       
28.     //Master的构造，需要一个Worker进程逻辑，和需要Worker进程数量  
29.     **public** Master(Worker worker,**int** countWorker)\{  
30.           
31.         worker.setWorkQueue(workQueue);  
32.         worker.setResultMap(resultMap);  
33.         **for**(**int** i=0;i<countWorker;i++)\{  
34.             threadMap.put(Integer.toString(i),  **new** Thread(worker, Integer.toString(i)));  
35.         \}  
36.           
37.     \}  
38.       
39.     //提交一个任务  
40.     **public** **void** submit(Object job)\{  
41.         workQueue.add(job);  
42.     \}  
43.       
44.       
45.     //返回子任务结果集  
46.     **public** Map<String ,Object> getResultMap()\{  
47.         **return** resultMap;  
48.     \}  
49.       
50.       
51.     //开始运行所有的Worker进程，进行处理  
52.     **public**  **void** execute()\{  
53.          **for**(Map.Entry<String , Thread> entry:threadMap.entrySet())\{  
54.              entry.getValue().start();  
55.                
56.          \}  
57.     \}  
58.       
59.       
60. \}

（2）Worker进程实现：

**\[java\]** [view plain][] [copy][view plain]

1.  **package** MasterWorker;  
2.    
3.  **import** java.util.Map;  
4.  **import** java.util.Queue;  
5.    
6.  **public** **class** Worker  **implements** Runnable\{  
7.    
8.      //任务队列，用于取得子任务  
9.      **protected** Queue<Object> workQueue;  
10.     //子任务处理结果集  
11.     **protected** Map<String ,Object> resultMap;  
12.     **public** **void** setWorkQueue(Queue<Object> workQueue)\{  
13.         **this**.workQueue= workQueue;  
14.     \}  
15.       
16.     **public** **void** setResultMap(Map<String ,Object> resultMap)\{  
17.         **this**.resultMap=resultMap;  
18.     \}  
19.     //子任务处理的逻辑，在子类中实现具体逻辑  
20.     **public** Object handle(Object input)\{  
21.         **return** input;  
22.     \}  
23.       
24.       
25.     @Override  
26.     **public** **void** run() \{  
27.           
28.         **while**(**true**)\{  
29.             //获取子任务  
30.             Object input= workQueue.poll();  
31.             **if**(input==**null**)\{  
32.                 **break**;  
33.             \}  
34.             //处理子任务  
35.             Object re = handle(input);  
36.             resultMap.put(Integer.toString(input.hashCode()), re);  
37.         \}  
38.     \}  
39.   
40. \}

（3）运用这个小框架计算1——100的立方和，PlusWorker的实现：

**\[java\]** [view plain][] [copy][view plain]

1.  **package** MasterWorker;  
2.    
3.  **public** **class** PlusWorker **extends** Worker \{  
4.    
5.      @Override  
6.      **public** Object handle(Object input) \{  
7.            
8.          Integer i =(Integer)input;  
9.          **return** i\*i\*i;  
10.     \}  
11.   
12.       
13. \}

（4）进行计算的Main函数：

**\[java\]** [view plain][] [copy][view plain]

1.  **package** MasterWorker;  
2.    
3.  **import** java.util.Map;  
4.  **import** java.util.Set;  
5.    
6.  **public** **class** Main \{  
7.    
8.        
9.      /\*\* 
10.      \* @param args 
11.      \*/  
12.     **public** **static** **void** main(String\[\] args) \{  
13.         //固定使用5个Worker，并指定Worker  
14.         Master m = **new** Master(**new** PlusWorker(), 5);  
15.         //提交100个子任务  
16.         **for**(**int** i=0;i<100;i++)\{  
17.             m.submit(i);  
18.         \}  
19.         //开始计算  
20.         m.execute();  
21.         **int** re= 0;  
22.         //保存最终结算结果  
23.         Map<String ,Object> resultMap =m.getResultMap();  
24.           
25.         //不需要等待所有Worker都执行完成，即可开始计算最终结果  
26.         **while**(resultMap.size()>0 || !m.isComplete())\{  
27.             Set<String> keys = resultMap.keySet();  
28.             String key =**null**;  
29.             **for**(String k:keys)\{  
30.                 key=k;  
31.                 **break**;  
32.             \}  
33.             Integer i =**null**;  
34.             **if**(key!=**null**)\{  
35.                 i=(Integer)resultMap.get(key);  
36.             \}  
37.             **if**(i!=**null**)\{  
38.                 //最终结果  
39.                 re+=i;  
40.             \}  
41.             **if**(key!=**null**)\{  
42.                 //移除已经被计算过的项  
43.                 resultMap.remove(key);  
44.             \}  
45.               
46.         \}  
47.           
48.   
49.     \}  
50.   
51. \}

**三、总结：**

Master-Worker模式是一种将串行任务并行化的方案，被分解的子任务在系统中可以被并行处理，同时，如果有需要，Master进程不需要等待所有子任务都完成计算，就可以根据已有的部分结果集计算最终结果集。

[SouthEast]: /images/20220606/401d53397c9e4a75bced864c98ef2d00.png
[SouthEast 1]: /images/20220606/378364e72a1b4534ac4e23c14c9993ac.png
[view plain]: http://blog.csdn.net/lmdcszh/article/details/39698189#