根据企业code查询企业联系人

ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/app/AppCompanyContactController.java

@@ -3,17 +3,22 @@ package com.ruoyi.cms.controller.app;
 
 import com.ruoyi.cms.domain.vo.CompanyContactVo;
 import com.ruoyi.cms.service.CompanyContactService;
+import com.ruoyi.cms.service.ICompanyService;
 import com.ruoyi.cms.util.ListUtil;
 import com.ruoyi.common.annotation.BussinessLog;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.Company;
 import com.ruoyi.common.core.domain.entity.CompanyContact;
 import com.ruoyi.common.core.page.TableDataInfo;
+import com.ruoyi.common.utils.SiteSecurityUtils;
+import com.ruoyi.common.utils.StringUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.ArrayList;
 import java.util.List;
 
 
@@ -30,6 +35,8 @@ import java.util.List;
 public class AppCompanyContactController extends BaseController {
     @Autowired
     private CompanyContactService companyContactService;
+    @Autowired
+    private ICompanyService iCompanyService;
 
     /**
      * 列表
@@ -37,6 +44,18 @@ public class AppCompanyContactController extends BaseController {
     @ApiOperation("公司联系人列表")
     @GetMapping("/list")
     public TableDataInfo list(CompanyContact companyContact){
+        if (SiteSecurityUtils.isLogin()) {
+            Long userId = SiteSecurityUtils.getUserId();
+            if(userId != null){
+                String idCard = SiteSecurityUtils.getIdCard();
+                if (StringUtils.isNotBlank(idCard)) {
+                    Company company = iCompanyService.queryCodeCompany(idCard);
+                    if(company == null){
+                        return getDataTable(new ArrayList<>());
+                    }
+                }
+            }
+        }
         List<CompanyContact> list=companyContactService.getSelectList(companyContact);
         return getDataTable(list);
     }

ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/cms/CompanyContactController.java

@@ -1,16 +1,21 @@
 package com.ruoyi.cms.controller.cms;
 
 
+import com.ruoyi.cms.service.ICompanyService;
+import com.ruoyi.common.core.domain.entity.Company;
 import com.ruoyi.common.core.domain.entity.CompanyContact;
 import com.ruoyi.cms.service.CompanyContactService;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.page.TableDataInfo;
+import com.ruoyi.common.utils.SecurityUtils;
+import com.ruoyi.common.utils.StringUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.ArrayList;
 import java.util.List;
 
 
@@ -27,6 +32,8 @@ import java.util.List;
 public class CompanyContactController extends BaseController {
     @Autowired
     private CompanyContactService companyContactService;
+    @Autowired
+    private ICompanyService iCompanyService;
 
     /**
      * 列表
@@ -35,8 +42,21 @@ public class CompanyContactController extends BaseController {
     @PreAuthorize("@ss.hasPermi('cms:companycontact:list')")
     @RequestMapping("/list")
     public TableDataInfo list(CompanyContact companyContact){
+        if(SecurityUtils.isLogin()){
+            if (SecurityUtils.isLogin()) {
+                Long userId = SecurityUtils.getUserId();
+                if(userId != null && !SecurityUtils.isAdmin(userId)){
+                    String idCard = SecurityUtils.getIdcard();
+                    if (StringUtils.isNotBlank(idCard)) {
+                        Company company = iCompanyService.queryCodeCompany(idCard);
+                        if(company == null){
+                            return getDataTable(new ArrayList<>());
+                        }
+                    }
+                }
+            }
+        }
         List<CompanyContact> list=companyContactService.getSelectList(companyContact);
         return getDataTable(list);
     }
-
 }

ruoyi-common/src/main/java/com/ruoyi/common/utils/SiteSecurityUtils.java

@@ -135,4 +135,19 @@ public class SiteSecurityUtils
             throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED);
         }
     }
+
+    /**
+     * 获取idcard
+     * @return
+     */
+    public static String getIdCard(){
+        try
+        {
+            return getLoginSiteUser().getUser().getIdCard();
+        }
+        catch (Exception e)
+        {
+            throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED);
+        }
+    }
 }