From 33ce4859a7c561e51fa49c4600e14d85705edd59 Mon Sep 17 00:00:00 2001
From: baoshiwei <baoshiwei@shlanbao.cn>
Date: 星期五, 08 十一月 2024 08:34:54 +0800
Subject: [PATCH] 租户端接收服务端请求,返回设备信息

---
 jeecg-module-dry/jeecg-module-dry-biz/src/main/java/org/jeecg/modules/dry/mqtt/MqttSampleCallback.java |   77 ++++++++++++++++++++++++++++++++++++++
 1 files changed, 77 insertions(+), 0 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..ab6e2a1 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
@@ -3,22 +3,32 @@
 import cn.hutool.core.thread.ThreadUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
+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.entity.DryEqpType;
+import org.jeecg.modules.dry.entity.DryEquipment;
+import org.jeecg.modules.dry.entity.DryShop;
+import org.jeecg.modules.dry.service.IDryEqpTypeService;
+import org.jeecg.modules.dry.service.IDryEquipmentService;
 import org.jeecg.modules.dry.service.IDryRealTimeDataService;
+import org.jeecg.modules.dry.service.IDryShopService;
+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;
 
+@Slf4j
 @Component
 @Scope("prototype")
 public class MqttSampleCallback implements MqttCallback {
@@ -35,6 +45,18 @@
 
   @Autowired
   private IDryRealTimeDataService realTimeDataService;
+
+
+  @Autowired
+  private IDryEquipmentService equipmentService;
+
+  @Autowired
+  private IDryEqpTypeService eqpTypeService;
+
+  @Autowired
+  private IDryShopService dryShopService;
+
+
 
 
   @Override
@@ -174,7 +196,26 @@
         }
 
         break;
+      case MqttConstant.TENANT_UP_PREFIX_EQU:
+        try {
+          DryEquipment equipment = (DryEquipment) messageJson.get("equipment");
 
+          DryEquipment dryEquipment = equipmentService.selectByTenantIdEquipmentId(equipment.getTenantId() + "", equipment.getCode());
+
+          if (dryEquipment == null) {
+            equipmentService.save(equipment);
+          }
+
+
+
+          // 鑾峰彇璁惧鎵�灞炶溅闂�
+          DryShop shop = (DryShop) messageJson.get("shop");
+          shop.setTenantId(equipment.getTenantId());
+          dryShopService.save(shop);
+
+        } catch (Exception e) {
+          e.printStackTrace();
+        }
     }
 
   }
@@ -218,6 +259,42 @@
 
 
         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)) {
+            // 鏍规嵁璁惧缂栫爜鏌ヨ璁惧淇℃伅
+            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", clientId);
+            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