From 8a66f76e0b0af13ccdba14f1d38eed91139ca124 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期四, 23 三月 2023 11:41:43 +0800 Subject: [PATCH] 处理TODO,导出部分修改 --- 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