| | |
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
| | | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; |
| | | import com.ruoyi.basic.dto.CoalPlanDto; |
| | | import com.ruoyi.basic.entity.CoalField; |
| | | import com.ruoyi.basic.entity.CoalPlan; |
| | | import com.ruoyi.basic.mapper.CoalFieldMapper; |
| | | import com.ruoyi.basic.mapper.CoalPlanMapper; |
| | | import com.ruoyi.basic.service.CoalPlanService; |
| | | import com.ruoyi.basic.vo.CoalFieldVo; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | import com.ruoyi.common.utils.bean.BeanUtils; |
| | | import lombok.RequiredArgsConstructor; |
| | | import org.springframework.stereotype.Service; |
| | | |
| | | import java.util.Arrays; |
| | | import java.util.List; |
| | | import java.util.Objects; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * <p> |
| | |
| | | |
| | | private final CoalPlanMapper coalPlanMapper; |
| | | |
| | | private final CoalFieldMapper coalFieldMapper; |
| | | |
| | | @Override |
| | | public IPage<CoalPlan> selectCoalFieldList(Page page, CoalPlanDto coalPlanDto) { |
| | | LambdaQueryWrapper<CoalPlan> queryWrapper = new LambdaQueryWrapper<>(); |
| | | return coalPlanMapper.selectPage(page,queryWrapper); |
| | | if (StringUtils.hasText(coalPlanDto.getSearchAll())){ |
| | | queryWrapper.like(CoalPlan::getPlan,coalPlanDto.getSearchAll()); |
| | | } |
| | | return coalPlanMapper.selectPage(page, queryWrapper); |
| | | } |
| | | |
| | | @Override |
| | |
| | | public List<CoalPlan> selectAllList() { |
| | | return coalPlanMapper.selectList(null); |
| | | } |
| | | |
| | | @Override |
| | | public List<CoalFieldVo> selectCoalPlanById(Long id) { |
| | | CoalPlan coalPlan = coalPlanMapper.selectById(id); |
| | | LambdaQueryWrapper<CoalField> coalFieldLambdaQueryWrapper = new LambdaQueryWrapper<>(); |
| | | List<Long> ids = Arrays.stream(coalPlan.getFieldIds().split(",")).map(String::trim).map(Long::parseLong).collect(Collectors.toList()); |
| | | coalFieldLambdaQueryWrapper.in(CoalField::getId, ids); |
| | | List<CoalField> coalFields = coalFieldMapper.selectList(coalFieldLambdaQueryWrapper); |
| | | return coalFields.stream() |
| | | .map(coalField -> { |
| | | CoalFieldVo dto = new CoalFieldVo(); |
| | | dto.setId(coalField.getId()); |
| | | dto.setFields(coalField.getFields()); |
| | | dto.setFieldName(coalField.getFieldName()); |
| | | return dto; |
| | | }) |
| | | .collect(Collectors.toList()); |
| | | } |
| | | } |