From b088628cf9be70c36e34032bb0d024548351541b Mon Sep 17 00:00:00 2001 From: puzhibing <393733352@qq.com> Date: 星期六, 11 三月 2023 17:37:12 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java | 29 +++++++++++++++++++++++++---- 1 files changed, 25 insertions(+), 4 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..ad9e895 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()); @@ -113,11 +125,11 @@ tDriverResp.setBranchName(tBranchOffice.getPrincipal()); } //累计订单量 - List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId()) + List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId()) && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode())).collect(Collectors.toList()); tDriverResp.setCumulativeOrderCount(cumulativeOrderCount.size()); //当月订单量 - List<TOrder> monthOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId()) + List<TOrder> monthOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId()) && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode()) && new SimpleDateFormat("yyyyMM").format(order.getCreateTime()).equals(monthDate)).collect(Collectors.toList()); tDriverResp.setMonthOrderCount(monthOrderCount.size()); @@ -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