From 258e65fd1b6799d7e9b55f11f08efb33bb009166 Mon Sep 17 00:00:00 2001
From: chenhao <852066789@qq.com>
Date: Fri, 5 Sep 2025 09:37:52 +0800
Subject: [PATCH] =?UTF-8?q?feat(nex-be):=20=E6=9B=B4=E6=96=B0=E9=95=9C?=
 =?UTF-8?q?=E5=83=8F=E7=B1=BB=E5=9E=8B=E5=B9=B6=E6=B7=BB=E5=8A=A0=E9=80=9A?=
 =?UTF-8?q?=E7=9F=A5=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 修改 Linux 镜像类型为 qcow2，Windows 镜像类型为 vhd
- 在设备镜像映射新增操作中添加发送通知的功能
---
 .../controller/DeviceImageMappingController.java     | 12 ++++++++++--
 .../impl/ImageVirtualMachinesServiceImpl.java        |  2 +-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/nex-be/src/main/java/com/unisinsight/project/controller/DeviceImageMappingController.java b/nex-be/src/main/java/com/unisinsight/project/controller/DeviceImageMappingController.java
index 4694bbb..da3145e 100644
--- a/nex-be/src/main/java/com/unisinsight/project/controller/DeviceImageMappingController.java
+++ b/nex-be/src/main/java/com/unisinsight/project/controller/DeviceImageMappingController.java
@@ -5,16 +5,20 @@ import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.json.JSONUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.unisinsight.project.entity.dao.DeviceImageMapping;
+import com.unisinsight.project.entity.enums.GrpcTypeEnum;
 import com.unisinsight.project.entity.req.DeviceImageMappingReq;
 import com.unisinsight.project.entity.res.DeviceImageMappingRes;
 import com.unisinsight.project.entity.res.ListReq;
 import com.unisinsight.project.exception.BaseErrorCode;
 import com.unisinsight.project.exception.Result;
+import com.unisinsight.project.grpc.generate.NotificationMessage;
+import com.unisinsight.project.grpc.service.SendNotificationService;
 import com.unisinsight.project.service.DeviceImageMappingService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ObjectUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -39,7 +43,8 @@ public class DeviceImageMappingController {
 
     @Resource
     private DeviceImageMappingService deviceImageMappingService;
-
+    @Autowired
+    private SendNotificationService notificationService;
     @ApiOperation(value = "终端镜像映射新增")
     @PostMapping("/add")
     public Result<?> insert(@RequestBody ListReq<DeviceImageMappingReq> deviceImageMappingReq) {
@@ -95,7 +100,10 @@ public class DeviceImageMappingController {
                 return Result.errorResult(BaseErrorCode.HTTP_ERROR_CODE_500);
             }
         }
-
+        //发送消息
+        notificationService.sendNotification(deviceImageMappingReq.getDeviceId(), NotificationMessage.newBuilder()
+                .setType(GrpcTypeEnum.IMAGE_UPDATE.getType())
+                .setContent(GrpcTypeEnum.IMAGE_UPDATE.getDesc()).build());
         return Result.successResult();
     }
 
diff --git a/nex-be/src/main/java/com/unisinsight/project/service/impl/ImageVirtualMachinesServiceImpl.java b/nex-be/src/main/java/com/unisinsight/project/service/impl/ImageVirtualMachinesServiceImpl.java
index 788bcb7..7e91e1c 100644
--- a/nex-be/src/main/java/com/unisinsight/project/service/impl/ImageVirtualMachinesServiceImpl.java
+++ b/nex-be/src/main/java/com/unisinsight/project/service/impl/ImageVirtualMachinesServiceImpl.java
@@ -298,7 +298,7 @@ public class ImageVirtualMachinesServiceImpl extends ServiceImpl<ImageVirtualMac
 
         String diskPath = vmInfo.getDiskPath();
 
-        String type = "linux".equalsIgnoreCase(imageVirtualMachines.getOsVersion()) ? "vhd" : "qcow2";
+        String type = "linux".equalsIgnoreCase(imageVirtualMachines.getOsVersion()) ? "qcow2" : "vhd";
 
         // 根据虚拟机信息调用远程虚拟机信息
         String fileName = imageVirtualMachines.getOsVersion() + "_desktop_" + System.currentTimeMillis();