aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:test；

Browse Source
This commit is contained in:
shahaibo
2025-05-30 11:42:27 +08:00
parent d9e25fd0e4
commit d6cc8ab9a2
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/com/ai/da/common/config/MyTaskScheduler.java
View File

@@ -6,6 +6,7 @@ import com.ai.da.mapper.primary.entity.*;
import com.ai.da.mapper.primary.entity.Collection; import com.ai.da.mapper.primary.entity.Collection;
import com.ai.da.mapper.secondary.AttributeRetrievalMapper; import com.ai.da.mapper.secondary.AttributeRetrievalMapper;
import com.ai.da.model.dto.*; 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.enums.DesignProcess;
import com.ai.da.model.vo.ValidateElementVO; import com.ai.da.model.vo.ValidateElementVO;
import com.ai.da.python.PythonService; import com.ai.da.python.PythonService;
@@ -60,8 +61,30 @@ public class MyTaskScheduler {
// runComparisonAndRecordResults(5); // runComparisonAndRecordResults(5);
// saveDataToNumpyFormat(); // saveDataToNumpyFormat();
// projectDataCreate(); // 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 @Resource
private AccountMapper accountMapper; private AccountMapper accountMapper;