package cn.iocoder.yudao.module.mes.api.itemreceipt;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.mes.api.itemreceipt.dto.MesWmItemReceiptLineRespDTO;
|
import cn.iocoder.yudao.module.mes.api.itemreceipt.dto.MesWmItemReceiptRespDTO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.itemreceipt.MesWmItemReceiptDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.itemreceipt.MesWmItemReceiptLineDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.wm.itemreceipt.MesWmItemReceiptLineMapper;
|
import cn.iocoder.yudao.module.mes.dal.mysql.wm.itemreceipt.MesWmItemReceiptMapper;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Service;
|
|
import java.math.BigDecimal;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* MES 采购入库单 API 实现类
|
*/
|
@Service
|
public class MesWmItemReceiptApiImpl implements MesWmItemReceiptApi {
|
|
@Resource
|
private MesWmItemReceiptMapper itemReceiptMapper;
|
@Resource
|
private MesWmItemReceiptLineMapper itemReceiptLineMapper;
|
|
@Override
|
public List<MesWmItemReceiptRespDTO> getItemReceiptListByPurchaseOrderId(Long purchaseOrderId) {
|
if (purchaseOrderId == null) {
|
return List.of();
|
}
|
List<MesWmItemReceiptDO> receipts = itemReceiptMapper.selectListByPurchaseOrderId(purchaseOrderId);
|
if (CollUtil.isEmpty(receipts)) {
|
return List.of();
|
}
|
Map<Long, List<MesWmItemReceiptLineDO>> lineMap = itemReceiptLineMapper
|
.selectListByReceiptIds(receipts.stream().map(MesWmItemReceiptDO::getId).toList())
|
.stream().collect(Collectors.groupingBy(MesWmItemReceiptLineDO::getReceiptId));
|
return receipts.stream()
|
.map(receipt -> convert(receipt, lineMap.getOrDefault(receipt.getId(), List.of())))
|
.toList();
|
}
|
|
@Override
|
public MesWmItemReceiptRespDTO getItemReceipt(Long id) {
|
MesWmItemReceiptDO receipt = itemReceiptMapper.selectById(id);
|
if (receipt == null) {
|
return null;
|
}
|
return convert(receipt, itemReceiptLineMapper.selectListByReceiptId(id));
|
}
|
|
private MesWmItemReceiptRespDTO convert(MesWmItemReceiptDO receipt, List<MesWmItemReceiptLineDO> lines) {
|
MesWmItemReceiptRespDTO resp = BeanUtils.toBean(receipt, MesWmItemReceiptRespDTO.class);
|
List<MesWmItemReceiptLineRespDTO> lineResps = BeanUtils.toBean(lines, MesWmItemReceiptLineRespDTO.class);
|
resp.setLines(lineResps);
|
resp.setTotalPrice(lineResps.stream()
|
.map(line -> line.getTotalPrice() != null ? line.getTotalPrice() : calcLineAmount(line))
|
.filter(amount -> amount != null)
|
.reduce(BigDecimal.ZERO, BigDecimal::add));
|
return resp;
|
}
|
|
private BigDecimal calcLineAmount(MesWmItemReceiptLineRespDTO line) {
|
if (line.getPrice() == null || line.getReceivedQuantity() == null) {
|
return null;
|
}
|
return line.getPrice().multiply(line.getReceivedQuantity());
|
}
|
|
}
|
