From 470aa479695a7821ef58457db9bd44dabba4acbf Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期三, 17 八月 2022 15:38:06 +0800
Subject: [PATCH] 优化队列的处理逻辑

---
 src/main/java/com/genersoft/iot/vmp/service/impl/DeviceServiceImpl.java                                                            |    1 
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/RecordInfoResponseMessageHandler.java |   13 ++-
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/NotifyRequestProcessor.java                                |    7 +
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java    |  140 ++++++++++++++++++++++++----------------------
 4 files changed, 86 insertions(+), 75 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/NotifyRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/NotifyRequestProcessor.java
index 4ce30a2..3dd8fcb 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/NotifyRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/NotifyRequestProcessor.java
@@ -116,14 +116,17 @@
 								logger.info("鎺ユ敹鍒版秷鎭細" + cmd);
 							}
 						} catch (DocumentException e) {
-							throw new RuntimeException(e);
+							logger.error("澶勭悊NOTIFY娑堟伅鏃堕敊璇�", e);
+						} finally {
+							taskQueueHandlerRun = false;
 						}
 					}
-				taskQueueHandlerRun = false;
 				});
 			}
 		} catch (SipException | InvalidArgumentException | ParseException e) {
 			e.printStackTrace();
+		} finally {
+			taskQueueHandlerRun = false;
 		}
 	}
 
diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
index ff71a92..a278c08 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
@@ -35,6 +35,9 @@
 import java.util.List;
 import java.util.concurrent.ConcurrentLinkedQueue;
 
+/**
+ * 鐩綍鏌ヨ鐨勫洖澶�
+ */
 @Component
 public class CatalogResponseMessageHandler extends SIPRequestProcessorParent implements InitializingBean, IMessageHandler {
 
@@ -85,82 +88,83 @@
         // 鍥炲200 OK
         try {
             responseAck(evt, Response.OK);
+            if (!taskQueueHandlerRun) {
+                taskQueueHandlerRun = true;
+                taskExecutor.execute(()-> {
+                    while (!taskQueue.isEmpty()) {
+                        HandlerCatchData take = taskQueue.poll();
+                        try {
+                            Element rootElement = getRootElement(take.getEvt(), take.getDevice().getCharset());
+                            Element deviceListElement = rootElement.element("DeviceList");
+                            Element sumNumElement = rootElement.element("SumNum");
+                            Element snElement = rootElement.element("SN");
+                            if (snElement == null || sumNumElement == null || deviceListElement == null) {
+                                responseAck(take.getEvt(), Response.BAD_REQUEST, "xml error");
+                                return;
+                            }
+                            int sumNum = Integer.parseInt(sumNumElement.getText());
+
+                            if (sumNum == 0) {
+                                logger.info("[鏀跺埌閫氶亾]璁惧:{}鐨�: 0涓�", take.getDevice().getDeviceId());
+                                // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                                storager.cleanChannelsForDevice(take.getDevice().getDeviceId());
+                                catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
+                            }else {
+                                Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
+                                if (deviceListIterator != null) {
+                                    List<DeviceChannel> channelList = new ArrayList<>();
+                                    // 閬嶅巻DeviceList
+                                    while (deviceListIterator.hasNext()) {
+                                        Element itemDevice = deviceListIterator.next();
+                                        Element channelDeviceElement = itemDevice.element("DeviceID");
+                                        if (channelDeviceElement == null) {
+                                            continue;
+                                        }
+                                        DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice, device, null);
+                                        deviceChannel.setDeviceId(take.getDevice().getDeviceId());
+
+                                        channelList.add(deviceChannel);
+                                    }
+                                    int sn = Integer.parseInt(snElement.getText());
+                                    catalogDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, take.getDevice(), channelList);
+                                    logger.info("[鏀跺埌閫氶亾]璁惧: {} -> {}涓紝{}/{}", take.getDevice().getDeviceId(), channelList.size(), catalogDataCatch.get(take.getDevice().getDeviceId()) == null ? 0 :catalogDataCatch.get(take.getDevice().getDeviceId()).size(), sumNum);
+                                    if (catalogDataCatch.get(take.getDevice().getDeviceId()).size() == sumNum) {
+                                        // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                                        boolean resetChannelsResult = storager.resetChannels(take.getDevice().getDeviceId(), catalogDataCatch.get(take.getDevice().getDeviceId()));
+                                        if (!resetChannelsResult) {
+                                            String errorMsg = "鎺ユ敹鎴愬姛锛屽啓鍏ュけ璐ワ紝鍏�" + sumNum + "鏉★紝宸叉帴鏀�" + catalogDataCatch.get(take.getDevice().getDeviceId()).size() + "鏉�";
+                                            catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), errorMsg);
+                                        }else {
+                                            catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
+                                        }
+                                    }
+                                }
+
+                            }
+                        } catch (DocumentException e) {
+                            e.printStackTrace();
+                        } catch (InvalidArgumentException e) {
+                            e.printStackTrace();
+                        } catch (ParseException e) {
+                            e.printStackTrace();
+                        } catch (SipException e) {
+                            e.printStackTrace();
+                        } finally {
+                            taskQueueHandlerRun = false;
+                        }
+                    }
+                });
+            }
         } catch (SipException e) {
             throw new RuntimeException(e);
         } catch (InvalidArgumentException e) {
             throw new RuntimeException(e);
         } catch (ParseException e) {
             throw new RuntimeException(e);
+        } finally {
+            taskQueueHandlerRun = false;
         }
-        if (!taskQueueHandlerRun) {
-            taskQueueHandlerRun = true;
-            taskExecutor.execute(()-> {
-                while (!taskQueue.isEmpty()) {
-                    HandlerCatchData take = taskQueue.poll();
-                    String key = DeferredResultHolder.CALLBACK_CMD_CATALOG + take.getDevice().getDeviceId();
-                    Element rootElement = null;
-                    try {
-                        rootElement = getRootElement(take.getEvt(), take.getDevice().getCharset());
-                        Element deviceListElement = rootElement.element("DeviceList");
-                        Element sumNumElement = rootElement.element("SumNum");
-                        Element snElement = rootElement.element("SN");
-                        if (snElement == null || sumNumElement == null || deviceListElement == null) {
-                            responseAck(take.getEvt(), Response.BAD_REQUEST, "xml error");
-                            return;
-                        }
-                        int sumNum = Integer.parseInt(sumNumElement.getText());
 
-                        if (sumNum == 0) {
-                            logger.info("[鏀跺埌閫氶亾]璁惧:{}鐨�: 0涓�", take.getDevice().getDeviceId());
-                            // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
-                            storager.cleanChannelsForDevice(take.getDevice().getDeviceId());
-                            catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
-                        }else {
-                            Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
-                            if (deviceListIterator != null) {
-                                List<DeviceChannel> channelList = new ArrayList<>();
-                                // 閬嶅巻DeviceList
-                                while (deviceListIterator.hasNext()) {
-                                    Element itemDevice = deviceListIterator.next();
-                                    Element channelDeviceElement = itemDevice.element("DeviceID");
-                                    if (channelDeviceElement == null) {
-                                        continue;
-                                    }
-                                    DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice, device, null);
-                                    deviceChannel.setDeviceId(take.getDevice().getDeviceId());
-
-                                    channelList.add(deviceChannel);
-                                }
-                                int sn = Integer.parseInt(snElement.getText());
-                                catalogDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, take.getDevice(), channelList);
-                                logger.info("[鏀跺埌閫氶亾]璁惧: {} -> {}涓紝{}/{}", take.getDevice().getDeviceId(), channelList.size(), catalogDataCatch.get(take.getDevice().getDeviceId()) == null ? 0 :catalogDataCatch.get(take.getDevice().getDeviceId()).size(), sumNum);
-                                if (catalogDataCatch.get(take.getDevice().getDeviceId()).size() == sumNum) {
-                                    // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
-                                    boolean resetChannelsResult = storager.resetChannels(take.getDevice().getDeviceId(), catalogDataCatch.get(take.getDevice().getDeviceId()));
-                                    if (!resetChannelsResult) {
-                                        String errorMsg = "鎺ユ敹鎴愬姛锛屽啓鍏ュけ璐ワ紝鍏�" + sumNum + "鏉★紝宸叉帴鏀�" + catalogDataCatch.get(take.getDevice().getDeviceId()).size() + "鏉�";
-                                        catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), errorMsg);
-                                    }else {
-                                        catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
-                                    }
-                                }
-                            }
-
-                        }
-                    } catch (DocumentException e) {
-                        e.printStackTrace();
-                    } catch (InvalidArgumentException e) {
-                        e.printStackTrace();
-                    } catch (ParseException e) {
-                        e.printStackTrace();
-                    } catch (SipException e) {
-                        e.printStackTrace();
-                    }
-                }
-                taskQueueHandlerRun = false;
-            });
-
-        }
     }
 
     @Override
diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/RecordInfoResponseMessageHandler.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/RecordInfoResponseMessageHandler.java
index 57e8045..7915319 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/RecordInfoResponseMessageHandler.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/RecordInfoResponseMessageHandler.java
@@ -76,8 +76,8 @@
             if (!taskQueueHandlerRun) {
                 taskQueueHandlerRun = true;
                 taskExecutor.execute(()->{
-                    try {
-                        while (!taskQueue.isEmpty()) {
+                    while (!taskQueue.isEmpty()) {
+                        try {
                             HandlerCatchData take = taskQueue.poll();
                             Element rootElementForCharset = getRootElement(take.getEvt(), take.getDevice().getCharset());
                             String sn = getText(rootElementForCharset, "SN");
@@ -141,10 +141,11 @@
                                     releaseRequest(take.getDevice().getDeviceId(), sn);
                                 }
                             }
+                        } catch (DocumentException e) {
+                            throw new RuntimeException(e);
+                        } finally {
+                            taskQueueHandlerRun = false;
                         }
-                        taskQueueHandlerRun = false;
-                    }catch (DocumentException e) {
-                        throw new RuntimeException(e);
                     }
                 });
             }
@@ -155,6 +156,8 @@
             e.printStackTrace();
         } catch (ParseException e) {
             e.printStackTrace();
+        }finally {
+            taskQueueHandlerRun = false;
         }
     }
 
diff --git a/src/main/java/com/genersoft/iot/vmp/service/impl/DeviceServiceImpl.java b/src/main/java/com/genersoft/iot/vmp/service/impl/DeviceServiceImpl.java
index 66f678a..1d54b5c 100644
--- a/src/main/java/com/genersoft/iot/vmp/service/impl/DeviceServiceImpl.java
+++ b/src/main/java/com/genersoft/iot/vmp/service/impl/DeviceServiceImpl.java
@@ -105,6 +105,7 @@
                 redisCatchStorage.updateDevice(device);
                 commander.deviceInfoQuery(device);
                 sync(device);
+                // TODO 濡傛灉璁惧涓嬬殑閫氶亾绾ц仈鍒颁簡鍏朵粬骞冲彴锛岄偅涔堥渶瑕佸彂閫佷簨浠舵垨鑰卬otify缁欎笂绾у钩鍙�
             }else {
                 deviceMapper.update(device);
                 redisCatchStorage.updateDevice(device);

--
Gitblit v1.8.0