TASK:1、印花允许修改并保存

BUGFIX: 1、批量生成有的类型没有任务开始执行时间，任务结束时间有误
2025-07-10 17:50:31 +08:00
parent 766e75f2ed
commit daa88889d0
5 changed files with 132 additions and 36 deletions
--- a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
@@ -2588,14 +2588,20 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
    public void processDesignBatch(Map<String, Object> designBatchResult) {
        Object progress = designBatchResult.get("progress");
        if (progress instanceof String) {
+            String taskId = (String) designBatchResult.get("task_id");
+            QueryWrapper<CloudTask> qw = new QueryWrapper<>();
+            qw.lambda().eq(CloudTask::getTaskId, taskId);
+            List<CloudTask> cloudTaskList = cloudTaskMapper.selectList(qw);
            if (((String) progress).startsWith("0/")) {
+                if (CollectionUtil.isNotEmpty(cloudTaskList) && Objects.nonNull(cloudTaskList.get(0).getStartTime())) {
+                    CloudTask cloudTask = cloudTaskList.get(0);
+                    cloudTask.setStartTime(LocalDateTime.now());
+                    cloudTask.setUpdateTime(LocalDateTime.now());
+                    cloudTaskMapper.updateById(cloudTask);
+                }
                return;
            }
            if (progress.equals("ok")) {
-                String taskId = (String) designBatchResult.get("task_id");
-                QueryWrapper<CloudTask> qw = new QueryWrapper<>();
-                qw.lambda().eq(CloudTask::getTaskId, taskId);
-                List<CloudTask> cloudTaskList = cloudTaskMapper.selectList(qw);
                if (CollectionUtil.isNotEmpty(cloudTaskList)) {
                    CloudTask cloudTask = cloudTaskList.get(0);
                    cloudTask.setCompletedNum(cloudTask.getNums());
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -1086,45 +1086,101 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
    @Override
    @Transactional(rollbackFor = Exception.class)
    public GenerateResultVO modifySketch(GenerateModifyDTO generateModifyDTO) {
-        log.info("修改提取出的sketch,并加入到library");
+        log.info("修改生成或library中的sketch或print");
+
+        // 提取常用参数
        Long accountId = UserContext.getUserHolder().getId();
        String base64 = generateModifyDTO.getBase64();
        String gender = generateModifyDTO.getGender();
        String category = generateModifyDTO.getCategory();
+        Long originalId = generateModifyDTO.getOriginalId();
+        boolean isOverride = generateModifyDTO.getIsOverride();
+        boolean isFromLibrary = !StringUtil.isNullOrEmpty(generateModifyDTO.getOriginalIdSource())
+                && generateModifyDTO.getOriginalIdSource().equals("Library");
+        boolean isSketch = generateModifyDTO.getType().equals(SKETCH_BOARD.getRealName());

-        // 将base64上传到minio
-        String path;
-        GenerateDetail originalDetail = generateDetailMapper.selectById(generateModifyDTO.getOriginalId());
-        Long generateDetailId = originalDetail.getId();
-        if (generateModifyDTO.getIsOverride()) {
-            path = originalDetail.getUrl().replaceFirst("^[^/]+/", "").replaceFirst("\\.[^.]+$", "");
-        } else {
-            path = accountId + "/sketchboard/" + gender.toLowerCase() + "/" + category + "/" + UUID.randomUUID();
+        // 获取原始路径和可能的generateId
+        PathInfo pathInfo = getOriginalPathAndGenerateId(isFromLibrary, originalId);
+
+        // 确定存储路径
+        String storagePath = isOverride
+                ? pathInfo.originalPath.replaceFirst("^[^/]+/", "").replaceFirst("\\.[^.]+$", "")
+                : isSketch
+                ? accountId + "/sketchboard/" + gender + "/" + category + "/" + UUID.randomUUID()
+                : accountId + "/printboard/" + UUID.randomUUID();
+
+        // 上传到MinIO
+        String minioPath = minioUtil.base64UploadToPath(base64, userBucket, storagePath);
+        log.info("修改后的图片：{}", minioPath);
+
+        // 保存到数据库并返回结果
+        return isFromLibrary
+                ? handleLibrarySave(accountId, originalId, minioPath, category, gender, isOverride, generateModifyDTO.getType())
+                : handleGenerateSave(originalId, pathInfo.generateId, minioPath, category, isOverride);
+    }
+
+    private static class PathInfo {
+        String originalPath;
+        Long generateId;
+
+        PathInfo(String originalPath, Long generateId) {
+            this.originalPath = originalPath;
+            this.generateId = generateId;
        }
+    }

-        String minioPath = minioUtil.base64UploadToPath(base64, userBucket, path);
+    private PathInfo getOriginalPathAndGenerateId(boolean isFromLibrary, Long originalId) {
+        if (isFromLibrary) {
+            return new PathInfo(libraryService.getById(originalId).getUrl(), null);
+        } else {
+            GenerateDetail detail = generateDetailMapper.selectById(originalId);
+            return new PathInfo(detail.getUrl(), detail.getGenerateId());
+        }
+    }

-        log.info("修改后的图片 ： {}", minioPath);
+    private GenerateResultVO handleLibrarySave(Long accountId, Long libraryId, String minioPath,
+                                               String category, String gender, boolean isOverride, String type) {
+        Library library;
+        if (isOverride) {
+            library = new Library();
+            library.setId(libraryId);
+            library.setUrl(minioPath);
+            library.setUpdateDate(new Date());
+            libraryService.updateById(library);
+        } else {
+            library = new Library(accountId, type, category, gender, minioPath,
+                    MD5Utils.encryptFile(minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME), false),
+                    new Date());
+            libraryService.save(library);
+            libraryId = library.getId();
+        }
+        return buildResultVO(libraryId, minioPath, category);
+    }

-        // 存入db 保存到t_generate_detail
-        if (!generateModifyDTO.getIsOverride()) {
-            GenerateDetail generateDetail = new GenerateDetail();
-            generateDetail.setGenerateId(originalDetail.getGenerateId());
+    private GenerateResultVO handleGenerateSave(Long originalId, Long generateId, String minioPath,
+                                                String category, boolean isOverride) {
+        GenerateDetail generateDetail = new GenerateDetail();
+        if (isOverride) {
+            generateDetail.setId(originalId);
            generateDetail.setUrl(minioPath);
-            generateDetail.setIsLike((byte) 0);
-            generateDetail.setMd5(MD5Utils.encryptFile(minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME, true), Boolean.FALSE));
+            generateDetail.setUpdateDate(new Date());
+            generateDetailMapper.updateById(generateDetail);
+        } else {
+            generateDetail.setGenerateId(generateId);
+            generateDetail.setUrl(minioPath);
+            generateDetail.setIsLike((byte)0);
+            generateDetail.setMd5(MD5Utils.encryptFile(minioUtil.getPreSignedUrl(minioPath,
+                    CommonConstant.MINIO_IMAGE_EXPIRE_TIME, true), Boolean.FALSE));
            generateDetail.setCreateDate(LocalDateTime.now());
            generateDetailMapper.insert(generateDetail);
-
-            generateDetailId = generateDetail.getId();
-        } else {
-            GenerateDetail generateDetail = new GenerateDetail();
-            generateDetail.setUrl(minioPath);
-            generateDetail.setId(generateDetailId);
-            generateDetailMapper.updateById(generateDetail);
+            originalId = generateDetail.getId();
        }
+        return buildResultVO(originalId, minioPath, category);
+    }

-        return new GenerateResultVO(generateDetailId, minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME, true), "Success", category);
+    private GenerateResultVO buildResultVO(Long id, String minioPath, String category) {
+        String url = minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME, true);
+        return new GenerateResultVO(id, url, "Success", category);
    }

    @Resource
@@ -1544,9 +1600,17 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
        poseTransformationMapper.updateById(poseTransformation);

        String taskIdBatch = poseTransformation.getTaskIdBatch();
-//        QueryWrapper<CloudTask> cloudTaskQueryWrapper = new QueryWrapper<>();
-//        cloudTaskQueryWrapper.lambda().eq(CloudTask::getTaskId, taskIdBatch);
-//        CloudTask cloudTask = cloudTaskMapper.selectOne(cloudTaskQueryWrapper);
+        QueryWrapper<CloudTask> cloudTaskQueryWrapper = new QueryWrapper<>();
+        cloudTaskQueryWrapper.lambda().eq(CloudTask::getTaskId, taskIdBatch);
+        CloudTask cloudTask = cloudTaskMapper.selectOne(cloudTaskQueryWrapper);
+        if (Objects.nonNull(cloudTask) && Objects.nonNull(cloudTask.getStartTime())){
+            cloudTask.setStartTime(LocalDateTime.now());
+            cloudTask.setUpdateTime(LocalDateTime.now());
+            cloudTaskMapper.updateById(cloudTask);
+        }else if (Objects.nonNull(cloudTask)){
+            cloudTask.setUpdateTime(LocalDateTime.now());
+            cloudTaskMapper.updateById(cloudTask);
+        }
 //        if (Objects.nonNull(cloudTask)) {
 //            if (cloudTask.getCompletedNum() == null) {
 //                cloudTask.setCompletedNum(1);
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -2921,6 +2921,7 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
        CloudTask cloudTask = cloudTaskMapper.selectOne(qw);
        cloudTask.setStatus(1);
        cloudTask.setCompletedNum(cloudTask.getNums());
+        cloudTask.setUpdateTime(LocalDateTime.now());
        cloudTaskMapper.updateById(cloudTask);

        // 扣除积分