From 41b60f3df5f3054aad44307c13a26b14f3b32ac0 Mon Sep 17 00:00:00 2001
From: nickchange <126672920+nickchange@users.noreply.github.com>
Date: 星期二, 21 十一月 2023 09:46:03 +0800
Subject: [PATCH] 11.21

---
 cloud-server-course/src/main/java/com/dsh/course/controller/CancelledClassesController.java |   41 ++++++++++++++++++++++++++++++++++++-----
 1 files changed, 36 insertions(+), 5 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 4da42ab..d3082bc 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,8 +1,7 @@
 package com.dsh.course.controller;
 
-import com.dsh.course.entity.CancelledClasses;
-import com.dsh.course.entity.CoursePackageScheduling;
-import com.dsh.course.entity.TCoursePackagePayment;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.dsh.course.entity.*;
 import com.dsh.course.feignclient.account.AppUserClient;
 import com.dsh.course.feignclient.account.CoachClient;
 import com.dsh.course.feignclient.account.StudentClient;
@@ -22,6 +21,7 @@
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author zhibing.pu
@@ -81,10 +81,29 @@
                 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);
+            }
+        }
+        if(query.getStoreIds()!=null && query.getStoreIds().size()!=0){
+            List<TCoursePackage> list = coursePackageService
+                    .list(new LambdaQueryWrapper<TCoursePackage>()
+                            .in(TCoursePackage::getStoreId, query.getStoreIds()));
+            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());
+            if (byId1==null){
+                continue;
+            }
             SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
             String string = format.format(byId1.getClassDate());
             String string1 = format.format(byId1.getEndDate());
@@ -103,6 +122,9 @@
         return result;
     }
 
+
+    @Autowired
+    private CourseCounsumService courseCounsumService;
     /**
      * 添加数据
      * @param cancelledClasses
@@ -111,6 +133,15 @@
     @PostMapping("/cancelledClasses/addCancelledClasses")
     public void addCancelledClasses(@RequestBody CancelledClasses cancelledClasses){
         cancelledClassesService.save(cancelledClasses);
+
+        TCoursePackagePayment byId = coursePackagePaymentService.getById(cancelledClasses.getCoursePackagePaymentId());
+        CourseCounsum courseCounsum = new CourseCounsum();
+        courseCounsum.setPaymentId(byId.getId());
+        courseCounsum.setChangeType(0);
+        courseCounsum.setNum(cancelledClasses.getCancelledClassesNumber());
+        courseCounsum.setInsertTime(new Date());
+        courseCounsum.setReason("消课");
+        courseCounsumService.save(courseCounsum);
     }
 
 }

--
Gitblit v1.7.1