From 15c1781360d95387ccd8322c53752d59ccc8609d Mon Sep 17 00:00:00 2001
From: wuchunlei <mailcauchy@gmail.com>
Date: Tue, 18 Jun 2024 13:48:16 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AF=BE=E7=A8=8B=E6=8C=87=E5=8D=97-=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E8=AF=BE=E7=A8=8B=E4=BB=B7=E6=A0=BC-=E7=9B=AE?=
 =?UTF-8?q?=E5=BD=95=E6=B7=BB=E5=8A=A0=E6=98=AF=E5=90=A6=E8=B4=AD=E4=B9=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/peanut/common/utils/ShiroUtils.java    |  4 ++++
 .../service/impl/CourseMedicalServiceImpl.java | 18 +++++++++++++++++-
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/peanut/common/utils/ShiroUtils.java b/src/main/java/com/peanut/common/utils/ShiroUtils.java
index 411749be..c1f46f85 100644
--- a/src/main/java/com/peanut/common/utils/ShiroUtils.java
+++ b/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();
diff --git a/src/main/java/com/peanut/modules/medical/service/impl/CourseMedicalServiceImpl.java b/src/main/java/com/peanut/modules/medical/service/impl/CourseMedicalServiceImpl.java
index 1b325659..b0dc916c 100644
--- a/src/main/java/com/peanut/modules/medical/service/impl/CourseMedicalServiceImpl.java
+++ b/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);
     }