package cn.iocoder.yudao.module.mes.service.wm.barcode;
|
|
import cn.hutool.core.codec.Base64;
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.ObjUtil;
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.mes.controller.admin.wm.barcode.vo.MesWmBarcodePageReqVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.wm.barcode.vo.MesWmBarcodeSaveReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.barcode.MesWmBarcodeConfigDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.barcode.MesWmBarcodeDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.wm.barcode.MesWmBarcodeMapper;
|
import cn.iocoder.yudao.module.mes.enums.wm.BarcodeFormatEnum;
|
import com.google.zxing.BarcodeFormat;
|
import com.google.zxing.MultiFormatWriter;
|
import com.google.zxing.client.j2se.MatrixToImageWriter;
|
import com.google.zxing.common.BitMatrix;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.validation.annotation.Validated;
|
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.ByteArrayOutputStream;
|
import java.util.ArrayList;
|
import java.util.Collection;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
|
|
/**
|
* MES 条码清单 Service 实现类
|
*
|
* @author 超级管理员
|
*/
|
@Service
|
@Validated
|
@Slf4j
|
public class MesWmBarcodeServiceImpl implements MesWmBarcodeService {
|
|
/**
|
* 内容格式模板中的占位符识别正则
|
*/
|
private static final Pattern PLACEHOLDER_PATTERN = Pattern.compile("\\{[A-Za-z0-9_]+\\}");
|
|
@Resource
|
private MesWmBarcodeMapper barcodeMapper;
|
|
@Resource
|
private MesWmBarcodeConfigService barcodeConfigService;
|
|
@Override
|
public Long createBarcode(MesWmBarcodeSaveReqVO createReqVO) {
|
// 1. 校验数据
|
MesWmBarcodeConfigDO config = validateBarcodeSaveData(createReqVO.getId(),
|
createReqVO.getBizType(), createReqVO.getBizId());
|
|
// 2. 生成条码内容，并校验唯一性
|
String content = generateAndValidateContent(createReqVO.getId(),
|
createReqVO.getContent(), config.getContentFormat(), createReqVO.getBizCode(),
|
createReqVO.getBizName());
|
|
// 3. 保存条码记录
|
MesWmBarcodeDO barcode = BeanUtils.toBean(createReqVO, MesWmBarcodeDO.class).setContent(content)
|
.setConfigId(config.getId()).setFormat(config.getFormat());
|
barcodeMapper.insert(barcode);
|
return barcode.getId();
|
}
|
|
@Override
|
public String generateBarcodeContent(Integer bizType, String bizCode, String bizName) {
|
// 1.1 校验参数
|
if (bizType == null) {
|
throw exception(BARCODE_BIZ_TYPE_NOT_EXISTS);
|
}
|
if (StrUtil.isBlank(bizCode)) {
|
throw exception(BARCODE_BIZ_CODE_NOT_EXISTS);
|
}
|
// 1.2 查询条码配置
|
MesWmBarcodeConfigDO config = barcodeConfigService.validateBarcodeConfigByBizType(bizType);
|
|
// 2. 生成条码内容
|
return generateBarcodeContent(config.getContentFormat(), bizCode, bizName);
|
}
|
|
@Override
|
public void updateBarcode(MesWmBarcodeSaveReqVO updateReqVO) {
|
// 1.1 校验存在
|
validateBarcodeExists(updateReqVO.getId());
|
// 1.2 校验数据
|
MesWmBarcodeConfigDO config = validateBarcodeSaveData(updateReqVO.getId(),
|
updateReqVO.getBizType(), updateReqVO.getBizId());
|
|
// 2. 生成条码内容，并校验唯一性
|
String content = generateAndValidateContent(updateReqVO.getId(),
|
updateReqVO.getContent(), config.getContentFormat(), updateReqVO.getBizCode(),
|
updateReqVO.getBizName());
|
|
// 3. 更新（刷新 configId 和 format，确保与最新配置一致）
|
MesWmBarcodeDO updateObj = BeanUtils.toBean(updateReqVO, MesWmBarcodeDO.class)
|
.setConfigId(config.getId()).setFormat(config.getFormat());
|
if (StrUtil.isNotBlank(content)) {
|
updateObj.setContent(content);
|
}
|
barcodeMapper.updateById(updateObj);
|
}
|
|
/**
|
* 校验条码保存数据：条码配置 + 业务对象唯一性
|
*
|
* @param id 条码编号（新增时为 null）
|
* @param bizType 业务类型
|
* @param bizId 业务编号
|
* @return 条码配置
|
*/
|
private MesWmBarcodeConfigDO validateBarcodeSaveData(Long id, Integer bizType, Long bizId) {
|
// 1. 校验条码配置
|
MesWmBarcodeConfigDO config = barcodeConfigService.validateBarcodeConfigByBizType(bizType);
|
// 2. 校验业务对象唯一性
|
validateBarcodeUnique(id, bizType, bizId);
|
return config;
|
}
|
|
/**
|
* 校验业务对象（bizType + bizId）唯一性
|
*
|
* @param id 条码编号（新增时为 null，更新时排除自身）
|
* @param bizType 业务类型
|
* @param bizId 业务编号
|
*/
|
private void validateBarcodeUnique(Long id, Integer bizType, Long bizId) {
|
MesWmBarcodeDO barcode = barcodeMapper.selectByBizTypeAndBizId(bizType, bizId);
|
if (barcode == null) {
|
return;
|
}
|
if (ObjUtil.notEqual(barcode, id)) {
|
throw exception(WM_BARCODE_ALREADY_EXISTS);
|
}
|
}
|
|
/**
|
* 生成条码内容（如果前端未传递则自动生成），并校验内容唯一性
|
*
|
* @param id 条码编号（新增时为 null，更新时排除自身）
|
* @param content 前端传递的条码内容（可为空）
|
* @param contentFormat 内容格式模板
|
* @param bizCode 业务编码
|
* @return 最终条码内容
|
*/
|
private String generateAndValidateContent(Long id, String content, String contentFormat, String bizCode, String bizName) {
|
// 1. 如果前端未传递内容，则自动生成
|
if (StrUtil.isBlank(content)) {
|
content = generateBarcodeContent(contentFormat, bizCode, bizName);
|
}
|
// 2. 校验条码内容唯一性（排除自身）
|
if (StrUtil.isNotBlank(content)) {
|
MesWmBarcodeDO contentBarcode = barcodeMapper.selectByContent(content);
|
if (contentBarcode != null && ObjUtil.notEqual(contentBarcode.getId(), id)) {
|
throw exception(WM_BARCODE_CONTENT_DUPLICATE);
|
}
|
}
|
return content;
|
}
|
|
/**
|
* 生成条码内容
|
*
|
* @param contentFormat 内容格式模板
|
* @param bizCode 业务编码
|
* @param bizName 业务名称
|
* @return 条码内容
|
*/
|
private String generateBarcodeContent(String contentFormat, String bizCode, String bizName) {
|
if (StrUtil.isBlank(contentFormat)) {
|
return bizCode;
|
}
|
// 1. 替换已支持的占位符
|
String content = contentFormat
|
.replace(MesWmBarcodeConfigDO.PLACEHOLDER_BUSINESS_CODE, StrUtil.nullToEmpty(bizCode))
|
.replace(MesWmBarcodeConfigDO.PLACEHOLDER_BUSINESS_NAME, StrUtil.nullToEmpty(bizName));
|
// 2. 清空模板中其它未填充的未知占位符，避免 {XXX} 残留影响扫码
|
return PLACEHOLDER_PATTERN.matcher(content).replaceAll("");
|
}
|
|
@Override
|
public String renderBarcodeImage(String content, Integer format, Integer width, Integer height) {
|
if (StrUtil.isBlank(content)) {
|
throw exception(BARCODE_CONTENT_EMPTY);
|
}
|
int w = width != null && width > 0 ? width : 200;
|
int h = height != null && height > 0 ? height : 200;
|
try {
|
BitMatrix bitMatrix = new MultiFormatWriter().encode(content, toZxingFormat(format), w, h);
|
BufferedImage image = MatrixToImageWriter.toBufferedImage(bitMatrix);
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
ImageIO.write(image, "png", out);
|
return "data:image/png;base64," + Base64.encode(out.toByteArray());
|
} catch (Exception e) {
|
log.error("[renderBarcodeImage] 条码渲染失败，content={}, format={}", content, format, e);
|
throw new RuntimeException("条码渲染失败");
|
}
|
}
|
|
/**
|
* 将项目条码格式枚举值映射为 zxing 条码格式
|
*
|
* @param format 项目条码格式 {@link BarcodeFormatEnum}
|
* @return zxing 条码格式
|
*/
|
private BarcodeFormat toZxingFormat(Integer format) {
|
if (format == null) {
|
return BarcodeFormat.CODE_128;
|
}
|
switch (format) {
|
case 1: // BarcodeFormatEnum.QR_CODE
|
return BarcodeFormat.QR_CODE;
|
case 2: // BarcodeFormatEnum.EAN13
|
return BarcodeFormat.EAN_13;
|
case 3: // BarcodeFormatEnum.CODE39
|
return BarcodeFormat.CODE_39;
|
case 4: // BarcodeFormatEnum.UPC_A
|
return BarcodeFormat.UPC_A;
|
default:
|
throw exception(BARCODE_FORMAT_INVALID);
|
}
|
}
|
|
@Override
|
public Map<String, Object> parseBarcodeContent(String content) {
|
if (StrUtil.isBlank(content)) {
|
throw exception(BARCODE_CONTENT_EMPTY);
|
}
|
// 1. 优先按条码内容精确命中已有条码记录
|
MesWmBarcodeDO barcode = barcodeMapper.selectByContent(content);
|
if (barcode != null) {
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("matched", true);
|
result.put("barcodeId", barcode.getId());
|
result.put("configId", barcode.getConfigId());
|
result.put("format", barcode.getFormat());
|
result.put("bizType", barcode.getBizType());
|
result.put("bizId", barcode.getBizId());
|
result.put("bizCode", barcode.getBizCode());
|
result.put("bizName", barcode.getBizName());
|
result.put("content", barcode.getContent());
|
return result;
|
}
|
// 2. 未命中时，遍历启用状态的配置按内容格式模板反解
|
List<MesWmBarcodeConfigDO> configs = barcodeConfigService.getEnableBarcodeConfigList();
|
for (MesWmBarcodeConfigDO config : configs) {
|
Map<String, Object> fields = matchByTemplate(config.getContentFormat(), content);
|
if (fields != null) {
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("matched", false);
|
result.put("configId", config.getId());
|
result.put("format", config.getFormat());
|
result.put("bizType", config.getBizType());
|
result.put("bizCode", fields.get(MesWmBarcodeConfigDO.PLACEHOLDER_BUSINESS_CODE));
|
result.put("bizName", fields.get(MesWmBarcodeConfigDO.PLACEHOLDER_BUSINESS_NAME));
|
result.put("fields", fields);
|
return result;
|
}
|
}
|
// 3. 均未命中
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("matched", false);
|
result.put("content", content);
|
return result;
|
}
|
|
/**
|
* 按内容格式模板反解条码内容
|
*
|
* @param contentFormat 内容格式模板
|
* @param content 待解析的条码内容
|
* @return 占位符 -> 实际值的映射；不匹配返回 null
|
*/
|
private Map<String, Object> matchByTemplate(String contentFormat, String content) {
|
if (StrUtil.isBlank(contentFormat) || !contentFormat.contains("{")) {
|
return null;
|
}
|
// 1. 将模板转为正则：字面量转义，占位符转为捕获组
|
List<String> placeholders = new ArrayList<>();
|
StringBuilder regexSb = new StringBuilder();
|
Matcher matcher = PLACEHOLDER_PATTERN.matcher(contentFormat);
|
int lastEnd = 0;
|
while (matcher.find()) {
|
regexSb.append(Pattern.quote(contentFormat.substring(lastEnd, matcher.start())));
|
placeholders.add(matcher.group());
|
regexSb.append("(.+?)");
|
lastEnd = matcher.end();
|
}
|
if (placeholders.isEmpty()) {
|
return null;
|
}
|
regexSb.append(Pattern.quote(contentFormat.substring(lastEnd)));
|
|
// 2. 全量匹配并提取各占位符值
|
Matcher contentMatcher = Pattern.compile(regexSb.toString()).matcher(content);
|
if (!contentMatcher.matches()) {
|
return null;
|
}
|
Map<String, Object> fields = new LinkedHashMap<>();
|
for (int i = 0; i < placeholders.size(); i++) {
|
fields.put(placeholders.get(i), contentMatcher.group(i + 1));
|
}
|
return fields;
|
}
|
|
@Override
|
public void deleteBarcode(Long id) {
|
// 校验存在
|
validateBarcodeExists(id);
|
// 删除
|
barcodeMapper.deleteById(id);
|
}
|
|
@Override
|
public void deleteBarcodeByBizTypeAndBizIds(Integer bizType, Collection<Long> bizIds) {
|
if (CollUtil.isEmpty(bizIds)) {
|
return;
|
}
|
barcodeMapper.deleteByBizTypeAndBizIds(bizType, bizIds);
|
}
|
|
private MesWmBarcodeDO validateBarcodeExists(Long id) {
|
MesWmBarcodeDO barcode = barcodeMapper.selectById(id);
|
if (barcode == null) {
|
throw exception(WM_BARCODE_NOT_EXISTS);
|
}
|
return barcode;
|
}
|
|
@Override
|
public MesWmBarcodeDO getBarcode(Long id) {
|
return barcodeMapper.selectById(id);
|
}
|
|
@Override
|
public PageResult<MesWmBarcodeDO> getBarcodePage(MesWmBarcodePageReqVO pageReqVO) {
|
return barcodeMapper.selectPage(pageReqVO);
|
}
|
|
@Override
|
public MesWmBarcodeDO getBarcodeByBizTypeAndBizId(Integer bizType, Long bizId) {
|
return barcodeMapper.selectByBizTypeAndBizId(bizType, bizId);
|
}
|
|
@Override
|
public void autoGenerateBarcode(Integer bizType, Long bizId, String bizCode, String bizName) {
|
// 1.1 检查是否配置自动生成
|
MesWmBarcodeConfigDO config = barcodeConfigService.getBarcodeConfigByBizType(bizType);
|
if (config == null || Boolean.FALSE.equals(config.getAutoGenerateFlag())
|
|| CommonStatusEnum.isDisable(config.getStatus())) {
|
return;
|
}
|
// 1.2 检查是否已存在条码
|
MesWmBarcodeDO existBarcode = barcodeMapper.selectByBizTypeAndBizId(bizType, bizId);
|
if (existBarcode != null) {
|
return;
|
}
|
|
// 2. 创建条码记录
|
MesWmBarcodeSaveReqVO createReqVO = new MesWmBarcodeSaveReqVO()
|
.setBizType(bizType).setBizId(bizId).setBizCode(bizCode).setBizName(bizName)
|
.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
createBarcode(createReqVO);
|
}
|
|
@Override
|
public long getBarcodeCountByConfigId(Long configId) {
|
return barcodeMapper.selectCountByConfigId(configId);
|
}
|
|
}
|
