haigeek blog


并发登录下session的处理

在进行性能测试的时候并发请求登录接口时候出现失败的情况,对排查记录做一下总结。

并发登录下session的处理

问题

在进行性能测试的时候并发请求登录接口,除登录接口之外的接口保护策略是使用过滤器从cookie获取用户信息并和服务端进行后续的数据交互。

设置一个线程组,线程主要有两个接口

  1. 登录接口,登录成功之后系统会返回一个set-cookie的响应头
  2. 调用被保护的接口,接口在请求时会携带cookie来让服务器鉴别当前请求是否合法

线程组的具体设置如下:

image-20190925090956992

标记为3 的接口访问进行了用户鉴权。

当执行此线程组一次的时候,运行正常,运行结果如下:

image-20190924231443836

将线程组的线程数调整为2,Rame-Up(线程启动时间间隔)设置为1s,两组线程执行正常,运行结果如下:

image-20190925091109920

可以看到执行顺序是按照我们1、2、3依次执行的顺序,因为1s的间隔较长,在线程1执行完毕之后线程2执行。

将线程组的线程数调整为2,Rame-Up(线程启动时间间隔)设置为0.1s,接口请求出现异常,运行结果如下:

image-20190925091517378

在这次请求可以看到,两个线程同时出发,模拟了系统可以存在的并发情况,可以确认的是第二个用户的确登录成功但是在与后台进行校验的时候出现了问题。

分析

通过报错的接口的异常信息定位到代码的位置是在从session中取用户信息的时候出现了错误。

image-20190925092015591

框选的代码为报错行,联系上下文猜测session并没有获取成功或session不存在。因此加入箭头所指代码进行debug。

发现在此处会报空指针异常,但是session是确实存在的。

检查redis中存储的session信息,发现有一个session缺少要存储的用户等关键信息

不完整的session信息:

不完整session

存储正常的session信息:

image-20190925092738929

因此定位到是session在存储的时候出了问题。

解决

检查session存储的代码,发现登录接口使用的session是一个共享的变量,因为 spring 的每个 controller 默认都是单例的,这个session 会被其他线程给共享,在多线程的情况下,极容易出现线程不安全的问题。

现有的session处理方式如下:

 1private HttpSession session;
 2private HttpServletRequest request;
 3
 4@ModelAttribute
 5public void init(HttpSession httpSession, HttpServletRequest request) {
 6	this.session = httpSession;
 7	this.request = request;
 8}
 9//登录controller
10 @PostMapping(value = "public/client-account/v1/login-dev")
11@ApiOperation(value = "用户登录校验(明文)")
12public ResponseData loginDev(@RequestBody OmsLoginDTO loginDTO) throws Exception {
13  OmsUserVO omsUserVO = this.userFacade.checkLogin(loginDTO, false);
14  this.userFacade.log(omsUserVO, request);
15  sessionCache(omsUserVO, session);
16  return ResponseUtil.success(omsUserVO);
17}
18/**
19     * 缓存用户session
20     *
21     * @param omsUserVO
22     * @throws Exception
23     */
24private void sessionCache(OmsUserVO omsUserVO) throws Exception {
25  if (repeatLogin.equals("true")) {
26    httpSession.setAttribute(FindByIndexNameSessionRepository.PRINCIPAL_NAME_INDEX_NAME, omsUserVO.getLoginName());
27  }
28  httpSession.setAttribute("user", JSONObject.toJSONString(omsUserVO));
29  if (null != omsUserVO.getRoles() && omsUserVO.getRoles().size() > 0) {
30    List<String> roleNameList = omsUserVO.getRoles().stream()
31      .map(OmsRoleVO::getRoleName)
32      .collect(Collectors.toList());
33    session.setAttribute("role", JSONObject.toJSONString(roleNameList));
34  }
35}

解决方式有三种:

首先介绍最简单的一种,

使用@Autowired注解

使用@Autowired代替@ModelAttribute

 1@Autowired
 2private HttpServletRequest request;
 3@Autowired
 4private HttpSession session;
 5    
 6//@ModelAttribute
 7//public void init(HttpSession httpSession, HttpServletRequest request) {
 8  //this.session = httpSession;
 9  //this.request = request;
10//}

对上述代码进行调试

image-20190925110213553

加上了@Autowired注解之后进行debug发现,request并不是原始的HttpServletRequest对象,而是HttpServletRequest的一个代理类。找到代理类的实现如下

image-20190925110822925

实际上Autowire进来的并不是原始的HttpServletRequest对象,而是HttpServletRequest的一个代理类。实际上它会通过

((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest()

到这一步骤已经很明显了,那如何看出是线程安全的,继续往下看。

image-20190925110859572

进一步去看RequestContextHolder发现RequestContextHolder是通过ThreadLocal来实现的,可以保证每个线程获取得到的Request对象一定是当前请求的Request对象,从而保证线程安全。

image-20190925111025094

一种是在登录相关接口将session作为参数传递到方法中使用

在请求和存储的时候加上单独的HttpSession session

 1@PostMapping(value = "public/client-account/v1/login-dev")
 2@ApiOperation(value = "用户登录校验(明文)")
 3public ResponseData loginDev(@RequestBody OmsLoginDTO loginDTO,HttpSession httpSession) throws Exception {
 4  OmsUserVO omsUserVO = this.userFacade.checkLogin(loginDTO, false);
 5  this.userFacade.log(omsUserVO, request);
 6  sessionCache(omsUserVO, httpSession);
 7  return ResponseUtil.success(omsUserVO);
 8}
 9
10/**
11     * 缓存用户session
12     *
13     * @param omsUserVO
14     * @throws Exception
15     */
16private void sessionCache(OmsUserVO omsUserVO) throws Exception {
17  if (repeatLogin.equals("true")) {
18    session.setAttribute(FindByIndexNameSessionRepository.PRINCIPAL_NAME_INDEX_NAME, omsUserVO.getLoginName());
19  }
20  session.setAttribute("user", JSONObject.toJSONString(omsUserVO));
21  if (null != omsUserVO.getRoles() && omsUserVO.getRoles().size() > 0) {
22    List<String> roleNameList = omsUserVO.getRoles().stream()
23      .map(OmsRoleVO::getRoleName)
24      .collect(Collectors.toList());
25    session.setAttribute("role", JSONObject.toJSONString(roleNameList));
26  }
27}

另外一种是将session使用ThreadLocal来处理

重点在于session的初始化设置

 1private static final ThreadLocal<HttpSession> session = new ThreadLocal<>();
 2@ModelAttribute
 3public void init(HttpSession httpSession, HttpServletRequest request) {
 4  ClientAccountController.session.set(httpSession);
 5  this.request = request;
 6}
 7//在涉及到session的获取时候需要使用session.get()
 8/**
 9     * 缓存用户session
10     * @param omsUserVO
11     * @throws Exception
12     */
13private void sessionCache(OmsUserVO omsUserVO) throws Exception {
14  if (repeatLogin.equals("true")) {
15    session.get().setAttribute(FindByIndexNameSessionRepository.PRINCIPAL_NAME_INDEX_NAME, omsUserVO.getLoginName());
16  }
17  session.get().setAttribute("user", JSONObject.toJSONString(omsUserVO));
18  if (null != omsUserVO.getRoles() && omsUserVO.getRoles().size() > 0) {
19    List<String> roleNameList = omsUserVO.getRoles().stream()
20      .map(OmsRoleVO::getRoleName)
21      .collect(Collectors.toList());
22    session.get().setAttribute("role", JSONObject.toJSONString(roleNameList));
23  }
24}

使用上述三种方式任意方式之后,最终请求正常

image-20190925094039469

继续调整请求线程为10,问题没有出现。

总结

1、在并发的情况下要注意考虑共享变量的问题

2、request、Response、session在单例controller是不安全的,采用ThreadLocal可以解决该问题。推荐直接使用spring帮助我们处理好的来使用。

3、建议修改现有的baseController。

参考链接:

https://www.cnblogs.com/kismetv/p/8757260.html#t4

https://segmentfault.com/q/1010000005139036