From 05dbb19d1826ae4b888340d53b56746c1efa66cc Mon Sep 17 00:00:00 2001
From: guoyuji <guoyujie@ng.com>
Date: 星期六, 11 五月 2024 14:55:57 +0800
Subject: [PATCH] 登录时获取用户权限
---
north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserService.java | 9 ++++++++-
1 files changed, 8 insertions(+), 1 deletions(-)
diff --git a/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserService.java b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserService.java
index 5b4ede4..17e1f71 100644
--- a/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserService.java
+++ b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserService.java
@@ -5,8 +5,11 @@
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.example.erp.controller.dto.UserDTO;
+import com.example.erp.entity.userInfo.PermissionBasic;
import com.example.erp.entity.userInfo.Role;
import com.example.erp.entity.userInfo.User;
+import com.example.erp.mapper.userInfo.PermissionBasicMapper;
+import com.example.erp.mapper.userInfo.PermissionRoleMapper;
import com.example.erp.mapper.userInfo.RoleMapper;
import com.example.erp.mapper.userInfo.UserMapper;
import com.example.erp.tools.TokenTools;
@@ -28,9 +31,12 @@
private final UserMapper userMapper;
private final RoleMapper roleMapper;
- public UserService(UserMapper userMapper, RoleMapper roleMapper) {
+ private final PermissionRoleMapper permissionRoleMapper;
+
+ public UserService(UserMapper userMapper, RoleMapper roleMapper, PermissionRoleMapper permissionRoleMapper) {
this.userMapper = userMapper;
this.roleMapper = roleMapper;
+ this.permissionRoleMapper = permissionRoleMapper;
}
@@ -88,6 +94,7 @@
userDTO.setId(user.getId());
userDTO.setUserName(user.getUserName());
userDTO.setToken(StpUtil.getTokenValue());
+ userDTO.setPermissions(permissionRoleMapper.getUserEditPermission(user.getId()));
return userDTO;
}else{
return null;
--
Gitblit v1.8.0