/*
|
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the dreamlu.net developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: Chill 庄骞 (smallchill@163.com)
|
*/
|
package org.sxkj.gd.workorder.wrapper;
|
|
import org.springblade.core.mp.support.BaseEntityWrapper;
|
import org.springblade.core.tool.utils.BeanUtil;
|
import org.sxkj.gd.workorder.dto.GdTaskResultDTO;
|
import org.sxkj.gd.workorder.entity.GdTaskResultEntity;
|
import org.sxkj.gd.workorder.vo.GdTaskResultVO;
|
|
import java.util.List;
|
import java.util.Objects;
|
import java.util.stream.Collectors;
|
|
/**
|
* 成果表 包装类,返回视图层所需的字段
|
*
|
* @author lw
|
* @since 2026-01-14
|
*/
|
public class GdTaskResultWrapper extends BaseEntityWrapper<GdTaskResultEntity, GdTaskResultVO> {
|
|
public static GdTaskResultWrapper build() {
|
return new GdTaskResultWrapper();
|
}
|
|
@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 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.getAttachmentType() != null) {
|
entity.setAttachmentType(dto.getAttachmentType());
|
} else if (dto.getDateType() != 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;
|
}
|
}
|
|
}
|
