TASK:test;

This commit is contained in:
shahaibo
2025-05-30 11:42:27 +08:00
parent d9e25fd0e4
commit d6cc8ab9a2

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.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;