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 qw = new QueryWrapper<>(); + + List userLikeSorts = userLikeSortMapper.selectList(qw); + for (UserLikeSort userLikeSort : userLikeSorts) { + Long userLikeGroupId = userLikeSort.getUserLikeGroupId(); + QueryWrapper 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;