| | |
| | | |
| | | import java.util.List; |
| | | import java.util.Objects; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * 成果表 包装类,返回视图层所需的字段 |
| | |
| | | |
| | | @Override |
| | | public GdTaskResultVO entityVO(GdTaskResultEntity gdTaskResult) { |
| | | |
| | | return Objects.requireNonNull(BeanUtil.copy(gdTaskResult, GdTaskResultVO.class)); |
| | | } |
| | | |
| | | /** |
| | | * 将DTO列表转换为Entity列表 |
| | | * |
| | | * @param gdTaskResultDTOList DTO列表 |
| | | * @return Entity列表 |
| | | */ |
| | | public List<GdTaskResultEntity> listEntity(List<GdTaskResultDTO> gdTaskResultDTOList) { |
| | | return BeanUtil.copyWithConvert(gdTaskResultDTOList, GdTaskResultEntity.class); |
| | | return gdTaskResultDTOList.stream() |
| | | .map(this::convertDtoToEntity) |
| | | .collect(Collectors.toList()); |
| | | } |
| | | |
| | | /** |
| | | * 将单个DTO转换为Entity |
| | | * |
| | | * @param dto DTO对象 |
| | | * @return Entity对象 |
| | | */ |
| | | private GdTaskResultEntity convertDtoToEntity(GdTaskResultDTO dto) { |
| | | GdTaskResultEntity entity = BeanUtil.copy(dto, GdTaskResultEntity.class); |
| | | |
| | | if (dto.getFileType() != null) { |
| | | entity.setAttachmentType(dto.getDateType()); |
| | | } |
| | | |
| | | return entity; |
| | | } |
| | | |
| | | /** |
| | | * 将文件类型字符串转换为附件类型编码 |
| | | * |
| | | * @param fileType 文件类型字符串(图片/视频/照片) |
| | | * @return 附件类型编码:0-图片,1-视频,2-其他 |
| | | */ |
| | | private Integer convertFileTypeToAttachmentType(String fileType) { |
| | | if (fileType == null) { |
| | | return 2; |
| | | } |
| | | switch (fileType.trim()) { |
| | | case "图片": |
| | | case "照片": |
| | | return 0; |
| | | case "视频": |
| | | return 1; |
| | | default: |
| | | return 2; |
| | | } |
| | | } |
| | | |
| | | } |