From 9a50a7ea9688a1e777f98713560e9c599d5064ae Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期三, 08 三月 2023 18:01:34 +0800 Subject: [PATCH] 省市区三级联动修改 --- management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java | 25 +++++++++++++++++++++++-- 1 files changed, 23 insertions(+), 2 deletions(-) diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java index cf65d45..b580ea6 100644 --- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java +++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java @@ -3,6 +3,7 @@ import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.stylefeng.guns.core.base.tips.SuccessTip; +import com.stylefeng.guns.core.shiro.ShiroKit; import com.stylefeng.guns.core.util.DateUtil; import com.stylefeng.guns.modular.system.controller.resp.TDriverCommissionResp; import com.stylefeng.guns.modular.system.controller.resp.TDriverResp; @@ -84,6 +85,15 @@ wrapper.between("createTime",startTime,endTime); } wrapper.orderBy(true,"approvalStatus"); + // 判断是否为代理商或者分公司 + if(Objects.requireNonNull(ShiroKit.getUser()).getRoleType() == 2){ + // 分公司 + wrapper.eq("branchOfficeId",ShiroKit.getUser().getObjectId()); + } + if(Objects.requireNonNull(ShiroKit.getUser()).getRoleType() == 3){ + // 代理商 + wrapper.eq("agentId",ShiroKit.getUser().getObjectId()); + } return wrapper; } @@ -99,8 +109,10 @@ TDriverResp tDriverResp = new TDriverResp(); BeanUtils.copyProperties(tDriver,tDriverResp); - // 设置余额 - tDriverResp.setBalance(tDriver.getBackgroundBalance().add(tDriver.getBalance())); + if(Objects.nonNull(tDriver.getBackgroundBalance()) && Objects.nonNull(tDriver.getBalance())){ + // 设置余额 + tDriverResp.setBalance(tDriver.getBackgroundBalance().add(tDriver.getBalance())); + } // 所属代理商 TAgent tAgent = tAgentMapper.selectById(tDriver.getAgentId()); @@ -331,6 +343,15 @@ wrapper.like("name",name); } wrapper.orderBy(true,"approvalStatus"); + // 判断是否为代理商或者分公司 + if(Objects.requireNonNull(ShiroKit.getUser()).getRoleType() == 2){ + // 分公司 + wrapper.eq("branchOfficeId",ShiroKit.getUser().getObjectId()); + } + if(Objects.requireNonNull(ShiroKit.getUser()).getRoleType() == 3){ + // 代理商 + wrapper.eq("agentId",ShiroKit.getUser().getObjectId()); + } return wrapper; } -- Gitblit v1.7.1