From 8740cdef0b72e5d2563650fede7156f16ae04488 Mon Sep 17 00:00:00 2001
From: 44323 <443237572@qq.com>
Date: 星期三, 25 十月 2023 10:02:38 +0800
Subject: [PATCH] 运营商权限

---
 cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java |   19 +++++++++++++++++--
 1 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java b/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java
index 6e8ad57..a0d9606 100644
--- a/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java
+++ b/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java
@@ -1,6 +1,7 @@
 package com.dsh.guns.modular.system.controller.code;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.dsh.course.feignClient.account.AppUserClient;
 import com.dsh.course.feignClient.account.CityManagerClient;
 import com.dsh.course.feignClient.account.model.CityManager;
@@ -27,9 +28,10 @@
 
 import java.math.BigDecimal;
 import java.util.*;
+import java.util.stream.Collectors;
 
 /**
- * 广告管理控制器
+ * 购课记录控制器
  *
  */
 @Controller
@@ -94,7 +96,7 @@
         // 获取课包id
         Integer coursePackageId = tCoursePackagePayment.getCoursePackageId();
         List<TCoursePackageDiscount> list = coursePackageDiscountClient.queryByCoursePackageId(coursePackageId);
-        TCoursePackageDiscount tCoursePackageDiscount1 = list.get(0);
+
         // 获取当前课包价格配置
         CoursePackagePaymentConfig coursePackagePaymentConfigs = coursePackagePaymentConfigClient.getById(id);
         Integer coursePackageId1 = tCoursePackagePayment.getCoursePackageId();
@@ -120,6 +122,19 @@
     @RequestMapping(value = "/listAll")
     @ResponseBody
     public List<CoursePackagePaymentVO> listAll(CoursePackagePaymentQuery query) {
+        Integer objectType = UserExt.getUser().getObjectType();
+        if (objectType == 2){
+            Integer objectId = UserExt.getUser().getObjectId();
+            List<Integer> operatorId = storeService.list(new QueryWrapper<TStore>().eq("operatorId", objectId))
+                    .stream().map(TStore::getId).collect(Collectors.toList());
+            query.setStoreIds(operatorId);
+        }
+        if (objectType == 3){
+            Integer objectId = UserExt.getUser().getObjectId();
+            List<Integer> list = new ArrayList<>();
+            list.add(objectId);
+            query.setStoreIds(list);
+        }
         return coursePackagePaymentClient.listAll(query);
     }
 

--
Gitblit v1.7.1