From dc9239d73b15b9a51c46a9e8d25c0d4400e613ce Mon Sep 17 00:00:00 2001 From: luodangjia <luodangjia> Date: 星期一, 29 七月 2024 09:17:54 +0800 Subject: [PATCH] 7.29 --- ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/HomePageController.java | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 48 insertions(+), 0 deletions(-) diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/HomePageController.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/HomePageController.java index cfcf91e..c1f197a 100644 --- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/HomePageController.java +++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/HomePageController.java @@ -8,9 +8,11 @@ import com.ruoyi.admin.vo.UserTrendsVO; import com.ruoyi.common.core.domain.R; import com.ruoyi.common.security.annotation.RequiresPermissions; +import com.ruoyi.common.security.service.TokenService; import com.ruoyi.order.api.entity.OrderCountRequest; import com.ruoyi.order.api.entity.OrderResultVO; import com.ruoyi.order.api.feignClient.OrderClient; +import com.ruoyi.system.api.model.LoginUser; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.web.bind.annotation.PostMapping; @@ -37,6 +39,8 @@ private UserService userService; @Resource private OrderClient orderClient; + @Resource + private TokenService tokenService; /** * 用户数据统计 @@ -47,6 +51,17 @@ @ApiOperation(value = "后台-用户数据统计", tags = {"后台-首页"}) @PostMapping(value = "/userDataCount") public R<UserDataCountVO> userDataCount(@RequestBody UserDataCountRequest userDataCountRequest) { + LoginUser loginUser = tokenService.getLoginUser(); + if (null == loginUser) { + return R.loginExpire("登录失效!"); + } + if (loginUser.getIsFranchisee()) { + if (null == userDataCountRequest.getCityList() || userDataCountRequest.getCityList().isEmpty()) { + userDataCountRequest.setCityList(loginUser.getCityList()); + } else { + userDataCountRequest.getCityList().addAll(loginUser.getCityList()); + } + } return R.ok(userService.userDataCount(userDataCountRequest)); } @@ -59,6 +74,17 @@ @ApiOperation(value = "后台-用户增量", tags = {"后台-首页"}) @PostMapping(value = "/userTrends") public R<List<UserTrendsVO>> userTrends(@RequestBody List<String> cityList) { + LoginUser loginUser = tokenService.getLoginUser(); + if (null == loginUser) { + return R.loginExpire("登录失效!"); + } + if (loginUser.getIsFranchisee()) { + if (null == cityList || cityList.isEmpty()) { + cityList = loginUser.getCityList(); + } else { + cityList.addAll(loginUser.getCityList()); + } + } return R.ok(userService.userTrends(cityList)); } @@ -71,6 +97,17 @@ @ApiOperation(value = "后台-回收订单统计", tags = {"后台-首页"}) @PostMapping(value = "/orderCountHome") public R<OrderResultVO> orderCountHome(@RequestBody OrderCountRequest orderCount) { + LoginUser loginUser = tokenService.getLoginUser(); + if (null == loginUser) { + return R.loginExpire("登录失效!"); + } + if (loginUser.getIsFranchisee()) { + if (null == orderCount.getCityList() || orderCount.getCityList().isEmpty()) { + orderCount.setCityList(loginUser.getCityList()); + } else { + orderCount.getCityList().addAll(loginUser.getCityList()); + } + } return orderClient.orderCountHome(orderCount); } @@ -83,6 +120,17 @@ @ApiOperation(value = "后台-师傅接单排行", tags = {"后台-首页"}) @PostMapping(value = "/workerRanking") public R<List<MasterWorkerRankVO>> workerRanking(@RequestBody List<String> cityList) { + LoginUser loginUser = tokenService.getLoginUser(); + if (null == loginUser) { + return R.loginExpire("登录失效!"); + } + if (loginUser.getIsFranchisee()) { + if (null == cityList || cityList.isEmpty()) { + cityList = loginUser.getCityList(); + } else { + cityList.addAll(loginUser.getCityList()); + } + } return R.ok(masterWorkerService.workerRanking(cityList)); } -- Gitblit v1.7.1