package com.ruoyi.business.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.ruoyi.business.dto.EquipmentManagementDto;
|
import com.ruoyi.business.entity.EquipmentManagement;
|
import com.ruoyi.business.entity.EquipmentUsageRecord;
|
import com.ruoyi.business.mapper.EquipmentManagementMapper;
|
import com.ruoyi.business.mapper.EquipmentUsageRecordMapper;
|
import com.ruoyi.business.service.EquipmentManagementService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.business.vo.EquipmentManagementVo;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.common.utils.bean.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import lombok.RequiredArgsConstructor;
|
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Objects;
|
|
/**
|
* <p>
|
* 设备管理表 服务实现类
|
* </p>
|
*
|
* @author ld
|
* @since 2025-07-01
|
*/
|
@Service
|
@RequiredArgsConstructor
|
public class EquipmentManagementServiceImpl extends ServiceImpl<EquipmentManagementMapper, EquipmentManagement> implements EquipmentManagementService {
|
|
private final EquipmentManagementMapper equipmentManagementMapper;
|
|
private final EquipmentUsageRecordMapper equipmentUsageRecordMapper;
|
|
public IPage<EquipmentManagementDto> selectProductionList(Page<EquipmentManagement> page, EquipmentManagementDto equipmentManagementDto) {
|
// 1. 构建查询条件（searchAll 模糊查询）
|
LambdaQueryWrapper<EquipmentManagement> queryWrapper = new LambdaQueryWrapper<>();
|
|
if (StringUtils.isNotBlank(equipmentManagementDto.getSearchAll())) {
|
String searchValue = equipmentManagementDto.getSearchAll();
|
queryWrapper.and(wrapper ->
|
wrapper.like(EquipmentManagement::getEquipmentNo, searchValue)
|
.or()
|
.like(EquipmentManagement::getEquipmentName, searchValue)
|
.or()
|
.like(EquipmentManagement::getSpecification, searchValue)
|
);
|
}
|
|
// 2. 执行分页查询
|
Page<EquipmentManagement> entityPage = equipmentManagementMapper.selectPage(page, queryWrapper);
|
|
// 3. 转换为 DTO 分页
|
IPage<EquipmentManagementDto> dtoPage = new Page<>();
|
BeanUtils.copyProperties(entityPage, dtoPage, "records");
|
|
// 4. 查询每个设备的使用总量（usageQuantity）并设置到 DTO
|
List<EquipmentManagementDto> dtoRecords = entityPage.getRecords().stream()
|
.map(entity -> {
|
EquipmentManagementDto dto = new EquipmentManagementDto();
|
BeanUtils.copyProperties(entity, dto);
|
|
// 查询该设备的使用总量（usageQuantity）
|
LambdaQueryWrapper<EquipmentUsageRecord> usageQueryWrapper = new LambdaQueryWrapper<>();
|
usageQueryWrapper.eq(EquipmentUsageRecord::getEquipmentId, entity.getId());
|
|
// 计算总使用量
|
Integer totalUsage = equipmentUsageRecordMapper.selectList(usageQueryWrapper).stream()
|
.mapToInt(EquipmentUsageRecord::getUsageQuantity)
|
.sum();
|
|
// 设置使用总量
|
dto.setUsedNo(totalUsage);
|
return dto;
|
})
|
.toList();
|
|
dtoPage.setRecords(dtoRecords);
|
return dtoPage;
|
}
|
|
@Override
|
public int addOrEditEquipment(EquipmentManagementDto equipmentManagementDto) {
|
EquipmentManagement equipmentManagement = new EquipmentManagement();
|
BeanUtils.copyProperties(equipmentManagementDto, equipmentManagement);
|
if (Objects.isNull(equipmentManagementDto.getId())) {
|
return equipmentManagementMapper.insert(equipmentManagement);
|
} else {
|
return equipmentManagementMapper.updateById(equipmentManagement);
|
}
|
}
|
|
@Override
|
public List<EquipmentManagementVo> selectEquipmentList() {
|
List<EquipmentManagement> equipmentManagements = equipmentManagementMapper.selectList(null);
|
return equipmentManagements.stream()
|
.map(equipment -> {
|
EquipmentManagementVo vo = new EquipmentManagementVo();
|
vo.setId(equipment.getId()); // 设备id
|
vo.setEquipmentName(equipment.getEquipmentName()); // 设备名称
|
vo.setQuantity(equipment.getQuantity()); // 数量
|
return vo;
|
})
|
.toList();
|
}
|
|
@Override
|
public int delByIds(Long[] ids) {
|
return equipmentManagementMapper.deleteByIds(Arrays.asList(ids));
|
}
|
}
|
