Spring中獲取request的幾種方法,及線程安全性

作者:編程迷思

概述

在使用Spring MVC開發Web系統時,經常需要在處理請求時使用request對象,比如獲取客戶端ip地址、請求的url、header中的屬性(如cookie、授權信息)、body中的數據等。由於在Spring MVC中,處理請求的Controller、Service等對象都是單例的,因此獲取request對象時最需要注意的問題,便是request對象是否是線程安全的:當有大量併發請求時,能否保證不同請求/線程中使用不同的request對象。

這裡還有一個問題需要注意:前面所說的“在處理請求時”使用request對象,究竟是在哪裡使用呢?考慮到獲取request對象的方法有微小的不同,大體可以分為兩類:

1) 在Spring的Bean中使用request對象:既包括Controller、Service、Repository等MVC的Bean,也包括了Component等普通的Spring Bean。為了方便說明,後文中Spring中的Bean一律簡稱為Bean。

2) 在非Bean中使用request對象:如普通的Java對象的方法中使用,或在類的靜態方法中使用。

此外,本文討論是圍繞代表請求的request對象展開的,但所用方法同樣適用於response對象、InputStream/Reader、OutputStream/ Writer等;其中InputStream/Reader可以讀取請求中的數據,OutputStream/ Writer可以向響應寫入數據。

最後,獲取request對象的方法與Spring及MVC的版本也有關係;本文基於Spring4進行討論,且所做的實驗都是使用4.1.1版本。

如何測試線程安全

既然request對象的線程安全問題需要特別關注,為了便於後面的討論,下面先說明如何測試request對象是否是線程安全的。

測試的基本思路,是模擬客戶端大量併發請求,然後在服務器判斷這些請求是否使用了相同的request對象。

判斷request對象是否相同,最直觀的方式是打印出request對象的地址,如果相同則說明使用了相同的對象。然而,在幾乎所有web服務器的實現中,都使用了線程池,這樣就導致先後到達的兩個請求,可能由同一個線程處理:在前一個請求處理完成後,線程池收回該線程,並將該線程重新分配給了後面的請求。而在同一線程中,使用的request對象很可能是同一個(地址相同,屬性不同)。因此即便是對於線程安全的方法,不同的請求使用的request對象地址也可能相同。

為了避免這個問題,一種方法是在請求處理過程中使線程休眠幾秒,這樣可以讓每個線程工作的時間足夠長,從而避免同一個線程分配給不同的請求;另一種方法,是使用request的其他屬性(如參數、header、body等)作為request是否線程安全的依據,因為即便不同的請求先後使用了同一個線程(request對象地址也相同),只要使用不同的屬性分別構造了兩次request對象,那麼request對象的使用就是線程安全的。本文使用第二種方法進行測試。

客戶端測試代碼如下(創建1000個線程分別發送請求):

public class Test {
 public static void main(String[] args) throws Exception {
 String prefix = UUID.randomUUID().toString().replaceAll("-", "") + "::";
 for (int i = 0; i < 1000; i++) {
 final String value = prefix + i;
 new Thread() {
 @Override
 public void run() {
 try {
 CloseableHttpClient httpClient = HttpClients.createDefault();
 HttpGet httpGet = new HttpGet("http://localhost:8080/test?key=" + value);
 httpClient.execute(httpGet);
 httpClient.close();
 } catch (IOException e) {
 e.printStackTrace();
 }
 }
 }.start();
 }
 }
}


服務器中Controller代碼如下(暫時省略了獲取request對象的代碼):

@Controller
public class TestController {
 // 存儲已有參數,用於判斷參數是否重複,從而判斷線程是否安全
 public static Set set = new ConcurrentSkipListSet<>();
 @RequestMapping("/test")
 public void test() throws InterruptedException {
 // …………………………通過某種方式獲得了request對象………………………………
 // 判斷線程安全
 String value = request.getParameter("key");
 if (set.contains(value)) {
 System.out.println(value + "\t重複出現,request併發不安全!");
 } else {
 System.out.println(value);
 set.add(value);
 }
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


補充:上述代碼原使用HashSet來判斷value是否重複,經網友批評指正,使用線程不安全的集合類驗證線程安全性是欠妥的,現已改為ConcurrentSkipListSet。

如果request對象線程安全,服務器中打印結果如下所示:


Spring中獲取request的幾種方法,及線程安全性


如果存在線程安全問題,服務器中打印結果可能如下所示:

Spring中獲取request的幾種方法,及線程安全性

如無特殊說明,本文後面的代碼中將省略掉測試代碼。

方法1:Controller中加參數

代碼示例:

這種方法實現最簡單,直接上Controller代碼:

@Controller
public class TestController {
 @RequestMapping("/test")
 public void test(HttpServletRequest request) throws InterruptedException {
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


該方法實現的原理是,在Controller方法開始處理請求時,Spring會將request對象賦值到方法參數中。除了request對象,可以通過這種方法獲取的參數還有很多,具體可以參見:
https://docs.spring.io/spring/docs/current/spring-framework-reference/web.html#mvc-ann-methods

Controller中獲取request對象後,如果要在其他方法中(如service方法、工具類方法等)使用request對象,需要在調用這些方法時將request對象作為參數傳入。

線程安全:

測試結果:線程安全

分析:此時request對象是方法參數,相當於局部變量,毫無疑問是線程安全的。

優缺點:

這種方法的主要缺點是request對象寫起來冗餘太多,主要體現在兩點:

1) 如果多個controller方法中都需要request對象,那麼在每個方法中都需要添加一遍request參數

2) request對象的獲取只能從controller開始,如果使用request對象的地方在函數調用層級比較深的地方,那麼整個調用鏈上的所有方法都需要添加request參數

實際上,在整個請求處理的過程中,request對象是貫穿始終的;也就是說,除了定時器等特殊情況,request對象相當於線程內部的一個全局變量。而該方法,相當於將這個全局變量,傳來傳去。

方法2:自動注入

代碼示例:

先上代碼:

@Controller
public class TestController{
 @Autowired
 private HttpServletRequest request; //自動注入request
 @RequestMapping("/test")
 public void test() throws InterruptedException{
 //模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


線程安全:

測試結果:線程安全

分析:在Spring中,Controller的scope是singleton(單例),也就是說在整個web系統中,只有一個TestController;但是其中注入的request卻是線程安全的,原因在於:

使用這種方式,當Bean(本例的TestController)初始化時,Spring並沒有注入一個request對象,而是注入了一個代理(proxy);當Bean中需要使用request對象時,通過該代理獲取request對象。

下面通過具體的代碼對這一實現進行說明。

在上述代碼中加入斷點,查看request對象的屬性,如下圖所示:

Spring中獲取request的幾種方法,及線程安全性


在圖中可以看出,request實際上是一個代理:代理的實現參見AutowireUtils的內部類
ObjectFactoryDelegatingInvocationHandler:

/**
 * Reflective InvocationHandler for lazy access to the current target object.
 */
@SuppressWarnings("serial")
private static class ObjectFactoryDelegatingInvocationHandler implements InvocationHandler, Serializable {
 private final ObjectFactory> objectFactory;
 public ObjectFactoryDelegatingInvocationHandler(ObjectFactory> objectFactory) {
 this.objectFactory = objectFactory;
 }
 @Override
 public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
 // ……省略無關代碼
 try {
 return method.invoke(this.objectFactory.getObject(), args); // 代理實現核心代碼
 }
 catch (InvocationTargetException ex) {
 throw ex.getTargetException();
 }
 }
}

也就是說,當我們調用request的方法method時,實際上是調用了由objectFactory.getObject()生成的對象的method方法;objectFactory.getObject()生成的對象才是真正的request對象。

繼續觀察上圖,發現objectFactory的類型為
WebApplicationContextUtils的內部類RequestObjectFactory;而RequestObjectFactory代碼如下:

/**
 * Factory that exposes the current request object on demand.
 */
@SuppressWarnings("serial")
private static class RequestObjectFactory implements ObjectFactory, Serializable {
 @Override
 public ServletRequest getObject() {
 return currentRequestAttributes().getRequest();
 }
 @Override
 public String toString() {
 return "Current HttpServletRequest";
 }
}


其中,要獲得request對象需要先調用currentRequestAttributes()方法獲得RequestAttributes對象,該方法的實現如下:

/**
 * Return the current RequestAttributes instance as ServletRequestAttributes.
 */
private static ServletRequestAttributes currentRequestAttributes() {
 RequestAttributes requestAttr = RequestContextHolder.currentRequestAttributes();
 if (!(requestAttr instanceof ServletRequestAttributes)) {
 throw new IllegalStateException("Current request is not a servlet request");
 }
 return (ServletRequestAttributes) requestAttr;
}


生成RequestAttributes對象的核心代碼在類RequestContextHolder中,其中相關代碼如下(省略了該類中的無關代碼):

public abstract class RequestContextHolder {
 public static RequestAttributes currentRequestAttributes() throws IllegalStateException {
 RequestAttributes attributes = getRequestAttributes();
 // 此處省略不相關邏輯…………
 return attributes;
 }
 public static RequestAttributes getRequestAttributes() {
 RequestAttributes attributes = requestAttributesHolder.get();
 if (attributes == null) {
 attributes = inheritableRequestAttributesHolder.get();
 }
 return attributes;
 }
 private static final ThreadLocal requestAttributesHolder =
 new NamedThreadLocal("Request attributes");
 private static final ThreadLocal inheritableRequestAttributesHolder =
 new NamedInheritableThreadLocal("Request context");
}


通過這段代碼可以看出,生成的RequestAttributes對象是線程局部變量(ThreadLocal),因此request對象也是線程局部變量;這就保證了request對象的線程安全性。

優缺點:

該方法的主要優點:

1) 注入不侷限於Controller中:在方法1中,只能在Controller中加入request參數。而對於方法2,不僅可以在Controller中注入,還可以在任何Bean中注入,包括Service、Repository及普通的Bean。

2) 注入的對象不限於request:除了注入request對象,該方法還可以注入其他scope為request或session的對象,如response對象、session對象等;並保證線程安全。

3) 減少代碼冗餘:只需要在需要request對象的Bean中注入request對象,便可以在該Bean的各個方法中使用,與方法1相比大大減少了代碼冗餘。

但是,該方法也會存在代碼冗餘。考慮這樣的場景:web系統中有很多controller,每個controller中都會使用request對象(這種場景實際上非常頻繁),這時就需要寫很多次注入request的代碼;如果還需要注入response,代碼就更繁瑣了。下面說明自動注入方法的改進方法,並分析其線程安全性及優缺點。

方法3:基類中自動注入

代碼示例:

與方法2相比,將注入部分代碼放入到了基類中。

基類代碼:

public class BaseController {
 @Autowired
 protected HttpServletRequest request; 
}

Controller代碼如下;這裡列舉了BaseController的兩個派生類,由於此時測試代碼會有所不同,因此服務端測試代碼沒有省略;客戶端也需要進行相應的修改(同時向2個url發送大量併發請求)。

@Controller
public class TestController extends BaseController {
 // 存儲已有參數,用於判斷參數value是否重複,從而判斷線程是否安全
 public static Set set = new ConcurrentSkipListSet<>();
 @RequestMapping("/test")
 public void test() throws InterruptedException {
 String value = request.getParameter("key");
 // 判斷線程安全
 if (set.contains(value)) {
 System.out.println(value + "\t重複出現,request併發不安全!");
 } else {
 System.out.println(value);
 set.add(value);
 }
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}
@Controller
public class Test2Controller extends BaseController {
 @RequestMapping("/test2")
 public void test2() throws InterruptedException {
 String value = request.getParameter("key");
 // 判斷線程安全(與TestController使用一個set進行判斷)
 if (TestController.set.contains(value)) {
 System.out.println(value + "\t重複出現,request併發不安全!");
 } else {
 System.out.println(value);
 TestController.set.add(value);
 }
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


線程安全:

測試結果:線程安全

分析:在理解了方法2的線程安全性的基礎上,很容易理解方法3是線程安全的:當創建不同的派生類對象時,基類中的域(這裡是注入的request)在不同的派生類對象中會佔據不同的內存空間,也就是說將注入request的代碼放在基類中對線程安全性沒有任何影響;測試結果也證明了這一點。

優缺點:

與方法2相比,避免了在不同的Controller中重複注入request;但是考慮到java只允許繼承一個基類,所以如果Controller需要繼承其他類時,該方法便不再好用。

無論是方法2和方法3,都只能在Bean中注入request;如果其他方法(如工具類中static方法)需要使用request對象,則需要在調用這些方法時將request參數傳遞進去。下面介紹的方法4,則可以直接在諸如工具類中的static方法中使用request對象(當然在各種Bean中也可以使用)。

方法4:手動調用

代碼示例:

@Controller
public class TestController {
 @RequestMapping("/test")
 public void test() throws InterruptedException {
 HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest();
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


線程安全:

測試結果:線程安全

分析:該方法與方法2(自動注入)類似,只不過方法2中通過自動注入實現,本方法通過手動方法調用實現。因此本方法也是線程安全的。

優缺點:

優點:可以在非Bean中直接獲取。

缺點:如果使用的地方較多,代碼非常繁瑣;因此可以與其他方法配合使用。

方法5:@ModelAttribute方法

代碼示例:

下面這種方法及其變種(變種:將request和bindRequest放在子類中)在網上經常見到:

@Controller
public class TestController {
 private HttpServletRequest request;
 @ModelAttribute
 public void bindRequest(HttpServletRequest request) {
 this.request = request;
 }
 @RequestMapping("/test")
 public void test() throws InterruptedException {
 // 模擬程序執行了一段時間
 Thread.sleep(1000);
 }
}


線程安全:

測試結果:線程不安全

分析:@ModelAttribute註解用在Controller中修飾方法時,其作用是Controller中的每個@RequestMapping方法執行前,該方法都會執行。因此在本例中,bindRequest()的作用是在test()執行前為request對象賦值。雖然bindRequest()中的參數request本身是線程安全的,但由於TestController是單例的,request作為TestController的一個域,無法保證線程安全。

總結:

綜上所述,Controller中加參數(方法1)、自動注入(方法2和方法3)、手動調用(方法4)都是線程安全的,都可以用來獲取request對象。如果系統中request對象使用較少,則使用哪種方式均可;如果使用較多,建議使用自動注入(方法2 和方法3)來減少代碼冗餘。如果需要在非Bean中使用request對象,既可以在上層調用時通過參數傳入,也可以直接在方法中通過手動調用(方法4)獲得。

此外,本文在討論獲取request對象的方法時,重點討論該方法的線程安全性、代碼的繁瑣程度等;在實際的開發過程中,還必須考慮所在項目的規範、代碼維護等問題(此處感謝網友的批評指正)。


分享到:


相關文章: