From d5165d2cdaeab2be112b805addf438466b59bedd Mon Sep 17 00:00:00 2001
From: zhuo <2089219845@qq.com>
Date: 星期二, 18 三月 2025 10:36:01 +0800
Subject: [PATCH] 单点登录完成

---
 cnas-device/src/main/java/com/ruoyi/device/service/impl/DeviceRecordServiceImpl.java |   18 ++++++++++++++++--
 1 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/cnas-device/src/main/java/com/ruoyi/device/service/impl/DeviceRecordServiceImpl.java b/cnas-device/src/main/java/com/ruoyi/device/service/impl/DeviceRecordServiceImpl.java
index f507f30..349ce8a 100644
--- a/cnas-device/src/main/java/com/ruoyi/device/service/impl/DeviceRecordServiceImpl.java
+++ b/cnas-device/src/main/java/com/ruoyi/device/service/impl/DeviceRecordServiceImpl.java
@@ -5,6 +5,7 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.deepoove.poi.XWPFTemplate;
 import com.deepoove.poi.config.Configure;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.device.dto.DeviceRecordDto;
 import com.ruoyi.device.excel.DeviceRecordExportWord;
 import com.ruoyi.device.mapper.DeviceMapper;
@@ -13,6 +14,8 @@
 import com.ruoyi.device.pojo.DeviceRecord;
 import com.ruoyi.device.service.DeviceRecordService;
 import com.ruoyi.inspect.util.HackLoopTableRenderPolicy;
+import com.ruoyi.system.domain.vo.SysRoleVo;
+import com.ruoyi.system.mapper.SysRoleMenuMapper;
 import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 
@@ -40,12 +43,23 @@
     @Resource
     private DeviceMapper deviceMapper;
 
+    @Resource
+    private SysRoleMenuMapper sysRoleMenuMapper;
+
     @Override
     public IPage<DeviceRecordDto> deviceRecordPage(Integer deviceId, Page page, String sampleCode, String managementNumber) {
         // todo:浠呯湅鑷繁
-        Integer userId = null;
+        Long userId = SecurityUtils.getUserId();
+        // 鑾峰彇鏉冮檺鍒ゆ柇鏄惁鏄粎鐪嬫垜
+        SysRoleVo sysRoleVo = sysRoleMenuMapper.selectRoleMenu("get:device:record", userId);
 
-        return baseMapper.deviceRecordPage(deviceId, page, sampleCode, managementNumber, userId);
+        Integer testUserId = null;
+        //娌℃湁浠呯湅鎴戞潈闄愯烦杩�
+        if (sysRoleVo != null && sysRoleVo.getIsRersonal() != null && sysRoleVo.getIsRersonal().equals(1)) {
+            testUserId = SecurityUtils.getUserId().intValue();
+        }
+
+        return baseMapper.deviceRecordPage(deviceId, page, sampleCode, managementNumber, testUserId);
     }
 
 

--
Gitblit v1.9.3