异步执行获取不到登录态的问题排查(RequestContextHolder.getRequestAttributes()子线程获取问题)

问题描述

有一个同步接口,里面执行了OpenFeign远程调用,其中远程调用的系统和本系统是同一个SSO系统下的两个子系统,因为查询性能问题,其中优化的一个点是同步改异步,可是改成异步之后,一直报错远程Feign调用的时候是未登录状态。

先说原因

由于是SSO统一认证,因此远程调用必须有登录状态,而我们的登录会话保存在当前请求(也就是主线程)的HttpServletRequest中,通过RequestContextHolder.getRequestAttributes().getRequest()来获取,而通过查看源码,发现RequestAttributes对象是存在ThreadLocal线程本地变量的,那问题可想而知了,问题根本原因就是,创建的子线程无法获取到会话信息,因为会话信息存在在主线程的本地变量里。

第一处转载

转载自:https://blog.csdn.net/ypp91zr/article/details/114395178

RequestContextHolder.getRequestAttributes()子线程获取问题

相信很多开发过程中都用过RequestContextHolder.getRequestAttributes(),没错,我也经常用,但今天出现了问题,获取到的实例是空的,原因是因为我新开了一个子线程,在子线程调用了RequestContextHolder.getRequestAttributes()。实际获取到的是空的,然后查看了源码

ThreadLocal获取。一个请求到达容器后,Spring会把该请求Request实例通过setRequestAttributes方法 把Request实例放入该请求线程内ThreadLocalMap中,然后就可以通过静态方法取到。原理就是ThreadLocal,但ThreadLocal不能让子线程继承ThreadLocalMap信息,可以使用InherbritableThreadLocal实现子线程信息传递。
Spring Boot 默认使用ThreadLocal把Request设置进请求线程中,这样如果在请求方法里面另起一个子线程然后再通过getRequestAttributes方法获取,是获取不到的

如果想要在子线程获取,设置inheritable=true即可,但我一直没找到在哪里可以设置,于是自己就写了个工具类来让子线程获取,思路是自定义一个注解,拦截注解,将父线程的ServletRequestAttributes给InheritableThreadLocal,然后在子线程即可获取

package com.shinedata.annotation;
 
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
 
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface RequestInheritableThread {
}

AOP

@Aspect
@Component
public class RequestHolderAspect {
 
    @Before("@annotation(com.shinedata.annotation.RequestInheritableThread)")
    public void doBefore(JoinPoint joinPoint) {
        ServletRequestAttributes servletRequestAttributes = RequestHolder.getServletRequestAttributes();
        RequestHolder.setServletRequestAttributes(servletRequestAttributes);
    }
 
    @After("@annotation(com.shinedata.annotation.RequestInheritableThread)")
    public void doAfter(JoinPoint joinPoint) {
        RequestHolder.removeServletRequestAttributes();
    }
}

工具类

package com.shinedata.util.context;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.HashMap;
import java.util.Map;
import java.util.StringTokenizer;

public class RequestHolder {
 
    private static final Logger logger = LoggerFactory.getLogger(RequestHolder.class);
 
    public static final String GET = "GET";
    public static final String POST = "POST";
 
    public static final String UTF8 = "UTF-8";
 
    public static InheritableThreadLocal<ServletRequestAttributes> servletRequestAttributesInheritableThreadLocal= new InheritableThreadLocal();
 
    public static HttpServletRequest getRequest() {
        return getServletRequestAttributes().getRequest();
    }
 
    public static HttpServletResponse getResponse() {
        return getServletRequestAttributes().getResponse();
    }
 
    public static ServletRequestAttributes setServletRequestAttributes(ServletRequestAttributes servletRequestAttributes) {
        servletRequestAttributesInheritableThreadLocal.set(servletRequestAttributes);
        return servletRequestAttributes;
    }
 
    public static void removeServletRequestAttributes() {
        servletRequestAttributesInheritableThreadLocal.remove();
    }
 
    public static ServletRequestAttributes getServletRequestAttributes() {
        ServletRequestAttributes requestAttributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();
        if(requestAttributes==null){
            requestAttributes=servletRequestAttributesInheritableThreadLocal.get();
        }
        return requestAttributes;
    }
}

需要在子线程获取ServletRequestAttributes使用的时候在父方法贴个注解@RequestInheritableThread就行了,父方法里面即使开子线程,子线程里面也能获取ServletRequestAttributes

补充第二种解决办法,在开启新线程之前,将RequestAttributes对象设置为子线程共享

开启新线程之前,添加代码:
ServletRequestAttributes sra = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
RequestContextHolder.setRequestAttributes(sra, true);

第二处转载

转载自:https://blog.csdn.net/VIP099/article/details/108479717

ThreadLocal和InheritableThreadLocal的区别

InheritableThreadLocal的实现

在Thread的init方法中

复制线程数据

原理分析

  1. 知识点普及:Thread类有两个ThreadLocalMap局部变量(threadLocals,inheritableThreadLocals)
  2. 知识点普及:InheritableThreadLocal类继承了ThreadLocal类(此处普及主要是和上面自定义注解联系)
  3. 主线程Thread的局部变量存储在threadLocals里
  4. 创建子线程时,会判断主(父)线程的inheritableThreadLocals变量是否存在值,如果存在,则复制自己的ThreadLocalMap到子线程的ThreadLocalMap中
  5. 因此要想要子线程拥有主(父)线程的ThreadLocalMap的值,只要把主(父)线程的inheritableThreadLocals变量设置上就行了
posted @ 2023-07-07 15:25  修电脑的  阅读(14)  评论(0)    收藏  举报