From e030255c23c7ba3e2cbad1036a810d6d72fa864f Mon Sep 17 00:00:00 2001
From: liujie <liujie>
Date: 星期四, 26 十月 2023 18:04:58 +0800
Subject: [PATCH] 修改bug

---
 cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java |   55 +++++++++++++++++++++++++++++++++++--------------------
 1 files changed, 35 insertions(+), 20 deletions(-)

diff --git a/cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java b/cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java
index ce1da41..17c7e01 100644
--- a/cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java
+++ b/cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java
@@ -1,7 +1,9 @@
 package com.dsh.course.controller;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.dsh.course.entity.CancelledClasses;
 import com.dsh.course.entity.CoursePackageScheduling;
+import com.dsh.course.entity.TCoursePackage;
 import com.dsh.course.entity.TCoursePackagePayment;
 import com.dsh.course.feignclient.account.AppUserClient;
 import com.dsh.course.feignclient.account.CoachClient;
@@ -22,6 +24,7 @@
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author zhibing.pu
@@ -54,31 +57,43 @@
     @ResponseBody
     @PostMapping("/cancelledClasses/listAll")
     public List<CancelClassesVO> listAll(@RequestBody CancelClassesQuery query){
-
         List<CancelClassesVO> res = new ArrayList<>();
-        if (!query.getStudentName().equals("")){
-            List<Integer> s = new ArrayList<>();
-            List<Student> students = studentClient.queryStudentListByName(query.getStudentName());
-            if (students.size()==0){
-                return new ArrayList<>();
+        if (query.getStudentName()!=null){
+            if (!query.getStudentName().equals("")){
+                List<Integer> s = new ArrayList<>();
+                List<Student> students = studentClient.queryStudentListByName(query.getStudentName());
+                if (students.size()==0){
+                    return new ArrayList<>();
+                }
+                for (Student coach : students) {
+                    s.add(coach.getId());
+                }
+                query.setStudentIds(s);
             }
-            for (Student coach : students) {
-                s.add(coach.getId());
-            }
-            query.setStudentIds(s);
         }
-        if (!query.getCoachName().equals("")){
-            List<Integer> c = new ArrayList<>();
-            List<Coach> coaches = coachClient.queryCoachListByName(query.getCoachName());
-            if (coaches.size()==0){
-                return new ArrayList<>();
+        if (query.getCoachName()!=null){
+            if (!query.getCoachName().equals("")){
+                List<Integer> c = new ArrayList<>();
+                List<Coach> coaches = coachClient.queryCoachListByName(query.getCoachName());
+                if (coaches.size()==0){
+                    return new ArrayList<>();
+                }
+                for (Coach coach : coaches) {
+                    c.add(coach.getId());
+                }
+                query.setCoachIds(c);
             }
-            for (Coach coach : coaches) {
-                c.add(coach.getId());
-            }
-            query.setCoachIds(c);
         }
-        List<CancelClassesVO> result = cancelledClassesService.listAll(query);
+        List<Integer> collect =new ArrayList<>();
+        if(query.getStoreId()!=null){
+            List<TCoursePackage> list = coursePackageService.list(new LambdaQueryWrapper<TCoursePackage>().eq(TCoursePackage::getStoreId, query.getStoreId()));
+             collect = list.stream().map(TCoursePackage::getId).collect(Collectors.toList());
+            if(collect.size()==0){
+                collect.add(-1);
+            }
+        }
+
+        List<CancelClassesVO> result = cancelledClassesService.listAll(query,collect);
         for (CancelClassesVO cancelClassesVO : result) {
             CoursePackageScheduling byId1 = coursePackageSchedulingService.getById(cancelClassesVO.getCoursePackageSchedulingId());
             SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

--
Gitblit v1.7.1