| | |
| | | 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; |
| | |
| | | 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; |
| | | } |
| | |
| | | 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) { |
| | |
| | | data.setUpdateBy(ContextUtil.getLoginUserName()); |
| | | data.setUpdateTime(new Date()); |
| | | |
| | | companyRepository.save(data); |
| | | coreCompanyMapper.updateById( data); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @return |
| | | */ |
| | | public void delData(String deptId) { |
| | | companyRepository.deleteById(deptId); |
| | | coreCompanyMapper.deleteById(deptId); |
| | | } |
| | | } |