From 2d933930ae05ec45a209d815cb03595ad3ec486e Mon Sep 17 00:00:00 2001 From: chenshaohua <635616957@qq.com> Date: Mon, 20 Jul 2026 20:03:20 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A0=B9=E6=8D=AE=E4=BC=81=E4=B8=9Acode?= =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E4=BC=81=E4=B8=9A=E8=81=94=E7=B3=BB=E4=BA=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../app/AppCompanyContactController.java | 19 ++++++++++++++++ .../cms/CompanyContactController.java | 22 ++++++++++++++++++- .../ruoyi/common/utils/SiteSecurityUtils.java | 15 +++++++++++++ 3 files changed, 55 insertions(+), 1 deletion(-) diff --git a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/app/AppCompanyContactController.java b/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/app/AppCompanyContactController.java index 71f8073..3ddc173 100644 --- a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/app/AppCompanyContactController.java +++ b/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 list=companyContactService.getSelectList(companyContact); return getDataTable(list); } diff --git a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/cms/CompanyContactController.java b/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/cms/CompanyContactController.java index c3d55c9..2ceced7 100644 --- a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/cms/CompanyContactController.java +++ b/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 list=companyContactService.getSelectList(companyContact); return getDataTable(list); } - } diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/SiteSecurityUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/SiteSecurityUtils.java index a9e5e72..0630df5 100644 --- a/ruoyi-common/src/main/java/com/ruoyi/common/utils/SiteSecurityUtils.java +++ b/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); + } + } }