维修模块优化

This commit is contained in:
sxu 2024-11-23 11:46:15 +08:00
parent a3163ab156
commit 709f445119
1 changed files with 9 additions and 0 deletions

View File

@ -3,6 +3,7 @@ package com.bonus.material.repair.service.impl;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.NumberUtil;
import com.bonus.common.biz.enums.MaTypeManageTypeEnum;
import com.bonus.common.biz.enums.RepairTaskStatusEnum;
import com.bonus.common.biz.enums.TmTaskTypeEnum;
import com.bonus.common.core.exception.ServiceException;
@ -108,6 +109,14 @@ public class RepairServiceImpl implements RepairService {
});
});
for (Long key : resultMap.keySet()) {
RepairDeviceListVo vo = resultMap.get(key);
if (MaTypeManageTypeEnum.CODE_DEVICE.getTypeId().equals(vo.getManageType())) {
vo.setRepairNum((int) vo.getRepairDeviceList().stream().filter(o -> o.getRepairNum() > 0).count());
vo.setRepairedNum((int) vo.getRepairDeviceList().stream().filter(o -> o.getRepairedNum() > 0).count());
}
}
return new ArrayList<>(resultMap.values());
}