关于Spring Boot动态权限变更问题的实现方案

时间:2021-09-17 11:50:24

1、前言

​  在Web项目中,权限管理即权限访问控制为网站访问安全提供了保障,并且很多项目使用了Session作为缓存,结合AOP技术进行token认证和权限控制。权限控制流程大致如下图所示:

关于Spring Boot动态权限变更问题的实现方案

​  现在,如果管理员修改了用户的角色,或修改了角色的权限,都会导致用户权限发生变化,此时如何实现动态权限变更,使得前端能够更新用户的权限树,后端访问鉴权AOP模块能够知悉这种变更呢?

2、问题及解决方案

​​  现在的问题是,管理员没法访问用户Session,因此没法将变更通知此用户。而用户如果已经登录,或直接关闭浏览器页面而不是登出操作,Session没有过期前,用户访问接口时,访问鉴权AOP模块仍然是根据之前缓存的Session信息进行处理,没法做到动态权限变更。

​​  使用Security+WebSocket是一个方案,但没法处理不在线用户。

​​  ​解决方案的核心思想是利用ServletContext对象的共享特性,来实现用户权限变更的信息传递。然后在AOP类中查询用户是否有变更通知记录需要处理,如果权限发生变化,则修改response消息体,添加附加通知信息给前端。前端收到附加的通知信息,可更新功能权限树,并进行相关处理。

​​​  这样,利用的变更通知服务,不仅后端的用户url访问接口可第一时间获悉变更,还可以通知到前端,从而实现了动态权限变更。

3、方案实现

3.1、开发变更通知类

​​​  服务接口类ChangeNotifyService,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package com.abc.questInvest.service;
 
/**
 * @className       : ChangeNotifyService
 * @description     : 变更通知服务
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/28   1.0.0       sheng.zheng     初版
 *
 */
public interface ChangeNotifyService {
 
    /**
     *
     * @methodName      : getChangeNotifyInfo
     * @description     : 获取指定用户ID的变更通知信息
     * @param userId    : 用户ID
     * @return      : 返回0表示无变更通知信息,其它值按照bitmap编码。目前定义如下:
     *      bit0:   : 修改用户的角色组合值,从而导致权限变更;
     *      bit1:   : 修改角色的功能项,从而导致权限变更;
     *      bit2:   : 用户禁用,从而导致权限变更;
     *      bit3:   : 用户调整部门,从而导致数据权限变更;
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/28   1.0.0       sheng.zheng     初版
     *
     */
    public Integer getChangeNotifyInfo(Integer userId);
    
    /**
     *
     * @methodName      : setChangeNotifyInfo
     * @description     : 设置变更通知信息
     * @param userId    : 用户ID
     * @param changeNotifyInfo  : 变更通知值
     *      bit0:   : 修改用户的角色组合值,从而导致权限变更;
     *      bit1:   : 修改角色的功能项,从而导致权限变更;
     *      bit2:   : 用户禁用,从而导致权限变更;
     *      bit3:   : 用户调整部门,从而导致数据权限变更;
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/28   1.0.0       sheng.zheng     初版
     *
     */
    public void setChangeNotifyInfo(Integer userId,Integer changeNotifyInfo);  
}

​​​  服务实现类ChangeNotifyServiceImpl,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package com.abc.questInvest.service.impl;
 
import java.util.HashMap;
import java.util.Map;
 
import org.springframework.stereotype.Service;
 
import com.abc.questInvest.service.ChangeNotifyService;
 
/**
 * @className       : ChangeNotifyServiceImpl
 * @description     : ChangeNotifyService实现类
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/28   1.0.0       sheng.zheng     初版
 *
 */
@Service
public class ChangeNotifyServiceImpl implements ChangeNotifyService {
    
    //用户ID与变更过通知信息映射表
    private Map<Integer,Integer> changeNotifyMap = new HashMap<Integer,Integer>();
    
    /**
     *
     * @methodName      : getChangeNotifyInfo
     * @description     : 获取指定用户ID的变更通知信息
     * @param userId    : 用户ID
     * @return      : 返回0表示无变更通知信息,其它值按照bitmap编码。目前定义如下:
     *      bit0:   : 修改用户的角色组合值,从而导致权限变更;
     *      bit1:   : 修改角色的功能项,从而导致权限变更;
     *      bit2:   : 用户禁用,从而导致权限变更;
     *      bit3:   : 用户调整部门,从而导致数据权限变更;
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/28   1.0.0       sheng.zheng     初版
     *
     */
    @Override
    public Integer getChangeNotifyInfo(Integer userId) {
        Integer changeNotifyInfo = 0;
        //检查该用户是否有变更通知信息
        if (changeNotifyMap.containsKey(userId)) {
            changeNotifyInfo = changeNotifyMap.get(userId);
            //移除数据,加锁保护
            synchronized(changeNotifyMap) {
                changeNotifyMap.remove(userId);
            }
        }
        return changeNotifyInfo;
    }
    
    /**
     *
     * @methodName      : setChangeNotifyInfo
     * @description     : 设置变更通知信息,该功能一般由管理员触发调用
     * @param userId    : 用户ID
     * @param changeNotifyInfo  : 变更通知值
     *      bit0:   : 修改用户的角色组合值,从而导致权限变更;
     *      bit1:   : 修改角色的功能项,从而导致权限变更;
     *      bit2:   : 用户禁用,从而导致权限变更;
     *      bit3:   : 用户调整部门,从而导致数据权限变更;
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/28   1.0.0       sheng.zheng     初版
     *
     */
    @Override
    public void setChangeNotifyInfo(Integer userId,Integer changeNotifyInfo) {
        //检查该用户是否有变更通知信息
        if (changeNotifyMap.containsKey(userId)) {
            //如果有,表示之前变更通知未处理
            //获取之前的值
            Integer oldChangeNotifyInfo = changeNotifyMap.get(userId);
            //计算新值。bitmap编码,或操作
            Integer newChangeNotifyInfo = oldChangeNotifyInfo | changeNotifyInfo;
            //设置数据,加锁保护
            synchronized(changeNotifyMap) {
                changeNotifyMap.put(userId,newChangeNotifyInfo);
            }
        }else {
            //如果没有,设置一条
            changeNotifyMap.put(userId,changeNotifyInfo);
        }
    }
}

​​  此处,变更通知类型,与使用的demo项目有关,目前定义了4种变更通知类型。实际上,除了权限相关的变更,还有与Session缓存字段相关的变更,也需要通知,否则用户还是在使用旧数据。

3.2、将变更通知类对象,纳入全局配置服务对象中进行管理

​​​  全局配置服务类GlobalConfigService,负责管理全局的配置服务对象,服务接口类代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package com.abc.questInvest.service;
 
/**
 * @className       : GlobalConfigService
 * @description     : 全局变量管理类
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/02   1.0.0       sheng.zheng     初版
 *
 */
public interface GlobalConfigService {
    
    /**
     *
     * @methodName      : loadData
     * @description     : 加载数据
     * @return      : 成功返回true,否则返回false
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/02   1.0.0       sheng.zheng     初版
     *
     */
    public boolean loadData();
    
    //获取TableCodeConfigService对象
    public TableCodeConfigService getTableCodeConfigService(); 
    
    //获取SysParameterService对象
    public SysParameterService getSysParameterService();
    
    //获取FunctionTreeService对象
    public FunctionTreeService getFunctionTreeService();
 
    //获取RoleFuncRightsService对象
    public RoleFuncRightsService getRoleFuncRightsService();
    
    //获取ChangeNotifyService对象
    public ChangeNotifyService getChangeNotifyService();
    
}

​​​  服务实现类GlobalConfigServiceImpl,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package com.abc.questInvest.service.impl;
 
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
import com.abc.questInvest.service.ChangeNotifyService;
import com.abc.questInvest.service.FunctionTreeService;
import com.abc.questInvest.service.GlobalConfigService;
import com.abc.questInvest.service.RoleFuncRightsService;
import com.abc.questInvest.service.SysParameterService;
import com.abc.questInvest.service.TableCodeConfigService;
 
/**
 * @className       : GlobalConfigServiceImpl
 * @description     : GlobalConfigService实现类
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/02   1.0.0       sheng.zheng     初版
 *
 */
@Service
public class GlobalConfigServiceImpl implements GlobalConfigService{
    
    //ID编码配置表数据服务
    @Autowired
    private TableCodeConfigService tableCodeConfigService;
    
    //系统参数表数据服务
    @Autowired
    private SysParameterService sysParameterService;
    
    //功能树表数据服务
    @Autowired
    private FunctionTreeService functionTreeService;
    
    //角色权限表数据服务
    @Autowired
    private RoleFuncRightsService roleFuncRightsService;
    
    //变更通知服务
    @Autowired
    private ChangeNotifyService changeNotifyService;
    
    
    /**
     *
     * @methodName      : loadData
     * @description     : 加载数据
     * @return      : 成功返回true,否则返回false
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/02   1.0.0       sheng.zheng     初版
     *
     */
    @Override
    public boolean loadData() {
        boolean bRet = false;
        
        //加载table_code_config表记录
        bRet = tableCodeConfigService.loadData();
        if (!bRet) {
            return bRet;
        }
        
        //加载sys_parameters表记录
        bRet = sysParameterService.loadData();
        if (!bRet) {
            return bRet;
        }
        
        //changeNotifyService目前没有持久层,无需加载
        //如果服务重启,信息丢失,也没关系,因为此时Session也会失效
        
        //加载function_tree表记录
        bRet = functionTreeService.loadData();
        if (!bRet) {
            return bRet;
        }
        
        //加载role_func_rights表记录
        //先设置完整功能树
        roleFuncRightsService.setFunctionTree(functionTreeService.getFunctionTree());
        //然后加载数据
        bRet = roleFuncRightsService.loadData();
        if (!bRet) {
            return bRet;
        }
        
        return bRet;
    }
    
    //获取TableCodeConfigService对象
    @Override
    public TableCodeConfigService getTableCodeConfigService() {
        return tableCodeConfigService;
    }
    
    //获取SysParameterService对象
    @Override
    public SysParameterService getSysParameterService() {
        return sysParameterService;
    }
    
    //获取FunctionTreeService对象
    @Override
    public FunctionTreeService getFunctionTreeService() {
        return functionTreeService;
    }  
    
    //获取RoleFuncRightsService对象
    @Override
    public RoleFuncRightsService getRoleFuncRightsService() {
        return roleFuncRightsService;
    }
    
    //获取ChangeNotifyService对象
    @Override
    public ChangeNotifyService getChangeNotifyService() {
        return changeNotifyService;
    }
 
}

​​  GlobalConfigServiceImpl类,管理了很多配置服务类,此处主要关注ChangeNotifyService类对象。

3.3、使用ServletContext,管理全局配置服务类对象

​​​  全局配置服务类在应用启动时加载到Spring容器中,这样可实现共享,减少对数据库的访问压力。

​​​  实现一个ApplicationListener类,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package com.abc.questInvest;
 
import javax.servlet.ServletContext;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Component;
import org.springframework.web.context.WebApplicationContext;
 
import com.abc.questInvest.service.GlobalConfigService;
 
/**
 * @className   : ApplicationStartup
 * @description : 应用侦听器
 *
 */
@Component
public class ApplicationStartup implements ApplicationListener<ContextRefreshedEvent>{
    //全局变量管理对象,此处不能自动注入
    private GlobalConfigService globalConfigService = null;
    
    @Override
    public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
        try {
            if(contextRefreshedEvent.getApplicationContext().getParent() == null){
                //root application context 没有parent.
                
                System.out.println("========定义全局变量==================");
                // 将 ApplicationContext 转化为 WebApplicationContext
                WebApplicationContext webApplicationContext =
                        (WebApplicationContext)contextRefreshedEvent.getApplicationContext();
                // 从 webApplicationContext 中获取  servletContext
                ServletContext servletContext = webApplicationContext.getServletContext();
                
                //加载全局变量管理对象
                globalConfigService = (GlobalConfigService)webApplicationContext.getBean(GlobalConfigService.class);
                //加载数据
                boolean bRet = globalConfigService.loadData();
                if (false == bRet) {
                    System.out.println("加载全局变量失败");
                    return;
                }       
                //======================================================================
                // servletContext设置值
                servletContext.setAttribute("GLOBAL_CONFIG_SERVICE", globalConfigService); 
                
            }
        } catch (Exception e) {
            e.printStackTrace();
        }       
    }
}

​​​  在启动类中,加入该应用侦听器ApplicationStartup。

?
1
2
3
4
5
public static void main(String[] args) {
        SpringApplication springApplication = new SpringApplication(QuestInvestApplication.class);
        springApplication.addListeners(new ApplicationStartup());
        springApplication.run(args); 
    }

​​  现在,有了一个GlobalConfigService类型的全局变量globalConfigService。

3.4、发出变更通知

​​​  此处举2个例子,说明发出变更通知的例子,这两个例子,都在用户管理模块,UserManServiceImpl类中。

​​​  1)管理员修改用户信息,可能导致权限相关项发生变动,2)禁用用户,发出变更过通知。

​​​  发出通知的相关代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/**
     *
     * @methodName      : editUser
     * @description     : 修改用户信息
     * @param userInfo  : 用户信息对象
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/08   1.0.0       sheng.zheng     初版
     * 2021/06/28   1.0.1       sheng.zheng     增加变更通知的处理
     *
     */
    @Override
    public void editUser(HttpServletRequest request,UserInfo userInfo) {
        //输入参数校验
        checkValidForParams("editUser",userInfo);
        
        //获取操作人账号
        String operatorName = (String) request.getSession().getAttribute("username");
        userInfo.setOperatorName(operatorName);    
 
        //登录名和密码不修改
        userInfo.setLoginName(null);
        userInfo.setSalt(null);
        userInfo.setPasswd(null);
        
        //获取修改之前的用户信息
        Integer userId = userInfo.getUserId();
        UserInfo oldUserInfo = userManDao.selectUserByKey(userId);
 
        //修改用户记录
        try {
            userManDao.updateSelective(userInfo);          
        }catch(Exception e) {
            e.printStackTrace();
            log.error(e.getMessage());
            throw new BaseException(ExceptionCodes.USERS_EDIT_USER_FAILED);
        }
        
        //检查是否有需要通知的变更
        Integer changeFlag = 0;
        if (userInfo.getRoles() != null) {
            if(oldUserInfo.getRoles() != userInfo.getRoles()) {
                //角色组合有变化,bit0
                changeFlag |= 0x01;
            }
        }
        if (userInfo.getDeptId() != null) {
            if (oldUserInfo.getDeptId() != userInfo.getDeptId()) {
                //部门ID有变化,bit3
                changeFlag |= 0x08;
            }
        }
        if (changeFlag > 0) {
            //如果有变更过通知项
            //获取全局变量
            ServletContext servletContext = request.getServletContext();
            GlobalConfigService globalConfigService = (GlobalConfigService)servletContext.getAttribute("GLOBAL_CONFIG_SERVICE");
            globalConfigService.getChangeNotifyService().setChangeNotifyInfo(userId, changeFlag);          
        }
    }
 
    /**
     *
     * @methodName      : disableUser
     * @description     : 禁用用户
     * @param params    : map对象,形式如下:
     *  {
     *      "userId"    : 1
     *  }
     * @history     :
     * ------------------------------------------------------------------------------
     * date         version     modifier        remarks                  
     * ------------------------------------------------------------------------------
     * 2021/06/08   1.0.0       sheng.zheng     初版
     * 2021/06/28   1.0.1       sheng.zheng     增加变更通知的处理
     *
     */
    @Override
    public void disableUser(HttpServletRequest request,Map<String,Object> params) {
        //输入参数校验
        checkValidForParams("disableUser",params);
        
        UserInfo userInfo = new UserInfo();
        
        //获取操作人账号
        String operatorName = (String) request.getSession().getAttribute("username");
        
        //设置userInfo信息
        Integer userId = (Integer)params.get("userId");
        userInfo.setUserId(userId);
        userInfo.setOperatorName(operatorName);
        //设置禁用标记
        userInfo.setDeleteFlag((byte)1);
        
        //修改密码
        try {
            userManDao.updateEnable(userInfo);         
        }catch(Exception e) {
            e.printStackTrace();
            log.error(e.getMessage());
            throw new BaseException(ExceptionCodes.USERS_EDIT_USER_FAILED);
        }      
        
        //禁用用户,发出变更通知
        //获取全局变量
        ServletContext servletContext = request.getServletContext();
        GlobalConfigService globalConfigService = (GlobalConfigService)servletContext.getAttribute("GLOBAL_CONFIG_SERVICE");
        //禁用用户:bit2
        globalConfigService.getChangeNotifyService().setChangeNotifyInfo(userId, 0x04);            
    }

​​  本demo项目的角色相对较少,没有使用用户角色关系表,而是使用了bitmap编码,角色ID取值为2^n,用户角色组合roles字段为一个Integer值。如roles=7,表示角色ID组合=[1,2,4]。
​​  另外,如果修改了角色的功能权限集合,则需要查询受影响的用户ID列表,依次发出通知,可类似处理。

3.5、修改Response响应消息体

​​​  Response响应消息体,为BaseResponse,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
package com.abc.questInvest.vo.common;
 
import lombok.Data;
 
/**
 * @className       : BaseResponse
 * @description     : 基本响应消息体对象
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/05/31   1.0.0       sheng.zheng     初版
 * 2021/06/28   1.0.1       sheng.zheng     增加变更通知的附加信息
 *
 */
@Data
public class BaseResponse<T> {
    //响应码
    private int code;
 
    //响应消息
    private String message;
        
    //响应实体信息
    private T data;
 
    //分页信息
    private Page page;
 
    //附加通知信息
    private Additional additional;
}

​​  BaseResponse类增加了Additional类型的additional属性字段,用于输出附加信息。

​​  Additional类的定义如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package com.abc.questInvest.vo.common;
 
import lombok.Data;
 
/**
 * @className       : Additional
 * @description     : 附加信息
 * @summary     :
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/28   1.0.0       sheng.zheng     初版
 *
 */
@Data
public class Additional {
    //通知码,附加信息
    private int notifycode;
 
    //通知码对应的消息
    private String notification;
    
    //更新的token
    private String token;
    
    //更新的功能权限树
    private String rights;
 
}

​​  附加信息类Additional中,各属性字段的说明:

  • notifycode,为通知码,即可对应通知消息的类型,目前只有一种,可扩展。
  • notification,为通知码对应的消息。

​​  通知码,在ExceptionCodes枚举文件中定义:

?
1
2
3
4
5
6
7
8
9
//变更通知信息
  USER_RIGHTS_CHANGED(51, "message.USER_RIGHTS_CHANGED", "用户权限发生变更"),
  //end enum
 
  ExceptionCodes(int code, String messageId, String message) {
      this.code = code;
      this.messageId = messageId;
      this.message = message;
  }
  • token,用于要求前端更新token。更新token的目的是确认前端已经收到权限变更通知。因为下次url请求将使用新的token,如果前端未收到或未处理,仍然用旧的token访问,就要跳到登录页了。
  • rights,功能树的字符串输出,是树型结构的JSON字符串。

3.6、AOP鉴权处理

​​​  AuthorizationAspect为鉴权认证的切面类,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
package com.abc.questInvest.aop;
 
import java.util.List;
 
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
 
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import com.abc.questInvest.common.constants.Constants;
import com.abc.questInvest.common.utils.Utility;
import com.abc.questInvest.dao.UserManDao;
import com.abc.questInvest.entity.FunctionInfo;
import com.abc.questInvest.entity.UserInfo;
import com.abc.questInvest.exception.BaseException;
import com.abc.questInvest.exception.ExceptionCodes;
import com.abc.questInvest.service.GlobalConfigService;
import com.abc.questInvest.service.LoginService;
import com.abc.questInvest.vo.TreeNode;
import com.abc.questInvest.vo.common.Additional;
import com.abc.questInvest.vo.common.BaseResponse;
 
/**
 * @className       : AuthorizationAspect
 * @description     : 接口访问鉴权切面类
 * @summary     : 使用AOP,进行token认证以及用户对接口的访问权限鉴权
 * @history     :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/06   1.0.0       sheng.zheng     初版
 * 2021/06/28   1.0.1       sheng.zheng     增加变更通知的处理,增加了afterReturning增强
 *
 */
@Aspect
@Component
@Order(2)
public class AuthorizationAspect {
    @Autowired
    private UserManDao userManDao;
    
    //设置切点
    @Pointcut("execution(public * com.abc.questInvest.controller..*.*(..))" +
    "&& !execution(public * com.abc.questInvest.controller.LoginController.*(..))" +
    "&& !execution(public * com.abc.questInvest.controller.QuestInvestController.*(..))")   
    public void verify(){}
    
    @Before("verify()")
    public void doVerify(){
        ServletRequestAttributes attributes=(ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
 
        HttpServletRequest request=attributes.getRequest();
        
        // ================================================================================
        // token认证
        
        //从header中获取token值
        String token = request.getHeader("Authorization");
        if (null == token || token.equals("")){
            //return;
            throw new BaseException(ExceptionCodes.TOKEN_IS_NULL);
        }
        
        //从session中获取token和过期时间
        String sessionToken = (String)request.getSession().getAttribute("token");
        
        //判断session中是否有信息,可能是非登录用户
        if (null == sessionToken || sessionToken.equals("")) {
            throw new BaseException(ExceptionCodes.TOKEN_WRONG);
        }
        
        //比较token
        if(!token.equals(sessionToken)) {
            //如果请求头中的token与存在session中token两者不一致
            throw new BaseException(ExceptionCodes.TOKEN_WRONG);           
        }
        
        long expireTime = (long)request.getSession().getAttribute("expireTime");
        //检查过期时间
        long time = System.currentTimeMillis();
        if (time > expireTime) {
            //如果token过期
            throw new BaseException(ExceptionCodes.TOKEN_EXPIRED);
        }else {
            //token未过期,更新过期时间
            long newExpiredTime = time + Constants.TOKEN_EXPIRE_TIME * 1000;
            request.getSession().setAttribute("expireTime", newExpiredTime);
        }
        
        // ============================================================================
        // 接口调用权限
        //获取用户ID
        Integer userId = (Integer)request.getSession().getAttribute("userId");
        //获取全局变量
        ServletContext servletContext = request.getServletContext();
        GlobalConfigService globalConfigService = (GlobalConfigService)servletContext.getAttribute("GLOBAL_CONFIG_SERVICE");
        
        //===================变更通知处理开始==============================================
        //检查有无变更通知信息
        Integer changeNotifyInfo = globalConfigService.getChangeNotifyService().getChangeNotifyInfo(userId);
        //通知前端权限变更的标记
        boolean rightsChangedFlag = false;     
        if (changeNotifyInfo > 0) {
            //有通知信息
            if ((changeNotifyInfo & 0x09) > 0) {
                //bit0:修改用户的角色组合值,从而导致权限变更
                //bit3:用户调整部门,从而导致数据权限变更
                //mask 0b1001 = 0x09
                //都需要查询用户表,并更新信息;合在一起查询。
                UserInfo userInfo = userManDao.selectUserByKey(userId);
                //更新Session
                        request.getSession().setAttribute("roles", userInfo.getRoles());
                        request.getSession().setAttribute("deptId", userInfo.getDeptId()); 
                        if ((changeNotifyInfo & 0x01) > 0) {
                            //权限变更标志置位
                            rightsChangedFlag = true;
                        }
            }else if((changeNotifyInfo & 0x02) > 0) {
                //bit1:修改角色的功能值,从而导致权限变更
                        //权限变更标志置位
                      rightsChangedFlag = true;
            }else if((changeNotifyInfo & 0x04) > 0) {
                //bit2:用户禁用,从而导致权限变更
                //设置无效token,可阻止该用户访问系统
                request.getSession().setAttribute("token", "");
                //直接抛出异常,由前端显示:Forbidden页面
                throw new BaseException(ExceptionCodes.ACCESS_FORBIDDEN);
            }
            if (rightsChangedFlag == true) {
                //写Session,用于将信息传递到afterReturning方法中
                request.getSession().setAttribute("rightsChanged", 1);
            }
        }
        //===================变更通知处理结束==============================================
                
        //从session中获取用户权限值
        Integer roles = (Integer)request.getSession().getAttribute("roles");
        //获取当前接口url值
        String servletPath = request.getServletPath();
                
        //获取该角色对url的访问权限
        Integer rights = globalConfigService.getRoleFuncRightsService().getRoleUrlRights(Utility.parseRoles(roles), servletPath);
        if (rights == 0) {
            //如果无权限访问此接口,抛出异常,由前端显示:Forbidden页面
            throw new BaseException(ExceptionCodes.ACCESS_FORBIDDEN);
        }      
    }   
    
    @AfterReturning(value="verify()" ,returning="result")
    public void afterReturning(BaseResponse result) {
        //限制必须是BaseResponse类型,其它类型的返回值忽略
        //获取Session
        ServletRequestAttributes sra = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = sra.getRequest();
        Integer rightsChanged = (Integer)request.getSession().getAttribute("rightsChanged");
        if (rightsChanged != null && rightsChanged == 1) {
            //如果有用户权限变更,通知前端来刷新该用户的功能权限树
            //构造附加信息
            Additional additional = new Additional();
            additional.setNotifycode(ExceptionCodes.USER_RIGHTS_CHANGED.getCode());
            additional.setNotification(ExceptionCodes.USER_RIGHTS_CHANGED.getMessage());
            //更新token
            String loginName = (String)request.getSession().getAttribute("username");
            String token = LoginService.generateToken(loginName);
            additional.setToken(token);
            //更新token,要求下次url访问使用新的token
            request.getSession().setAttribute("token", token);
            //获取用户的功能权限树
            Integer roles = (Integer)request.getSession().getAttribute("roles");
            ServletContext servletContext = request.getServletContext();
            GlobalConfigService globalConfigService = (GlobalConfigService)servletContext.getAttribute("GLOBAL_CONFIG_SERVICE");
            //获取用户权限的角色功能树
            List<Integer> roleList = Utility.parseRoles(roles);
            TreeNode<FunctionInfo> rolesFunctionTree =
                    globalConfigService.getRoleFuncRightsService().
                    getRoleRights(roleList);
            additional.setRights(rolesFunctionTree.toString());
            //修改response信息
            result.setAdditional(additional);
            //移除Session的rightsChanged项
            request.getSession().removeAttribute("rightsChanged");
        }
    }
}

​​  AuthorizationAspect类定义了切点verify(),@Before增强用于鉴权验证,增加了对变更通知信息的处理。并利用Session,用rightsChanged属性字段记录需要通知前端的标志,在@AfterReturning后置增强中根据该属性字段的值,进行一步的处理。

​​  @Before增强的doVerify方法中,如果发现角色组合有改变,但仍有访问此url权限时,会继续后续处理,这样不会中断业务;如果没有访问此url权限,则返回访问受限异常信息,由前端显示访问受限页码(类似403 Forbidden 页码)。

​​  在后置增强@AfterReturning中,限定了返回值类型,如果该请求响应的类型是BaseResponse类型,则修改reponse消息体,附加通知信息;如果不是,则不处理,会等待下一个url请求,直到返回类型是BaseResponse类型。也可以采用自定义response的header的方式,这样,就无需等待了。

​​  generateToken方法,是LoginService类的静态方法,用于生成用户token。

​​  至于Utility的parseRoles方法,是将bitmap编码的roles解析为角色ID的列表,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
//========================= 权限组合值解析 ======================================     
/**
 *
 * @methodName      : parseRoles
 * @description     : 解析角色组合值
 * @param roles     : 按位设置的角色组合值
 * @return          : 角色ID列表
 * @history         :
 * ------------------------------------------------------------------------------
 * date         version     modifier        remarks                  
 * ------------------------------------------------------------------------------
 * 2021/06/24   1.0.0       sheng.zheng     初版
 *
 */
public static List<Integer> parseRoles(int roles){
    List<Integer> roleList = new ArrayList<Integer>();
 
    int newRoles = roles;
    int bit0 = 0;
    int roleId = 0;
    for (int i = 0; i < 32; i++) {
        //如果组合值的余位都为0,则跳出
        if (newRoles == 0) {
            break;
        }
        
        //取得最后一位
        bit0 = newRoles & 0x01;
        if (bit0 == 1) {
            //如果该位为1,左移i位
            roleId = 1 << i;
            roleList.add(roleId);
        }
        
        //右移一位
        newRoles = newRoles >> 1;
    }
    return roleList;
}  

​​  getRoleRights方法,是角色功能权限服务类RoleFuncRightsService的方法,它提供了根据List类型的角色ID列表,快速获取功能权限树的功能。
​​  关于功能权限树TreeNode类型,请参阅:《Java通用树结构数据管理》。

到此这篇关于Spring Boot动态权限变更实现的整体方案的文章就介绍到这了,更多相关Spring Boot动态权限内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://www.cnblogs.com/alabo1999/archive/2021/06/29/14948914.html