From b9da8289051937fbbbd47bbda97414ad6c769a90 Mon Sep 17 00:00:00 2001 From: hjl <1657978663@qq.com> Date: 星期四, 18 七月 2024 14:37:00 +0800 Subject: [PATCH] feat: 代码重构 --- ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/WorkerProcessController.java | 24 ++++++++++++++++++++++++ 1 files changed, 24 insertions(+), 0 deletions(-) diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/WorkerProcessController.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/WorkerProcessController.java index c40f934..c6b5a34 100644 --- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/WorkerProcessController.java +++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/WorkerProcessController.java @@ -10,6 +10,8 @@ import com.ruoyi.common.core.domain.R; import com.ruoyi.common.core.utils.StringUtils; import com.ruoyi.common.security.annotation.RequiresPermissions; +import com.ruoyi.common.security.service.TokenService; +import com.ruoyi.system.api.model.LoginUser; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; @@ -17,6 +19,7 @@ import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; +import java.util.ArrayList; import java.util.Arrays; import java.util.List; import java.util.stream.Collectors; @@ -36,6 +39,8 @@ @Resource private WorkerProcessService workerProcessService; + @Resource + private TokenService tokenService; /** * 师傅入驻审核分页列表 @@ -57,11 +62,30 @@ public R<IPage<WorkerProcess>> queryPageList(String id, String realName, String city, Integer state, @RequestParam(name = "pageNum", defaultValue = "1") Integer pageNum, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize) { + LoginUser loginUser = tokenService.getLoginUser(); + if (null == loginUser) { + return R.loginExpire("登录失效!"); + } + List<String> cityList = new ArrayList<>(); + if (loginUser.getIsFranchisee()) { + cityList = loginUser.getCityList(); + } + // 构建初始条件 LambdaQueryChainWrapper<WorkerProcess> wrapper = workerProcessService.lambdaQuery(); wrapper = StringUtils.isNotBlank(id) ? wrapper.like(WorkerProcess::getId, id) : wrapper; wrapper = StringUtils.isNotBlank(realName) ? wrapper.like(WorkerProcess::getRealName, realName) : wrapper; wrapper = StringUtils.isNotBlank(city) ? wrapper.like(WorkerProcess::getCity, city) : wrapper; wrapper = null != state ? wrapper.eq(WorkerProcess::getState, state) : wrapper; + // 管辖城市 + for (int i = 0; i < cityList.size(); i++) { + // 最后一跳城市不匹配or条件 + String s = cityList.get(i); + if (cityList.size() - 1 == i) { + wrapper.like(WorkerProcess::getCity, s); + } else { + wrapper.like(WorkerProcess::getCity, s).or(); + } + } return R.ok(wrapper.eq(WorkerProcess::getIsDelete, 0) .orderByDesc(WorkerProcess::getCreateTime).page(Page.of(pageNum, pageSize))); } -- Gitblit v1.7.1