吉安感知网项目-后端
linwei
2026-06-04 ad935c07a6cabf05fed9c615c7bd4f67f6d65293
drone-service/drone-gd/src/main/java/org/sxkj/gd/workorder/wrapper/GdTaskResultWrapper.java
@@ -24,6 +24,7 @@
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
/**
 * 成果表 包装类,返回视图层所需的字段
@@ -39,12 +40,56 @@
   @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;
      }
   }
}