add: 供应商子账号

cloudroam

2024-12-09 2c5a0e133b64bb0a46b1419c48e9b62a4779e268

 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;
    }