From be8968605b53fce6edac376c5b4658bd3912fb62 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Thu, 4 Jul 2024 16:52:06 +0800
Subject: [PATCH] BUGFIX:aida;

---
 src/main/java/com/ai/da/model/vo/MagicToolResultVO.java       | 4 ++++
 .../java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/src/main/java/com/ai/da/model/vo/MagicToolResultVO.java b/src/main/java/com/ai/da/model/vo/MagicToolResultVO.java
index fb313f1a..1c26db5a 100644
--- a/src/main/java/com/ai/da/model/vo/MagicToolResultVO.java
+++ b/src/main/java/com/ai/da/model/vo/MagicToolResultVO.java
@@ -24,4 +24,8 @@ public class MagicToolResultVO {
     private String sourceUrl;
 
     private String resultType;
+
+    private String elementType;
+
+    private Long elementId;
 }
diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
index 6a178daa..cd8c38bd 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -356,6 +356,8 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
                         throw new BusinessException("The source image does not exist.");
                     }
                     magicToolResultVO.setResultType(toProductImageResult.getResultType());
+                    magicToolResultVO.setElementId(toProductImageResult.getElementId());
+                    magicToolResultVO.setElementType(toProductImageResult.getElementType());
                     if (toProductImageResult.getElementType().equals("ProductElement")) {
                         ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId());
                         magicToolResultVO.setSourceUrl(minioUtil.getPresignedUrl(toProductElement.getUrl(), 24 * 60));
@@ -535,6 +537,8 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
                         throw new BusinessException("The source image does not exist.");
                     }
                     magicToolResultVO.setResultType(toProductImageResult.getResultType());
+                    magicToolResultVO.setElementId(toProductImageResult.getElementId());
+                    magicToolResultVO.setElementType(toProductImageResult.getElementType());
                     if (toProductImageResult.getElementType().equals("ProductElement")) {
                         ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId());
                         magicToolResultVO.setSourceUrl(minioUtil.getPresignedUrl(toProductElement.getUrl(), 24 * 60));