From 3c8f3a7b74f62a9053cd9d860d8f1d37454712eb Mon Sep 17 00:00:00 2001
From: 郑永安 <zyazyz250@sina.com>
Date: 星期一, 04 九月 2023 17:53:59 +0800
Subject: [PATCH] back

---
 src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java |   55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 55 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java b/src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java
index da98dfb..7260444 100644
--- a/src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java
+++ b/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;
+    }
 
 }

--
Gitblit v1.9.2