diff --git a/src/main/java/com/keyware/composeanalysis/service/impl/AnalysisTaskServiceImpl.java b/src/main/java/com/keyware/composeanalysis/service/impl/AnalysisTaskServiceImpl.java index 9fde7bb..f61cd13 100644 --- a/src/main/java/com/keyware/composeanalysis/service/impl/AnalysisTaskServiceImpl.java +++ b/src/main/java/com/keyware/composeanalysis/service/impl/AnalysisTaskServiceImpl.java @@ -2,6 +2,7 @@ package com.keyware.composeanalysis.service.impl; import cn.hutool.core.date.DateUnit; import cn.hutool.core.date.DateUtil; +import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.keyware.common.constant.RedisConst; import com.keyware.common.constant.enums.AnalysisStatusEnum; @@ -69,8 +70,11 @@ public class AnalysisTaskServiceImpl extends ServiceImpl updateWrapper = new LambdaUpdateWrapper<>(); + updateWrapper.set(AnalysisTask::getId, analysisTask.getId()) + .set(AnalysisTask::getComposeFlag, AnalysisStatusEnum.ANALYSISING.getCode()); + this.update(null,updateWrapper); + MongoTemplate mongoTemplate = new MongoTemplate(mongoClient, MongoDBConst.DB_NAME_PREFIX + analysisTask.getId()); AnalysisLogUtil.insert(mongoTemplate, "【成分分析】开始:" + analysisTask.getFileName()); @@ -85,8 +89,10 @@ public class AnalysisTaskServiceImpl extends ServiceImpl updateWrapper = new LambdaUpdateWrapper<>(); + updateWrapper.set(AnalysisTask::getId, analysisTask.getId()) + .set(AnalysisTask::getAnalysisStatus, AnalysisStatusEnum.FAIL_ANALYSIS.getCode()); + analysisService.update(null,updateWrapper); } }