sgj
2026-01-06 65f6665b28af5912b8fbacb43cf111087fbf812d
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;
@@ -21,6 +24,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);
    }
    /**
     * 根据条件查询公司信息
@@ -47,6 +92,22 @@
    }
    /**
     * 查询公司数量
     * @param companyId
     * @return
     */
    public Integer getCompanyNum(String companyId) {
        QueryWrapper<Company> queryWrapper = new QueryWrapper<>();
        if(StringUtils.isNotBlank(companyId)){
            queryWrapper.eq("company_id", companyId);
        }
        return coreCompanyMapper.selectCount(queryWrapper);
    }
    /**
     * 更新或保存公司信息
     * @param sysDept
     * @param isAdd