From d6cc8ab9a296fad8c7b3f390bf28890c1a9d6ce7 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Fri, 30 May 2025 11:42:27 +0800
Subject: [PATCH] =?UTF-8?q?TASK:test=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/da/common/config/MyTaskScheduler.java  | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/main/java/com/ai/da/common/config/MyTaskScheduler.java b/src/main/java/com/ai/da/common/config/MyTaskScheduler.java
index cb8d9c0d..4fc93c46 100644
--- a/src/main/java/com/ai/da/common/config/MyTaskScheduler.java
+++ b/src/main/java/com/ai/da/common/config/MyTaskScheduler.java
@@ -6,6 +6,7 @@ import com.ai.da.mapper.primary.entity.*;
 import com.ai.da.mapper.primary.entity.Collection;
 import com.ai.da.mapper.secondary.AttributeRetrievalMapper;
 import com.ai.da.model.dto.*;
+import com.ai.da.model.enums.CollectionType;
 import com.ai.da.model.enums.DesignProcess;
 import com.ai.da.model.vo.ValidateElementVO;
 import com.ai.da.python.PythonService;
@@ -60,8 +61,30 @@ public class MyTaskScheduler {
 //        runComparisonAndRecordResults(5);
 //        saveDataToNumpyFormat();
 //        projectDataCreate();
+//        collectionSortData();
     }
 
+    private void collectionSortData() {
+        QueryWrapper<UserLikeSort> qw = new QueryWrapper<>();
+
+        List<UserLikeSort> userLikeSorts = userLikeSortMapper.selectList(qw);
+        for (UserLikeSort userLikeSort : userLikeSorts) {
+            Long userLikeGroupId = userLikeSort.getUserLikeGroupId();
+            QueryWrapper<UserLikeGroup> userLikeGroupQueryWrapper = new QueryWrapper<>();
+            userLikeGroupQueryWrapper.lambda().eq(UserLikeGroup::getId, userLikeGroupId);
+            UserLikeGroup userLikeGroup = userLikeGroupMapper.selectOne(userLikeGroupQueryWrapper);
+            if (Objects.nonNull(userLikeGroup)) {
+                userLikeSort.setProjectId(userLikeGroup.getProjectId());
+                userLikeSort.setRelationId(userLikeSort.getUserLikeId());
+                userLikeSort.setRelationType(CollectionType.DESIGN.getValue());
+                userLikeSortMapper.updateById(userLikeSort);
+            }
+        }
+    }
+
+    @Resource
+    private UserLikeSortMapper userLikeSortMapper;
+
     @Resource
     private AccountMapper accountMapper;