From 58998d471cd4334e2763b4c87a1901cb73e2b8ff Mon Sep 17 00:00:00 2001
From: liujie <liujie>
Date: 星期三, 25 十月 2023 17:34:28 +0800
Subject: [PATCH] 10.13bug2

---
 cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/code/CoursePackagePaymentController.java |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 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 175193d..f5e6940 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,6 +28,7 @@
 
 import java.math.BigDecimal;
 import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * 广告管理控制器
@@ -78,7 +80,23 @@
         List<TCity> list = cityService.list(new LambdaQueryWrapper<TCity>().eq(TCity::getParentId, 0));
         model.addAttribute("list",list);
         Integer roleType = UserExt.getUser().getObjectType();
+        if (roleType==2){
+            Integer objectId = UserExt.getUser().getObjectId();
+            List<TStore> operatorId = storeService.list(new QueryWrapper<TStore>().eq("operatorId", objectId));
+            model.addAttribute("stores",operatorId);
+        }else{
+            model.addAttribute("stores",new ArrayList<>());
+        }
+        // 获取所有课包类型
+        List<TCoursePackageType> tCoursePackageTypes = coursePackageTypeClient.queryAllCoursePackageType();
         model.addAttribute("roleType",roleType);
+        if(UserExt.getUser().getObjectType()==3){
+            List<TStore> list1 = storeService.list(new LambdaQueryWrapper<TStore>().eq(TStore::getId, UserExt.getUser().getObjectId()));
+            model.addAttribute("stores",list1);
+        }else {
+            model.addAttribute("stores",new ArrayList<>());
+        }
+        model.addAttribute("coursePackageType",tCoursePackageTypes);
         return PREFIX + "cpPayment_add.html";
     }
     /**
@@ -127,6 +145,12 @@
             ids =  coursePackageClient.queryIdsByStore(UserExt.getUser().getObjectId());
             query.setIds(ids);
         }
+        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);
+        }
         return coursePackagePaymentClient.listAll(query);
     }
 

--
Gitblit v1.7.1