架构师养成记--10.master-worker模式

　　master-worker模式是一种并行计算模式，分为master进程和worker进程两个部分，master是担任总管角色，worker才是执行具体任务的地方。

　　总体流程应该是这样的：

具体一点，代码实现流程应该是这样的：

client：

import java.util.Random;

public class Main {

    public static void main(String[] args) {

        Master master = new Master(new Worker(), 20);//并发数20，也就是有20个Worker在工作

        Random r = new Random();

        for(int i = 1; i <= 100; i++){//总共有100个任务

            Task t = new Task();

            t.setId(i);

            t.setPrice(r.nextInt(1000));

            master.submit(t);//提交任务，向WorkerQueue<Task> 中加入元素

        }

        master.execute();//启动所有的worker

        long start = System.currentTimeMillis();

        while(true){

            if(master.isComplete()){//100个任务执行完成

                long end = System.currentTimeMillis() - start;

                int priceResult = master.getResult();//获取所有任务的执行结果

                System.out.println("最终结果：" + priceResult + ", 执行时间：" + end);

                break;

            }

        }

    }

}

Master：

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> workQueue = new ConcurrentLinkedQueue<Task>();

    //2 需要有一个盛放worker的集合

    private HashMap<String, Thread> workers = new HashMap<String, Thread>();

    //3 需要有一个盛放每一个worker执行任务的结果集合

    private ConcurrentHashMap<String, Object> resultMap = new ConcurrentHashMap<String, Object>();

    //4 构造方法

    public Master(Worker worker , int workerCount){

        worker.setWorkQueue(this.workQueue);

        worker.setResultMap(this.resultMap);

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

            this.workers.put(Integer.toString(i), new Thread(worker));

        }

    }

    //5 需要一个提交任务的方法

    public void submit(Task task){

        this.workQueue.add(task);

    }

    //6 需要有一个执行的方法，启动所有的worker方法去执行任务

    public void execute(){

        for(Map.Entry<String, Thread> me : workers.entrySet()){

            me.getValue().start();

        }

    }

    //7 判断是否运行结束的方法

    public boolean isComplete() {

        for(Map.Entry<String, Thread> me : workers.entrySet()){

            if(me.getValue().getState() != Thread.State.TERMINATED){

                return false;

            }

        }

        return true;

    }

    //8 计算结果方法

    public int getResult() {

        int priceResult = 0;

        for(Map.Entry<String, Object> me : resultMap.entrySet()){

            priceResult += (Integer)me.getValue();

        }

        return priceResult;

    }

Worker：

 import java.util.concurrent.ConcurrentHashMap;

 import java.util.concurrent.ConcurrentLinkedQueue;

 public class Worker implements Runnable {

     private ConcurrentLinkedQueue<Task> workQueue;

     private ConcurrentHashMap<String, Object> resultMap;

     public void setWorkQueue(ConcurrentLinkedQueue<Task> workQueue) {

         this.workQueue = workQueue;

     }

     public void setResultMap(ConcurrentHashMap<String, Object> resultMap) {

         this.resultMap = resultMap;

     }

     @Override

     public void run() {

         while(true){

             Task input = this.workQueue.poll();

             if(input == null) break;

             Object output = handle(input);

             this.resultMap.put(Integer.toString(input.getId()), output);

         }

     }

     private Object handle(Task input) {

         Object output = null;

         try {

             //处理任务的耗时。。 比如说进行操作数据库。。。

             Thread.sleep(500);

             output = input.getPrice();

         } catch (InterruptedException e) {

             e.printStackTrace();

         }

         return output;

     }

 }

Task：

public class Task {

    private int id;

    private int price ;

    public int getId() {

        return id;

    }

    public void setId(int id) {

        this.id = id;

    }

    public int getPrice() {

        return price;

    }

    public void setPrice(int price) {

        this.price = price;

    } 

}

巴特西

架构师养成记--10.master-worker模式

最新文章

热门文章