From 8ac4d9334df66aaf3f06c00d6587d683d033bfc0 Mon Sep 17 00:00:00 2001
From: baoshiwei <baoshiwei@shlanbao.cn>
Date: 星期四, 07 十一月 2024 11:10:26 +0800
Subject: [PATCH] 优化设备信息获取主题处理逻辑

---
 jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/service/impl/DryRealTimeDataServiceImpl.java |   74 ++++++++++++++++++++++++++++++-------
 1 files changed, 60 insertions(+), 14 deletions(-)

diff --git a/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/service/impl/DryRealTimeDataServiceImpl.java b/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/service/impl/DryRealTimeDataServiceImpl.java
index c2c7ff0..e3f64dd 100644
--- a/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/service/impl/DryRealTimeDataServiceImpl.java
+++ b/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/service/impl/DryRealTimeDataServiceImpl.java
@@ -8,16 +8,23 @@
 import org.apache.mina.core.service.IoAcceptor;
 import org.apache.mina.core.service.IoHandler;
 import org.apache.mina.core.session.IoSession;
+import org.apache.shiro.SecurityUtils;
+import org.eclipse.paho.client.mqttv3.MqttClient;
+import org.eclipse.paho.client.mqttv3.MqttException;
+import org.eclipse.paho.client.mqttv3.MqttMessage;
 import org.jeecg.common.api.CommonAPI;
 import org.jeecg.common.api.vo.Result;
 import org.jeecg.common.config.TenantContext;
 import org.jeecg.common.config.mqtoken.UserTokenContext;
 import org.jeecg.common.constant.CommonConstant;
+import org.jeecg.common.constant.MqttConstant;
 import org.jeecg.common.system.util.JwtUtil;
+import org.jeecg.common.system.vo.LoginUser;
 import org.jeecg.common.util.RedisUtil;
 import org.jeecg.common.util.SpringContextUtils;
 import org.jeecg.modules.dry.common.CacheConstants;
 import org.jeecg.modules.dry.entity.*;
+import org.jeecg.modules.dry.mqtt.MqttUtil;
 import org.jeecg.modules.dry.service.*;
 import org.jeecg.modules.dry.socket.ServerHandler;
 import org.jeecg.modules.dry.socket.SocketServerConfig;
@@ -25,9 +32,11 @@
 import org.jeecg.modules.dry.vo.*;
 import org.jetbrains.annotations.NotNull;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.security.auth.login.LoginContext;
 import java.io.IOException;
 import java.io.ObjectOutputStream;
 import java.net.Socket;
@@ -63,7 +72,11 @@
 
     private String token;
 
+    @Value(value = "${jeecg.mqtt.role}")
+    private String role;
 
+    @Autowired
+    private MqttUtil mqttUtil;
 
     public String getTemporaryToken() {
         if (token == null) {
@@ -97,6 +110,10 @@
             // 鏍规嵁绉熸埛id鍜屽伐鍗曞彿鏌ヨ鏁版嵁搴撴槸鍚︽湁璁板綍锛屾湁鍒欒繑鍥烇紝娌℃湁鍒欐柊澧炰竴鏉″啀杩斿洖
             orderVo = getOrSaveDryOrderVoDB(realTimeDataVo);
         }
+        if (orderVo == null) {
+            log.error("宸ュ崟涓嶅瓨鍦紝宸ュ崟鍙凤細"+realTimeDataVo.getWorkorder()+",璁惧锛�" + realTimeDataVo.getMachineid() +",鑽潗锛�" + realTimeDataVo.getName());
+            return Result.error("宸ュ崟涓嶅瓨鍦�");
+        }
 
         // 2 鏇存柊宸ュ崟瀹炴椂鏁版嵁
         // 2.1 灏嗗伐鍗曚腑鐨勬暟鎹浛鎹负鏈�鏂版暟鎹�
@@ -126,7 +143,7 @@
         orderVo.setState_valve(realTimeDataVo.getState_valve());
         orderVo.setOrderStatus(realTimeDataVo.getWorkorder_status());
         orderVo.setEqp_status(realTimeDataVo.getEqp_status());
-        orderVo.setEqp_state(realTimeDataVo.getEqp_state());
+//        orderVo.setEqp_state(realTimeDataVo.getEqp_state());
         orderVo.setWarning(realTimeDataVo.getEqp_warning());
         orderVo.setFault(realTimeDataVo.getEqp_fault());
         orderVo.setLevel(realTimeDataVo.getLevel());
@@ -232,16 +249,25 @@
      * @param realTimeDataVo
      * @return
      */
-    @NotNull
     private DryOrderVo saveNewOrder(RealTimeDataVo realTimeDataVo) {
         DryOrderVo orderVo;
-        // 鏌ヨ鑽潗
-        DryHerbFormula herbFormula =  queryHerbByIndexTenant(realTimeDataVo);
+
         // 鏌ヨ璁惧
         DryEquipment equ = queryEquipmentByCodeTenant(realTimeDataVo);
+        if (equ == null) {
+            log.error("鏈壘鍒拌澶囷細"+realTimeDataVo.getIndex() +","+realTimeDataVo.getName() +",鏈哄彴锛�" + realTimeDataVo.getMachineid());
+            return null;
+        }
+        // 鏌ヨ鑽潗
+        DryHerbFormula herbFormula =  queryHerbByIndexTenant(realTimeDataVo);
 
+        if (herbFormula == null) {
+            log.error("鏈壘鍒拌嵂鏉愶細"+realTimeDataVo.getIndex() +","+realTimeDataVo.getName() +",鏈哄彴锛�" + realTimeDataVo.getMachineid());
+            return null;
+        }
         // 鍒涘缓鏂板伐鍗�
         orderVo = new DryOrderVo(realTimeDataVo);
+
         orderVo.setHerbId(herbFormula.getId());
         orderVo.setEquId(equ.getId());
         DryOrder dryOrder = BeanUtil.toBean(orderVo, DryOrder.class);
@@ -261,8 +287,24 @@
         queryWrapper.eq(DryEquipment::getCode, realTimeDataVo.getMachineid());
         DryEquipment one = equipmentService.getOne(queryWrapper);
         if (one == null) {
-            one = new DryEquipment(realTimeDataVo);
-            equipmentService.save(one);
+            log.error(role+"淇濆瓨瀹炴椂鏁版嵁锛屾湭鎵惧埌璁惧锛�"+realTimeDataVo.getMachineid());
+//            one = new DryEquipment(realTimeDataVo);
+//            equipmentService.save(one);
+            if (MqttConstant.ROLE_ADMIN.equals(role)) {
+                MqttMessage mqttMessage = new MqttMessage();
+                mqttMessage.setQos(0);
+                JSONObject object = new JSONObject();
+                object.put("code", realTimeDataVo.getMachineid());
+                object.put("tenantId", realTimeDataVo.getTenantid());
+                mqttMessage.setPayload(object.toJSONString().getBytes());
+                try {
+                    mqttUtil.getMqttClient().publish(MqttConstant.SERVICE_REQ_PREFIX ,mqttMessage);
+                }catch (MqttException e) {
+                    e.printStackTrace();
+                }
+
+            }
+            return null;
         }
         return one;
     }
@@ -275,12 +317,14 @@
     private DryHerbFormula queryHerbByIndexTenant(RealTimeDataVo realTimeDataVo) {
         LambdaQueryWrapper<DryHerbFormula> queryWrapper = new LambdaQueryWrapper<>();
         queryWrapper.eq(DryHerbFormula::getTenantId, realTimeDataVo.getTenantid());
-        queryWrapper.eq(DryHerbFormula::getCode, realTimeDataVo.getIndex());
+        queryWrapper.eq(DryHerbFormula::getName, realTimeDataVo.getName());
         DryHerbFormula one = dryHerbFormulaService.getOne(queryWrapper);
-//        if (one == null) {
-//            one = new DryHerb(realTimeDataVo);
-//            herbService.save(one);
-//        }
+        if (one == null) {
+            one = new DryHerbFormula(realTimeDataVo);
+            DryEquipment dryEquipment = equipmentService.selectByTenantIdEquipmentId(realTimeDataVo.getTenantid() + "", realTimeDataVo.getMachineid());
+            one.setEqpType(dryEquipment.getType());
+            dryHerbFormulaService.save(one);
+        }
         return one;
     }
 
@@ -318,7 +362,7 @@
     public Result<?> queryMachineRealTImeData(RealTimeDataVo realTimeDataVo) {
         TenantContext.setTenant(realTimeDataVo.getTenantid()+"");
 
-        // 鏌ヨ鎵�鏈夋満鍙版煡璇㈢粍瑁�
+        // 鏌ヨ鎵�鏈夋満鍙�,鏌ヨ璇彞缁勮
         LambdaQueryWrapper<DryEquipment> queryWrapper = new LambdaQueryWrapper<>();
         queryWrapper.eq(DryEquipment::getTenantId, realTimeDataVo.getTenantid());
         queryWrapper.eq(DryEquipment::getEnable, "Y");
@@ -351,9 +395,11 @@
                             dList.add(50d);
                         }
 
+                    } else {
+                        // 濡傛灉娌℃湁鐢熶骇锛屾晥鐜囩疆0
+                        dList.add(0d);
                     }
-                    // 濡傛灉娌℃湁鐢熶骇锛屾晥鐜囩疆0
-                    dList.add(0d);
+
                 });
             }
 

--
Gitblit v1.9.3