/*
|
* 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 com.alibaba.fastjson.JSON;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springblade.core.mp.support.BaseEntityWrapper;
|
import org.springblade.core.tool.utils.BeanUtil;
|
import org.springblade.core.tool.utils.StringUtil;
|
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
|
*/
|
@Slf4j
|
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
|
* 处理geojson字段：将字符串数组格式转换为正确的JSON对象数组格式
|
*
|
* @param dto DTO对象
|
* @return Entity对象
|
*/
|
private GdTaskResultEntity convertDtoToEntity(GdTaskResultDTO dto) {
|
GdTaskResultEntity entity = BeanUtil.copy(dto, GdTaskResultEntity.class);
|
|
// 步骤1：设置附件类型
|
if (dto.getDateType() != null) {
|
if (dto.getDateType().equals(1) && dto.getFileType().equals("视频")) {
|
// 录像的视频
|
entity.setAttachmentType(3);
|
} else if (dto.getDateType().equals(3) && dto.getFileType().equals("视频")) {
|
// 机巢+无人机录像视频
|
entity.setAttachmentType(4);
|
} else {
|
entity.setAttachmentType(dto.getDateType());
|
}
|
}
|
|
// 步骤2：处理geojson字段，将字符串数组转换为正确的JSON格式
|
String processedGeojson = processGeojson(dto.getGeojson());
|
entity.setGeojson(processedGeojson);
|
|
return entity;
|
}
|
|
/**
|
* 处理geojson字段，将字符串数组格式转换为正确的JSON对象数组格式
|
* 解决问题：["{\"score\":0.88}"] -> [{"score":0.88}]
|
*
|
* @param geojson 原始geojson字符串
|
* @return 处理后的geojson字符串，如果处理失败则返回原值
|
*/
|
private String processGeojson(String geojson) {
|
// 步骤1：空值检查
|
if (StringUtil.isEmpty(geojson)) {
|
return geojson;
|
}
|
|
try {
|
// 步骤2：尝试解析为JSONArray
|
com.alibaba.fastjson.JSONArray jsonArray = JSON.parseArray(geojson);
|
if (jsonArray == null || jsonArray.isEmpty()) {
|
return geojson;
|
}
|
|
// 步骤3：检查第一个元素是否为字符串（说明是字符串数组格式）
|
Object firstElement = jsonArray.get(0);
|
if (firstElement instanceof String) {
|
// 步骤4：这是一个字符串数组，需要解析每个字符串元素
|
com.alibaba.fastjson.JSONArray resultArray = new com.alibaba.fastjson.JSONArray();
|
for (int i = 0; i < jsonArray.size(); i++) {
|
String elementStr = jsonArray.getString(i);
|
if (StringUtil.hasText(elementStr)) {
|
try {
|
// 步骤5：解析字符串为JSON对象
|
com.alibaba.fastjson.JSONObject jsonObj = JSON.parseObject(elementStr);
|
if (jsonObj != null) {
|
resultArray.add(jsonObj);
|
}
|
} catch (Exception e) {
|
// 单个元素解析失败，记录日志但继续处理其他元素
|
log.warn("解析geojson数组元素失败: {}", elementStr);
|
}
|
}
|
}
|
// 步骤6：返回正确的JSON数组格式
|
return resultArray.toJSONString();
|
}
|
|
// 如果不是字符串数组，直接返回原值
|
return geojson;
|
} catch (Exception e) {
|
// 解析失败，返回原值
|
log.warn("处理geojson失败，返回原值: {}", geojson, e);
|
return geojson;
|
}
|
}
|
|
/**
|
* 将文件类型字符串转换为附件类型编码
|
*
|
* @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;
|
}
|
}
|
|
}
|
