src/main/java/com/ruoyi/device/dto/DeviceMaintenanceDto.java
@@ -66,4 +66,11 @@ @ApiModelProperty("创建人名称") private String createUserName; @ApiModelProperty("领用备件ids") private String sparePartsIds; private String sparePartsNames; } src/main/java/com/ruoyi/device/dto/DeviceRepairDto.java
@@ -69,6 +69,10 @@ @TableField(fill = FieldFill.INSERT) private Long tenantId; @ApiModelProperty("领用备件ids") private String sparePartsIds; private String sparePartsNames; }
@@ -84,4 +84,7 @@ @ApiModelProperty("租户id") @TableField(fill = FieldFill.INSERT) private Long tenantId; @ApiModelProperty("领用备件ids") private String sparePartsIds; } src/main/java/com/ruoyi/device/pojo/DeviceRepair.java
@@ -73,4 +73,8 @@ private Long tenantId; @ApiModelProperty("领用备件ids") private String sparePartsIds; } src/main/java/com/ruoyi/device/service/impl/DeviceMaintenanceServiceImpl.java
@@ -4,26 +4,34 @@ import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.github.xiaoymin.knife4j.core.util.StrUtil; import com.ruoyi.common.utils.bean.BeanUtils; import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.device.dto.DeviceMaintenanceDto; import com.ruoyi.device.dto.DeviceRepairDto; import com.ruoyi.device.execl.DeviceMaintenanceExeclDto; import com.ruoyi.device.mapper.DeviceMaintenanceMapper; import com.ruoyi.device.pojo.DeviceMaintenance; import com.ruoyi.device.service.IDeviceMaintenanceService; import com.ruoyi.framework.web.domain.AjaxResult; import com.ruoyi.measuringinstrumentledger.mapper.SparePartsMapper; import com.ruoyi.measuringinstrumentledger.pojo.SpareParts; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import javax.servlet.http.HttpServletResponse; import java.util.ArrayList; import java.util.List; import java.util.*; import java.util.stream.Collectors; @Service public class DeviceMaintenanceServiceImpl extends ServiceImpl<DeviceMaintenanceMapper, DeviceMaintenance> implements IDeviceMaintenanceService { @Autowired private SparePartsMapper sparePartsMapper; @Autowired private DeviceMaintenanceMapper deviceMaintenanceMapper; @@ -31,7 +39,43 @@ @Override public IPage<DeviceMaintenanceDto> queryPage(Page page, DeviceMaintenanceDto deviceMaintenanceDto) { return deviceMaintenanceMapper.queryPage(page, deviceMaintenanceDto); IPage<DeviceMaintenanceDto> deviceMaintenanceDtoIPage = deviceMaintenanceMapper.queryPage(page, deviceMaintenanceDto); List<DeviceMaintenanceDto> records = deviceMaintenanceDtoIPage.getRecords(); if (!CollectionUtils.isEmpty(records)) { // 1. 获取所有唯一的备件ID Set<String> allIds = records.stream() .map(DeviceMaintenanceDto::getSparePartsIds) .filter(StrUtil::isNotBlank) .flatMap(ids -> Arrays.stream(StringUtils.split(ids, ","))) .map(String::trim) .filter(StrUtil::isNotBlank) .collect(Collectors.toSet()); if (!allIds.isEmpty()) { // 2. 查询并创建ID到名称的映射 Map<Long, String> idToNameMap = sparePartsMapper.selectBatchIds(new ArrayList<>(allIds)) .stream() .collect(Collectors.toMap( SpareParts::getId, SpareParts::getName, (v1, v2) -> v1 )); // 3. 为每个记录设置备件名称 records.forEach(record -> { String names = Optional.ofNullable(record.getSparePartsIds()) .filter(StrUtil::isNotBlank) .map(ids -> Arrays.stream(StringUtils.split(ids, ",")) .map(String::trim) .map(idToNameMap::get) .filter(Objects::nonNull) .collect(Collectors.joining(","))) .orElse(""); record.setSparePartsNames(names); }); } } return deviceMaintenanceDtoIPage; } @Override src/main/java/com/ruoyi/device/service/impl/DeviceRepairServiceImpl.java
@@ -1,8 +1,11 @@ package com.ruoyi.device.service.impl; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.github.xiaoymin.knife4j.core.util.StrUtil; import com.google.common.collect.Lists; import com.ruoyi.common.utils.bean.BeanUtils; import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.device.dto.DeviceDefectRecordDto; @@ -15,15 +18,20 @@ import com.ruoyi.device.service.IDeviceLedgerService; import com.ruoyi.device.service.IDeviceRepairService; import com.ruoyi.framework.web.domain.AjaxResult; import com.ruoyi.measuringinstrumentledger.mapper.SparePartsMapper; import com.ruoyi.measuringinstrumentledger.pojo.SpareParts; import com.ruoyi.measuringinstrumentledger.service.SparePartsService; import lombok.AllArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.tuple.Pair; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import javax.servlet.http.HttpServletResponse; import java.util.ArrayList; import java.util.Arrays; import java.util.List; import java.util.*; import java.util.stream.Collectors; @Service @AllArgsConstructor @@ -36,10 +44,51 @@ private DeviceRepairMapper deviceRepairMapper; @Autowired private IDeviceLedgerService deviceLedgerService; @Autowired private SparePartsMapper sparePartsMapper; @Override public IPage<DeviceRepairDto> queryPage(Page page, DeviceRepairDto deviceRepairDto) { IPage<DeviceRepairDto> deviceRepairDtoIPage = deviceRepairMapper.queryPage(page, deviceRepairDto); return deviceRepairMapper.queryPage(page, deviceRepairDto); List<DeviceRepairDto> records = deviceRepairDtoIPage.getRecords(); if (!CollectionUtils.isEmpty(records)) { // 1. 获取所有唯一的备件ID Set<String> allIds = records.stream() .map(DeviceRepairDto::getSparePartsIds) .filter(StrUtil::isNotBlank) .flatMap(ids -> Arrays.stream(StringUtils.split(ids, ","))) .map(String::trim) .filter(StrUtil::isNotBlank) .collect(Collectors.toSet()); if (!allIds.isEmpty()) { // 2. 查询并创建ID到名称的映射 Map<Long, String> idToNameMap = sparePartsMapper.selectBatchIds(new ArrayList<>(allIds)) .stream() .collect(Collectors.toMap( SpareParts::getId, SpareParts::getName, (v1, v2) -> v1 )); // 3. 为每个记录设置备件名称 records.forEach(record -> { String names = Optional.ofNullable(record.getSparePartsIds()) .filter(StrUtil::isNotBlank) .map(ids -> Arrays.stream(StringUtils.split(ids, ",")) .map(String::trim) .map(Long::valueOf) .map(idToNameMap::get) .filter(Objects::nonNull) .collect(Collectors.joining(","))) .orElse(""); record.setSparePartsNames(names); }); } } return deviceRepairDtoIPage; } @Override src/main/java/com/ruoyi/measuringinstrumentledger/controller/SparePartsController.java
@@ -58,4 +58,10 @@ return AjaxResult.success(sparePartsService.removeById(id)); } @GetMapping("getByIdDeviceId") @ApiOperation("备件分类-查询") public AjaxResult getByDeviceId(Long deviceLedgerId){ return AjaxResult.success(sparePartsService.getByDeviceId(deviceLedgerId)); } } src/main/java/com/ruoyi/measuringinstrumentledger/mapper/SparePartsMapper.java
@@ -8,7 +8,11 @@ import io.lettuce.core.dynamic.annotation.Param; import org.apache.ibatis.annotations.Mapper; import java.util.List; @Mapper public interface SparePartsMapper extends BaseMapper<SpareParts> { IPage<SparePartsDto> listPage(Page page,@Param("spareParts") SpareParts spareParts); List<SparePartsDto> getByDeviceId(@Param("id") Long id); } src/main/java/com/ruoyi/measuringinstrumentledger/service/SparePartsService.java
@@ -13,4 +13,6 @@ IPage<SparePartsDto> listPage(Page page, SpareParts spareParts); List<SparePartsDto> getTree(); List<SparePartsDto> getByDeviceId(Long id); } src/main/java/com/ruoyi/measuringinstrumentledger/service/impl/SparePartsServiceImpl.java
@@ -15,10 +15,7 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.*; import java.util.stream.Collectors; @Service @@ -52,6 +49,11 @@ return buildTree(records); } @Override public List<SparePartsDto> getByDeviceId(Long id) { return sparePartsMapper.getByDeviceId(id); } /** * 将扁平结构数据转换为树形结构 src/main/resources/mapper/device/DeviceRepairMapper.xml
@@ -21,7 +21,8 @@ dr.update_user, dr.tenant_id, dl.device_name, dl.device_model dl.device_model, dr.spare_parts_ids from device_repair dr left join device_ledger dl on dr.device_ledger_id = dl.id <where> src/main/resources/mapper/measuringinstrumentledger/SparePartsMapper.xml
@@ -11,4 +11,9 @@ </if> </where> </select> <select id="getByDeviceId" resultType="com.ruoyi.measuringinstrumentledger.dto.SparePartsDto"> SELECT * FROM spare_parts WHERE status = '正常' AND FIND_IN_SET(#{id}, device_ids) > 0 </select> </mapper>
