From 91cabe852620e6a2f67344aa74363226152906c4 Mon Sep 17 00:00:00 2001
From: 44323 <443237572@qq.com>
Date: 星期四, 26 十月 2023 08:54:42 +0800
Subject: [PATCH] 运营商权限

---
 cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/system/BallController.java |   62 +++++++++++++++++++++++--------
 1 files changed, 46 insertions(+), 16 deletions(-)

diff --git a/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/system/BallController.java b/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/system/BallController.java
index a1e3151..41b066c 100644
--- a/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/system/BallController.java
+++ b/cloud-server-management/src/main/java/com/dsh/guns/modular/system/controller/system/BallController.java
@@ -57,6 +57,7 @@
 import java.time.LocalTime;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
+import java.util.stream.Collectors;
 
 @Controller
 @RequestMapping("/ball")
@@ -881,6 +882,13 @@
     @ResponseBody
     public List<Map<String,Object>> yuyuelist(BookingQuery bookingQuery) {
         Integer objectType = UserExt.getUser().getObjectType();
+
+        if(objectType==2){
+            List<Integer> operatorId = storeService.list(new QueryWrapper<TStore>()
+                    .eq("operatorId", UserExt.getUser().getObjectId()))
+                    .stream().map(TStore::getId).collect(Collectors.toList());
+            bookingQuery.setStoreIds(operatorId);
+        }
         if(objectType==3){
             bookingQuery.setStoreId(UserExt.getUser().getObjectId());
         }
@@ -1044,20 +1052,36 @@
 
         String cityCode = null;
         if(objectType == 2){//城市管理员
-            CityManager cityManager = cityManagerClient.queryCityManagerById(objectId);
-            cityCode = cityManager.getCityCode();
+            // 获取到这个运营商下面的所有门店
+            List<TStore> operatorId = storeService.list(new QueryWrapper<TStore>().eq("operatorId",objectId));
+            model.addAttribute("store", operatorId);
+            List<Coach> coach = coachClient.queryCoachByOperatorId(objectId);
+            model.addAttribute("coach", coach);
+            // 如果该运营商下面没有门店
+            if (operatorId.size()==0){
+                List<TStore> o = new ArrayList<>();
+                List<TSite> tSites = new ArrayList<>();
+                model.addAttribute("store", o);
+                model.addAttribute("site", tSites);
+            }else{
+                List<TSite> storeId = siteService.list(new QueryWrapper<TSite>()
+                        .eq("storeId", operatorId.get(0).getId()));
+                model.addAttribute("site", storeId);
+            }
+        }else {
+            List<Map<String, Object>> list = storeService.queryProvince(cityCode);
+            model.addAttribute("province", list);
+            Object code = list.get(0).get("code");
+            List<Map<String, Object>> list1 = storeService.queryCity(code.toString(), cityCode);
+            model.addAttribute("city", list1);
+            String code1 = list1.get(0).get("code").toString();
+            List<TStore> list2 = storeService.list(new QueryWrapper<TStore>().eq("cityCode", code1).eq("state", 1));
+            model.addAttribute("store", list2);
+            TStore store = list2.get(0);
+            List<TSite> list3 = siteService.list(new QueryWrapper<TSite>().eq("storeId", store.getId()).eq("state", 1));
+            model.addAttribute("site", list3);
         }
-        List<Map<String, Object>> list = storeService.queryProvince(cityCode);
-        model.addAttribute("province", list);
-        Object code = list.get(0).get("code");
-        List<Map<String, Object>> list1 = storeService.queryCity(code.toString(), cityCode);
-        model.addAttribute("city", list1);
-        String code1 = list1.get(0).get("code").toString();
-        List<TStore> list2 = storeService.list(new QueryWrapper<TStore>().eq("cityCode", code1).eq("state", 1));
-        model.addAttribute("store", list2);
-        TStore store = list2.get(0);
-        List<TSite> list3 = siteService.list(new QueryWrapper<TSite>().eq("storeId", store.getId()).eq("state", 1));
-        model.addAttribute("site", list3);
+        model.addAttribute("objectType",objectType);
         return PREFIX+"yuyue_add.html";
     }
 
@@ -1069,10 +1093,16 @@
         siteBooking.setInsertTime(new Date());
         siteBooking.setState(1);
         siteBooking.setStatus(0);
-
-    siteClient.addSiteBooking(siteBooking);
+        // 如果是运营商添加的话 根据门店的省市来存储
+        if (UserExt.getUser().getObjectType() == 2){
+            TStore byId = storeService.getById(siteBooking.getStoreId());
+            siteBooking.setProvince(byId.getProvince());
+            siteBooking.setProvinceCode(byId.getProvinceCode());
+            siteBooking.setCity(byId.getCity());
+            siteBooking.setCityCode(byId.getCityCode());
+        }
+        siteClient.addSiteBooking(siteBooking);
         System.out.println("================="+siteBooking);
-
         return new ResultUtil<>(null,200,null,null,null);
     }
 

--
Gitblit v1.7.1