From 765f87b17ff1709d5307a6c01bbc40bcfd1c4d48 Mon Sep 17 00:00:00 2001
From: bsw215583320 <baoshiwei121@163.com>
Date: 星期一, 18 十一月 2024 16:58:41 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/herb' into herb

---
 jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java |  156 ++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 138 insertions(+), 18 deletions(-)

diff --git a/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java b/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java
index e1ccd32..1efee35 100644
--- a/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java
+++ b/jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java
@@ -2,23 +2,35 @@
 
 import cn.hutool.core.thread.ThreadUtil;
 import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
+import com.alibaba.fastjson.TypeReference;
+import lombok.extern.slf4j.Slf4j;
 import org.eclipse.paho.client.mqttv3.IMqttDeliveryToken;
 import org.eclipse.paho.client.mqttv3.MqttCallback;
 import org.eclipse.paho.client.mqttv3.MqttMessage;
+import org.jeecg.common.config.TenantContext;
 import org.jeecg.common.constant.CommonConstant;
 import org.jeecg.common.constant.MqttConstant;
 import org.jeecg.common.util.DateUtils;
 import org.jeecg.common.util.RedisUtil;
 import org.jeecg.modules.base.service.BaseCommonService;
 import org.jeecg.modules.dry.api.EmqxApi;
-import org.jeecg.modules.dry.service.IDryRealTimeDataService;
+import org.jeecg.modules.dry.entity.DryEqpType;
+import org.jeecg.modules.dry.entity.DryEquipment;
+import org.jeecg.modules.dry.entity.DryFaultRecord;
+import org.jeecg.modules.dry.entity.DryShop;
+import org.jeecg.modules.dry.service.*;
+import org.jeecg.modules.dry.vo.DryEquipmentVo;
 import org.jeecg.modules.dry.vo.RealTimeDataVo;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Scope;
 import org.springframework.stereotype.Component;
 
+import java.util.List;
+
+@Slf4j
 @Component
 @Scope("prototype")
 public class MqttSampleCallback implements MqttCallback {
@@ -37,13 +49,29 @@
   private IDryRealTimeDataService realTimeDataService;
 
 
+  @Autowired
+  private IDryEquipmentService equipmentService;
+
+  @Autowired
+  private IDryEqpTypeService eqpTypeService;
+
+  @Autowired
+  private IDryShopService dryShopService;
+
+  @Autowired
+  private IDryFaultRecordService faultRecordService;
+
+
+
+
   @Override
   public void connectionLost(Throwable throwable) {
     System.err.println("杩炴帴鏂紑锛氾細鎺夌嚎");
+    System.err.println("杩炴帴鏂紑锛氾細"+throwable.toString());
   }
 
   @Override
-  public void messageArrived(String topic, MqttMessage mqttMessage) throws Exception {
+  public void messageArrived(String topic, MqttMessage mqttMessage) {
     System.out.println("鏀跺埌娑堟伅: \n  topic锛�" + topic + "\n  Qos锛�" + mqttMessage.getQos() + "\n  payload锛�"
       + new String(mqttMessage.getPayload()));
 
@@ -68,21 +96,29 @@
             item.put("clientid", clientid);
             //鏄惁杩炴帴
             item.put("connected", true);
-            //
-            String[] info = clientid.split("-");
-            item.put("type", info[0]);
-            item.put("tenantId", info[1]);
-            item.put("code", info[2]);
-
+            //鏍规嵁clientid瑙f瀽(娉ㄦ剰閰嶇疆鏂囦欢涓璫lientid鏍煎紡  渚嬶細client-1000)
+            try {
+              String[] info = clientid.split("-");
+              item.put("type", info[0]);
+              item.put("tenantId", info[1]);
+              //item.put("code", info[2]);
+            }catch (Exception e){
+              e.printStackTrace();
+            }
             redisUtil.set(MqttConstant.MQTT_ONLINE_CLIENT + clientid, item);
             System.err.println(String.format("璁惧: %s涓婄嚎", clientid));
           }
 
         }
         if (topic.startsWith("$SYS/brokers/") && topic.endsWith("disconnected")) {
-          String clientid = messageJson.getString("clientid");
-          redisUtil.del(MqttConstant.MQTT_ONLINE_CLIENT + clientid);
-          System.err.println(String.format("璁惧: %s涓嬬嚎", clientid));
+          try {
+            String clientid = messageJson.getString("clientid");
+            redisUtil.del(MqttConstant.MQTT_ONLINE_CLIENT + clientid);
+            System.err.println(String.format("璁惧: %s涓嬬嚎", clientid));
+          } catch (Exception e) {
+            e.printStackTrace();
+          }
+
         }
         parseAdminCommand(topic, mqttMessage);
 
@@ -123,7 +159,7 @@
       messageJson.put("timestamp", DateUtils.zone2Str(messageJson.get("timestamp").toString()));
     }
     // 瀹炴椂鏁版嵁涓婁紶澶绻佷笖鏁版嵁鍐呭瓒呰繃瀛楁澶у皬涓嶈褰曟棩蹇�
-    if (!topic.equals(MqttConstant.TENANT_UP_PREFIX_REALTIME_DATA)){
+    if (!topic.equals(MqttConstant.TENANT_UP_PREFIX_REALTIME_DATA) && !topic.equals(MqttConstant.TENANT_UP_PREFIX_FAULT_DATA)){
       baseCommonService.addLog(message, CommonConstant.LOG_TYPE_MQTT, CommonConstant.OPERATE_MQTT_1);
     }
 
@@ -166,14 +202,61 @@
 
         // 鎺ユ敹璁惧瀹炴椂鏁版嵁
       case MqttConstant.TENANT_UP_PREFIX_REALTIME_DATA:
-        try {
-          RealTimeDataVo vo = JSON.parseObject(message, RealTimeDataVo.class);
-          realTimeDataService.realTimeDataHandle(vo);
-        } catch (Exception e) {
-          e.printStackTrace();
-        }
+        ThreadUtil.execute(() -> {
+          try {
+            RealTimeDataVo vo = JSON.parseObject(message, RealTimeDataVo.class);
+            realTimeDataService.realTimeDataHandle(vo);
+          } catch (Exception e) {
+            e.printStackTrace();
+          }
+        });
 
         break;
+      // 鎺ユ敹璁惧鎶ヨ鏁版嵁
+      case MqttConstant.TENANT_UP_PREFIX_FAULT_DATA:
+        ThreadUtil.execute(() -> {
+          try {
+
+            JSONObject jsonObject = JSON.parseObject(message);
+            List<DryFaultRecord> faultRecords = JSON.parseArray(jsonObject.get("data").toString(), DryFaultRecord.class);
+            System.err.println(faultRecords.toString());
+            faultRecordService.saveBatch(faultRecords);
+
+          } catch (Exception e) {
+            e.printStackTrace();
+          }
+        });
+
+        break;
+
+
+      case MqttConstant.TENANT_UP_PREFIX_EQU:
+        ThreadUtil.execute(() -> {
+          try {
+            Object equObj = messageJson.get("equipment");
+            DryEquipment equipment = JSON.parseObject(equObj.toString(), DryEquipment.class);
+            TenantContext.setTenant(equipment.getTenantId()+"");
+            DryEquipment dryEquipment = equipmentService.selectByTenantIdEquipmentId(equipment.getTenantId() + "", equipment.getCode());
+            if (dryEquipment == null) {
+              equipmentService.save(equipment);
+            }
+            Object typeObj = messageJson.get("eqpType");
+            DryEqpType eqpType = JSON.parseObject(typeObj.toString(), DryEqpType.class);
+            DryEqpType dryEqpType = eqpTypeService.getById(eqpType.getId());
+            if (dryEqpType == null) {
+              eqpTypeService.save(eqpType);
+            }
+            // 鑾峰彇璁惧鎵�灞炶溅闂�
+            Object shopObj = messageJson.get("shop");
+            DryShop shop = JSON.parseObject(shopObj.toString(), DryShop.class);
+            DryShop dryShop = dryShopService.getById(shop.getId());
+            if (dryShop == null) {
+              dryShopService.save(shop);
+            }
+          } catch (Exception e) {
+            e.printStackTrace();
+          }
+        });
 
     }
 
@@ -218,6 +301,43 @@
 
 
         break;
+      case MqttConstant.SERVICE_REQ_PREFIX:
+        log.debug("鏀跺埌璁惧璇︾粏淇℃伅鏌ヨ璇锋眰");
+        ThreadUtil.execute(() -> {
+          String tenantId = messageJson.getString("tenantId");
+          String clientId = mqttUtil.getMqttClient().getClientId();
+          String tenant = clientId.substring(clientId.lastIndexOf("_")+1);
+          if (tenantId!=null && tenantId.equals(tenant)) {
+            TenantContext.setTenant(tenantId);
+            // 鏍规嵁璁惧缂栫爜鏌ヨ璁惧淇℃伅
+            String code = messageJson.getString("code");
+            DryEquipment equipmentVo = equipmentService.selectByTenantIdEquipmentId(tenantId,code);
+            // 鏍规嵁璁惧杞﹂棿id鏌ヨ杞﹂棿淇℃伅
+            DryShop shop = dryShopService.getById(equipmentVo.getShopId());
+            // 鏍规嵁璁惧绫诲瀷ID鏌ヨ璁惧绫诲瀷淇℃伅
+            DryEqpType eqpType = eqpTypeService.getById(equipmentVo.getType());
+
+            JSONObject res = new JSONObject();
+
+            res.put("tenant", tenantId);
+            res.put("equipment", equipmentVo);
+            res.put("shop", shop);
+            res.put("eqpType", eqpType);
+            try {
+              MqttMessage sendMessage = new MqttMessage(res.toJSONString().getBytes());
+              sendMessage.setQos(0);
+              mqttUtil.getMqttClient().publish(MqttConstant.TENANT_UP_PREFIX_EQU, sendMessage);
+              // baseCommonService.addLog(res.toString(), CommonConstant.LOG_TYPE_MQTT, CommonConstant.OPERATE_MQTT_2);
+            } catch (Exception e) {
+              e.printStackTrace();
+            }
+          }
+        });
+
+
+
+        break;
+
     }
 
   }

--
Gitblit v1.9.3