权限进行判断

zs-manager/src/main/java/com/ruoyi/zhushi/service/impl/DkRecordServiceImpl.java

@@ -5,7 +5,10 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.ruoyi.common.core.domain.PageQuery;
+import com.ruoyi.common.core.domain.dto.RoleDTO;
+import com.ruoyi.common.core.domain.model.LoginUser;
 import com.ruoyi.common.core.page.TableDataInfo;
+import com.ruoyi.common.helper.LoginHelper;
 import com.ruoyi.zhushi.entity.*;
 import com.ruoyi.zhushi.mapper.DkRecordMapper;
 import com.ruoyi.zhushi.service.DkRecordService;
@@ -46,7 +49,21 @@ public class DkRecordServiceImpl implements DkRecordService {
 
     @Override
     public TableDataInfo<DkCheckInRecordDTO> queryPageList(DkCheckInRecordDTO dkCheckInRecordDTO, PageQuery pageQuery) {
+        LoginUser loginUser = LoginHelper.getLoginUser();
         LambdaQueryWrapper<DkCheckInRecord> queryWrapper = buildQueryWrapper(dkCheckInRecordDTO, null);
+        if (loginUser == null) {
+            return TableDataInfo.build();
+        }else {
+            // 获取用户角色
+            List<RoleDTO> roles = loginUser.getRoles();
+            if (roles != null && roles.size() > 0) {
+                for (RoleDTO role : roles) {
+                    if (role.getRoleKey().equals(Constans.ROLEKEY)) {
+                        queryWrapper.eq(DkCheckInRecord::getSysUserId, loginUser.getUserId());
+                    }
+                }
+            }
+        }
         Page<DkCheckInRecordDTO> result = dkMapper.selectVoPage(pageQuery.build(), queryWrapper);
         return TableDataInfo.build(result);
     }

zs-manager/src/main/java/com/ruoyi/zhushi/util/Constans.java

@@ -34,4 +34,6 @@ public class Constans {
     public static final int ANNUALLEAVETIERS_10 = 10;
 
     public static final int ANNUALLEAVETIERS_15 = 15;
+
+    public static final String ROLEKEY = "common";
 }