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