前言

最近做項目，springboot項目，本來我們在controller的requestmapping取參數(shù)值或者返回寫時，使用方法參數(shù)，但是發(fā)現(xiàn)老項目直接注入了成員變量，Spring本身是單例的，如果是成員變量注入，那么也是單例的，怎么實現(xiàn)不同的請求讀取不同的參數(shù)呢，如果實現(xiàn)線程安全呢，筆者立馬想到了ThreadLocal，但是如果要說就是這個原理，那么必須源碼證明。

準備demo

簡單寫一個demo

@RestController
public class DemoController {
 
    @Autowired
    private HttpServletRequest request;
 
    @GetMapping("/hello")
    public String demo(String param) {
        request.getParameterMap().forEach((k,v)-> System.out.println(k + " : " + Arrays.toString(v)));
        return param + ":hello";
    }
}

只寫了 HttpServletRequest，實際上HttpServletResponse亦是如此。

分析demo，注入的HttpServletRequest是接口類型，那么在Boot啟動中就會動態(tài)代理實現(xiàn)，由于是接口，可以推測是JDK動態(tài)代理，debug果然如此。

源碼分析

根據(jù)debug，JDK動態(tài)代理注入了接口的實現(xiàn)類，關(guān)鍵在于InvocationHandler

Spring使用如下：

org.springframework.beans.factory.support.AutowireUtils.ObjectFactoryDelegatingInvocationHandler

里面有關(guān)鍵代碼

return method.invoke(this.objectFactory.getObject(), args);

this.objectFactory.getObject()這句決定線程安全

看看Spring Bean下JDK是怎么動態(tài)代理注入的

可以看到JDK動態(tài)代理在Spring注入的時候，把這個factory注入了InvocationHandler

其中的handler的invoke方法，這里實際上還要其他類的讀取埋點。

這里的invoke僅僅是反射，關(guān)鍵還是 HttpServletRequest的對象來源，跟蹤讀取邏輯

org.springframework.web.context.request.RequestContextHolder

到此就明確了，注入的成員變量，動態(tài)代理后使用Threadlocal處理，所以線程安全，因為每次請求都是線程請求，那么原始的HttpServletRequest對象怎么塞進去的呢，就要看filter的了

org.springframework.web.filter.RequestContextFilter

在doFilter時，執(zhí)行

同理在org.springframework.web.servlet.FrameworkServlet也會再次讀取和寫入

這里是為了，如果filter被去除的時候可以有值，再次保底，并且在結(jié)束時rest

只不過這個rest有點奇怪

對象并沒有清除，還在，說明即使 FrameworkServlet后還可以獲取，因為有filter可能會在這個后面執(zhí)行，如果干掉，很可能就不能讀取了。

會在RequestContextFilter后面remove；如果我們?nèi)サ暨@個filter，那么需要自定義一個filter實現(xiàn)remove防止內(nèi)存泄漏。

清除當前線程及子線程ThreadLocal

public static void resetRequestAttributes() {
    requestAttributesHolder.remove();
    inheritableRequestAttributesHolder.remove();
}

至此 HttpServletRequest的成員變量注入邏輯，即ThreadLocal變量，所以請求可以正常訪問

總結(jié)

實際上這種用法很多項目都用了，只不過我們寫代碼下意思的通過方法參數(shù)來規(guī)避線程安全性，這種想法是有益的，可以從源頭規(guī)避風險，不過實際上Spring也幫我們考慮了這個問題，相當于使用RequestContextFilter做了保底措施。

源碼分析實際上是知所以然，尤其是我們自己寫公共SDK時，可以把這種設(shè)計放在代碼中，實現(xiàn)優(yōu)雅和保底邏輯。

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

最新評論