diff --git a/src/main/java/com/peanut/modules/psyche/controller/PsycheCourseController.java b/src/main/java/com/peanut/modules/psyche/controller/PsycheCourseController.java index d9889fdc..00f1d3cf 100644 --- a/src/main/java/com/peanut/modules/psyche/controller/PsycheCourseController.java +++ b/src/main/java/com/peanut/modules/psyche/controller/PsycheCourseController.java @@ -30,6 +30,8 @@ public class PsycheCourseController { private UserVipService userVipService; @Autowired private UserCourseStudyingService userCourseStudyingService; + @Autowired + private UserCourseBuyService userCourseBuyService; //我的课程-全部 @@ -88,7 +90,12 @@ public class PsycheCourseController { List> list = courseService.listMaps(wrapper); for (Map courseEntity : list) { if (!userVipService.isPsycheVip(ShiroUtils.getUId())){ - courseEntities.add(courseEntity); + int ucb = userCourseBuyService.count(new LambdaQueryWrapper() + .eq(UserCourseBuyEntity::getUserId,ShiroUtils.getUId()) + .eq(UserCourseBuyEntity::getCatalogueId,courseEntity.get("catalogueId"))); + if (ucb == 0) { + courseEntities.add(courseEntity); + } } } return R.ok().put("courseList",courseEntities);