Master-Worker 模式
Task.class
public class Task {
private int id;
private String name;
private int price;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getPrice() {
return price;
}
public void setPrice(int price) {
this.price = price;
}
}
Master.class
public class Master {
// 1. 應(yīng)該有一個承載任務(wù)的集合
private ConcurrentLinkedQueue<Task> workQueue = new ConcurrentLinkedQueue<>();
// 2. 使用 HashMap 去承載所有的 worker 對象
private HashMap<String, Thread> works = new HashMap<>();
// 3. 使用一個容器承載每一個 worker 并發(fā)執(zhí)行任務(wù)的結(jié)果集
private ConcurrentHashMap<String, Object> resultMap = new ConcurrentHashMap<>();
// 4. 構(gòu)造方法
public Master(Worker worker, int workerCount) {
// 每一個 worker 對象都需要有 Master 的引用图仓，workQueue 用于任務(wù)的領(lǐng)取，resultMap 用于任務(wù)的提交
worker.setWorkQueue(this.workQueue);
worker.setResultMap(this.resultMap);
for (int i = 0; i < workerCount; i++) {
// key 表示每一個 worker 的名字，value 表示線程執(zhí)行對象
works.put("子節(jié)點(diǎn)" + Integer.toString(i), new Thread(worker));
}
}
// 5. 提交方法
public void submit(Task task) {
this.workQueue.add(task);
}
// 6. 需要有一個執(zhí)行的方法（啟動應(yīng)用程序 讓所有的 worker 工作）
public void execute() {
for (Map.Entry<String, Thread> me : works.entrySet()) {
me.getValue().start();
}
}
// 7. 判斷線程是否執(zhí)行完畢
public boolean isComplete() {
for (Map.Entry<String, Thread> me : works.entrySet()) {
if (me.getValue().getState() != Thread.State.TERMINATED) {
return false;
}
}
return true;
}
// 8. 返回結(jié)果集數(shù)據(jù)
public Long getResult() {
Long ret = 0L;
for (Map.Entry<String, Object> me : resultMap.entrySet()) {
Integer value = (Integer) me.getValue();
ret += value;
}
return ret;
}
}
Worker.class
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;
}
// 真正的去做業(yè)務(wù)處理
Object output = handle(input);
this.resultMap.put(Integer.toString(input.getId()), output);
}
}
public Object handle(Task input) {
return null;
}
}
MyWorker.class
public class MyWorker extends Worker {
public Object handle(Task input) {
Object output = null;
try {
// 表示處理 task 任務(wù)的耗時，可能是做數(shù)據(jù)的加工戈次，也可能是操作數(shù)據(jù)庫...
Thread.sleep(5000);
output = input.getPrice();
} catch (InterruptedException e) {
e.printStackTrace();
}
return output;
}
}
Main.class
public class Main {
public static void main(String[] args) {
Master master = new Master(new MyWorker(), Runtime.getRuntime().availableProcessors());
System.out.println("機(jī)器線程數(shù)量：" + Runtime.getRuntime().availableProcessors());
Random random = new Random();
for (int i = 1; i <= 100; i++) {
Task task = new Task();
task.setId(i);
task.setName("任務(wù)" + i);
task.setPrice(random.nextInt(1000));
master.submit(task);
}
master.execute();
long start = System.currentTimeMillis();
while (true) {
if (master.isComplete()) {
long end = System.currentTimeMillis() - start;
Long ret = master.getResult();
System.out.println("最終結(jié)果：" + ret + "蔑歌，耗時：" + end);
break;
}
}
}
}
