From e1b0603648fc0e33645beac889403a95b89e90dd Mon Sep 17 00:00:00 2001
From: Mzaxd <caihaohan0712@foxmail.com>
Date: Tue, 21 Mar 2023 23:16:33 +0800
Subject: [PATCH] =?UTF-8?q?[Fix]=E4=BF=AE=E5=A4=8D=E2=80=9C=E5=88=A0?=
 =?UTF-8?q?=E9=99=A4=E6=9C=89=E6=8E=89=E7=BA=BF=E6=8F=90=E9=86=92=E7=9A=84?=
 =?UTF-8?q?=E5=AE=9E=E4=BE=8B=E6=97=B6=E4=BC=9A=E5=AF=BC=E8=87=B4=E6=89=80?=
 =?UTF-8?q?=E6=9C=89=E6=8F=90=E9=86=92=E9=83=BD=E6=97=A0=E6=B3=95=E6=98=BE?=
 =?UTF-8?q?=E7=A4=BA=E2=80=9D=E7=9A=84Bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ContainerServiceImpl.java    | 12 +++++++++
 .../service/impl/HostMachineServiceImpl.java  | 27 ++++++++++++++-----
 2 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/mzaxd/noodles/service/impl/ContainerServiceImpl.java b/src/main/java/com/mzaxd/noodles/service/impl/ContainerServiceImpl.java
index 0b4bdfd..b9fd191 100644
--- a/src/main/java/com/mzaxd/noodles/service/impl/ContainerServiceImpl.java
+++ b/src/main/java/com/mzaxd/noodles/service/impl/ContainerServiceImpl.java
@@ -11,12 +11,14 @@ import com.mzaxd.noodles.domain.entity.Container;
 import com.mzaxd.noodles.domain.vo.*;
 import com.mzaxd.noodles.enums.AppHttpCodeEnum;
 import com.mzaxd.noodles.mapper.HostMachineMapper;
+import com.mzaxd.noodles.mapper.NotificationMapper;
 import com.mzaxd.noodles.service.ContainerService;
 import com.mzaxd.noodles.mapper.ContainerMapper;
 import com.mzaxd.noodles.service.HostMachineService;
 import com.mzaxd.noodles.service.SshLinkService;
 import com.mzaxd.noodles.util.BeanCopyUtils;
 import com.mzaxd.noodles.util.SshLinkUtil;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 import org.springframework.util.StringUtils;
 
@@ -32,6 +34,7 @@ import java.util.Objects;
  * @createDate 2023-02-02 06:26:00
  */
 @Service
+@Slf4j
 public class ContainerServiceImpl extends ServiceImpl<ContainerMapper, Container>
         implements ContainerService {
 
@@ -47,6 +50,9 @@ public class ContainerServiceImpl extends ServiceImpl<ContainerMapper, Container
     @Resource
     private SshLinkService sshLinkService;
 
+    @Resource
+    private NotificationMapper notificationMapper;
+
     @Override
     public ResponseResult containerListSummaryStatistics() {
         //设置查询所有在线vm的条件
@@ -109,6 +115,12 @@ public class ContainerServiceImpl extends ServiceImpl<ContainerMapper, Container
     @Override
     public ResponseResult deleteContainerById(Integer id) {
         if (removeById(id)) {
+            //删除掉线提醒
+            LambdaQueryWrapper<Notification> notificationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+            notificationLambdaQueryWrapper.eq(Notification::getInstanceId, id);
+            notificationLambdaQueryWrapper.eq(Notification::getInstanceType, SystemConstant.INSTANCETYPE_CONTAINER);
+            notificationMapper.delete(notificationLambdaQueryWrapper);
+            log.info("删除id为{}的虚拟机的所有掉线提醒", id);
             return ResponseResult.okResult("删除成功");
         }
         return ResponseResult.errorResult(AppHttpCodeEnum.SYSTEM_ERROR);
diff --git a/src/main/java/com/mzaxd/noodles/service/impl/HostMachineServiceImpl.java b/src/main/java/com/mzaxd/noodles/service/impl/HostMachineServiceImpl.java
index 806d644..bb79bf4 100644
--- a/src/main/java/com/mzaxd/noodles/service/impl/HostMachineServiceImpl.java
+++ b/src/main/java/com/mzaxd/noodles/service/impl/HostMachineServiceImpl.java
@@ -13,10 +13,7 @@ import com.mzaxd.noodles.domain.message.Server;
 import com.mzaxd.noodles.domain.vo.*;
 import com.mzaxd.noodles.enums.AppHttpCodeEnum;
 import com.mzaxd.noodles.exception.SystemException;
-import com.mzaxd.noodles.mapper.HostDetectorMapper;
-import com.mzaxd.noodles.mapper.HostMachineMapper;
-import com.mzaxd.noodles.mapper.OsMapper;
-import com.mzaxd.noodles.mapper.SshLinkMapper;
+import com.mzaxd.noodles.mapper.*;
 import com.mzaxd.noodles.service.*;
 import com.mzaxd.noodles.util.BeanCopyUtils;
 import com.mzaxd.noodles.util.SshLinkUtil;
@@ -74,6 +71,9 @@ public class HostMachineServiceImpl extends ServiceImpl<HostMachineMapper, HostM
     @Resource
     private SshLinkMapper sshLinkMapper;
 
+    @Resource
+    private NotificationMapper notificationMapper;
+
     @Override
     public ResponseResult getHostDrawer() {
         LambdaQueryWrapper<HostMachine> wrapper = new LambdaQueryWrapper<>();
@@ -340,8 +340,16 @@ public class HostMachineServiceImpl extends ServiceImpl<HostMachineMapper, HostM
             log.info("删除hostId为{}的host信息", id);
             osMapper.deleteById(hostMachine.getOsId());
         }
-        log.info("删除id为{}的host", id);
+        //删除掉线提醒
+        LambdaQueryWrapper<Notification> notificationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        notificationLambdaQueryWrapper.eq(Notification::getInstanceId, id);
+        notificationLambdaQueryWrapper.eq(Notification::getInstanceType, SystemConstant.INSTANCETYPE_HOST);
+        notificationMapper.delete(notificationLambdaQueryWrapper);
+        log.info("删除id为{}的物理机的所有掉线提醒", id);
+
+        //删除物理机
         hostMachineMapper.deleteById(id);
+        log.info("删除id为{}的host", id);
         return ResponseResult.okResult(SystemConstant.SUCCESS_CODE, "删除成功");
     }
 
@@ -453,8 +461,15 @@ public class HostMachineServiceImpl extends ServiceImpl<HostMachineMapper, HostM
         if (Objects.nonNull(getById(id).getSshId())) {
             sshLinkMapper.deleteById(getById(id).getSshId());
         }
-        log.info("删除id为{}的vm", id);
+        //删除掉线提醒
+        LambdaQueryWrapper<Notification> notificationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        notificationLambdaQueryWrapper.eq(Notification::getInstanceId, id);
+        notificationLambdaQueryWrapper.eq(Notification::getInstanceType, SystemConstant.INSTANCETYPE_VM);
+        notificationMapper.delete(notificationLambdaQueryWrapper);
+        log.info("删除id为{}的虚拟机的所有掉线提醒", id);
+
         hostMachineMapper.deleteById(id);
+        log.info("删除id为{}的vm", id);
         return ResponseResult.okResult(SystemConstant.SUCCESS_CODE, "删除成功");
     }