package com.ard.sample.annotation.service.impl;
|
|
import com.ard.common.utils.bean.BeanUtils;
|
import com.ard.sample.annotation.mapper.AnnotationMapper;
|
import com.ard.sample.annotation.model.dto.AnnotationDTO;
|
import com.ard.sample.annotation.model.dto.AnnotationDirectoryDTO;
|
import com.ard.sample.annotation.model.entity.AnnotationDirectoryEntity;
|
import com.ard.sample.annotation.model.entity.AnnotationEntity;
|
import com.ard.sample.annotation.service.IAnnotationDirectoryService;
|
import com.ard.sample.annotation.service.IAnnotationService;
|
import com.ard.sample.annotation.util.JsonSerializeUtil;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ard.sample.annotation.mapper.AnnotationDirectoryMapper;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class AnnotationDirectoryServiceImpl extends ServiceImpl<AnnotationDirectoryMapper, AnnotationDirectoryEntity>
|
implements IAnnotationDirectoryService {
|
|
// 注入已实现的AnnotationService（而非直接用Mapper），复用其查询逻辑
|
private final IAnnotationService annotationService;
|
|
private final AnnotationMapper annotationMapper;
|
|
@Override
|
public boolean addDirectory(AnnotationDirectoryEntity entity) {
|
return save(entity);
|
}
|
|
@Override
|
public boolean editDirectory(AnnotationDirectoryEntity entity) {
|
if (entity.getId() == null) {
|
log.error("修改目录失败：ID为空");
|
return false;
|
}
|
return updateById(entity);
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class) // 事务保证级联删除
|
public boolean deleteDirectory(Integer id) {
|
if (id == null) {
|
log.error("删除目录失败：ID为空");
|
return false;
|
}
|
// 1. 删除目录下所有标记
|
annotationMapper.deleteByMap(Map.of("directory_id", id));
|
// 2. 删除目录
|
boolean result = removeById(id);
|
log.info("删除目录{}，结果：{}", id, result);
|
return result;
|
}
|
|
/**
|
* 查询全部目录（含下属标注数据）
|
* 修复点：替换自定义Mapper方法为MyBatis-Plus通用查询，避免SQL映射缺失问题
|
*/
|
@Override
|
public List<AnnotationDirectoryDTO> listAllDirectoryWithAnnotations() {
|
// 1. 查询所有目录
|
List<AnnotationDirectoryEntity> directoryEntities = list();
|
if (directoryEntities.isEmpty()) {
|
return List.of();
|
}
|
|
// 2. 批量查询所有标注，按目录ID分组（减少数据库查询次数）
|
List<AnnotationEntity> allAnnotations = annotationService.list();
|
Map<Integer, List<AnnotationEntity>> annotationMap = allAnnotations.stream()
|
.collect(Collectors.groupingBy(AnnotationEntity::getDirectoryId));
|
|
// 3. 遍历目录，组装DTO（关联下属标注）
|
return directoryEntities.stream().map(directoryEntity -> {
|
AnnotationDirectoryDTO directoryDTO = AnnotationDirectoryDTO.builder().build();
|
BeanUtils.copyProperties(directoryEntity, directoryDTO);
|
|
// 4. 获取当前目录下的标注（从分组Map中获取，避免循环查库）
|
List<AnnotationEntity> annotationEntities = annotationMap.getOrDefault(directoryEntity.getId(), List.of());
|
|
// 5. 标注实体转DTO（反序列化坐标数据）
|
List<AnnotationDTO> annotationDTOList = annotationEntities.stream().map(annotationEntity -> {
|
AnnotationDTO annotationDTO = AnnotationDTO.builder().build();
|
BeanUtils.copyProperties(annotationEntity, annotationDTO);
|
try {
|
// 反序列化坐标数据
|
annotationDTO.setCoordinateInfo(
|
JsonSerializeUtil.deserialize(annotationEntity.getCoordinateInfo())
|
);
|
} catch (Exception e) {
|
log.error("反序列化标注坐标失败，标注ID：{}", annotationEntity.getId(), e);
|
annotationDTO.setCoordinateInfo(null);
|
}
|
return annotationDTO;
|
}).collect(Collectors.toList());
|
|
// 6. 设置目录下的标注列表
|
directoryDTO.setAnnotations(annotationDTOList);
|
return directoryDTO;
|
}).collect(Collectors.toList());
|
}
|
|
// /**
|
// * 查询全部目录（含下属标注数据）
|
// */
|
// @Override
|
// public List<AnnotationDirectoryDTO> listAllDirectoryWithAnnotations() {
|
// // 1. 查询所有目录
|
// List<AnnotationDirectoryEntity> directoryEntities = list();
|
//
|
// // 2. 遍历目录，关联查询标注数据并组装DTO
|
// return directoryEntities.stream().map(directoryEntity -> {
|
// AnnotationDirectoryDTO directoryDTO = AnnotationDirectoryDTO.builder().build();
|
// BeanUtils.copyProperties(directoryEntity, directoryDTO);
|
//
|
// // 3. 查询当前目录下的所有标注
|
// List<AnnotationEntity> annotationEntities = annotationMapper.selectByDirectoryId(directoryEntity.getId());
|
//
|
// // 4. 标注实体转DTO（反序列化坐标数据）
|
// List<AnnotationDTO> annotationDTOList = annotationEntities.stream().map(annotationEntity -> {
|
// AnnotationDTO annotationDTO = AnnotationDTO.builder().build();
|
// BeanUtils.copyProperties(annotationEntity, annotationDTO);
|
// try {
|
// // 反序列化坐标数据
|
// annotationDTO.setCoordinateInfo(
|
// JsonSerializeUtil.deserialize(annotationEntity.getCoordinateInfo())
|
// );
|
// } catch (Exception e) {
|
// log.error("反序列化标注坐标失败，标注ID：{}", annotationEntity.getId(), e);
|
// annotationDTO.setCoordinateInfo(null);
|
// }
|
// return annotationDTO;
|
// }).collect(Collectors.toList());
|
//
|
// // 5. 给目录DTO设置下属标注列表（需先给AnnotationDirectoryDTO新增annotations字段）
|
// directoryDTO.setAnnotations(annotationDTOList);
|
// return directoryDTO;
|
// }).collect(Collectors.toList());
|
// }
|
}
|
