springboot 使用ThreadLocal的實(shí)例代碼

更新時(shí)間：2021年12月06日 15:13:30 作者：沒(méi)錯(cuò)，是一只菜鳥(niǎo)啦

這篇文章主要介紹了springboot 使用ThreadLocal的實(shí)例代碼，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

springboot 使用ThreadLocal

本文參考慕課教程給出一個(gè)在spring boot中使用ThreadLocal實(shí)現(xiàn)線程封閉的實(shí)例。

首先創(chuàng)建一個(gè)包含ThreadLocal成員變量的實(shí)例：

public class RequestHolder { 
    private final static ThreadLocal<Long> requestHolder = new ThreadLocal<>();
    public static void add(Long id) {
        requestHolder.set(id);
    }
 
    public static Long getId() {
        return requestHolder.get();
    }
 
    public static void remove() {
        requestHolder.remove();
    }
}

編寫一個(gè)Controller類，請(qǐng)求該類的test()方法獲取ThreadLocal中存儲(chǔ)的id：

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
 
@Controller
@RequestMapping("/threadLocal")
public class ThreadLocalController {
 
    @RequestMapping("/test")
    @ResponseBody
    public Long test() {
        return RequestHolder.getId();
    }
}

編寫過(guò)濾器，在請(qǐng)求到達(dá)Servlet之前（請(qǐng)求->tomcat容器->filter->servlet->inteceptor->controller），將當(dāng)前線程的id添加到ThreadLocal中：

import com.mmall.concurrency.example.threadLocal.RequestHolder;
import lombok.extern.slf4j.Slf4j; 
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
 
@Slf4j
public class HttpFilter implements Filter {
 
    @Override
    public void init(FilterConfig filterConfig) throws ServletException { 
    }
    
    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) servletRequest;
        log.info("do filter, {}, {}", Thread.currentThread().getId(), request.getServletPath());
        //在ThreadLocal中添加當(dāng)前線程的id
        RequestHolder.add(Thread.currentThread().getId());
        filterChain.doFilter(servletRequest, servletResponse);
    } 
    @Override
    public void destroy() { 
    }
}

編寫攔截器，當(dāng)請(qǐng)求處理完成后（從Controller返回后），清除ThreadLocal中的id，避免內(nèi)存泄漏。

import com.mmall.concurrency.example.threadLocal.RequestHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
@Slf4j
public class HttpInterceptor extends HandlerInterceptorAdapter {
 
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        log.info("preHandle");
        return true;
    }
 
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        log.info("ThreadId:"+RequestHolder.getId());
        RequestHolder.remove();
        log.info("afterCompletion");
        return;
    }
}

最后，我們需要在spring boot啟動(dòng)類上注冊(cè)我們定義的Filer及Inteceptor，并設(shè)置攔截路徑。

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
 
@SpringBootApplication
public class ConcurrencyApplication extends WebMvcConfigurerAdapter{ 
 public static void main(String[] args) {
  SpringApplication.run(ConcurrencyApplication.class, args);
 }
 
 @Bean
 public FilterRegistrationBean httpFilter() {
  FilterRegistrationBean registrationBean = new FilterRegistrationBean();
  registrationBean.setFilter(new HttpFilter());
  registrationBean.addUrlPatterns("/threadLocal/*");
  return registrationBean;
 }
 
 @Override
 public void addInterceptors(InterceptorRegistry registry) {
  registry.addInterceptor(new HttpInterceptor()).addPathPatterns("/**");
 }
}

在瀏覽器或者postman中輸入http://localhost:8080/threadLocal/test

觀察輸出結(jié)果：

2018-11-09 11:16:51.287? INFO 34076 --- [?????????? main] s.b.c.e.t.TomcatEmbeddedServletContainer : Tomcat started on port(s): 8080 (http)
2018-11-09 11:16:51.290? INFO 34076 --- [?????????? main] c.m.concurrency.ConcurrencyApplication?? : Started ConcurrencyApplication in 1.718 seconds (JVM running for 2.132)
2018-11-09 11:17:03.060? INFO 34076 --- [nio-8080-exec-2] o.a.c.c.C.[Tomcat].[localhost].[/]?????? : Initializing Spring FrameworkServlet 'dispatcherServlet'
2018-11-09 11:17:03.060? INFO 34076 --- [nio-8080-exec-2] o.s.web.servlet.DispatcherServlet??????? : FrameworkServlet 'dispatcherServlet': initialization started
2018-11-09 11:17:03.072? INFO 34076 --- [nio-8080-exec-2] o.s.web.servlet.DispatcherServlet??????? : FrameworkServlet 'dispatcherServlet': initialization completed in 12 ms
2018-11-09 11:17:03.078? INFO 34076 --- [nio-8080-exec-2] com.mmall.concurrency.HttpFilter???????? : do filter, 29, /threadLocal/test
2018-11-09 11:17:03.090? INFO 34076 --- [nio-8080-exec-2] com.mmall.concurrency.HttpInterceptor??? : preHandle
2018-11-09 11:17:03.124? INFO 34076 --- [nio-8080-exec-2] com.mmall.concurrency.HttpInterceptor??? : ThreadId:29
2018-11-09 11:17:03.124? INFO 34076 --- [nio-8080-exec-2] com.mmall.concurrency.HttpInterceptor??? : afterCompletion

從打印的日志結(jié)果中，我們看到在Filter中我們將當(dāng)前線程的id 29添加到了ThreadLocal中，隨后在Inteceptor中打印并刪除了id。

ThreadLocal在springboot使用中的坑

ThreadLocal 適用于變量在線程間隔離，而在方法或類間共享的場(chǎng)景。現(xiàn)在在Springboot中我做如下場(chǎng)景的使用：

使用 Spring Boot 創(chuàng)建一個(gè) Web 應(yīng)用程序，使用 ThreadLocal 存放一個(gè) Integer 的值，來(lái)暫且代表需要在線程中保存的用戶信息，這個(gè)值初始是 null。在業(yè)務(wù)邏輯中，我先從 ThreadLocal 獲取一次值，然后把外部傳入的參數(shù)設(shè)置到 ThreadLocal 中，來(lái)模擬從當(dāng)前上下文獲取到用戶信息的邏輯，隨后再獲取一次值，最后輸出兩次獲得的值和線程名稱。

@RestController
public class threadLocal {
    private ThreadLocal<Integer> currentUser = ThreadLocal.withInitial(() -> null);
    @RequestMapping("wrong")
    public Map wrong(@RequestParam("userId") Integer userId) {
        //設(shè)置用戶信息之前先查詢一次ThreadLocal中的用戶信息
        String before = Thread.currentThread().getName() + ":" + currentUser.get();
        //設(shè)置用戶信息到ThreadLocal
        currentUser.set(userId);
      
            //設(shè)置用戶信息之后再查詢一次ThreadLocal中的用戶信息
            String after = Thread.currentThread().getName() + ":" + currentUser.get();
            //匯總輸出兩次查詢結(jié)果
            Map result = new HashMap();
            result.put("before", before);
            result.put("after", after);
            return result;      
    }
}

為了讓問(wèn)題快速的重現(xiàn)，我在配置文件中設(shè)置一下 Tomcat 的參數(shù)，把工作線程池最大線程數(shù)設(shè)置為 1，這樣始終是同一個(gè)線程在處理請(qǐng)求：

server.tomcat.max-threads=1

運(yùn)行程序后先讓用戶 1 來(lái)請(qǐng)求接口，可以看到第一和第二次獲取到用戶 ID 分別是 null 和 1，符合預(yù)期：隨后用戶 2 來(lái)請(qǐng)求接口，這次就出現(xiàn)了 Bug，第一和第二次獲取到用戶 ID 分別是 1 和 2，顯然第一次獲取到了用戶 1 的信息，原因就是 Tomcat 的線程池重用了線程。

在 Tomcat 這種 Web 服務(wù)器下跑的業(yè)務(wù)代碼，本來(lái)就運(yùn)行在一個(gè)多線程環(huán)境中，并不能認(rèn)為沒(méi)有顯式開(kāi)啟多線程就不會(huì)有線程安全問(wèn)題，所以使用類似 ThreadLocal 工具來(lái)存放一些數(shù)據(jù)時(shí)，需要特別注意在代碼運(yùn)行完后，顯式地去清空設(shè)置的數(shù)據(jù)。如果在代碼中使用了自定義的線程池，也同樣會(huì)遇到這個(gè)問(wèn)題。修改后代碼如下：

@RestController
public class threadLocal {
    private ThreadLocal<Integer> currentUser = ThreadLocal.withInitial(() -> null);
    @RequestMapping("wrong")
    public Map wrong(@RequestParam("userId") Integer userId) {
        //設(shè)置用戶信息之前先查詢一次ThreadLocal中的用戶信息
        String before = Thread.currentThread().getName() + ":" + currentUser.get();
        //設(shè)置用戶信息到ThreadLocal
        currentUser.set(userId);
        try {
            //設(shè)置用戶信息之后再查詢一次ThreadLocal中的用戶信息
            String after = Thread.currentThread().getName() + ":" + currentUser.get();
            //匯總輸出兩次查詢結(jié)果
            Map result = new HashMap();
            result.put("before", before);
            result.put("after", after);
            return result;
        } finally {
         //增加移除處理
            currentUser.remove();
        }
    }
}

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: