package com.gk.hotwork.Service.ServiceImpl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.gk.hotwork.Domain.Exception.BusinessException;
|
import com.gk.hotwork.Domain.SubTaskInfo;
|
import com.gk.hotwork.Domain.TaskEnclosure;
|
import com.gk.hotwork.Domain.UserInfo;
|
import com.gk.hotwork.Mapper.SubTaskInfoMapper;
|
import com.gk.hotwork.Service.SubTaskService;
|
import com.gk.hotwork.Service.TaskEnclosureService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
/**
|
* @author : jingjy
|
* @date : 2021/10/11 16:43
|
*/
|
@Service("SubTaskService")
|
public class SubTaskServiceImpl extends ServiceImpl<SubTaskInfoMapper, SubTaskInfo> implements SubTaskService {
|
@Autowired
|
private TaskEnclosureService taskEnclosureService;
|
@Override
|
public List<SubTaskInfo> getSubTasksByCode(String code) {
|
LambdaQueryWrapper<SubTaskInfo> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(SubTaskInfo::getTaskcode,code);
|
List<SubTaskInfo> subTaskInfos = baseMapper.selectList(wrapper);
|
for (SubTaskInfo subTaskInfo : subTaskInfos){
|
if (subTaskInfo.getType().equals("动土作业")){
|
List<TaskEnclosure> enclosures = taskEnclosureService.getFileByCodeAndType(code,"动土方案文件");
|
enclosures.forEach(item->{
|
item.setUrl(item.getFileurl());
|
});
|
subTaskInfo.setSoilfile(enclosures);
|
}
|
if (subTaskInfo.getType().equals("断路作业")){
|
List<TaskEnclosure> enclosures = taskEnclosureService.getFileByCodeAndType(code,"断路地段示意图");
|
enclosures.forEach(item->{
|
item.setUrl(item.getFileurl());
|
});
|
subTaskInfo.setBreakfile(enclosures);
|
}
|
if (subTaskInfo.getType().equals("盲板抽堵作业")){
|
List<TaskEnclosure> enclosures = taskEnclosureService.getFileByCodeAndType(code,"盲板位置图");
|
enclosures.forEach(item->{
|
item.setUrl(item.getFileurl());
|
});
|
subTaskInfo.setBlindboardlocation(enclosures);
|
}
|
}
|
return subTaskInfos;
|
}
|
|
@Override
|
public void removeByCode(String taskCode) {
|
LambdaQueryWrapper<SubTaskInfo> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(SubTaskInfo::getTaskcode,taskCode);
|
baseMapper.delete(wrapper);
|
}
|
|
@Override
|
public void saveSubTasks(List<SubTaskInfo> subTaskInfos, String taskCode, UserInfo userInfo, String option) {
|
{
|
if (subTaskInfos == null || subTaskInfos.size() == 0) {throw new BusinessException("请选择涉及的作业信息");}
|
List<TaskEnclosure> enclosures = new ArrayList<>();
|
subTaskInfos.forEach(item-> {
|
item.setTaskcode(taskCode);
|
List<TaskEnclosure> breakFiles = item.getBreakfile();
|
if (breakFiles != null){
|
breakFiles.forEach(item1->{
|
item1.setTaskcode(taskCode);
|
item1.setOriginalname(item1.getName());
|
item1.setCreatedat(new Date());
|
item1.setCreatedby(userInfo.getRealname());
|
item1.setType("断路地段示意图");
|
});
|
enclosures.addAll(breakFiles);
|
}
|
|
List<TaskEnclosure> soilFiles = item.getSoilfile();
|
if (soilFiles != null){
|
soilFiles.forEach(item2->{
|
item2.setTaskcode(taskCode);
|
item2.setOriginalname(item2.getName());
|
item2.setCreatedat(new Date());
|
item2.setCreatedby(userInfo.getRealname());
|
item2.setType("动土方案文件");
|
});
|
enclosures.addAll(soilFiles);
|
}
|
|
|
List<TaskEnclosure> blindBoardLocations = item.getBlindboardlocation();
|
if (blindBoardLocations != null){
|
blindBoardLocations.forEach(item3->{
|
item3.setTaskcode(taskCode);
|
item3.setOriginalname(item3.getName());
|
item3.setCreatedat(new Date());
|
item3.setCreatedby(userInfo.getRealname());
|
item3.setType("盲板位置图");
|
});
|
enclosures.addAll(blindBoardLocations);
|
}
|
|
});
|
if (option.equals("UPDATE")){
|
taskEnclosureService.removeSubFile(taskCode);
|
removeByCode(taskCode);
|
}
|
|
taskEnclosureService.saveBatch(enclosures);
|
saveBatch(subTaskInfos);
|
}
|
}
|
}
|
