Commit 2a96c4ef authored by R10's avatar R10

add

parent aa58bb63
......@@ -11,6 +11,8 @@ public interface BizAiDataAuditTaskService extends BaseService {
BizAiDataAuditTaskEntity get(Long id) throws Exception;
BizAiDataAuditTaskEntity getWithToken(Long id) throws Exception;
void delete(Long id) throws Exception;
List<BizAiDataAuditTaskEntity> findByExample(BizAiDataAuditTaskEntity example, PagingInfo pagingInfo) throws Exception;
......
......@@ -61,6 +61,19 @@ public class BizAiDataAuditTaskServiceImpl extends BaseServiceImpl
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
public void delete(java.lang.Long id) throws Exception {
Assert.notNull(id);
......
......@@ -86,7 +86,7 @@ public class AirportConsumerServiceImpl implements AirportConsumerService {
@Override
@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任务是否已经全部完成]
redisService.set("AI_DATA_AUDIT_TASK:" + message.getTaskId(), "Get");
......@@ -155,37 +155,38 @@ public class AirportConsumerServiceImpl implements AirportConsumerService {
@Override
@Consumer(topic = AirportTopic.AIRPORT_PDF_PAPER_TOPIC)
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查找任务
BizAiDataAuditTaskEntity bizAiDataAuditTaskEntity = bizAiDataAuditTaskService.get(taskId);
//根据任务id查找任务
BizAiDataAuditTaskEntity bizAiDataAuditTaskEntity = bizAiDataAuditTaskService.getWithToken(taskId);
if (bizAiDataAuditTaskEntity == null) {
throw new BusinessException("任务不存在");
}
if (bizAiDataAuditTaskEntity == null) {
throw new BusinessException("任务不存在");
}
//查找任务下的所有文件
BizAiDataAuditFileEntity bizAiDataAuditFileEntity = new BizAiDataAuditFileEntity();
bizAiDataAuditFileEntity.setTaskId(taskId);
List<BizAiDataAuditFileEntity> bizAiDataAuditFileEntities = bizAiDataAuditFileService.findByExample(bizAiDataAuditFileEntity, null);
//查找任务下的所有文件
BizAiDataAuditFileEntity bizAiDataAuditFileEntity = new BizAiDataAuditFileEntity();
bizAiDataAuditFileEntity.setTaskId(taskId);
List<BizAiDataAuditFileEntity> bizAiDataAuditFileEntities = bizAiDataAuditFileService.findByExample(bizAiDataAuditFileEntity, null);
if (CollectionUtils.isEmpty(bizAiDataAuditFileEntities)) {
throw new BusinessException("找不到该任务下的文件");
}
if (CollectionUtils.isEmpty(bizAiDataAuditFileEntities)) {
throw new BusinessException("找不到该任务下的文件");
}
Map<String, List<BizAiDataAuditFileEntity>> groupedByTemplateCode = bizAiDataAuditFileEntities.stream()
.collect(Collectors.groupingBy(BizAiDataAuditFileEntity::getOrginalTemplateCode));
Map<String, List<BizAiDataAuditFileEntity>> groupedByTemplateCode = bizAiDataAuditFileEntities.stream()
.collect(Collectors.groupingBy(BizAiDataAuditFileEntity::getOrginalTemplateCode));
//获取所有的规则
AiDataAuditRuleQueryCondition aiDataAuditRuleQueryCondition = new AiDataAuditRuleQueryCondition();
List<AiDataAuditRuleQueryItem> aiDataAuditRuleQueryItems = bizAiDataAuditRuleService.aiDataAuditRuleQueryItemList(aiDataAuditRuleQueryCondition, null);
//获取所有的规则
AiDataAuditRuleQueryCondition aiDataAuditRuleQueryCondition = new AiDataAuditRuleQueryCondition();
List<AiDataAuditRuleQueryItem> aiDataAuditRuleQueryItems = bizAiDataAuditRuleService.aiDataAuditRuleQueryItemList(aiDataAuditRuleQueryCondition, null);
CountDownLatch countDownLatch = new CountDownLatch(aiDataAuditRuleQueryItems.size());
try {
CountDownLatch countDownLatch = new CountDownLatch(aiDataAuditRuleQueryItems.size());
//
for (AiDataAuditRuleQueryItem aiDataAuditRuleQueryItem : aiDataAuditRuleQueryItems) {
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