代码优化

2024-10-21 20:00:51 +08:00
parent 4a20892ab3
commit 6d7000ddc2
7 changed files with 76 additions and 27 deletions
--- a/iot-access/access-api/src/main/java/com/njcn/access/utils/ChannelObjectUtil.java
+++ b/iot-access/access-api/src/main/java/com/njcn/access/utils/ChannelObjectUtil.java
@@ -1,7 +1,10 @@
 package com.njcn.access.utils;

+import com.njcn.redis.pojo.enums.AppRedisKey;
+import com.njcn.redis.utils.RedisUtil;
 import org.springframework.stereotype.Component;

+import javax.annotation.Resource;
 import java.util.ArrayList;
 import java.util.List;

@@ -11,6 +14,9 @@ import java.util.List;
@Component
 public class ChannelObjectUtil {

+    @Resource
+    private RedisUtil redisUtil;
+
    /**
     * 将list转成对应实体
     * @param object
@@ -42,4 +48,8 @@ public class ChannelObjectUtil {
        // 或者抛出异常，根据您的需求
        return null;
    }
+
+    public Object getDeviceMid(String nDid) {
+        return redisUtil.getObjectByKey(AppRedisKey.DEVICE_MID + nDid);
+    }
 }
--- a/iot-access/access-boot/src/main/java/com/njcn/access/handler/MqttMessageHandler.java
+++ b/iot-access/access-boot/src/main/java/com/njcn/access/handler/MqttMessageHandler.java
@@ -27,6 +27,7 @@ import com.njcn.access.service.ICsDeviceOnlineLogsService;
 import com.njcn.access.service.ICsEquipmentDeliveryService;
 import com.njcn.access.service.ICsLineModelService;
 import com.njcn.access.service.ICsTopicService;
+import com.njcn.access.utils.ChannelObjectUtil;
 import com.njcn.common.pojo.dto.DeviceLogDTO;
 import com.njcn.common.pojo.exception.BusinessException;
 import com.njcn.csdevice.api.*;
@@ -43,7 +44,8 @@ import com.njcn.mq.template.AppFileMessageTemplate;
 import com.njcn.mq.template.AppFileStreamMessageTemplate;
 import com.njcn.redis.pojo.enums.AppRedisKey;
 import com.njcn.redis.utils.RedisUtil;
-import com.njcn.web.utils.RequestUtil;
+import com.njcn.zlevent.api.WaveFeignClient;
+import com.njcn.zlevent.pojo.dto.WaveTimeDto;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.eclipse.paho.client.mqttv3.MqttMessage;
@@ -93,6 +95,8 @@ public class MqttMessageHandler {
    private final DevCapacityFeignClient devCapacityFeignClient;
    private final EquipmentFeignClient equipmentFeignClient;
    private final OverlimitMapper overlimitMapper;
+    private final ChannelObjectUtil channelObjectUtil;
+    private final WaveFeignClient waveFeignClient;
    @Autowired
    Validator validator;

@@ -667,15 +671,34 @@ public class MqttMessageHandler {
        switch (fileDto.getType()){
            case 4657:
                log.info("获取文件信息" + fileDto);
-                String key = AppRedisKey.PROJECT_INFO + nDid;
-                if (Objects.isNull(fileDto.getMsg().getType())) {
-                    handleDefaultCase(fileDto, nDid);
-                } else {
-                    if (Objects.equals("dir", fileDto.getMsg().getType())) {
-                        saveDirectoryInfo(fileDto.getMsg().getDirInfo(), key);
-                    } else if (Objects.equals("file", fileDto.getMsg().getType())){
-                        saveFileInfo(fileDto.getMsg().getFileInfo(), key);
-                        appFileMessageTemplate.sendMember(appFileMessage);
+                if (Objects.equals(fileDto.getCode(),AccessEnum.SUCCESS.getCode())) {
+                    String key = AppRedisKey.PROJECT_INFO + nDid;
+                    if (Objects.isNull(fileDto.getMsg().getType())) {
+                        handleDefaultCase(fileDto, nDid);
+                    } else {
+                        if (Objects.equals("dir", fileDto.getMsg().getType())) {
+                            saveDirectoryInfo(fileDto.getMsg().getDirInfo(), key);
+                        } else if (Objects.equals("file", fileDto.getMsg().getType())){
+                            saveFileInfo(fileDto.getMsg().getFileInfo(), key);
+                            appFileMessageTemplate.sendMember(appFileMessage);
+                        }
+                    }
+                } else if (Objects.equals(fileDto.getCode(),AccessEnum.NOT_FIND.getCode())) {
+                    Object object = redisUtil.getObjectByKey("fileMid:" + nDid);
+                    if (Objects.nonNull(object)) {
+                        String data = redisUtil.getObjectByKey("fileMid:" + nDid).toString();
+                        String [] arr = data.split("concat");
+                        Integer mid = Integer.parseInt(arr[0]);
+                        String fileName = arr[1];
+                        if (Objects.equals(mid,fileDto.getMid())) {
+                            List<WaveTimeDto> list = channelObjectUtil.objectToList( redisUtil.getObjectByKey("eventFile:" + nDid),WaveTimeDto.class);
+                            list.removeIf(item -> item.getFileName().equals(fileName));
+                            redisUtil.saveByKey("eventFile:" + nDid, list);
+                            if (CollectionUtil.isNotEmpty(list)) {
+                                redisUtil.delete("handleEvent:" + nDid);
+                                waveFeignClient.channelWave(nDid);
+                            }
+                        }
                    }
                }
                break;
--- a/iot-access/access-boot/src/main/java/com/njcn/access/listener/RedisKeyExpirationListener.java
+++ b/iot-access/access-boot/src/main/java/com/njcn/access/listener/RedisKeyExpirationListener.java
@@ -105,17 +105,6 @@ public class RedisKeyExpirationListener extends KeyExpirationEventMessageListene
            ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
            executeMainTask(scheduler,nDid,version);
        }
-//        //自动接入
-//        else if (expiredKey.startsWith("autoAccess")) {
-//            List<CsEquipmentDeliveryPO> list = csEquipmentDeliveryService.getAll();
-//            list.forEach(item->{
-//                String version = csTopicService.getVersion(item.getNdid());
-//                if (!Objects.isNull(version)){
-//                    csDeviceService.devAccessAskTemplate(item.getNdid(),version,1);
-//                    redisUtil.saveByKey(AppRedisKey.DEVICE_MID + item.getNdid(),1);
-//                }
-//            });
-//        }
    }

    //主任务
--- a/iot-analysis/analysis-zl-event/zl-event-api/src/main/java/com/njcn/zlevent/api/WaveFeignClient.java
+++ b/iot-analysis/analysis-zl-event/zl-event-api/src/main/java/com/njcn/zlevent/api/WaveFeignClient.java
@@ -6,6 +6,7 @@ import com.njcn.mq.message.AppEventMessage;
 import com.njcn.zlevent.api.fallback.WaveClientFallbackFactory;
 import org.springframework.cloud.openfeign.FeignClient;
 import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;

 /**
 * @author xy
@@ -15,4 +16,7 @@ public interface WaveFeignClient {

    @PostMapping("/analysis")
    HttpResult<String> analysis(AppEventMessage appEventMessage);
+
+    @PostMapping("/channelWave")
+    HttpResult<String> channelWave(@RequestParam("nDid") String nDid);
 }
--- a/iot-analysis/analysis-zl-event/zl-event-api/src/main/java/com/njcn/zlevent/api/fallback/WaveClientFallbackFactory.java
+++ b/iot-analysis/analysis-zl-event/zl-event-api/src/main/java/com/njcn/zlevent/api/fallback/WaveClientFallbackFactory.java
@@ -4,7 +4,6 @@ import com.njcn.common.pojo.enums.response.CommonResponseEnum;
 import com.njcn.common.pojo.exception.BusinessException;
 import com.njcn.common.pojo.response.HttpResult;
 import com.njcn.mq.message.AppEventMessage;
-import com.njcn.zlevent.api.EventFeignClient;
 import com.njcn.zlevent.api.WaveFeignClient;
 import feign.hystrix.FallbackFactory;
 import lombok.extern.slf4j.Slf4j;
@@ -31,6 +30,12 @@ public class WaveClientFallbackFactory implements FallbackFactory<WaveFeignClien
                log.error("{}异常，降级处理，异常为：{}","波形报文解析",cause.toString());
                throw new BusinessException(finalExceptionEnum);
            }
+
+            @Override
+            public HttpResult<String> channelWave(String nDid) {
+                log.error("{}异常，降级处理，异常为：{}","处理录波事件",cause.toString());
+                throw new BusinessException(finalExceptionEnum);
+            }
        };
    }
 }
--- a/iot-analysis/analysis-zl-event/zl-event-boot/src/main/java/com/njcn/zlevent/controller/WaveController.java
+++ b/iot-analysis/analysis-zl-event/zl-event-boot/src/main/java/com/njcn/zlevent/controller/WaveController.java
@@ -13,10 +13,7 @@ import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiOperation;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;

 /**
 * 类的介绍：
@@ -44,4 +41,14 @@ public class WaveController extends BaseController {
        return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, null, methodDescribe);
    }

+    @OperateInfo(info = LogEnum.BUSINESS_COMMON)
+    @PostMapping("/channelWave")
+    @ApiOperation("处理录波事件")
+    @ApiImplicitParam(name = "nDid", value = "nDid", required = true)
+    public HttpResult<String> channelWave(@RequestParam("nDid") String nDid){
+        String methodDescribe = getMethodDescribe("channelWave");
+        csWaveService.channelWave(nDid);
+        return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, null, methodDescribe);
+    }
+
 }
--- a/iot-analysis/analysis-zl-event/zl-event-boot/src/main/java/com/njcn/zlevent/service/impl/CsWaveAnalysisServiceImpl.java
+++ b/iot-analysis/analysis-zl-event/zl-event-boot/src/main/java/com/njcn/zlevent/service/impl/CsWaveAnalysisServiceImpl.java
@@ -54,6 +54,7 @@ public class CsWaveAnalysisServiceImpl implements ICsWaveAnalysisService {
    private final DicDataFeignClient dicDataFeignClient;
    private final ChannelObjectUtil channelObjectUtil;
    private final MqttUtil mqttUtil;
+    private static Integer mid = 1;

    @Override
    public void analysis(AppEventMessage appEventMessage) {
@@ -112,8 +113,18 @@ public class CsWaveAnalysisServiceImpl implements ICsWaveAnalysisService {
            if (Objects.isNull(obj)) {
                List<WaveTimeDto> list = channelObjectUtil.objectToList( redisUtil.getObjectByKey("eventFile:" + nDid),WaveTimeDto.class);
                if (CollectionUtil.isNotEmpty(list)) {
+                    Object object = channelObjectUtil.getDeviceMid(nDid);
+                    if (!Objects.isNull(object)) {
+                        mid = (Integer) object;
+                    }
                    WaveTimeDto dto = list.get(0);
-                    askFileInfo(nDid,1,dto.getFileName());
+                    redisUtil.saveByKeyWithExpire("fileMid:" + nDid,mid + "concat" +dto.getFileName(),60L);
+                    askFileInfo(nDid,mid,dto.getFileName());
+                    mid = mid + 1;
+                    if (mid > 10000) {
+                        mid = 1;
+                    }
+                    redisUtil.saveByKey(AppRedisKey.DEVICE_MID + nDid,mid);
                }
            } else {
                throw new BusinessException(AlgorithmResponseEnum.FILE_DOWNLOADING);