Browse Source

保存结果

main
maochaoying 2 years ago
parent
commit
2c640ea908
  1. 2
      src/main/java/com/iflytop/nuclear/handler/MessageHandler.java
  2. 23
      src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java

2
src/main/java/com/iflytop/nuclear/handler/MessageHandler.java

@ -44,6 +44,8 @@ public class MessageHandler {
boolean updateResult = nuclearCoreConfigService.updateDetectionResult(imgUrl, detectionResult, coord, taskId); boolean updateResult = nuclearCoreConfigService.updateDetectionResult(imgUrl, detectionResult, coord, taskId);
if (updateResult) { if (updateResult) {
checkServiceImpl.changeMessageReceived(true); checkServiceImpl.changeMessageReceived(true);
}else {
// 通过websocket向前端发送错误并中断本次自动检测
} }
break; break;
default: default:

23
src/main/java/com/iflytop/nuclear/service/impl/NuclearCoreConfigServiceImpl.java

@ -1,12 +1,14 @@
package com.iflytop.nuclear.service.impl; package com.iflytop.nuclear.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; 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.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.iflytop.nuclear.mapper.NuclearCoreConfigMapper; import com.iflytop.nuclear.mapper.NuclearCoreConfigMapper;
import com.iflytop.nuclear.model.NuclearCoreConfig; import com.iflytop.nuclear.model.NuclearCoreConfig;
import com.iflytop.nuclear.service.NuclearCoreConfigService; import com.iflytop.nuclear.service.NuclearCoreConfigService;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.Date;
import java.util.List; import java.util.List;
/** /**
@ -33,6 +35,25 @@ public class NuclearCoreConfigServiceImpl extends ServiceImpl<NuclearCoreConfigM
*/ */
@Override @Override
public boolean updateDetectionResult(String imgUrl, String detectionResult, String coord, int taskId) { public boolean updateDetectionResult(String imgUrl, String detectionResult, String coord, int taskId) {
return true;
UpdateWrapper<NuclearCoreConfig> 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<NuclearCoreConfig> 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;
} }
} }
Loading…
Cancel
Save