add 功能安全中心成员-企业端-分页列表查询

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);
+    }
+
     /**
      * 列表查询
      */

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获取所有联系人