Master-Worker
- Master-Worker模式是常用的并行计算模式。它的核心思想是系统由两类进程协作工作:Master进程和Worker进程
- Master负责接收和分配任务,Worker负责处理子任务
- 当各个Worker子进程处理完成后,会将结果返回给Master,由Master做归纳和总结。 其好处是能将一个大任务分解成若干个小任务,并行执行,从而提高系统的吞吐量
- master接收来自client的任务请求,将任务分发给不同的worker任务节点去执行任务,再将最终的任务结果返回给客户端
- 模拟如下:客户端、Master和Worker
- master里面用ConcurrentLinkedQueue盛放待处理的任务和HashMap<string,Thread>盛放每个线程,以及将每一个worker的执行结果存放在ConcurrentHashMap 中
- worker需要对任务队列和线程处理进行映射,并且实现Runnable接口,设立一个集合,存放任务处理完的结果,等处理完之后,将结果集合返还到master的ConcurrentHashMap中,再由Master将结果返回到客户端
具体代码如下
package com.example.core.masterworker;
public class Task {
private int id;
private int count;
public Task(){
}
public Task(int id,int count){
this.id = id;
this.count = count;
}
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public int getCount() {
return count;
}
public void setCount(int count) {
this.count = count;
}
}
package com.example.core.masterworker;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
public class Master {
//1 承装任务的一个容器
private ConcurrentLinkedQueue<Task> taskQueue = new ConcurrentLinkedQueue<>();
//2 承装worker执行器
private HashMap<String,Thread>workers = new HashMap<>();
//3 接受worker处理成功的结果集合
private ConcurrentHashMap<String,Object>resultMap = new ConcurrentHashMap<>();
//4 构造方法里面,要对worker进行一个初始化操作
public Master(Worker worker,int workerCount) {
//4.1 每一个worker 应该有master任务队列容器对引用
worker.setTaskQueue(this.taskQueue);
//4.2 每一个worker 应该有master结果集容器对的引用
worker.setResultMap(this.resultMap);
//4.3 将所有的worker进行初始化,放入workers容器中
for(int i=0;i<workerCount;i++){
this.workers.put(Integer.toString(i),new Thread(worker));
}
}
//5 需要一个提交任务的方法
public void submit(Task task){
this.taskQueue.add(task);
}
//6 需要一个真正Master所有worker进行工作的方法
public void execute(){
for(Map.Entry<String,Thread>me:this.workers.entrySet()){
me.getValue().start();
}
}
//7 需要一个统计的方法,用于合并结果结合
public int getResult(){
int sum=0;
for(Map.Entry<String,Object>me : resultMap.entrySet()){
sum += (Integer)me.getValue();
}
return sum;
}
//8,判断是否所有的worker都完成了工作,如果全部完成就返true
public boolean isComplete(){
for(Map.Entry<String,Thread> me : this.workers.entrySet()){
if(me.getValue().getState() != Thread.State.TERMINATED){
return false;
}
}
return true;
}
}
package com.example.core.masterworker;
import java.util.Random;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
public class Worker implements Runnable{
private ConcurrentLinkedQueue<Task> taskQueue;
private ConcurrentHashMap<String,Object> resultMap;
//设置任务集合
public void setTaskQueue(ConcurrentLinkedQueue<Task>taskQueue){
this.taskQueue = taskQueue;
}
//设置结果集合
public void setResultMap(ConcurrentHashMap<String,Object>resultMap){
this.resultMap = resultMap;
}
@Override
public void run(){
while(true){
Task task = this.taskQueue.poll();
if(task == null){
break;
}
try{
Object result = handle(task);
this.resultMap.put(Integer.toString(task.getId()),result);
}catch(Exception e){
e.printStackTrace();
}
}
}
private Random r = new Random();
//实际做每一个工作
private Object handle(Task task)throws Exception{
//每一个任务的处理时间
Thread.sleep(200);
int ret = task.getCount();
return ret;
}
}
package com.example.core.masterworker;
import java.util.Random;
public class Main {
public static void main(String[] args) {
System.out.println("线程数:"+Runtime.getRuntime().availableProcessors());
Master master = new Master(new Worker(),Runtime.getRuntime().availableProcessors());
Random r = new Random();
for(int i=0;i<100;i++){
Task t = new Task(i,r.nextInt(1000));
master.submit(t);
}
master.execute();
long start = System.currentTimeMillis();
while(true){
if(master.isComplete()){
long end = System.currentTimeMillis();
int result = master.getResult();
System.out.println("最终结果为:"+result+",总耗时:"+(end-start));
break;
}
}
}
}
/*
output:
线程数:12
最终结果为:48834,总耗时:1819
*/