From b73080a8f4692ae8fe3b8d54c35a8b5185e3371d Mon Sep 17 00:00:00 2001
From: huangfeng <huangfeng@xinyingpower.com>
Date: Thu, 4 Jan 2024 16:44:10 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=A2=9E=E5=8A=A0=E7=B1=BB=E5=9E=8B?=
 =?UTF-8?q?=E6=A0=A1=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/xydl/cac/service/impl/ZsbServiceImpl.java   | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/main/java/com/xydl/cac/service/impl/ZsbServiceImpl.java b/src/main/java/com/xydl/cac/service/impl/ZsbServiceImpl.java
index ea5f020..6026a07 100644
--- a/src/main/java/com/xydl/cac/service/impl/ZsbServiceImpl.java
+++ b/src/main/java/com/xydl/cac/service/impl/ZsbServiceImpl.java
@@ -1,7 +1,9 @@
 package com.xydl.cac.service.impl;
 
 import com.xydl.cac.entity.Jg;
+import com.xydl.cac.entity.Lx;
 import com.xydl.cac.entity.Zsb;
+import com.xydl.cac.repository.LxRepository;
 import com.xydl.cac.repository.ZsbRepository;
 import com.xydl.cac.service.JgService;
 import com.xydl.cac.service.ZsbService;
@@ -11,6 +13,7 @@ import org.springframework.transaction.annotation.Transactional;
 
 import javax.annotation.Resource;
 import java.util.List;
+import java.util.Optional;
 
 @Service
 @Slf4j
@@ -21,6 +24,8 @@ public class ZsbServiceImpl implements ZsbService {
     ZsbRepository repository;
     @Resource
     JgService jgService;
+    @Resource
+    LxRepository lxRepository;
 
     @Override
     public List<Zsb> listAll(Integer jgid) {
@@ -36,6 +41,10 @@ public class ZsbServiceImpl implements ZsbService {
         item.setId(null);
         Jg jg = jgService.detail(item.getJgid());
         item.setBdzid(jg.getBdzid());
+        Optional<Lx> optional = lxRepository.findById(item.getLxid());
+        if (!optional.isPresent()) {
+            throw new Exception("未找到该设备类型");
+        }
         return repository.save(item);
     }
 
@@ -43,6 +52,10 @@ public class ZsbServiceImpl implements ZsbService {
     public void update(Zsb item) throws Exception {
         Jg jg = jgService.detail(item.getJgid());
         item.setBdzid(jg.getBdzid());
+        Optional<Lx> optional = lxRepository.findById(item.getLxid());
+        if (!optional.isPresent()) {
+            throw new Exception("未找到该设备类型");
+        }
         repository.save(item);
     }