服务器之家：专注于VPS、云服务器配置技术及软件下载分享
分类导航

PHP教程|ASP.NET教程|Java教程|ASP教程|编程技术|正则表达式|C/C++|IOS|C#|Swift|Android|VB|R语言|JavaScript|易语言|vb.net|

服务器之家 - 编程语言 - Java教程 - SpringBoot详细讲解异步任务如何获取HttpServletRequest

SpringBoot详细讲解异步任务如何获取HttpServletRequest

2022-11-22 11:16code2roc Java教程

在使用框架日常开发中需要在controller中进行一些异步操作减少请求时间，但是发现在使用@Anysc注解后会出现Request对象无法获取的情况，本文就此情况给出完整的解决方案

原因分析

  • @Anysc注解会开启一个新的线程，主线程的Request和子线程是不共享的，所以获取为null
  • 在使用springboot的自定带的线程共享后，代码如下，Request不为null，但是偶发的其中body/head/urlparam内容出现获取不到的情况，是因为异步任务在未执行完毕的情况下，主线程已经返回，拷贝共享的Request对象数据被清空
?
1
2
3
4
ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
//设置子线程共享
RequestContextHolder.setRequestAttributes(servletRequestAttributes, true);
HttpServletRequest request = servletRequestAttributes.getRequest();

解决方案

前置条件

  • 启动类添加@EnableAsync注解
  • 标记@Async的异步方法不能和调用者在同一个class中

pom配置

?
1
2
3
4
5
6
<!-- 阿里线程共享 -->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>transmittable-thread-local</artifactId>
    <version>2.11.0</version>
</dependency>

requrest共享

通过TransmittableThreadLocal对象进行线程对象共享

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public class CommonUtil {
    public static TransmittableThreadLocal<HttpServletRequest> requestTransmittableThreadLocal = new TransmittableThreadLocal<HttpServletRequest>();
    public static void shareRequest(HttpServletRequest request){
        requestTransmittableThreadLocal.set(request);
    }
    public static HttpServletRequest getRequest(){
        HttpServletRequest request = requestTransmittableThreadLocal.get();
        if(request!=null){
            return requestTransmittableThreadLocal.get();
        }else{
            ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
            if(requestAttributes!=null){
                return  requestAttributes.getRequest();
            }else{
                return  null;
            }
        }
    }
    public static void remove(){
        requestTransmittableThreadLocal.remove();
    }
}

注：系统中所有Request获取需要统一从CommonUtil指定来源，例如token鉴权等

自定义request过滤器

通过自定义过滤器对Request的内容进行备份保存，主线程结束时Request清除结束不会影响到子线程的相应参数的获取，也适用于增加拦截器/过滤器后body参数无法重复获取的问题。需要注意的是对header参数处理时key要忽略大小写

?
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
public class HttpServletRequestReplacedFilter implements Filter, Ordered {
    @Override
    public void destroy() {
    }
    @Override
    public void doFilter(ServletRequest request, ServletResponse response,
                         FilterChain chain) throws IOException, ServletException {
        ServletRequest requestWrapper = null;
        if (request instanceof HttpServletRequest) {
            requestWrapper = new RequestWrapper((HttpServletRequest) request);
        }
        //获取请求中的流如何，将取出来的字符串，再次转换成流，然后把它放入到新request对象中。
        // 在chain.doFiler方法中传递新的request对象
        if (requestWrapper == null) {
            chain.doFilter(request, response);
        } else {
            chain.doFilter(requestWrapper, response);
        }
    }
    @Override
    public void init(FilterConfig arg0) throws ServletException {
    }
    @Override
    public int getOrder() {
        return 10;
    }
}
?
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
public class RequestWrapper extends HttpServletRequestWrapper{
    private final byte[] body;
    private final HashMap<String,String> headMap;
    private final HashMap<String,String> requestParamMap;
    public RequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        body = CommonUtil.getBodyString(request).getBytes(Charset.forName("UTF-8"));
        headMap = new HashMap();
        Enumeration<String> headNameList = request.getHeaderNames();
        while (headNameList.hasMoreElements()){
            String key = headNameList.nextElement();
            headMap.put(key.toLowerCase(),request.getHeader(key));
        }
        requestParamMap = new HashMap<>();
        Enumeration<String> parameterNameList = request.getParameterNames();
        while (parameterNameList.hasMoreElements()){
            String key = parameterNameList.nextElement();
            requestParamMap.put(key,request.getParameter(key));
        }
    }
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }
    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream bais = new ByteArrayInputStream(body);
        return new ServletInputStream() {
            @Override
            public int read() throws IOException {
                return bais.read();
            }
            @Override
            public boolean isFinished() {
                return false;
            }
            @Override
            public boolean isReady() {
                return false;
            }
            @Override
            public void setReadListener(ReadListener readListener) {
            }
        };
    }
    @Override
    public String getHeader(String name) {
        return headMap.get(name.toLowerCase());
    }
    @Override
    public String getParameter(String name) {
        return requestParamMap.get(name);
    }
}

自定义任务执行器

用于拦截异步任务执行，在任务执前统一进行Request共享操作，且可以定义多个，不影响原有的异步任务代码

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
public class CustomTaskDecorator implements TaskDecorator {
    @Override
    public Runnable decorate(Runnable runnable) {
        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = requestAttributes.getRequest();
        System.out.println("异步任务共享request");
        return () -> {
            try {
                CommonUtil.shareRequest(request);
                runnable.run();
            } finally {
                CommonUtil.remove();
            }
        };
    }
}
?
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
@Configuration
public class TaskExecutorConfig {
    @Bean()
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(20);
        executor.setQueueCapacity(200);
        executor.setKeepAliveSeconds(60);
        executor.setThreadNamePrefix("taskExecutor-");
        executor.setAwaitTerminationSeconds(60);
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.initialize();
        return executor;
    }
    @Bean("shareTaskExecutor")
    public Executor hpTaskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(20);
        executor.setQueueCapacity(200);
        executor.setKeepAliveSeconds(60);
        executor.setThreadNamePrefix("shareTaskExecutor-");
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setAwaitTerminationSeconds(60);
        // 增加 TaskDecorator 属性的配置
        executor.setTaskDecorator(new CustomTaskDecorator());
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.initialize();
        return executor;
    }
}

调用示例

给@Anysc注解指定进行共享拦截的任务执行器即可

?
1
2
3
4
5
6
7
@PostMapping("/testAsync")
@ResponseBody
public Object testAsync(@RequestBody Map<String, Object> params) throws Exception{
    Result result = Result.okResult();
    asyncUtil.executeAsync();
    return result;
}
?
1
2
3
4
5
6
7
8
9
@Component
public class AsyncUtil {
    @Async("shareTaskExecutor")
    public void executeAsync () throws InterruptedException {
        System.out.println("开始执行executeAsync");
        Thread.sleep(3000);
        System.out.println("结束执行executeAsync");
    }
}

到此这篇关于SpringBoot详细讲解异步任务如何获取HttpServletRequest的文章就介绍到这了,更多相关SpringBoot获取HttpServletRequest内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://blog.csdn.net/u013407099/article/details/124061950

  • springboot
  • 异步任务
  • HttpServletRequest

    • 延伸 · 阅读

    精彩推荐
    • Java教程Java程序设计之12个经典样例

      Java程序设计之12个经典样例

      这篇文章主要给大家分享关于Java程序设计11个经典样例，主要以举例的形式详细的讲解了Java程序设计的各种方法,需要的朋友可以参考一下文章具体的内容...

      zhulin10287002022-02-28
    • Java教程Java线程的start方法回调run方法的操作技巧

      Java线程的start方法回调run方法的操作技巧

      面试过程中经常会被面试官问到为什么我们调用start()方法时会执行run()方法，为什么不能直接调用run()方法，问的一头雾水，今天小编给大家介绍下Java线程...

      麦田12012021-02-02
    • Java教程Java之Algorithm_analysis案例详解

      Java之Algorithm_analysis案例详解

      这篇文章主要介绍了Java之Algorithm_analysis案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下...

      lxj_csdn4722021-12-28
    • Java教程idea 如何查找类中的某个方法

      idea 如何查找类中的某个方法

      这篇文章主要介绍了idea 如何查找类中的某个方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      李耦合9072021-08-16
    • Java教程Java经典面试题汇总:Spring

      Java经典面试题汇总:Spring

      本篇总结的是Spring框架相关的面试题，后续会持续更新，希望我的分享可以帮助到正在备战面试的实习生或者已经工作的同行，如果发现错误还望大家多多...

      让程序飞4462021-10-18
    • Java教程浅析Spring和MyBatis整合及逆向工程

      浅析Spring和MyBatis整合及逆向工程

      这篇文章主要介绍了Spring和MyBatis整合及逆向工程的相关资料,非常不错，具有参考借鉴价值,需要的朋友可以参考下 ...

      鹿天斐3802020-05-23
    • Java教程Spring Security实现自动登陆功能示例

      Spring Security实现自动登陆功能示例

      自动登录在很多网站和APP上都能用的到，解决了用户每次输入账号密码的麻烦。本文就使用Spring Security实现自动登陆功能，具有一定的参考价值，感兴趣的...

      Java Gosling11412022-03-11
    • Java教程Java多态的使用注意事项

      Java多态的使用注意事项

      本文讲解了什么是JAVA多态和Java多态是如何实现的，在使用Java多态时需要注意什么，具体大家看下面的内容 ...

      java教程网4862019-10-21
    最近更新
    编辑推荐
    阅读排行
    热门标签
    501 　 406 　 415 　 2000 　 1433 　 401 　 204 　 330 　 选择排序法 　 随机抽取 　 内存卡 　 下载器 　 十六进制字符 　 响应 　 静态工厂 　 构造函数 　 swt 　 Overview 　 对象池模式 　 Throwable 　 支票 　 金额大写转换 　 空心菱形 　 AspectJ 　 面向切面编程 　 方法名 　 网址信息 　 动态代理模式 　 线程安全 　 htmlparser 　
    © 2019-2022 服务器之家(www.tuohang.net) 版权所有关于我们联系我们版权申明网站地图