From 1b4fbdb92537036aed4d648967ef7e7ab8842aec Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期四, 13 十一月 2025 16:37:11 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/core/cache/MessageQueue.java |   79 +++++++++++++++++++++++++++++++++------
 1 files changed, 67 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/zy/core/cache/MessageQueue.java b/src/main/java/com/zy/core/cache/MessageQueue.java
index 20d588c..fe3638b 100644
--- a/src/main/java/com/zy/core/cache/MessageQueue.java
+++ b/src/main/java/com/zy/core/cache/MessageQueue.java
@@ -1,13 +1,12 @@
 package com.zy.core.cache;
 
-import com.zy.core.Slave;
-import com.zy.core.ThreadHandler;
 import com.zy.core.enums.SlaveType;
 import com.zy.core.model.Task;
 
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.concurrent.LinkedBlockingQueue;
 
 /**
  * 娑堟伅闃熷垪
@@ -16,35 +15,40 @@
 public class MessageQueue {
 
     // 鍫嗗灈鏈簃q浜ゆ崲鏈�
-    private static final Map<Integer, ConcurrentLinkedQueue<Task>> CRN_EXCHANGE = new ConcurrentHashMap<>();
+    private static final Map<Integer, LinkedBlockingQueue<Task>> CRN_EXCHANGE = new ConcurrentHashMap<>();
     // 杈撻�佺嚎mq浜ゆ崲鏈�
     private static final Map<Integer, ConcurrentLinkedQueue<Task>> DEVP_EXCHANGE = new ConcurrentHashMap<>();
     // 鏉$爜鎵弿浠猰q浜ゆ崲鏈�
     private static final Map<Integer, ConcurrentLinkedQueue<Task>> BARCODE_EXCHANGE = new ConcurrentHashMap<>();
     // Led鐏� mq浜ゆ崲鏈�
-    private static final Map<Integer, ConcurrentLinkedQueue<Task>> LED_EXCHANGE = new ConcurrentHashMap<>();
+    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, Slave slave) {
+     * mq 浜ゆ崲鏈哄垵濮嬪寲
+     */
+    public static void init(SlaveType type, Integer id) {
         switch (type) {
             case Crn:
-                CRN_EXCHANGE.put(slave.getId(), new ConcurrentLinkedQueue<>());
+                CRN_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
+                break;
+            case Rgv:
+                RGV_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
                 break;
             case Devp:
-                DEVP_EXCHANGE.put(slave.getId(), new ConcurrentLinkedQueue<>());
+                DEVP_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
                 break;
             case Barcode:
-                BARCODE_EXCHANGE.put(slave.getId(), new ConcurrentLinkedQueue<>());
+                BARCODE_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
                 break;
             case Led:
-                LED_EXCHANGE.put(slave.getId(), new ConcurrentLinkedQueue<>());
+                LED_EXCHANGE.put(id, new LinkedBlockingQueue<>(1));
                 break;
             case Scale:
-                SCALE_EXCHANGE.put(slave.getId(), new ConcurrentLinkedQueue<>());
+                SCALE_EXCHANGE.put(id, new ConcurrentLinkedQueue<>());
                 break;
             default:
                 break;
@@ -59,6 +63,8 @@
         switch (type) {
             case Crn:
                 return CRN_EXCHANGE.get(id).offer(task);
+            case Rgv:
+                return RGV_EXCHANGE.get(id).offer(task);
             case Devp:
                 return DEVP_EXCHANGE.get(id).offer(task);
             case Barcode:
@@ -80,6 +86,8 @@
         switch (type) {
             case Crn:
                 return CRN_EXCHANGE.get(id).poll();
+            case Rgv:
+                return RGV_EXCHANGE.get(id).poll();
             case Devp:
                 return DEVP_EXCHANGE.get(id).poll();
             case Barcode:
@@ -93,4 +101,51 @@
         }
     }
 
+    /**
+     * 鍙栧嚭鍏冪礌锛屽苟涓嶅垹闄�.
+     */
+    public static Task peek(SlaveType type, Integer id) {
+        switch (type) {
+            case Crn:
+                return CRN_EXCHANGE.get(id).peek();
+            case Rgv:
+                return RGV_EXCHANGE.get(id).peek();
+            case Devp:
+                return DEVP_EXCHANGE.get(id).peek();
+            case Barcode:
+                return BARCODE_EXCHANGE.get(id).peek();
+            case Led:
+                return LED_EXCHANGE.get(id).peek();
+            case Scale:
+                return SCALE_EXCHANGE.get(id).peek();
+            default:
+                return null;
+        }
+    }
+
+    public static void clear(SlaveType type, Integer id){
+        switch (type) {
+            case Crn:
+                CRN_EXCHANGE.get(id).clear();
+                break;
+            case Rgv:
+                RGV_EXCHANGE.get(id).clear();
+                break;
+            case Devp:
+                DEVP_EXCHANGE.get(id).clear();
+                break;
+            case Barcode:
+                BARCODE_EXCHANGE.get(id).clear();
+                break;
+            case Led:
+                LED_EXCHANGE.get(id).clear();
+                break;
+            case Scale:
+                SCALE_EXCHANGE.get(id).clear();
+                break;
+            default:
+                break;
+        }
+    }
+
 }

--
Gitblit v1.9.1