From 2c640ea9081029355793b12ea9954aa1987ff5b3 Mon Sep 17 00:00:00 2001 From: maochaoying <925670706@qq.com> Date: Mon, 3 Jul 2023 16:04:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=9D=E5=AD=98=E7=BB=93=E6=9E=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../iflytop/nuclear/handler/MessageHandler.java | 2 ++ .../service/impl/NuclearCoreConfigServiceImpl.java | 23 +++++++++++++++++++++- 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/iflytop/nuclear/handler/MessageHandler.java b/src/main/java/com/iflytop/nuclear/handler/MessageHandler.java index ffe9560..e5feb93 100644 --- a/src/main/java/com/iflytop/nuclear/handler/MessageHandler.java +++ b/src/main/java/com/iflytop/nuclear/handler/MessageHandler.java @@ -44,6 +44,8 @@ public class MessageHandler { boolean updateResult = nuclearCoreConfigService.updateDetectionResult(imgUrl, detectionResult, coord, taskId); if (updateResult) { checkServiceImpl.changeMessageReceived(true); + }else { + // 通过websocket向前端发送错误并中断本次自动检测 } break; default: diff --git a/src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java b/src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java index 62e73ac..3c696e4 100644 --- a/src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java +++ b/src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java @@ -1,12 +1,14 @@ package com.iflytop.nuclear.service.impl; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.iflytop.nuclear.mapper.NuclearCoreConfigMapper; import com.iflytop.nuclear.model.NuclearCoreConfig; import com.iflytop.nuclear.service.NuclearCoreConfigService; import org.springframework.stereotype.Service; +import java.util.Date; import java.util.List; /** @@ -33,6 +35,25 @@ public class NuclearCoreConfigServiceImpl extends ServiceImpl nuclearCoreConfigUpdateWrapper = new UpdateWrapper<>(); + nuclearCoreConfigUpdateWrapper.eq("task_id",taskId).eq("serial_number", coord); + NuclearCoreConfig nuclearCoreConfig = new NuclearCoreConfig(); + nuclearCoreConfig.setResultImg(imgUrl); + nuclearCoreConfig.setTaskId(taskId); + nuclearCoreConfig.setResultSerial(detectionResult); + QueryWrapper nuclearCoreConfigQueryWrapper = new QueryWrapper<>(); + nuclearCoreConfigQueryWrapper.eq("task_id",taskId).eq("serial_number", coord); + NuclearCoreConfig one = this.getOne(nuclearCoreConfigQueryWrapper); + String firstSign = one.getFirstSign(); + if (!"".equals(firstSign)){ + // 取后四位 + String lastFourStr = firstSign.substring(firstSign.length() - 4); + if (lastFourStr.equals(detectionResult)){ + // 对比源数据和检测数据 得出结果并保存 + nuclearCoreConfig.setResult(1); + } + } + boolean update = this.update(nuclearCoreConfig, nuclearCoreConfigUpdateWrapper); + return update; } }