Commit 2a96c4ef authored by R10's avatar R10

add

parent aa58bb63
...@@ -11,6 +11,8 @@ public interface BizAiDataAuditTaskService extends BaseService { ...@@ -11,6 +11,8 @@ public interface BizAiDataAuditTaskService extends BaseService {
BizAiDataAuditTaskEntity get(Long id) throws Exception; BizAiDataAuditTaskEntity get(Long id) throws Exception;
BizAiDataAuditTaskEntity getWithToken(Long id) throws Exception;
void delete(Long id) throws Exception; void delete(Long id) throws Exception;
List<BizAiDataAuditTaskEntity> findByExample(BizAiDataAuditTaskEntity example, PagingInfo pagingInfo) throws Exception; List<BizAiDataAuditTaskEntity> findByExample(BizAiDataAuditTaskEntity example, PagingInfo pagingInfo) throws Exception;
......
...@@ -61,6 +61,19 @@ public class BizAiDataAuditTaskServiceImpl extends BaseServiceImpl ...@@ -61,6 +61,19 @@ public class BizAiDataAuditTaskServiceImpl extends BaseServiceImpl
return BizAiDataAuditTaskConvert.modelToEntity(model); return BizAiDataAuditTaskConvert.modelToEntity(model);
} }
@Override
public BizAiDataAuditTaskEntity getWithToken(java.lang.Long id) throws Exception {
Assert.notNull(id);
BizAiDataAuditTaskModel model = this.repository.get(id);
if (model == null) {
return null;
}
if ("Y".equals(model.getIsDeleted())) {
return null;
}
return BizAiDataAuditTaskConvert.modelToEntity(model);
}
@Override @Override
public void delete(java.lang.Long id) throws Exception { public void delete(java.lang.Long id) throws Exception {
Assert.notNull(id); Assert.notNull(id);
......
...@@ -86,7 +86,7 @@ public class AirportConsumerServiceImpl implements AirportConsumerService { ...@@ -86,7 +86,7 @@ public class AirportConsumerServiceImpl implements AirportConsumerService {
@Override @Override
@Consumer(topic = AirportTopic.AIRPORT_CREATE_TOPIC) @Consumer(topic = AirportTopic.AIRPORT_CREATE_TOPIC)
public void createTask(AirportTaskCreateMessage message) throws Exception { public void createTask(AirportTaskCreateMessage message) throws Exception {
// 配置Redis AI_DATA_AUDIT_TASK , 表示消息队列已经获取到任务 [用于定时任务(AiDataAuditScheduler)判断该任务下的 OCR任务是否已经全部完成] // 配置Redis AI_DATA_AUDIT_TASK , 表示消息队列已经获取到任务 [用于定时任务(AiDataAuditScheduler)判断该任务下的 OCR任务是否已经全部完成]
redisService.set("AI_DATA_AUDIT_TASK:" + message.getTaskId(), "Get"); redisService.set("AI_DATA_AUDIT_TASK:" + message.getTaskId(), "Get");
...@@ -155,37 +155,38 @@ public class AirportConsumerServiceImpl implements AirportConsumerService { ...@@ -155,37 +155,38 @@ public class AirportConsumerServiceImpl implements AirportConsumerService {
@Override @Override
@Consumer(topic = AirportTopic.AIRPORT_PDF_PAPER_TOPIC) @Consumer(topic = AirportTopic.AIRPORT_PDF_PAPER_TOPIC)
public void pdfPaper(AirportPDFPaperMessage message) throws Exception { public void pdfPaper(AirportPDFPaperMessage message) throws Exception {
List<AuditResultDto> results = new CopyOnWriteArrayList<>(); try {
List<AuditResultDto> results = new CopyOnWriteArrayList<>();
Long taskId = message.getTaskId(); Long taskId = message.getTaskId();
//根据任务id查找任务 //根据任务id查找任务
BizAiDataAuditTaskEntity bizAiDataAuditTaskEntity = bizAiDataAuditTaskService.get(taskId); BizAiDataAuditTaskEntity bizAiDataAuditTaskEntity = bizAiDataAuditTaskService.getWithToken(taskId);
if (bizAiDataAuditTaskEntity == null) { if (bizAiDataAuditTaskEntity == null) {
throw new BusinessException("任务不存在"); throw new BusinessException("任务不存在");
} }
//查找任务下的所有文件 //查找任务下的所有文件
BizAiDataAuditFileEntity bizAiDataAuditFileEntity = new BizAiDataAuditFileEntity(); BizAiDataAuditFileEntity bizAiDataAuditFileEntity = new BizAiDataAuditFileEntity();
bizAiDataAuditFileEntity.setTaskId(taskId); bizAiDataAuditFileEntity.setTaskId(taskId);
List<BizAiDataAuditFileEntity> bizAiDataAuditFileEntities = bizAiDataAuditFileService.findByExample(bizAiDataAuditFileEntity, null); List<BizAiDataAuditFileEntity> bizAiDataAuditFileEntities = bizAiDataAuditFileService.findByExample(bizAiDataAuditFileEntity, null);
if (CollectionUtils.isEmpty(bizAiDataAuditFileEntities)) { if (CollectionUtils.isEmpty(bizAiDataAuditFileEntities)) {
throw new BusinessException("找不到该任务下的文件"); throw new BusinessException("找不到该任务下的文件");
} }
Map<String, List<BizAiDataAuditFileEntity>> groupedByTemplateCode = bizAiDataAuditFileEntities.stream() Map<String, List<BizAiDataAuditFileEntity>> groupedByTemplateCode = bizAiDataAuditFileEntities.stream()
.collect(Collectors.groupingBy(BizAiDataAuditFileEntity::getOrginalTemplateCode)); .collect(Collectors.groupingBy(BizAiDataAuditFileEntity::getOrginalTemplateCode));
//获取所有的规则 //获取所有的规则
AiDataAuditRuleQueryCondition aiDataAuditRuleQueryCondition = new AiDataAuditRuleQueryCondition(); AiDataAuditRuleQueryCondition aiDataAuditRuleQueryCondition = new AiDataAuditRuleQueryCondition();
List<AiDataAuditRuleQueryItem> aiDataAuditRuleQueryItems = bizAiDataAuditRuleService.aiDataAuditRuleQueryItemList(aiDataAuditRuleQueryCondition, null); List<AiDataAuditRuleQueryItem> aiDataAuditRuleQueryItems = bizAiDataAuditRuleService.aiDataAuditRuleQueryItemList(aiDataAuditRuleQueryCondition, null);
CountDownLatch countDownLatch = new CountDownLatch(aiDataAuditRuleQueryItems.size()); CountDownLatch countDownLatch = new CountDownLatch(aiDataAuditRuleQueryItems.size());
try {
// //
for (AiDataAuditRuleQueryItem aiDataAuditRuleQueryItem : aiDataAuditRuleQueryItems) { for (AiDataAuditRuleQueryItem aiDataAuditRuleQueryItem : aiDataAuditRuleQueryItems) {
String ruleType = aiDataAuditRuleQueryItem.getDataAuditRuleType(); String ruleType = aiDataAuditRuleQueryItem.getDataAuditRuleType();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment