sgj
2026-01-06 b4b15671788a5899ab144935c5084c2497acfd65
fzzy-igdss-core/src/main/java/com/fzzy/igds/service/CoreCompanyService.java
@@ -1,10 +1,13 @@
package com.fzzy.igds.service;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.fzzy.igds.constant.Constant;
import com.fzzy.igds.domain.Company;
import com.fzzy.igds.domain.PledgeContract;
import com.fzzy.igds.mapper.CoreCompanyMapper;
import com.fzzy.igds.utils.ContextUtil;
import com.ruoyi.common.core.domain.entity.SysDept;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.utils.StringUtils;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
@@ -22,6 +25,48 @@
    @Resource
    private CoreCompanyMapper coreCompanyMapper;
    /**
     * 根据用户类型获取对应公司列表信息
     * @return
     */
    public List<Company> getCompanyByUserType() {
        //获取当前登录人
        SysUser user = ContextUtil.getLoginUser();
        if (Constant.USER_TYPE_10.equals(user.getUserType())) {
            //监管用户,直接查询组织下所有公司
            return this.listCompany(null,user.getCompanyId(),null);
        }
        if (Constant.USER_TYPE_20.equals(user.getUserType())) {
            //银行用户,根据合同查询银行下所有库区
            return this.getCompanyByContract(user.getUserData());
        }
        if (Constant.USER_TYPE_30.equals(user.getUserType())) {
            //库区用户
            return this.listCompany(user.getDeptId()+"",null,null);
        }
        return this.listCompany(null,user.getCompanyId(),null);
    }
    /**
     * 根据银行id,关联合同表查询公司列表
     * @param bankId
     * @return
     */
    public List<Company> getCompanyByContract(String bankId) {
        if (StringUtils.isBlank(bankId)){
            return null;
        }
        QueryWrapper<PledgeContract> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("c.pledge_bank", bankId);
        return coreCompanyMapper.selectCompanyByBankId(queryWrapper);
    }
    /**
     * 根据条件查询公司信息
     * @param id