diff --git a/core/src/main/java/com/dite/znpt/domain/vo/ImageSimpleReq.java b/core/src/main/java/com/dite/znpt/domain/vo/ImageSimpleReq.java index e0000d0..8f4e8ce 100644 --- a/core/src/main/java/com/dite/znpt/domain/vo/ImageSimpleReq.java +++ b/core/src/main/java/com/dite/znpt/domain/vo/ImageSimpleReq.java @@ -26,4 +26,7 @@ public class ImageSimpleReq implements Serializable { @ApiModelProperty(name = "图像路径", required = true) private String imagePath; + + @ApiModelProperty("影像类型") + private String imageType; } diff --git a/core/src/main/java/com/dite/znpt/service/impl/DefectServiceImpl.java b/core/src/main/java/com/dite/znpt/service/impl/DefectServiceImpl.java index 91dfc47..a3dc96b 100644 --- a/core/src/main/java/com/dite/znpt/service/impl/DefectServiceImpl.java +++ b/core/src/main/java/com/dite/znpt/service/impl/DefectServiceImpl.java @@ -108,9 +108,12 @@ public class DefectServiceImpl extends ServiceImpl i @Transactional(rollbackFor = Exception.class) @Override public void save(String imageId, DefectReq req) { - if(null == imageService.getById(imageId)){ + ImageEntity image = imageService.getById(imageId); + if(null == image){ throw new ServiceException(Message.IMAGE_ID_IS_NOT_EXIST); } + image.setImageType(ImageTypeEnum.DEFECT.getCode()); + imageService.save(image); DefectEntity defectEntity = Converts.INSTANCE.toDefectEntity(req); defectEntity.setImageId(imageId); defectEntity.setLabelInfo(JSONUtil.toJsonStr(req.getMarkInfo())); @@ -128,6 +131,7 @@ public class DefectServiceImpl extends ServiceImpl i @Override public List saveOutWorkDefect(List list) { List imageSimpleReqList = Converts.INSTANCE.toImageSimpleReq(list); + imageSimpleReqList.forEach(req -> req.setImageType(ImageTypeEnum.DEFECT.getCode())); List imageEntityList = imageService.batchSaveByImageSimpleReq(imageSimpleReqList); Map imageMap = imageEntityList.stream().collect(Collectors.toMap(k->k.getPartId().concat(StrUtil.COLON).concat(k.getImagePath()), Function.identity())); List defectEntityList = new ArrayList<>(); @@ -192,6 +196,8 @@ public class DefectServiceImpl extends ServiceImpl i if (Objects.isNull(image)) { throw new ServiceException(Message.IMAGE_ID_IS_NOT_EXIST); } + image.setImageType(ImageTypeEnum.DEFECT.getCode()); + imageService.save(image); FilePathEnum pathEnum = image.getImagePath().contains("temp") ? FilePathEnum.IMAGE_TEMP : FilePathEnum.IMAGE; String inputPath = pathEnum.getFileAbsolutePath(image.getImagePath()); // 写入attach同层级文件夹下