检查统计

zf

2023-08-30 464fe41610a39c3a06f070d5a01c930134ff0163

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

@@ -5,15 +5,22 @@
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.Do.StatisticCompanyCount;
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.Domain.query.CountQuery;
import com.gk.hotwork.Mapper.CompanyInfoMapper;
import com.gk.hotwork.Service.CompanyService;
import io.swagger.models.auth.In;
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 +77,53 @@
        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;
    }

    @Override
    public Integer getCountByArea(Map<String,Object> params) {
        return companyInfoMapper.getCountByArea(params);
    }

    @Override
    public Page<CompanyInfo> getUncheckCompanyByArea(CountQuery query){
        Page<CompanyInfo> page = new Page<>(query.getPageIndex(),query.getPageSize());

        List<CompanyInfo> companyInfos = companyInfoMapper.getUncheckCompanyByArea(page,query);
        page.setRecords(companyInfos);
        return page;
    }


    @Override
    public Page<CompanyInfo> getByArea(CountQuery query){
       Page<CompanyInfo> page = new Page<>(query.getPageIndex(),query.getPageSize());
       CompanyQuery companyQuery = new CompanyQuery();
       companyQuery.setArea(query.getArea());
       List<CompanyInfo> companyInfos = companyInfoMapper.selectCompanyList(page,companyQuery);
       page.setRecords(companyInfos);
       return page;
    }

}