czt
2025-12-01 096296cd7485c5583c8194d88cca700e3c4d84a0
fzzy-igdss-core/src/main/java/com/fzzy/igds/service/CoreCompanyService.java
@@ -1,7 +1,8 @@
package com.fzzy.igds.service;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.fzzy.igds.domain.Company;
import com.fzzy.igds.repository.CompanyRepository;
import com.fzzy.igds.mapper.CoreCompanyMapper;
import com.fzzy.igds.utils.ContextUtil;
import com.ruoyi.common.core.domain.entity.SysDept;
import com.ruoyi.common.utils.StringUtils;
@@ -19,27 +20,39 @@
public class CoreCompanyService {
    @Resource
    private CompanyRepository companyRepository;
    private CoreCompanyMapper coreCompanyMapper;
    /**
     * jpa查询公司信息
     * 根据条件查询公司信息
     * @param id
     * @param companyId
     * @param parentId
     * @return
     */
    public List<Company> getAllData(String companyId) {
    public List<Company> listCompany(String id, String companyId, String parentId) {
        if (StringUtils.isEmpty(companyId)) {
            companyId = ContextUtil.getCompanyId();
        QueryWrapper<Company> queryWrapper = new QueryWrapper<>();
        if(StringUtils.isNotBlank(id)){
            queryWrapper.eq("id", id);
        }
        return companyRepository.getAllData(companyId);
        if(StringUtils.isNotBlank(companyId)){
            queryWrapper.eq("company_id", companyId);
        }
        if(StringUtils.isNotBlank(parentId)){
            queryWrapper.likeRight("id", parentId);
        }
        return coreCompanyMapper.selectList(queryWrapper);
    }
    /**
     * jpa新增公司信息
     * 更新或保存公司信息
     * @param sysDept
     * @param isAdd
     * @return
     */
    public void saveOrUpdate(SysDept sysDept) {
    public void saveOrUpdate(SysDept sysDept, Boolean isAdd) {
        if (null == sysDept) {
            return;
        }
@@ -52,15 +65,30 @@
        company.setDwzt(sysDept.getStatus());
        company.setCompanyId(sysDept.getCompanyId());
        //创建信息
        company.setCreateBy(ContextUtil.getLoginUserName());
        company.setCreateTime(new Date());
        this.update(company);
        if (isAdd) {
            this.save(company);
        }else {
            this.update(company);
        }
    }
    /**
     * jpa更新公司信息
     * 新增公司信息
     * @param data
     */
    public void save(Company data) {
        if (StringUtils.isEmpty(data.getCompanyId())) {
            data.setCompanyId(ContextUtil.getCompanyId());
        }
        //新增信息
        data.setCreateBy(ContextUtil.getLoginUserName());
        data.setCreateTime(new Date());
        coreCompanyMapper.insert(data);
    }
    /**
     * 更新公司信息
     * @param data
     */
    public void update(Company data) {
@@ -71,7 +99,7 @@
        data.setUpdateBy(ContextUtil.getLoginUserName());
        data.setUpdateTime(new Date());
        companyRepository.save(data);
        coreCompanyMapper.updateById( data);
    }
    /**
@@ -80,6 +108,6 @@
     * @return
     */
    public void delData(String deptId) {
        companyRepository.deleteById(deptId);
        coreCompanyMapper.deleteById(deptId);
    }
}