建站
咨詢
售后
建站咨詢
新聞中心
如果通過 IO 流來解析參數(shù)，默認(rèn)情況下，IO 流讀一次就結(jié)束了，就沒有了。而往往有些場景，需要我們多次讀取參數(shù)，我舉一個例子：
創(chuàng)新互聯(lián)建站不只是一家網(wǎng)站建設(shè)的網(wǎng)絡(luò)公司；我們對營銷、技術(shù)、服務(wù)都有自己獨(dú)特見解，公司采取“創(chuàng)意+綜合+營銷”一體化的方式為您提供更專業(yè)的服務(wù)！我們經(jīng)歷的每一步也許不一定是最完美的，但每一步都有值得深思的意義。我們珍視每一份信任，關(guān)注我們的成都網(wǎng)站制作、成都網(wǎng)站建設(shè)、外貿(mào)營銷網(wǎng)站建設(shè)質(zhì)量和服務(wù)品質(zhì)，在得到用戶滿意的同時，也能得到同行業(yè)的專業(yè)認(rèn)可，能夠?yàn)樾袠I(yè)創(chuàng)新發(fā)展助力。未來將繼續(xù)專注于技術(shù)創(chuàng)新，服務(wù)升級，滿足企業(yè)一站式網(wǎng)絡(luò)營銷推廣需求，讓再小的品牌網(wǎng)站設(shè)計(jì)也能產(chǎn)生價(jià)值！
接口冪等性的處理，同一個接口，在短時間內(nèi)接收到相同參數(shù)的請求，接口可能會拒絕處理。那么在判斷的時候，就需要先把請求的參數(shù)提取出來進(jìn)行判斷，如果是 JSON 參數(shù)，此時就會有問題，參數(shù)提前取出來了，將來在接口中再去獲取 JSON 參數(shù)，就會發(fā)現(xiàn)沒有了。
我們來看看這個問題怎么解決，這也是最近松哥在做的 TienChin 項(xiàng)目的一個小知識點(diǎn)，和大家分享下。
新建一個 Spring Boot 項(xiàng)目，引入 Web 依賴，我們一起來看下面的問題。
1. 問題演示
假設(shè)我現(xiàn)在有一個處理接口冪等性的攔截器，在這個攔截器中，我需要先獲取到請求的參數(shù)，然后進(jìn)行比對等等，我這里先簡單模擬一下，比如我們項(xiàng)目中有如下攔截器：
public class IdempotenceInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
String s = request.getReader().readLine();
System.out.println("s = " + s);
return true;
}
@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
}
}在這個攔截器中先把請求的參數(shù)拎出來，瞅一眼。通過 IO 流讀取出來的參數(shù)最大特點(diǎn)是一次性，也就是讀一次就失效了。
然后我們配置一下這個攔截器：
@Configuration
public class WebConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new IdempotenceInterceptor()).addPathPatterns("/**");
}
}
最后再來看看 Controller 接口：
@RestController
public class HelloController {
@PostMapping("/hello")
public void hello(@RequestBody String msg) throws IOException {
System.out.println("msg = " + msg);
}
}
在接口參數(shù)上我們加了 @RequestBody 注解，這個底層也是通過 IO 流來讀取數(shù)據(jù)的，但是由于 IO 流在攔截器中已經(jīng)被讀取過一次了，所以到了接口中再去讀取就會出錯。報(bào)錯信息如下：
然而很多時候，我們希望 IO 流能夠被多次讀取，那么怎么辦呢?
2. 問題解決
這里我們可以利用裝飾者模式對 HttpServletRequest 的功能進(jìn)行增強(qiáng)，具體做法也很簡單，我們重新定義一個 HttpServletRequest：
public class RepeatedlyRequestWrapper extends HttpServletRequestWrapper {
private final byte[] body;
public RepeatedlyRequestWrapper(HttpServletRequest request, ServletResponse response) throws IOException {
super(request);
request.setCharacterEncoding("UTF-8");
response.setCharacterEncoding("UTF-8");
body = request.getReader().readLine().getBytes("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 int available() throws IOException {
return body.length;
}
@Override
public boolean isFinished() {
return false;
}
@Override
public boolean isReady() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
};
}
}這段代碼并不難，很好懂。
首先在構(gòu)造 RepeatedlyRequestWrapper 的時候，就通過 IO 流將數(shù)據(jù)讀取出來并存入到一個 byte 數(shù)組中，然后重寫 getReader 和 getInputStream 方法，在這兩個讀取 IO 流的方法中，都從 byte 數(shù)組中返回 IO 流數(shù)據(jù)出來，這樣就實(shí)現(xiàn)了反復(fù)讀取了。
接下來我們定義一個過濾器，讓這個裝飾后的 Request 生效：
public class RepeatableFilter implements Filter {
@Override
public void init(FilterConfig filterConfig) throws ServletException {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
throws IOException, ServletException {
ServletRequest requestWrapper = null;
if (request instanceof HttpServletRequest
&& StringUtils.startsWithIgnoreCase(request.getContentType(), MediaType.APPLICATION_JSON_VALUE)) {
requestWrapper = new RepeatedlyRequestWrapper((HttpServletRequest) request, response);
}
if (null == requestWrapper) {
chain.doFilter(request, response);
} else {
chain.doFilter(requestWrapper, response);
}
}
@Override
public void destroy() {
}
}判斷一下，如果請求數(shù)據(jù)類型是 JSON 的話，就把 HttpServletRequest “偷梁換柱”改為 RepeatedlyRequestWrapper，然后讓過濾器繼續(xù)往下走。
最后再配置一下這個過濾器：
@Bean
FilterRegistrationBeanrepeatableFilterBean() {
FilterRegistrationBeanbean = new FilterRegistrationBean<>();
bean.addUrlPatterns("/*");
bean.setFilter(new RepeatableFilter());
return bean;
}
好啦大功告成。
以后，我們的 JSON 數(shù)據(jù)就可以通過 IO 流反復(fù)讀取了。
分享名稱：JSON數(shù)據(jù)讀一次就沒了，怎么辦？
網(wǎng)頁鏈接：http://m.fisionsoft.com.cn/article/cdpoddp.html
