From 58bd9860909a973a4a11e982ea40443e7a08f39a Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Wed, 23 Apr 2025 14:22:21 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9B=BE=E7=89=87=E5=88=86=E5=89=B2=EF=BC=8C?=
 =?UTF-8?q?=E5=8C=BA=E5=88=86=E7=B1=BB=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/da/service/impl/CollectionElementServiceImpl.java    | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/ai/da/service/impl/CollectionElementServiceImpl.java b/src/main/java/com/ai/da/service/impl/CollectionElementServiceImpl.java
index a9ad15da..575bb48c 100644
--- a/src/main/java/com/ai/da/service/impl/CollectionElementServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/CollectionElementServiceImpl.java
@@ -14,7 +14,6 @@ import com.ai.da.mapper.primary.entity.*;
 import com.ai.da.model.dto.*;
 import com.ai.da.model.enums.ModelType;
 import com.ai.da.model.enums.Sex;
-import com.ai.da.model.enums.StyleEnum;
 import com.ai.da.model.vo.*;
 import com.ai.da.python.PythonService;
 import com.ai.da.python.vo.DesignPythonItem;
@@ -31,7 +30,6 @@ import com.google.gson.Gson;
 import io.netty.util.internal.StringUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.BeanUtils;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -970,7 +968,7 @@ public class CollectionElementServiceImpl extends ServiceImpl<CollectionElementM
             String objectName = accountId + "/ImageSegment/input";
             for (MultipartFile file : files) {
                 String md5 = MD5Utils.encryptFile(file);
-                String segmentedResult = redisUtil.getFromString(RedisUtil.IMAGE_SEGMENTATION + md5);
+                String segmentedResult = redisUtil.getFromString(RedisUtil.IMAGE_SEGMENTATION + type + ":" + md5);
                 // 判断上传的图片是否有已分割的数据
                 if (StringUtil.isNullOrEmpty(segmentedResult)) {
                     String path = minioUtil.upload(userBucketName, objectName, file);
@@ -1012,7 +1010,7 @@ public class CollectionElementServiceImpl extends ServiceImpl<CollectionElementM
             for (ImageSegmentation.ImageDate imageData : segmented) {
                 if (isUploadMode) {
                     // 上传的图片需要添加到redis中存一天
-                    String key = RedisUtil.IMAGE_SEGMENTATION +
+                    String key = RedisUtil.IMAGE_SEGMENTATION + type + ":" +
                             MD5Utils.encryptFile
                                     (minioUtil.getPreSignedUrl
                                             (imageData.getImage_url(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME), false);