课程指南-获取课程价格-目录添加是否购买

src/main/java/com/peanut/common/utils/ShiroUtils.java

@@ -38,6 +38,10 @@ public class ShiroUtils {
 		return getUserEntity().getUserId();
 	}
 
+	public static MyUserEntity getUser() {
+		return (MyUserEntity)SecurityUtils.getSubject().getPrincipal();
+	}
+
 	public static Integer getUId() {
 		MyUserEntity user = (MyUserEntity)SecurityUtils.getSubject().getPrincipal();
 		return user.getId();

src/main/java/com/peanut/modules/medical/service/impl/CourseMedicalServiceImpl.java

@@ -4,9 +4,11 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.github.yulichang.wrapper.MPJLambdaWrapper;
+import com.peanut.common.utils.ShiroUtils;
 import com.peanut.modules.common.dao.CourseCatalogueDao;
 import com.peanut.modules.common.dao.CourseMedicineDao;
 import com.peanut.modules.common.dao.CourseToMedicineDao;
+import com.peanut.modules.common.dao.UserCourseBuyDao;
 import com.peanut.modules.common.entity.*;
 import com.peanut.modules.common.to.ParamTo;
 import com.peanut.modules.medical.service.CourseMedicalService;
@@ -28,6 +30,8 @@ public class CourseMedicalServiceImpl extends ServiceImpl<CourseMedicineDao, Cou
     private CourseToMedicineDao toMedicineDao;
     @Autowired
     private CourseCatalogueDao courseCatalogueDao;
+    @Autowired
+    private UserCourseBuyDao userCourseBuyDao;
 
     @Override
     public List<CourseMedicine> getMedicalLabels(Integer id) {
@@ -94,6 +98,7 @@ public class CourseMedicalServiceImpl extends ServiceImpl<CourseMedicineDao, Cou
     }
 
     private void getLastLevelCourseList(CourseMedicine c,Map<String,Object> map){
+        String vip = ShiroUtils.getUser().getVip();
         if (c.getChildren()!=null&&c.getChildren().size() > 0){
             for (CourseMedicine child : c.getChildren()){
                 getLastLevelCourseList(child,map);
@@ -108,9 +113,20 @@ public class CourseMedicalServiceImpl extends ServiceImpl<CourseMedicineDao, Cou
         if(courseEntities.size()>0){
             for (CourseEntity co:courseEntities){
                 List<CourseCatalogueEntity> courseCatalogueEntities = courseCatalogueDao.selectList(new LambdaQueryWrapper<CourseCatalogueEntity>().eq(CourseCatalogueEntity::getCourseId, co.getId()).orderByAsc(CourseCatalogueEntity::getSort));
+                if (courseCatalogueEntities.size() > 0){
+                    for (CourseCatalogueEntity cc:courseCatalogueEntities){
+                        if ("0".equals(vip)||"2".equals(vip)){
+                            Integer i = userCourseBuyDao.selectCount(new LambdaQueryWrapper<UserCourseBuyEntity>()
+                                    .eq(UserCourseBuyEntity::getCatalogueId, cc.getId())
+                                    .eq(UserCourseBuyEntity::getUserId, ShiroUtils.getUId()));
+                            cc.setIsBuy(i>0?1:0);
+                        }else {
+                            cc.setIsBuy(1);
+                        }
+                    }
+                }
                 co.setCourseCatalogueEntityList(courseCatalogueEntities);
             }
-//            c.setCourseList(courseEntities);
         }
         ((List<CourseEntity>)map.get("courseList")).addAll(courseEntities);
     }