From ef406c4432c290f2627fa3742a1489e660dcc239 Mon Sep 17 00:00:00 2001
From: zhangfeng <1603559716@qq.com>
Date: 星期六, 22 七月 2023 18:30:26 +0800
Subject: [PATCH] 检查修改

---
 src/main/java/com/gk/hotwork/Service/ServiceImpl/CompanyServiceImpl.java |   28 ++++++++++++++++++++++++++++
 1 files changed, 28 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..d424e76 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,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;
+    }
+
 
 }

--
Gitblit v1.9.2