java.lang.IllegalStateException：在异步方面使用 RequestContextHolder.currentRequestAttributes() 时未找到 thread-bound 请求

Question

我有下面这个方面，出于某些原因，您可以看到 here，不得不使用 @EnableAsync 和 @Async over aspect 方法，如下所示：

@Aspect
@Component
@EnableAsync
public class ApiCallLogAspect {

    @Async
    @AfterReturning(value = ("within(com.example..*.web.rest.api..*)"), returning = "returnValue")
    public void endpointAfterReturning(JoinPoint p, Object returnValue) throws InterruptedException {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
        System.out.println(request.getHeader("authorization"));
    }

}

现在，我得到这个例外：

java.lang.IllegalStateException: No thread-bound request found: Are you referring to request attributes outside of an actual web request, or processing a request outside of the originally receiving thread? If you are actually operating within a web request and still receive this message, your code is probably running outside of DispatcherServlet: In this case, use RequestContextListener or RequestContextFilter to expose the current request.
    at org.springframework.web.context.request.RequestContextHolder.currentRequestAttributes(RequestContextHolder.java:131) ~[spring-web-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at com.example.web.rest.api.ApiCallLogAspect.endpointAfterReturning(ApiCallLogAspect.java:40) ~[classes/:na]
    at com.example.web.rest.api.ApiCallLogAspect$$FastClassBySpringCGLIB$$f034ee12.invoke(<generated>) ~[classes/:na]
    at org.springframework.cglib.proxy.MethodProxy.invoke(MethodProxy.java:218) ~[spring-core-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at org.springframework.aop.framework.CglibAopProxy$CglibMethodInvocation.invokeJoinpoint(CglibAopProxy.java:771) ~[spring-aop-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at org.springframework.aop.framework.ReflectiveMethodInvocation.proceed(ReflectiveMethodInvocation.java:163) ~[spring-aop-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at org.springframework.aop.framework.CglibAopProxy$CglibMethodInvocation.proceed(CglibAopProxy.java:749) ~[spring-aop-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at org.springframework.aop.interceptor.AsyncExecutionInterceptor.lambda$invoke[=14=](AsyncExecutionInterceptor.java:115) ~[spring-aop-5.2.7.RELEASE.jar:5.2.7.RELEASE]
    at java.util.concurrent.FutureTask.run(FutureTask.java:266) ~[na:1.8.0_162]
    at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1149) ~[na:1.8.0_162]
    at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:624) ~[na:1.8.0_162]
    at java.lang.Thread.run(Thread.java:748) ~[na:1.8.0_162]

我该如何解决这个问题？有没有办法从

获取请求headers

Answer 1

我通过以下更改解决了这个问题，@M.Deinum 评论：

public class ThreadContextHolder {
    private ThreadContextHolder() {
    }

    private static final ThreadLocal<Map<String, Object>> ctx = new ThreadLocal<>();

    public static Map<String, Object> getContext() {
        return ctx.get();
    }

    public static void setContext(Map<String, Object> attrs) {
        ctx.set(attrs);
    }

    public static void removeContext() {
        ctx.remove();
    }
}

然后注入taskExecutor到ApiCallLogAspect class:

@Bean
public Executor taskExecutor() {
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();

    executor.setTaskDecorator(
            runnable -> {
                HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();

                Map<String, Object> headers = new HashMap<>();
                for (String header : Collections.list(request.getHeaderNames())) {
                    headers.put(header, request.getHeader(header));
                }

                return () -> {
                    try {
                        ThreadContextHolder.setContext(headers);
                        runnable.run();
                    } finally {
                        ThreadContextHolder.removeContext();
                    }
                };
            });

    executor.initialize();
    return executor;
}

然后将 endpointAfterReturning 方法更改为：

@Async
@AfterReturning(value = ("within(com.example..*.web.rest.api..*)"), returning = "returnValue")
public void endpointAfterReturning(JoinPoint p, Object returnValue) throws InterruptedException {
    Map<String, String> headers = new HashMap<>();
    ThreadContextHolder.getContext().forEach((s, o) -> headers.put(s, o.toString()));
    System.out.println(headers.get("authorization"));
}

现在，一切正常。如果此解决方案有问题，请通知我。

Answer 2

我通过在切入点表达式中包含以下注释解决了这个问题

    @Pointcut("within(@org.springframework.stereotype.Repository *)"
        + " || within(@org.springframework.stereotype.Service *)"
        + "|| within(@org.springframework.stereotype.Component *)"
        + " || within(@org.springframework.web.bind.annotation.RestController *)")
public void springBeanPointcut() {
}

并将其分配给如下建议

    @Around("myPointcut() && springBeanPointcut()")
    public Object applicationLogger(ProceedingJoinPoint pjp) throws Throwable {
   HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();

}

java.lang.IllegalStateException：在异步方面使用 RequestContextHolder.currentRequestAttributes() 时未找到 thread-bound 请求

java.lang.IllegalStateException: No thread-bound request found when using RequestContextHolder.currentRequestAttributes() in async aspect

java

spring

request

aspect

spring-boot