| | |
| | | package com.ruoyi.system.service.impl; |
| | | |
| | | import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
| | | import com.ruoyi.common.basic.PageInfo; |
| | | import com.ruoyi.system.mapper.TFaultAreaDicMapper; |
| | | import com.ruoyi.system.mapper.TFaultDescribeDicMapper; |
| | | import com.ruoyi.system.model.TFaultAreaDic; |
| | | import com.ruoyi.system.model.TFaultDescribeDic; |
| | | import com.ruoyi.system.query.TFaultAreaDicQuery; |
| | | import com.ruoyi.system.query.TFaultDescribeDicQuery; |
| | | import com.ruoyi.system.service.TFaultDescribeDicService; |
| | | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; |
| | | import com.ruoyi.system.vo.TFaultDescribeDicVO; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.stereotype.Service; |
| | | import org.springframework.util.CollectionUtils; |
| | | |
| | | import java.util.List; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * <p> |
| | |
| | | @Service |
| | | public class TFaultDescribeDicServiceImpl extends ServiceImpl<TFaultDescribeDicMapper, TFaultDescribeDic> implements TFaultDescribeDicService { |
| | | |
| | | @Autowired |
| | | private TFaultAreaDicMapper faultAreaDicMapper; |
| | | |
| | | @Override |
| | | public PageInfo<TFaultDescribeDicVO> pageList(TFaultDescribeDicQuery query) { |
| | | PageInfo<TFaultDescribeDicVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize()); |
| | | List<TFaultDescribeDicVO> list = this.baseMapper.pageList(query,pageInfo); |
| | | if(!CollectionUtils.isEmpty(list)){ |
| | | List<String> areaIds = list.stream().map(TFaultDescribeDicVO::getFaultId).collect(Collectors.toList()); |
| | | List<TFaultAreaDic> tFaultAreaDicList = faultAreaDicMapper.selectList(Wrappers.lambdaQuery(TFaultAreaDic.class) |
| | | .in(TFaultAreaDic::getId, areaIds)); |
| | | for (TFaultDescribeDicVO tFaultDescribeDicVO : list) { |
| | | tFaultDescribeDicVO.setFaultAreaName(tFaultAreaDicList.stream().filter(t -> t.getId().equals(tFaultDescribeDicVO.getFaultId())) |
| | | .findFirst().orElse(new TFaultAreaDic()).getFaultAreaName()); |
| | | } |
| | | } |
| | | pageInfo.setRecords(list); |
| | | return pageInfo; |
| | | } |
| | | } |