From 3bcf6a65dfb9813dff2986c9cc03b638742ce64e Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期一, 06 三月 2023 18:02:00 +0800 Subject: [PATCH] 司机管理,用户角色分类查询修改 --- management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TEvaluateServiceImpl.java | 13 ++++++++++++- 1 files changed, 12 insertions(+), 1 deletions(-) diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TEvaluateServiceImpl.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TEvaluateServiceImpl.java index 7aaa6bd..d796a6c 100644 --- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TEvaluateServiceImpl.java +++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TEvaluateServiceImpl.java @@ -1,5 +1,6 @@ package com.stylefeng.guns.modular.system.service.impl; +import com.stylefeng.guns.core.shiro.ShiroKit; import com.stylefeng.guns.modular.system.controller.resp.TEvaluateResp; import com.stylefeng.guns.modular.system.dao.*; import com.stylefeng.guns.modular.system.model.*; @@ -13,6 +14,7 @@ import java.text.SimpleDateFormat; import java.util.List; import java.util.Objects; +import java.util.stream.Collectors; /** * <p> @@ -47,7 +49,16 @@ startTime = split[0]; endTime = split[1]; } - return tEvaluateMapper.selectPageList(startTime,endTime,userName,driverName,orderType,score); + List<TEvaluateResp> list = tEvaluateMapper.selectPageList(startTime, endTime, userName, driverName, orderType, score); + Integer roleType = Objects.requireNonNull(ShiroKit.getUser()).getRoleType(); + Integer objectId = Objects.requireNonNull(ShiroKit.getUser()).getObjectId(); + if(3 == roleType){ + list = list.stream().filter(tEvaluate -> tEvaluate.getAgentId().equals(objectId)).collect(Collectors.toList()); + } + if(2 == roleType){ + list = list.stream().filter(tEvaluate -> tEvaluate.getBranchOfficeId().equals(objectId)).collect(Collectors.toList()); + } + return list; } @Override -- Gitblit v1.7.1