package cn.iocoder.yudao.module.mes.service.pro.qrexport;
|
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.bagcode.MesProBagCodeDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.batch.MesProBatchDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.pallet.MesProPalletDO;
|
import cn.iocoder.yudao.module.mes.enums.pro.MesProPalletStatusEnum;
|
import cn.iocoder.yudao.module.mes.enums.pro.MesProQrExportBizTypeEnum;
|
import cn.iocoder.yudao.module.mes.framework.qrcode.QrLabelInfo;
|
import cn.iocoder.yudao.module.mes.service.pro.bagcode.MesProBagCodeService;
|
import cn.iocoder.yudao.module.mes.service.pro.batch.MesProBatchService;
|
import cn.iocoder.yudao.module.mes.service.pro.pallet.MesProPalletService;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Component;
|
|
import java.util.List;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PRO_BATCH_NOT_EXISTS;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PRO_PALLET_NOT_EXISTS;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PRO_QR_EXPORT_PARAMS_INVALID;
|
|
/**
|
* 二维码标签数据源工厂
|
* <p>
|
* 把「业务类型 + 导出参数」翻译成统一的数据源，使后台导出任务不必关心标签是怎么来的，
|
* 也让同步导出路径与后台路径共用同一套取数与装配口径。
|
*/
|
@Component
|
public class MesProQrLabelSourceFactory {
|
|
@Resource
|
private MesProBagCodeService bagCodeService;
|
@Resource
|
private MesProPalletService palletService;
|
@Resource
|
private MesProBatchService batchService;
|
@Resource
|
private MesProQrLabelAssembler assembler;
|
|
/**
|
* 创建标签数据源；参数不合法时直接抛出业务异常
|
*
|
* @param bizType 业务类型
|
* @param uuid 生成组 UUID（袋码类导出可用）
|
* @param batchId 生产批次编号（袋码类可用，其余类型必填）
|
*/
|
public MesProQrLabelSource create(MesProQrExportBizTypeEnum bizType, String uuid, Long batchId) {
|
if (bizType == null) {
|
throw exception(PRO_QR_EXPORT_PARAMS_INVALID);
|
}
|
return switch (bizType) {
|
case BAG_LABEL -> bagSource(uuid, batchId, false);
|
case BAG_BUSINESS -> bagSource(uuid, batchId, true);
|
case PALLET_LABEL -> palletSource(requireBatchId(batchId), false);
|
case PALLET_BUSINESS -> palletSource(requireBatchId(batchId), true);
|
case BATCH_LABEL -> batchSource(requireBatchId(batchId), false);
|
case BATCH_BUSINESS -> batchSource(requireBatchId(batchId), true);
|
};
|
}
|
|
/**
|
* 统计标签张数（分流判断用，不装配标签内容）
|
*/
|
public long count(MesProQrExportBizTypeEnum bizType, String uuid, Long batchId) {
|
return create(bizType, uuid, batchId).count();
|
}
|
|
private Long requireBatchId(Long batchId) {
|
if (batchId == null) {
|
throw exception(PRO_QR_EXPORT_PARAMS_INVALID);
|
}
|
return batchId;
|
}
|
|
// ==================== 各业务类型的数据源 ====================
|
|
/**
|
* 袋码类数据源（标签 / 业务码）
|
*/
|
private MesProQrLabelSource bagSource(String uuid, Long batchId, boolean business) {
|
if (StrUtil.isBlank(uuid) && batchId == null) {
|
throw exception(PRO_QR_EXPORT_PARAMS_INVALID);
|
}
|
boolean byUuid = StrUtil.isNotBlank(uuid);
|
String zipName = (business ? "袋码业务二维码-" : "袋码二维码-") + (byUuid ? uuid : batchId);
|
return new MesProQrLabelSource() {
|
|
@Override
|
public long count() {
|
return byUuid ? bagCodeService.getPrintableBagCodeCountByUuid(uuid)
|
: bagCodeService.getPrintableBagCodeCountByBatchId(batchId);
|
}
|
|
@Override
|
public List<QrLabelInfo> fetchPage(int pageNo, int pageSize) {
|
List<MesProBagCodeDO> bags = byUuid
|
? bagCodeService.getPrintableBagCodePageByUuid(uuid, pageNo, pageSize)
|
: bagCodeService.getPrintableBagCodePageByBatchId(batchId, pageNo, pageSize);
|
return business ? assembler.buildBagBusinessLabels(bags) : assembler.buildBagCodeLabels(bags);
|
}
|
|
@Override
|
public String zipName() {
|
return zipName;
|
}
|
};
|
}
|
|
/**
|
* 托盘类数据源
|
* <p>
|
* 一个批次下的托盘数量有上界（受每托袋数约束），故一次性取出后按页切片，无需真分页查询。
|
*/
|
private MesProQrLabelSource palletSource(Long batchId, boolean business) {
|
List<MesProPalletDO> pallets = palletService.getPalletListByBatchId(batchId);
|
if (business) {
|
// 业务码导出需排除已作废托盘，避免作废码被印出
|
pallets = pallets.stream()
|
.filter(record -> !MesProPalletStatusEnum.CANCELED.getStatus().equals(record.getStatus()))
|
.toList();
|
}
|
if (pallets.isEmpty()) {
|
throw exception(PRO_PALLET_NOT_EXISTS);
|
}
|
List<QrLabelInfo> labels = business ? assembler.buildPalletBusinessLabels(pallets)
|
: assembler.buildPalletLabels(pallets);
|
String zipName = business ? "托盘业务二维码" : "托盘码二维码-" + batchId;
|
return fixedSource(labels, zipName);
|
}
|
|
/**
|
* 批次类数据源：批次图 + 该批次全部托盘图
|
*/
|
private MesProQrLabelSource batchSource(Long batchId, boolean business) {
|
MesProBatchDO batch = batchService.getBatch(batchId);
|
if (batch == null) {
|
throw exception(PRO_BATCH_NOT_EXISTS);
|
}
|
List<MesProPalletDO> pallets = palletService.getPalletListByBatchId(batchId);
|
if (business) {
|
pallets = pallets.stream()
|
.filter(record -> !MesProPalletStatusEnum.CANCELED.getStatus().equals(record.getStatus()))
|
.toList();
|
}
|
List<QrLabelInfo> labels = business ? assembler.buildBatchBusinessLabels(batch, pallets)
|
: assembler.buildBatchLabels(batch, pallets);
|
String zipName = business ? "批次业务二维码-" + batch.getBusinessCode()
|
: "批次二维码-" + StrUtil.blankToDefault(batch.getCode(), batch.getTempCode());
|
return fixedSource(labels, zipName);
|
}
|
|
/**
|
* 由已装配好的标签列表构造数据源（适用于总量有上界的场景）
|
*/
|
private MesProQrLabelSource fixedSource(List<QrLabelInfo> labels, String zipName) {
|
return new MesProQrLabelSource() {
|
|
@Override
|
public long count() {
|
return labels.size();
|
}
|
|
@Override
|
public List<QrLabelInfo> fetchPage(int pageNo, int pageSize) {
|
int from = (pageNo - 1) * pageSize;
|
if (from >= labels.size()) {
|
return List.of();
|
}
|
return labels.subList(from, Math.min(from + pageSize, labels.size()));
|
}
|
|
@Override
|
public String zipName() {
|
return zipName;
|
}
|
};
|
}
|
|
}
|
