package com.zy.core.cache;
|
|
import com.zy.core.enums.SlaveType;
|
import com.zy.core.model.Task;
|
|
import java.util.Map;
|
import java.util.Queue;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentLinkedQueue;
|
import java.util.concurrent.LinkedBlockingQueue;
|
|
/**
|
* 消息队列
|
* Created by vincent on 2020/8/5
|
*/
|
public class MessageQueue {
|
|
// 堆垛机mq交换机
|
private static final Map<Integer, LinkedBlockingQueue<Task>> CRN_EXCHANGE = new ConcurrentHashMap<>();
|
// 双工位堆垛机mq交换机
|
private static final Map<Integer, LinkedBlockingQueue<Task>> DUAL_CRN_EXCHANGE = new ConcurrentHashMap<>();
|
// 输送线mq交换机
|
private static final Map<Integer, ConcurrentLinkedQueue<Task>> DEVP_EXCHANGE = new ConcurrentHashMap<>();
|
// 条码扫描仪mq交换机
|
private static final Map<Integer, ConcurrentLinkedQueue<Task>> BARCODE_EXCHANGE = new ConcurrentHashMap<>();
|
// Led灯 mq交换机
|
private static final Map<Integer, LinkedBlockingQueue<Task>> LED_EXCHANGE = new ConcurrentHashMap<>();
|
// 磅称mq交换机
|
private static final Map<Integer, ConcurrentLinkedQueue<Task>> SCALE_EXCHANGE = new ConcurrentHashMap<>();
|
// RGVmq交换机
|
private static final Map<Integer, LinkedBlockingQueue<Task>> RGV_EXCHANGE = new ConcurrentHashMap<>();
|
|
/**
|
* mq 交换机初始化
|
*/
|
public static void init(SlaveType type, Integer id) {
|
switch (type) {
|
case Crn:
|
CRN_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
|
break;
|
case DualCrn:
|
DUAL_CRN_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
|
break;
|
case Rgv:
|
RGV_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
|
break;
|
case Devp:
|
DEVP_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
|
break;
|
case Barcode:
|
BARCODE_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
|
break;
|
case Led:
|
LED_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
|
break;
|
case Scale:
|
SCALE_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
|
break;
|
default:
|
break;
|
}
|
}
|
|
/**
|
* 添加元素
|
* 如果发现队列已满无法添加的话，会直接返回false。
|
*/
|
public static boolean offer(SlaveType type, Integer id, Task task) {
|
Queue<Task> queue = resolveQueue(type, id);
|
if (queue == null) {
|
return false;
|
}
|
switch (type) {
|
case Crn:
|
case DualCrn:
|
case Rgv:
|
case Devp:
|
case Barcode:
|
case Led:
|
case Scale:
|
return queue.offer(task);
|
default:
|
return false;
|
}
|
}
|
|
/**
|
* 移除元素
|
* 若队列为空，返回null。
|
*/
|
public static Task poll(SlaveType type, Integer id) {
|
Queue<Task> queue = resolveQueue(type, id);
|
if (queue == null) {
|
return null;
|
}
|
switch (type) {
|
case Crn:
|
case DualCrn:
|
case Rgv:
|
case Devp:
|
case Barcode:
|
case Led:
|
case Scale:
|
return queue.poll();
|
default:
|
return null;
|
}
|
}
|
|
/**
|
* 取出元素，并不删除.
|
*/
|
public static Task peek(SlaveType type, Integer id) {
|
Queue<Task> queue = resolveQueue(type, id);
|
if (queue == null) {
|
return null;
|
}
|
switch (type) {
|
case Crn:
|
case DualCrn:
|
case Rgv:
|
case Devp:
|
case Barcode:
|
case Led:
|
case Scale:
|
return queue.peek();
|
default:
|
return null;
|
}
|
}
|
|
public static void clear(SlaveType type, Integer id){
|
Queue<Task> queue = resolveQueue(type, id);
|
if (queue == null) {
|
return;
|
}
|
switch (type) {
|
case Crn:
|
case DualCrn:
|
case Rgv:
|
case Devp:
|
case Barcode:
|
case Led:
|
case Scale:
|
queue.clear();
|
break;
|
default:
|
break;
|
}
|
}
|
|
public static boolean hasExchange(SlaveType type, Integer id) {
|
return resolveQueue(type, id) != null;
|
}
|
|
public static int size(SlaveType type, Integer id) {
|
Queue<Task> queue = resolveQueue(type, id);
|
return queue == null ? 0 : queue.size();
|
}
|
|
private static Queue<Task> resolveQueue(SlaveType type, Integer id) {
|
if (type == null || id == null) {
|
return null;
|
}
|
switch (type) {
|
case Crn:
|
return CRN_EXCHANGE.get(id);
|
case DualCrn:
|
return DUAL_CRN_EXCHANGE.get(id);
|
case Rgv:
|
return RGV_EXCHANGE.get(id);
|
case Devp:
|
return DEVP_EXCHANGE.get(id);
|
case Barcode:
|
return BARCODE_EXCHANGE.get(id);
|
case Led:
|
return LED_EXCHANGE.get(id);
|
case Scale:
|
return SCALE_EXCHANGE.get(id);
|
default:
|
return null;
|
}
|
}
|
|
}
|
