Merge branch 'master' of https://sinanoaq.cn:8888/r/hazardInvestigationSyst...

郑永安

2023-07-24 0645153dfa233b51a749db73f9bd5a8c5127c595

 src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java

@@ -5,15 +5,19 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.hotwork.Domain.CompanyInfo;
import com.gk.hotwork.Domain.UserInfo;
import com.gk.hotwork.Domain.Utils.PageInfo;
import com.gk.hotwork.Domain.Utils.StringUtils;
import com.gk.hotwork.Domain.query.CompanyQuery;
import com.gk.hotwork.Mapper.CompanyInfoMapper;
import com.gk.hotwork.Service.CompanyService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service("CompanyService")
public class CompanyServiceImpl extends ServiceImpl<CompanyInfoMapper, CompanyInfo> implements CompanyService {
@@ -70,5 +74,29 @@
        return companyInfo.getIsmain() == 1;
    }

    @Override
    public List<CompanyInfo> selectCompanyList(String companyName, UserInfo user) {
        CompanyQuery query = new CompanyQuery();
        query.setCompany(companyName);


        Integer type = user.getType();
        //企业用户
        if (type.equals(3)) {
            Long companyid = user.getCompanyid();
            query.setCompanyid(companyid);
        }
        //监管用户
        if (type.equals(2)) {
            //获取企业信息
            query.setArea(user.getCounty());
            query.setProvince(user.getProvince());
            query.setCity(user.getCity());
        }

        List<CompanyInfo> companyInfos = companyInfoMapper.selectCompanyList(query);
        return companyInfos;
    }


}