diff --git a/klp-mes/src/main/java/com/klp/mes/eqp/domain/bo/EqpEquipmentChecklistBo.java b/klp-mes/src/main/java/com/klp/mes/eqp/domain/bo/EqpEquipmentChecklistBo.java index 0d34d6b3..5b11e70b 100644 --- a/klp-mes/src/main/java/com/klp/mes/eqp/domain/bo/EqpEquipmentChecklistBo.java +++ b/klp-mes/src/main/java/com/klp/mes/eqp/domain/bo/EqpEquipmentChecklistBo.java @@ -4,6 +4,7 @@ import com.klp.common.core.domain.BaseEntity; import lombok.Data; import lombok.EqualsAndHashCode; import javax.validation.constraints.*; +import java.util.List; /** @@ -62,5 +63,7 @@ public class EqpEquipmentChecklistBo extends BaseEntity { */ private String remark; + private List partIds; + } diff --git a/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentChecklistServiceImpl.java b/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentChecklistServiceImpl.java index 2535a45a..e5a164c1 100644 --- a/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentChecklistServiceImpl.java +++ b/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentChecklistServiceImpl.java @@ -12,7 +12,9 @@ import org.springframework.stereotype.Service; import com.klp.mes.eqp.domain.bo.EqpEquipmentChecklistBo; import com.klp.mes.eqp.domain.vo.EqpEquipmentChecklistVo; import com.klp.mes.eqp.domain.EqpEquipmentChecklist; +import com.klp.mes.eqp.domain.EqpEquipmentPart; import com.klp.mes.eqp.mapper.EqpEquipmentChecklistMapper; +import com.klp.mes.eqp.mapper.EqpEquipmentPartMapper; import com.klp.mes.eqp.service.IEqpEquipmentChecklistService; import java.util.List; @@ -30,6 +32,7 @@ import java.util.Collection; public class EqpEquipmentChecklistServiceImpl implements IEqpEquipmentChecklistService { private final EqpEquipmentChecklistMapper baseMapper; + private final EqpEquipmentPartMapper eqpEquipmentPartMapper; /** * 查询设备检验清单 @@ -46,6 +49,7 @@ public class EqpEquipmentChecklistServiceImpl implements IEqpEquipmentChecklistS public TableDataInfo queryPageList(EqpEquipmentChecklistBo bo, PageQuery pageQuery) { LambdaQueryWrapper lqw = buildQueryWrapper(bo); Page result = baseMapper.selectVoPage(pageQuery.build(), lqw); + fillPartName(result.getRecords()); return TableDataInfo.build(result); } @@ -55,7 +59,26 @@ public class EqpEquipmentChecklistServiceImpl implements IEqpEquipmentChecklistS @Override public List queryList(EqpEquipmentChecklistBo bo) { LambdaQueryWrapper lqw = buildQueryWrapper(bo); - return baseMapper.selectVoList(lqw); + List list = baseMapper.selectVoList(lqw); + fillPartName(list); + return list; + } + + /** + * 填充检验清单的检验部位名称 + */ + private void fillPartName(List list) { + if (list == null || list.isEmpty()) { + return; + } + list.forEach(vo -> { + if (vo.getPartId() != null) { + EqpEquipmentPart part = eqpEquipmentPartMapper.selectById(vo.getPartId()); + if (part != null) { + vo.setPartName(part.getInspectPart()); + } + } + }); } private LambdaQueryWrapper buildQueryWrapper(EqpEquipmentChecklistBo bo) { diff --git a/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentPartServiceImpl.java b/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentPartServiceImpl.java index c04f4a49..36f90973 100644 --- a/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentPartServiceImpl.java +++ b/klp-mes/src/main/java/com/klp/mes/eqp/service/impl/EqpEquipmentPartServiceImpl.java @@ -8,6 +8,7 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.klp.common.utils.StringUtils; import com.klp.mes.eqp.domain.bo.EqpEquipmentChecklistBo; +import com.klp.mes.eqp.service.IEqpEquipmentChecklistService; import lombok.RequiredArgsConstructor; import org.springframework.stereotype.Service; import com.klp.mes.eqp.domain.bo.EqpEquipmentPartBo; @@ -21,6 +22,7 @@ import com.klp.mes.eqp.service.IEqpEquipmentPartService; import java.util.*; import java.util.stream.Collectors; + import org.springframework.transaction.annotation.Transactional; /** @@ -34,6 +36,7 @@ import org.springframework.transaction.annotation.Transactional; public class EqpEquipmentPartServiceImpl implements IEqpEquipmentPartService { private final EqpEquipmentPartMapper baseMapper; + private final IEqpEquipmentChecklistService checklistService; private final EqpEquipmentChecklistMapper checklistMapper; /** @@ -54,16 +57,14 @@ public class EqpEquipmentPartServiceImpl implements IEqpEquipmentPartService { List partIds = result.getRecords().stream() .map(EqpEquipmentPartVo::getPartId).collect(Collectors.toList()); if (!partIds.isEmpty()) { - LambdaQueryWrapper checklistLqw = Wrappers.lambdaQuery(); - checklistLqw.in(EqpEquipmentChecklist::getPartId, partIds); - List checklistList = checklistMapper.selectVoList(checklistLqw); + EqpEquipmentChecklistBo eqpEquipmentChecklistBo = new EqpEquipmentChecklistBo(); + eqpEquipmentChecklistBo.setPartIds(partIds); + List checklistList = checklistService.queryList(eqpEquipmentChecklistBo); Map> checklistMap = checklistList.stream() .collect(Collectors.groupingBy(EqpEquipmentChecklistVo::getPartId)); - result.getRecords().forEach(vo -> { - List items = checklistMap.getOrDefault(vo.getPartId(), Collections.emptyList()); - items.forEach(checklist -> checklist.setPartName(vo.getInspectPart())); - vo.setChecklistList(items); - }); + result.getRecords().forEach(vo -> + vo.setChecklistList(checklistMap.getOrDefault(vo.getPartId(), Collections.emptyList())) + ); } return TableDataInfo.build(result); }