From f9731ce5fdc89a7d3bdfb472691c0fc7f9c9aee9 Mon Sep 17 00:00:00 2001
From: lijiarao <lijiarao@syxysoft.com>
Date: Fri, 5 Jul 2024 16:56:14 +0800
Subject: [PATCH] =?UTF-8?q?add=20=E5=8A=9F=E8=83=BD=E5=AE=89=E5=85=A8?=
 =?UTF-8?q?=E4=B8=AD=E5=BF=83=E6=88=90=E5=91=98-=E4=BC=81=E4=B8=9A?=
 =?UTF-8?q?=E7=AB=AF-=E5=88=86=E9=A1=B5=E5=88=97=E8=A1=A8=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/SafetyCenterSubitemController.java  | 14 ++++++++++++++
 .../service/SafetyCenterSubitemService.java        | 11 +++++++++++
 2 files changed, 25 insertions(+)

diff --git a/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/controller/SafetyCenterSubitemController.java b/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/controller/SafetyCenterSubitemController.java
index 43bb5cf6..ee7120e9 100644
--- a/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/controller/SafetyCenterSubitemController.java
+++ b/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/controller/SafetyCenterSubitemController.java
@@ -70,6 +70,20 @@ public class SafetyCenterSubitemController extends JeroController<SafetyCenterSu
         return Result.OK(pageList);
     }
 
+    /**
+     * 分页列表查询
+     */
+    @ApiOperation(value = "功能安全中心成员-企业端-分页列表查询", notes = "功能安全中心成员-分页列表查询")
+    @GetMapping(value = "/pageByCompany")
+    public Result<IPage<SafetyCenterSubitem>> queryCompanyPageList(SafetyCenterSubitem safetyCenterSubitem,
+                                                            @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
+                                                            @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
+                                                            HttpServletRequest req) {
+        Page<SafetyCenterSubitem> page = new Page<>(pageNo, pageSize);
+        IPage<SafetyCenterSubitem> pageList = safetyCenterSubitemService.pageList(page, safetyCenterSubitem);
+        return Result.OK(pageList);
+    }
+
     /**
      * 列表查询
      */
diff --git a/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/service/SafetyCenterSubitemService.java b/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/service/SafetyCenterSubitemService.java
index 13db7e75..5496a16c 100644
--- a/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/service/SafetyCenterSubitemService.java
+++ b/jero-boot-incoming-payment/src/main/java/com/jero/functionalsafetycenter/service/SafetyCenterSubitemService.java
@@ -228,6 +228,17 @@ public class SafetyCenterSubitemService extends ServiceImpl<SafetyCenterSubitemM
     }
 
     public IPage<SafetyCenterSubitem> pageList(Page<SafetyCenterSubitem> page, SafetyCenterSubitem safetyCenterSubitem) {
+        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        //如果是企业查询，判断是否有数据权限
+        if (sysUser.getIsCompanyUser()) {
+            LambdaQueryWrapper<SafetyCenterSubitem> wrapper = new LambdaQueryWrapper<>();
+            wrapper.eq(SafetyCenterSubitem::getCompanyId,sysUser.getCompanyId());
+            wrapper.eq(SafetyCenterSubitem::getProjectId,safetyCenterSubitem.getProjectId());
+            SafetyCenterSubitem subitem = this.getOne(wrapper);
+            if (subitem == null){
+                return new Page<>();
+            }
+        }
         IPage<SafetyCenterSubitem> pageList = baseMapper.pageList(page, safetyCenterSubitem);
         List<SafetyCenterSubitem> records = pageList.getRecords();
         //根据企业id获取所有联系人