SpringBoot 攔截器獲取http請求引數—— 所有騷操作基礎
獲取http請求引數是一種剛需
我想有的小夥伴肯定有過獲取http請求的需要，比如想
- 前置獲取引數，統計請求資料
- 做服務的介面簽名校驗
- 敏感介面監控日誌
- 敏感介面防重複提交
等等各式各樣的場景，這時你就需要獲取 HTTP 請求的引數或者請求body，一般思路有兩種，一種就是自定義個AOP去攔截目標方法，第二種就是使用攔截器。整體比較來說，使用攔截器更靈活些，因為每個介面的請求引數定義不同，使用AOP很難細粒度的獲取到變數引數，本文主線是採用攔截器來獲取HTTP請求。
定義攔截器獲取請求
基於 spring-boot-starter-parent 2.1.9.RELEASE
看起來這個很簡單，這裡就直接上code，定義個攔截器
/**
* @author axin
* @summary HTTP請求攔截器
*/
@Slf4j
public class RequestInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
//獲取請求引數
String queryString = request.getQueryString();
log.info("請求引數:{}", queryString);
//獲取請求body
byte[] bodyBytes = StreamUtils.copyToByteArray(request.getInputStream());
String body = new String(bodyBytes, request.getCharacterEncoding());
log.info("請求體：{}", body);
return true;
}
}
然後把這個攔截器配置一下中：
/**
* WebMVC配置，你可以集中在這裡配置攔截器、過濾器、靜態資源快取等
*/
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new RequestInterceptor()).addPathPatterns("/**");
}
}
定義個介面測試一下
/**
* @author axin
* @summary 提交測試介面
*/
@Slf4j
@RestController
public class MyHTTPController {
@GetMapping("/v1/get")
public void get(@RequestParam("one") String one,
@RequestParam("two") BigDecimal number) {
log.info("引數:{},{}", one, number);
}
@PostMapping("/v1/post")
public void check(@RequestBody User user) {
log.info("{}", JSON.toJSONString(user));
}
}
GET請求獲取請求引數示例：
POST請求獲取請求Body示例：
我們發現攔截器在獲取HTTP請求的body時出現了 400 （Required request body is missing: public void com.axin.world.controller.MyHTTPController.check(com.axin.world.domain.User)）；同時也發現攔截器竟然走了兩遍，這又是咋回事呢？
為什麼攔截器會重複調兩遍呢？
其實是因為 tomcat擷取到異常後就轉發到/error頁面，就在這個轉發的過程中導致了springmvc重新開始DispatcherServlet的整個流程，所以攔截器執行了兩次，我們可以看下第二次呼叫時的url路徑：
ServletInputStream(CoyoteInputStream) 輸入流無法重複呼叫
而之前出現的 Required request body is missing 錯誤 其實是ServletInputStream被讀取後無法第二次再讀取了，所以我們要把讀取過的內容存下來，然後需要的時候對外提供可被重複讀取的ByteArrayInputStream。
對於MVC的過濾器來說，我們就需要重寫 ServletInputStream 的 getInputStream()方法。
自定義 HttpServletRequestWrapper
為了 重寫 ServletInputStream 的 getInputStream()方法，我們需要自定義一個 HttpServletRequestWrapper ：
/**
* @author Axin
* @summary 自定義 HttpServletRequestWrapper 來包裝輸入流
*/
public class AxinHttpServletRequestWrapper extends HttpServletRequestWrapper {
/**
* 快取下來的HTTP body
*/
private byte[] body;
public AxinHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
body = StreamUtils.copyToByteArray(request.getInputStream());
}
/**
* 重新包裝輸入流
* @return
* @throws IOException
*/
@Override
public ServletInputStream getInputStream() throws IOException {
InputStream bodyStream = new ByteArrayInputStream(body);
return new ServletInputStream() {
@Override
public int read() throws IOException {
return bodyStream.read();
}
/**
* 下面的方法一般情況下不會被使用，如果你引入了一些需要使用ServletInputStream的外部元件，可以重點關注一下。
* @return
*/
@Override
public boolean isFinished() {
return false;
}
@Override
public boolean isReady() {
return true;
}
@Override
public void setReadListener(ReadListener readListener) {
}
};
}
@Override
public BufferedReader getReader() throws IOException {
InputStream bodyStream = new ByteArrayInputStream(body);
return new BufferedReader(new InputStreamReader(getInputStream()));
}
}
然後定義一個 DispatcherServlet子類來分派 上面自定義的 AxinHttpServletRequestWrapper ：
/**
* @author Axin
* @summary 自定義 DispatcherServlet 來分派 AxinHttpServletRequestWrapper
*/
public class AxinDispatcherServlet extends DispatcherServlet {
/**
* 包裝成我們自定義的request
* @param request
* @param response
* @throws Exception
*/
@Override
protected void doDispatch(HttpServletRequest request, HttpServletResponse response) throws Exception {
super.doDispatch(new AxinHttpServletRequestWrapper(request), response);
}
}
然後配置一下：
/**
* WebMVC配置，你可以集中在這裡配置攔截器、過濾器、靜態資源快取等
*/
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new RequestInterceptor()).addPathPatterns("/**");
}
@Bean
@Qualifier(DispatcherServletAutoConfiguration.DEFAULT_DISPATCHER_SERVLET_BEAN_NAME)
public DispatcherServlet dispatcherServlet() {
return new AxinDispatcherServlet();
}
}
再呼叫一下 POST請求：
請求成功！
總結一下 展望一下
如果你想對HTTP請求做些騷操作，那麼前置獲取HTTP請求引數是前提，為此文字給出了使用MVC攔截器獲取引數的樣例。
在獲取HTTP Body 的時候，出現了 Required request body is missing 的錯誤，同時攔截器還出現執行了兩遍的問題，這是因為 ServletInputStream被讀取了兩遍導致的，tomcat擷取到異常後就轉發到 /error 頁面 被攔截器攔截到了，攔截器也就執行了兩遍。
為此我們通過自定義 HttpServletRequestWrapper 來包裝一個可被重讀讀取的輸入流，來達到期望的攔截效果。
在獲取到HTTP的請求引數後，我們可以前置做很多操作，比如常用的服務端介面簽名驗證，敏感介面防重複請求等等。
個人水平有限，如果文章有邏輯錯誤或表述問題還請指出，歡迎一起交流。