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) { @@ -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