package cn.gistack.sm.sg.service.impl;
|
|
import cn.gistack.sm.sg.DO.SgGxDO;
|
import cn.gistack.sm.sg.DO.SgGxStepDO;
|
import cn.gistack.sm.sg.DTO.ProjectSearchDTO;
|
import cn.gistack.sm.sg.VO.SgGxVO;
|
import cn.gistack.sm.sg.mapper.SgGxMapper;
|
import cn.gistack.sm.sg.service.SgGxService;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 施工工序基本表 服务实现类
|
* </p>
|
*
|
* @author shenyijian
|
* @since 2024-06-13 15:46:18
|
*/
|
@Service
|
public class SgGxServiceImpl extends ServiceImpl<SgGxMapper, SgGxDO> implements SgGxService {
|
@Autowired
|
private SgGxMapper sgGxMapper;
|
@Override
|
public IPage<SgGxVO> queryPageList(IPage<SgGxVO> page, ProjectSearchDTO searchDTO) {
|
List<SgGxVO> list = sgGxMapper.queryPageList(searchDTO);
|
if (CollectionUtils.isNotEmpty(list)){
|
List<Long> ids = list.stream().map(SgGxDO::getId).collect(Collectors.toList());
|
List<SgGxStepDO> stepDOS = sgGxMapper.queryStepByIds(ids);
|
if (CollectionUtils.isNotEmpty(stepDOS)){
|
Map<Long, List<SgGxStepDO>> map = stepDOS.stream().collect(Collectors.groupingBy(SgGxStepDO::getGxInfoId));
|
for (SgGxVO sgGxDO : list) {
|
sgGxDO.setStepDOList(map.get(sgGxDO.getId()));
|
}
|
}
|
}
|
return page.setRecords(list);
|
}
|
|
@Override
|
public void add(SgGxDO sgGxDO) {
|
if (sgGxMapper.selectByGxType(sgGxDO.getGxType())!= null){
|
throw new RuntimeException("工序类型已存在");
|
}
|
this.save(sgGxDO);
|
}
|
|
@Override
|
public void update(SgGxDO sgGxDO) {
|
if (sgGxMapper.selectByGxType(sgGxDO.getGxType())!= null){
|
throw new RuntimeException("工序类型已存在");
|
}
|
this.updateById(sgGxDO);
|
}
|
|
@Override
|
public SgGxVO detail(Long id) {
|
SgGxVO byId = (SgGxVO) this.getById(id);
|
byId.setStepDOList(sgGxMapper.queryStepByIds(List.of(id)));
|
return byId;
|
}
|
}
|
