From b6096b76d28d2d47596194b85c39835a822e40af Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Wed, 8 May 2024 15:10:57 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=A2=9E=E5=8A=A0=E3=80=91ai=20image?=
 =?UTF-8?q?=20=E7=94=9F=E6=88=90=E8=AE=B0=E5=BD=95=E5=88=A0=E9=99=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/admin/image/AiImageController.java  |  2 ++
 .../image/vo/AiImageMidjourneyOperateReqVO.java    |  2 +-
 .../yudao/module/ai/service/AiImageService.java    |  7 +++++++
 .../module/ai/service/impl/AiImageServiceImpl.java | 14 +++++++++++---
 4 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
index 7f0620ab2..7df2592f9 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
@@ -59,12 +59,14 @@ public class AiImageController {
     @Operation(summary = "取消 midjourney 绘画", description = "取消 midjourney 绘画")
     @PostMapping("/cancel-midjourney")
     public CommonResult<Void> cancelMidjourney(@RequestParam("id") Long id) {
+        // @范 这里实现mj取消逻辑
         return success(null);
     }
 
     @Operation(summary = "删除绘画记录", description = "")
     @DeleteMapping("/delete")
     public CommonResult<Void> delete(@RequestParam("id") Long id) {
+        aiImageService.delete(id);
         return success(null);
     }
 }
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageMidjourneyOperateReqVO.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageMidjourneyOperateReqVO.java
index b49530425..994c1c5f0 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageMidjourneyOperateReqVO.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageMidjourneyOperateReqVO.java
@@ -19,7 +19,7 @@ public class AiImageMidjourneyOperateReqVO {
 
     @NotNull(message = "图片编号不能为空")
     @Schema(description = "编号")
-    private String id;
+    private Long id;
 
     @NotNull(message = "消息编号不能为空")
     @Schema(description = "消息编号")
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/AiImageService.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/AiImageService.java
index 33c4bf449..b58201056 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/AiImageService.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/AiImageService.java
@@ -44,4 +44,11 @@ public interface AiImageService {
      */
     void midjourneyOperate(AiImageMidjourneyOperateReqVO req);
 
+    /**
+     * 删除 - image 记录
+     *
+     * @param id
+     */
+    void delete(Long id);
+
 }
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/AiImageServiceImpl.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/AiImageServiceImpl.java
index c3fb4454d..e0d797d18 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/AiImageServiceImpl.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/AiImageServiceImpl.java
@@ -137,7 +137,7 @@ public class AiImageServiceImpl implements AiImageService {
     @Override
     public void midjourneyOperate(AiImageMidjourneyOperateReqVO req) {
         // 校验是否存在
-        AiImageDO aiImageDO = validateExists(req);
+        AiImageDO aiImageDO = validateExists(req.getId());
         // 获取 midjourneyOperations
         List<AiImageMidjourneyOperationsVO> midjourneyOperations = getMidjourneyOperations(aiImageDO);
         // 校验 OperateId 是否存在
@@ -158,6 +158,14 @@ public class AiImageServiceImpl implements AiImageService {
         );
     }
 
+    @Override
+    public void delete(Long id) {
+        // 校验记录是否存在
+        AiImageDO aiImageDO = validateExists(id);
+        // 删除记录
+        aiImageMapper.deleteById(id);
+    }
+
     private void validateMessageId(String mjMessageId, String messageId) {
         if (!mjMessageId.equals(messageId)) {
             throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MIDJOURNEY_MESSAGE_ID_INCORRECT);
@@ -181,8 +189,8 @@ public class AiImageServiceImpl implements AiImageService {
         return JsonUtils.parseArray(aiImageDO.getMjOperations(), AiImageMidjourneyOperationsVO.class);
     }
 
-    private AiImageDO validateExists(AiImageMidjourneyOperateReqVO req) {
-        AiImageDO aiImageDO = aiImageMapper.selectById(req.getId());
+    private AiImageDO validateExists(Long id) {
+        AiImageDO aiImageDO = aiImageMapper.selectById(id);
         if (aiImageDO == null) {
             throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MIDJOURNEY_IMAGINE_FAIL);
         }