From 1b6c4cd60f3882c400d30af367bd4c5960fbd4a1 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期一, 10 四月 2023 15:02:24 +0800 Subject: [PATCH] 优化管理 --- management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/THomePageController.java | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 55 insertions(+), 5 deletions(-) diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/THomePageController.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/THomePageController.java index 99cbd75..0033e18 100644 --- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/THomePageController.java +++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/THomePageController.java @@ -59,6 +59,13 @@ */ @RequestMapping("/map") public String map(Integer agentId,Integer type,Model model) { + + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } + // 查询所有代理商 List<TAgent> agentList = tAgentService.selectList(new EntityWrapper<TAgent>() .ne("status", StatusEnum.DELETE.getCode()) @@ -115,6 +122,11 @@ @RequestMapping("/getMap") @ResponseBody public Object getMap(Integer agentId,Integer type,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } HashMap<String, Object> map = new HashMap<>(); // 查询所有代理商 List<TAgent> agentList = tAgentService.selectList(new EntityWrapper<TAgent>() @@ -173,9 +185,13 @@ .eq("serverStatus", 2)); map.put("onLineDriverCount",onLineDriver.size()); - tOrderService.getDataStatisticsCountByIdsGetMap(ids,map); + + List<TAgent> tAgentAllList = tAgentService.selectList(new EntityWrapper<TAgent>()); + List<Integer> allIds = tAgentAllList.stream().map(TAgent::getId).collect(Collectors.toList()); + tOrderService.getDataStatisticsCountByIdsGetMap(allIds,map); + // 查询服务中的订单列表 - List<TOrderServerResp> orderServerRespList = tOrderService.getDataStatisticsServerListByIds(ids); + List<TOrderServerResp> orderServerRespList = tOrderService.getDataStatisticsServerListByIds(allIds); // 查询司机 for (TOrderServerResp tOrderServerResp : orderServerRespList) { TDriver tDriver = tDriverService.selectById(tOrderServerResp.getDriverId()); @@ -184,7 +200,7 @@ } map.put("serverList",orderServerRespList); // 查询今天所有订单 - List<TOrder> allList = tOrderService.getDataStatisticsAllListByIds(ids,type); + List<TOrder> allList = tOrderService.getDataStatisticsAllListByIds(allIds,type); List<List<String>> list = new ArrayList<>(allList.size()); for (TOrder tOrder : allList) { List<String> obj = new ArrayList<>(3); @@ -215,6 +231,11 @@ @RequestMapping("/getServerOrder") @ResponseBody public Object getServerOrder(Integer agentId,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } HashMap<String, Object> map = new HashMap<>(); if(Objects.nonNull(agentId)){ // 查询服务中的订单列表 @@ -227,8 +248,7 @@ } map.put("serverList",orderServerRespList); }else { - List<TAgent> tAgentList = tAgentService.selectList(new EntityWrapper<TAgent>() - .eq("provinceCode", "440000")); + List<TAgent> tAgentList = tAgentService.selectList(new EntityWrapper<TAgent>()); List<Integer> ids = tAgentList.stream().map(TAgent::getId).collect(Collectors.toList()); // 查询服务中的订单列表 List<TOrderServerResp> orderServerRespList = tOrderService.getDataStatisticsServerListByIds(ids); @@ -248,6 +268,11 @@ */ @RequestMapping("/statistics") public String statistics(Integer agentId,String dayDate,String yearDate,String monthDate,Integer type,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } if(Objects.isNull(type)){ type = 1; } @@ -317,6 +342,11 @@ @RequestMapping("/statisticsByAgentId") @ResponseBody public Map<String, Object> statisticsByAgentId(Integer agentId, String dayDate, String yearDate, String monthDate, Integer type, Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } Map<String, Object> map = new HashMap<>(); if(Objects.nonNull(agentId)){ // 统计代理商 @@ -350,6 +380,11 @@ @RequestMapping("/statisticsMonthByAgentId") @ResponseBody public Map<String, Object> statisticsMonthByAgentId(Integer agentId, String dayDate,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } dayDate = new SimpleDateFormat("yyyy-MM").format(new Date()); Map<String, Object> map = new HashMap<>(); if(Objects.nonNull(agentId)){ @@ -372,6 +407,11 @@ @RequestMapping("/statisticsYearByAgentId") @ResponseBody public Map<String, Object> statisticsYearByAgentId(Integer agentId, String yearDate,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } if(!StringUtils.hasLength(yearDate)){ yearDate = new SimpleDateFormat("yyyy").format(new Date()); } @@ -395,6 +435,11 @@ @RequestMapping("/statisticsOrderByAgentId") @ResponseBody public Map<String, Object> statisticsOrderByAgentId(Integer agentId, String monthDate,Integer type,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } if(Objects.isNull(type)){ type = 1; } @@ -434,6 +479,11 @@ @RequestMapping("/statisticsDriverByAgentId") @ResponseBody public Map<String, Object> statisticsDriverByAgentId(Integer agentId,Model model) { + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + if(roleType == 3 && Objects.nonNull(objectId)){ + agentId = objectId; + } Map<String, Object> map = new HashMap<>(); if(Objects.nonNull(agentId)){ // 统计司机数 -- Gitblit v1.7.1