package com.ruoyi.projectManagement.service.impl.handle;
|
|
import cn.hutool.core.bean.BeanUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.ruoyi.basic.service.CustomerFollowUpFileService;
|
import com.ruoyi.common.enums.IsDeleteEnum;
|
import com.ruoyi.common.enums.PlanStageEnum;
|
import com.ruoyi.projectManagement.dto.InfoStageDto;
|
import com.ruoyi.projectManagement.dto.PlanStageDto;
|
import com.ruoyi.projectManagement.dto.SaveInfoDto;
|
import com.ruoyi.projectManagement.mapper.InfoStageMapper;
|
import com.ruoyi.projectManagement.pojo.Info;
|
import com.ruoyi.projectManagement.pojo.InfoStage;
|
import com.ruoyi.projectManagement.vo.InfoStageVo;
|
import com.ruoyi.projectManagement.vo.SaveInfoStageVo;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.context.annotation.Lazy;
|
import org.springframework.stereotype.Component;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.validation.constraints.NotNull;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author buhuazhen
|
* @date 2026/3/11
|
* @email 3038525872@qq.com
|
*/
|
@Component
|
@RequiredArgsConstructor
|
@Transactional(rollbackFor = Exception.class, readOnly = true)
|
public class InfoStageHandleService {
|
private final InfoStageMapper infoStageMapper;
|
private final InfoHandleService infoHandleService;
|
private final CustomerFollowUpFileService customerFollowUpFileService;
|
@Autowired
|
@Lazy
|
private InfoStageHandleService infoStageHandleService;
|
|
@Transactional
|
public void save(@NotNull SaveInfoStageVo saveInfoStageVo) {
|
InfoStage infoStage = BeanUtil.copyProperties(saveInfoStageVo, InfoStage.class);
|
// 附件处理
|
String attachmentIds = StrUtil.join(",", Optional.ofNullable(saveInfoStageVo.getAttachmentIds()).orElse(Collections.emptyList()));
|
infoStage.setAttachment(attachmentIds);
|
|
if (infoStage.getId() == null) {
|
infoStageMapper.insert(infoStage);
|
} else {
|
infoStageMapper.updateById(infoStage);
|
}
|
|
infoStageHandleService.syncInfoStage(infoStage.getProjectManagementInfoId());
|
}
|
|
/**
|
* 同步项目阶段状态
|
*
|
* @param infoId
|
*/
|
@Transactional
|
public void syncInfoStage(@NotNull Long infoId) {
|
SaveInfoDto infoById = infoHandleService.getInfoById(infoId);
|
List<PlanStageDto> planStage = infoById.getPlanStage();
|
|
// 获取依旧存在的阶段信息
|
List<InfoStageDto> infoStageNow = getListDtoByInfoId(infoId);
|
|
Map<Long, Integer> maxProgressMap =
|
infoStageNow.stream()
|
.filter(e -> e.getProjectManagementPlanNodeId() != null)
|
.collect(Collectors.toMap(
|
InfoStageDto::getProjectManagementPlanNodeId,
|
e -> Optional.ofNullable(e.getProgress()).orElse(0),
|
Integer::max
|
));
|
for (PlanStageDto stage : planStage) {
|
int progress = maxProgressMap.getOrDefault(stage.getId(), 0);
|
|
stage.setPlanStageEnum(calcStage(progress));
|
}
|
|
// 进行调整
|
Info info = new Info();
|
info.setId(infoId);
|
info.setPlanStage(planStage);
|
infoHandleService.updateById(info);
|
}
|
|
private PlanStageEnum calcStage(int progress) {
|
if (progress <= 0) {
|
return PlanStageEnum.TO_BEGIN;
|
} else if (progress < 100) {
|
return PlanStageEnum.ON_GOING;
|
}
|
return PlanStageEnum.ENDED;
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
public void deleteById(@NotNull Long id) {
|
InfoStage infoStage = infoStageMapper.selectById(id);
|
if (infoStage == null) {
|
return;
|
}
|
LambdaUpdateWrapper<InfoStage> updateWrapper = new LambdaUpdateWrapper<InfoStage>();
|
updateWrapper.eq(InfoStage::getId, id);
|
updateWrapper.set(InfoStage::getIsDelete, IsDeleteEnum.DELETED.getCode());
|
infoStageMapper.update(null, updateWrapper);
|
|
infoStageHandleService.syncInfoStage(infoStage.getProjectManagementInfoId());
|
}
|
|
public List<InfoStageDto> getListDtoByInfoId(@NotNull Long infoId) {
|
LambdaQueryWrapper<InfoStage> queryWrapper = new LambdaQueryWrapper<InfoStage>();
|
queryWrapper.eq(InfoStage::getProjectManagementInfoId, infoId);
|
queryWrapper.eq(InfoStage::getIsDelete, IsDeleteEnum.NOT_DELETED.getCode());
|
List<InfoStageDto> infoStageDtos = BeanUtil.copyToList(infoStageMapper.selectList(queryWrapper), InfoStageDto.class);
|
|
/*
|
* 进行排序 先按照对应的sort进行排序，后按照创建时间排序
|
* 可能会出现后部情况 比如
|
* 立项 100%
|
* 立项 50% 这种情况，当然也支持修改操作
|
*/
|
SaveInfoDto infoById = infoHandleService.getInfoById(infoId);
|
List<PlanStageDto> planStage = infoById.getPlanStage();
|
|
// 构建 id -> sort 映射
|
Map<Long, Integer> stageSortMap = planStage.stream()
|
.collect(Collectors.toMap(
|
PlanStageDto::getId,
|
PlanStageDto::getSort
|
));
|
|
// 排序
|
infoStageDtos.sort(
|
Comparator
|
// 先按阶段排序
|
.comparing((InfoStageDto dto) ->
|
stageSortMap.getOrDefault(dto.getProjectManagementPlanNodeId(), Integer.MAX_VALUE)
|
)
|
// 再按创建时间排序
|
.thenComparing(InfoStageDto::getCreateTime)
|
);
|
return infoStageDtos;
|
}
|
|
public List<InfoStageVo> getListVoByInfoId(@NotNull Long infoId) {
|
List<InfoStageDto> listByInfoId = getListDtoByInfoId(infoId);
|
List<InfoStageVo> infoStageVos = BeanUtil.copyToList(listByInfoId, InfoStageVo.class);
|
// 处理附件
|
customerFollowUpFileService.fillAttachment(infoStageVos, InfoStageVo::getAttachment, InfoStageVo::setAttachmentList);
|
return infoStageVos;
|
}
|
|
|
}
|
