SpringBoot攔截器之如何解決java.io.IOException:Stream closed問題
1.SpringBoot攔截器是什么
SpringBoot攔截器和過濾器是Spring Boot的一種機制,用于對請求和響應進行操作的攔截,是AOP編程的一種體現(xiàn)。該方法可以在不改變代碼基本業(yè)務和邏輯的前提下對SpringBoot的一些操作進行攔截、過濾和更改。
SpirngBoot攔截器(Interceptor)和過濾器(Filter)的操作的如下圖所示。
如果不愿意仔細理解,也可以記住這個執(zhí)行步驟就可以:過濾器(Filter)——>攔截器(Interceptor)——>Controller Adavice——>Controller。

2.AOP編程
SpringBoot是AOP編程的思想體現(xiàn)。AOP編程即面向切面編程,是函數(shù)式編程的一種思想體現(xiàn)。AOP編程和OOP編程沒有沖突。
OOP編程的思路是封裝、繼承和多態(tài),而AOP編程的思路是不改變原有接口和功能的基礎上提供功能擴展。
一般來說,AOP編程的實現(xiàn)思路是提供代理,把一些常用功能如權限檢查、日志、事務等,從每個業(yè)務方法中剝離出來,而提供代理統(tǒng)一實現(xiàn)。
SpringBoot攔截器、過濾器就是AOP編程的體現(xiàn),即把部分功能統(tǒng)一剝離起來進行統(tǒng)一攔截、過濾和實現(xiàn)。
3.請求攔截和返回攔截
可以對請求和返回進行統(tǒng)一攔截。
一般來說,如果是非常簡單的操作,可以用Controller Adavice進行攔截。代碼如下:
@ControllerAdvice
public class TestResponseBodyAdvice implements ResponseBodyAdvice {
@Override
public boolean supports(MethodParameter returnType, Class converterType) {
Class targetClass = returnType.getMethod().getDeclaringClass();
return true;
}
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
//操作
return body;
}
}如果是更復雜的操作,可以用攔截器(Interceptor)進行攔截,代碼如下:
@Component
public class TokenInterceptor implements HandlerInterceptor {
/**
* 在訪問Controller某個方法之前這個方法會被調用。
*
* @param request
* @param response
* @param handler
* @return false則表示不執(zhí)行postHandle方法, true 表示執(zhí)行postHandle方法
* @throws Exception
*/
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
//false則表示不執(zhí)行postHandle方法,不執(zhí)行下一步chain鏈,直接返回response
if(/***相關操作***/) {
return false;
} else {
return true;
}
}
}該方法的要點是:
1)可以自己選擇preHandel和psotHanle操作;
2)此攔截器需要在Bean里進行注冊;
注冊代碼(Configuration注入下)如下:
@Autowired
TokenInterceptor tokenInterceptor;
/**
* 添加攔截器
* @param registry
*/
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(tokenInterceptor)
.addPathPatterns("/api/**")//指定該類攔截的url
.excludePathPatterns( "/static/**");//過濾靜態(tài)資源
}2)如果有對request操作,會造成java.io.IOException: Stream closed問題;
4.Java.io.IOException: Stream closed問題解決
Java.io.IOException: Stream closed出現(xiàn)的原因是HTTP的Stream只允許操作一次,如果操作多次就會失敗。
該問題的解決思路是:提供一個復制流,讓操作基于復制流。
解決代碼如下(原有攔截器保持不變):
1)提供文件解析:
public class HttpHelper {
public static String getBodyString(HttpServletRequest request) throws IOException {
StringBuilder sb = new StringBuilder();
InputStream inputStream = null;
BufferedReader reader = null;
try {
inputStream = request.getInputStream();
reader = new BufferedReader(new InputStreamReader(inputStream, Charset.forName("UTF-8")));
String line = "";
while ((line = reader.readLine()) != null) {
sb.append(line);
}
} catch (IOException e) {
e.printStackTrace();
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (reader != null) {
try {
reader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return sb.toString();
}
}2)提供Filter過濾:
public class HttpServletRequestReplacedFilter implements Filter {
@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 RequestReaderHttpServletRequestWrapper((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 {
}
}3)對請求的裝飾,用于建立復制流。
public class RequestReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {
private final byte[] body;
public RequestReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
body = HttpHelper.getBodyString(request).getBytes(Charset.forName("UTF-8"));
}
@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) {
}
};
}
}4)在Bean操作(Configuration注入)中加入相關操作:
@Bean
public FilterRegistrationBean httpServletRequestReplacedRegistration() {
FilterRegistrationBean registration = new FilterRegistrationBean();
registration.setFilter(new HttpServletRequestReplacedFilter());
registration.addUrlPatterns("/*");
registration.addInitParameter("paramName", "paramValue");
registration.setName("httpServletRequestReplacedFilter");
registration.setOrder(1);
return registration;
}這個問題解決方案,最關鍵的部分其實是 HttpServletRequestReplacedFilter里的
if(requestWrapper == null) {
chain.doFilter(request, response);
} else {
chain.doFilter(requestWrapper, response);
}這是讓復制流起作用的操作,從而避免流被關閉。
總結
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
JAVA中時間戳與LocalDateTime互相轉換代碼例子
最近在編碼過程中遇到將時間戳轉化為 LocalDateTime,所以這里給總結下,這篇文章主要給大家介紹了關于JAVA中時間戳與LocalDateTime互相轉換的相關資料,需要的朋友可以參考下2023-11-11
SpringBoot整合Liquibase實現(xiàn)對數(shù)據(jù)庫管理和遷移
Liquibase是一個用于用于跟蹤、管理和應用數(shù)據(jù)庫變化的開源工具,通過日志文件(changelog)的形式記錄數(shù)據(jù)庫的變更(changeset),然后執(zhí)行日志文件中的修改,將數(shù)據(jù)庫更新或回滾(rollback)到一致的狀態(tài),本文主要介紹SpringBoot與Liquibase的集成,需要的朋友可以參考下2024-11-11
VScode 打造完美java開發(fā)環(huán)境最新教程
這篇文章主要介紹了VScode 打造完美java開發(fā)環(huán)境最新教程,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12

