From 433f3631e70c86714ec7f0ce7a740050a51e6944 Mon Sep 17 00:00:00 2001 From: gongzuming <gongzuming> Date: 星期三, 11 九月 2024 15:17:19 +0800 Subject: [PATCH] 质检是否有总仓权限 --- src/main/java/com/mzl/flower/service/system/UserService.java | 5 +++++ 1 files changed, 5 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/mzl/flower/service/system/UserService.java b/src/main/java/com/mzl/flower/service/system/UserService.java index 3ddf79c..e5286b5 100644 --- a/src/main/java/com/mzl/flower/service/system/UserService.java +++ b/src/main/java/com/mzl/flower/service/system/UserService.java @@ -21,6 +21,7 @@ import com.mzl.flower.service.customer.CustomerService; import com.mzl.flower.service.partner.PartnerService; import com.mzl.flower.service.point.CustomerPointService; +import com.mzl.flower.service.supplier.StationService; import com.mzl.flower.service.supplier.SupplierService; import com.mzl.flower.utils.TreeBuilderUtil; import com.mzl.flower.utils.UUIDGenerator; @@ -79,6 +80,8 @@ @Autowired private CustomerPointMapper customerPointMapper; + @Autowired + private StationService stationService; public User findUserByOpenId(String openId, String sessionKey, String unionId @@ -208,6 +211,8 @@ result.setSupplierDTO(supplierService.getCurrentSupplier()); }else if(Constants.USER_TYPE.partner.name().equals(user.getType())){ result.setPartnerDTO(partnerService.getCurrentPartner()); + }else if (Constants.USER_TYPE.admin.name().equals(user.getType())){ + result.setMainWarehouse(stationService.getMainWarehouse(user.getId())); } result.setBindWechat(wechatMapper.selectCount(new LambdaQueryWrapper<UserWechat>() .eq(UserWechat::getUserId, userId)) > 0); -- Gitblit v1.9.3