From 2c5a0e133b64bb0a46b1419c48e9b62a4779e268 Mon Sep 17 00:00:00 2001 From: cloudroam <cloudroam> Date: 星期一, 09 十二月 2024 16:13:17 +0800 Subject: [PATCH] add: 供应商子账号 --- src/main/java/com/mzl/flower/service/supplier/SupplierService.java | 16 ++++++++++++++++ 1 files changed, 16 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/mzl/flower/service/supplier/SupplierService.java b/src/main/java/com/mzl/flower/service/supplier/SupplierService.java index 0a4582e..39e482a 100644 --- a/src/main/java/com/mzl/flower/service/supplier/SupplierService.java +++ b/src/main/java/com/mzl/flower/service/supplier/SupplierService.java @@ -10,7 +10,9 @@ import com.mzl.flower.dto.request.supplier.*; import com.mzl.flower.dto.response.supplier.SupplierDTO; import com.mzl.flower.entity.supplier.Supplier; +import com.mzl.flower.entity.supplier.SupplierSub; import com.mzl.flower.mapper.supplier.SupplierMapper; +import com.mzl.flower.mapper.supplier.SupplierSubMapper; import com.mzl.flower.service.BaseService; import com.mzl.flower.utils.DateUtils; import org.apache.commons.lang3.StringUtils; @@ -21,6 +23,7 @@ import org.springframework.security.oauth2.provider.token.TokenStore; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; +import org.springframework.util.ObjectUtils; import javax.annotation.Resource; import java.time.LocalDateTime; @@ -35,6 +38,9 @@ private final static String SUPPLIER_STATUS_REJECT = "R"; // 审核拒绝 private final SupplierMapper supplierMapper; + + @Resource + private SupplierSubMapper supplierSubMapper; @Resource private BaseService baseService; @@ -162,6 +168,16 @@ throw new ValidationException("当前登录用户不是供应商账号"); } SupplierDTO dto = supplierMapper.getCurrentSupplier(SecurityUtils.getUserId()); + //如果是空 查询子账号的供应商ID + if (ObjectUtils.isEmpty(dto)) { + SupplierSub sub = supplierSubMapper.getCurrentSupplier(SecurityUtils.getUserId()); + if (!ObjectUtils.isEmpty(sub)) { + dto = supplierMapper.findSupplierDetail(sub.getSupplierId()); + } + } + if(ObjectUtils.isEmpty(dto)){ + throw new ValidationException("供应商信息不存在"); + } return dto; } -- Gitblit v1.9.3