From 33597959e263703dacb41ee0fe28487f7845b9e7 Mon Sep 17 00:00:00 2001
From: yangsb <there-y@qq.com>
Date: Sun, 7 Apr 2024 17:00:30 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=A0=87=E7=AD=BE=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E7=BA=A6=E6=9D=9F=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module/system/enums/ErrorCodeConstants.java     |  1 +
 .../system/service/lable/LabelServiceImpl.java      | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/cf-module-system/cf-module-system-api/src/main/java/com/cf/imes/module/system/enums/ErrorCodeConstants.java b/cf-module-system/cf-module-system-api/src/main/java/com/cf/imes/module/system/enums/ErrorCodeConstants.java
index 7ca05a731..1390fd524 100644
--- a/cf-module-system/cf-module-system-api/src/main/java/com/cf/imes/module/system/enums/ErrorCodeConstants.java
+++ b/cf-module-system/cf-module-system-api/src/main/java/com/cf/imes/module/system/enums/ErrorCodeConstants.java
@@ -189,6 +189,7 @@ public interface ErrorCodeConstants {
     ErrorCode LABEL_TEMPLATE_NOT_EXISTS = new ErrorCode(1_002_300_000, "标签模板不存在");
     ErrorCode LABEL_NOT_EXISTS = new ErrorCode(1_002_300_001, "标签不存在");
     ErrorCode DEFAULT_LABEL_NOT_EXISTS = new ErrorCode(1_002_300_002, "该类型标签默认模板不存在");
+    ErrorCode MACHINE_USE_LABEL = new ErrorCode(1_002_300_003, "无法删除已有机台使用标签");
 
     // ========== 系统数据源 1_002_301_000 ==========
     ErrorCode DATA_SOURCE_NOT_EXISTS = new ErrorCode(1_002_301_000, "系统数据源不存在");
diff --git a/cf-module-system/cf-module-system-biz/src/main/java/com/cf/imes/module/system/service/lable/LabelServiceImpl.java b/cf-module-system/cf-module-system-biz/src/main/java/com/cf/imes/module/system/service/lable/LabelServiceImpl.java
index 2f4368caa..e1410df03 100644
--- a/cf-module-system/cf-module-system-biz/src/main/java/com/cf/imes/module/system/service/lable/LabelServiceImpl.java
+++ b/cf-module-system/cf-module-system-biz/src/main/java/com/cf/imes/module/system/service/lable/LabelServiceImpl.java
@@ -1,11 +1,14 @@
 package com.cf.imes.module.system.service.lable;
 
+import cn.hutool.core.collection.CollectionUtil;
 import com.cf.imes.framework.common.pojo.PageResult;
 import com.cf.imes.framework.common.util.object.BeanUtils;
 import com.cf.imes.framework.mybatis.core.query.LambdaQueryWrapperX;
 import com.cf.imes.module.system.controller.admin.label.vo.*;
 import com.cf.imes.module.system.dal.dataobject.lable.LabelDO;
+import com.cf.imes.module.system.dal.dataobject.machine.MachineDO;
 import com.cf.imes.module.system.dal.mysql.label.LabelMapper;
+import com.cf.imes.module.system.dal.mysql.machine.MachineMapper;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -15,6 +18,7 @@ import java.util.*;
 import java.util.stream.Collectors;
 import static com.cf.imes.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static com.cf.imes.module.system.enums.ErrorCodeConstants.LABEL_NOT_EXISTS;
+import static com.cf.imes.module.system.enums.ErrorCodeConstants.MACHINE_USE_LABEL;
 
 /**
  * 标签模板 Service 实现类
@@ -29,6 +33,8 @@ public class LabelServiceImpl implements LabelService {
     @Resource
     private LabelMapper labelMapper;
 
+    @Resource
+    private MachineMapper machineMapper;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -58,6 +64,13 @@ public class LabelServiceImpl implements LabelService {
         if (LabelDO == null) {
             throw exception(LABEL_NOT_EXISTS);
         }
+        List<MachineDO> machineDOS = machineMapper.selectList(new LambdaQueryWrapperX<MachineDO>()
+                        .select(MachineDO::getId)
+                        .eq(MachineDO::getLabelId, id)
+        );
+        if(CollectionUtil.isNotEmpty(machineDOS)) {
+            throw exception(MACHINE_USE_LABEL);
+        }
         // 删除标签模板表
         labelMapper.deleteById(id);