文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

springboot中inputStream神秘消失的示例分析

2023-06-20 20:53

关注

这篇文章主要为大家展示了“springboot中inputStream神秘消失的示例分析”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“springboot中inputStream神秘消失的示例分析”这篇文章吧。

HTTPClient 模拟调用

为了把这个问题说明,我们首先从最简单的 http 调用说起。

设置 body

服务端

服务端的代码如下:

@Controller@RequestMapping("/")public class ReqController {    @PostMapping(value = "/body")    @ResponseBody    public String body(HttpServletRequest httpServletRequest) {        try {            String body = StreamUtil.toString(httpServletRequest.getInputStream());            System.out.println("请求的 body: " + body);            // 从参数中获取            return body;        } catch (IOException e) {            e.printStackTrace();            return e.getMessage();        }    }}

java 客户端要如何请求才能让服务端读取到传递的 body 呢?

客户端

这个问题一定难不到你,实现的方式有很多种。

我们以 apache httpclient 为例:

//post请求,带集合参数public static String post(String url, String body) {    try {        // 通过HttpPost来发送post请求        HttpPost httpPost = new HttpPost(url);        StringEntity stringEntity = new StringEntity(body);        // 通过setEntity 将我们的entity对象传递过去        httpPost.setEntity(stringEntity);        return execute(httpPost);    } catch (UnsupportedEncodingException e) {        throw new RuntimeException(e);    }}//执行请求返回响应数据private static String execute(HttpRequestBase http) {    try {        CloseableHttpClient client = HttpClients.createDefault();        // 通过client调用execute方法        CloseableHttpResponse Response = client.execute(http);        //获取响应数据        HttpEntity entity = Response.getEntity();        //将数据转换成字符串        String str = EntityUtils.toString(entity, "UTF-8");        //关闭        Response.close();        return str;    } catch (IOException e) {        throw new RuntimeException(e);    }}

可以发现 httpclient 封装之后还是非常方便的。

我们设置 setEntity 为对应入参的 StringEntity 即可。

测试

为了验证正确性,小明本地实现了一个验证方法。

@Testpublic void bodyTest() {    String url = "http://localhost:8080/body";    String body = buildBody();    String result = HttpClientUtils.post(url, body);    Assert.assertEquals("body", result);}private String buildBody() {    return "body";}

很轻松,小明漏出了龙王的微笑。

设置 parameter

服务端

小明又看到有一个服务端的代码实现如下:

@PostMapping(value = "/param")@ResponseBodypublic String param(HttpServletRequest httpServletRequest) {    // 从参数中获取    String param = httpServletRequest.getParameter("id");    System.out.println("param: " + param);    return param;}private Map<String,String> buildParamMap() {    Map<String,String> map = new HashMap<>();    map.put("id", "123456");    return map;}

所有的参数是通过 getParameter 方法获取,应该如何实现呢?

客户端

这个倒也不难,小明心想。

因为以前很多代码都是这样实现的,于是 ctrl+CV 搞定了下面的代码:

//post请求,带集合参数public static String post(String url, Map<String, String> paramMap) {    List<NameValuePair> nameValuePairs = new ArrayList<>();    for (Map.Entry<String, String> entry : paramMap.entrySet()) {        NameValuePair pair = new BasicNameValuePair(entry.getKey(), entry.getValue());        nameValuePairs.add(pair);    }    return post(url, nameValuePairs);}//post请求,带集合参数private static String post(String url, List<NameValuePair> list) {    try {        // 通过HttpPost来发送post请求        HttpPost httpPost = new HttpPost(url);        // 我们发现Entity是一个接口,所以只能找实现类,发现实现类又需要一个集合,集合的泛型是NameValuePair类型        UrlEncodedFormEntity formEntity = new UrlEncodedFormEntity(list);        // 通过setEntity 将我们的entity对象传递过去        httpPost.setEntity(formEntity);        return execute(httpPost);    } catch (Exception exception) {        throw new RuntimeException(exception);    }}

如此是最常用的 paramMap,便于构建;和具体的实现方式脱离,也便于后期拓展。

servlet 标准

UrlEncodedFormEntity 看似平平无奇,表示这是一个 post 表单请求。

里面还涉及到 servlet 3.1 的一个标准,必须满足下面的标准,post 表单的 parameter 集合才可用。

请求是 http 或 https

请求的方法是 POST

content type 为: application/x-www-form-urlencoded

servlet 已经在 request 对象上调用了相关的 getParameter 方法。

当以上条件不满足时,POST 表单的数据并不会设置到 parameter 集合中,但依然可以通过 request 对象的 inputstream 来获取。

当以上条件满足时,POST 表单的数据在 request 对象的 inputstream 将不再可用了。

这是很重要的一个约定,导致很多小伙伴比较蒙圈。

测试

于是,小明也写好了对应的测试用例:

@Testpublic void paramTest() {    String url = "http://localhost:8080/param";    Map<String,String> map = buildParamMap();    String result = HttpClientUtils.post(url, map);    Assert.assertEquals("123456", result);}

如果谈恋爱能像编程一样,那该多好。

springboot中inputStream神秘消失的示例分析

小明想着,却不由得眉头一皱,发现事情并不简单。

设置 parameter 和 body

服务端

有一个请求的入参是比较大,所以放在 body 中,其他参数依然放在 paramter 中。

@PostMapping(value = "/paramAndBody")@ResponseBodypublic String paramAndBody(HttpServletRequest httpServletRequest) {    try {        // 从参数中获取        String param = httpServletRequest.getParameter("id");        System.out.println("param: " + param);        String body = StreamUtil.toString(httpServletRequest.getInputStream());        System.out.println("请求的 body: " + body);        // 从参数中获取        return param+"-"+body;    } catch (IOException e) {        e.printStackTrace();        return e.getMessage();    }}

其中 StreamUtil#toString 是一个对流简单处理的工具类。

public static String toString(final InputStream inputStream)  {    if (inputStream == null) {        return null;    }    try {        int length = inputStream.available();        final Reader reader = new InputStreamReader(inputStream, StandardCharsets.UTF_8);        final CharArrayBuffer buffer = new CharArrayBuffer(length);        final char[] tmp = new char[1024];        int l;        while((l = reader.read(tmp)) != -1) {            buffer.append(tmp, 0, l);        }        return buffer.toString();    } catch (Exception exception) {        throw new RuntimeException(exception);    }}

客户端

那么问题来了,如何同时在 HttpClient 中设置 parameter 和 body 呢?

机智的小伙伴们可以自己先尝试一下。

小明尝试了多种方法,发现一个残酷的现实—— httpPost 只能设置一个 Entity,也尝试看了各种子类,然并LUAN。

就在小明想要放弃的时候,小明忽然想到,paramter 完全可以通过拼接 URL 的方式实现。

也就是我们把 parameter 和 url 并且为一个新的 URL,body 和以前设置方式一样。

实现代码如下:

//post请求,带集合参数public static String post(String url, Map<String, String> paramMap,                          String body) {    try {        List<NameValuePair> nameValuePairs = new ArrayList<>();        for (Map.Entry<String, String> entry : paramMap.entrySet()) {            NameValuePair pair = new BasicNameValuePair(entry.getKey(), entry.getValue());            nameValuePairs.add(pair);        }        // 构建 url        //构造请求路径,并添加参数        URI uri = new URIBuilder(url).addParameters(nameValuePairs).build();        //构造HttpClient        CloseableHttpClient httpClient = HttpClients.createDefault();        // 通过HttpPost来发送post请求        HttpPost httpPost = new HttpPost(uri);        httpPost.setEntity(new StringEntity(body));        // 获取响应        // 通过client调用execute方法        CloseableHttpResponse Response = httpClient.execute(httpPost);        //获取响应数据        HttpEntity entity = Response.getEntity();        //将数据转换成字符串        String str = EntityUtils.toString(entity, "UTF-8");        //关闭        Response.close();        return str;    } catch (URISyntaxException | IOException | ParseException e) {        throw new RuntimeException(e);    }}

这里通过 new URIBuilder(url).addParameters(nameValuePairs).build() 构建新的 URL,当然你可以使用 &key=value 的方式自己拼接。

测试代码

@Testpublic void paramAndBodyTest() {    String url = "http://localhost:8080/paramAndBody";    Map<String,String> map = buildParamMap();    String body = buildBody();    String result = HttpClientUtils.post(url, map, body);    Assert.assertEquals("123456-body", result);}

测试通过,非常完美。

新的征程

当然,一般的文章到这里就该结束了。

不过上面并不是本文的重点,我们的故事才刚刚开始。

日志需求

大雁飞过,天空一定会留下他的痕迹。

程序更应如此。

为了方便的跟踪问题,我们一般都是对调用的入参进行日志留痕。

为了便于代码拓展和可维护性,小明当然采用拦截器的方式。

日志拦截器

import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.stereotype.Component;import org.springframework.util.StreamUtils;import org.springframework.web.servlet.HandlerInterceptor;import org.springframework.web.servlet.ModelAndView;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import java.nio.charset.StandardCharsets;import java.util.Enumeration;@Componentpublic class LogHandlerInterceptor implements HandlerInterceptor {    private Logger logger = LoggerFactory.getLogger(LogHandlerInterceptor.class);    @Override    public boolean preHandle(HttpServletRequest httpServletRequest,                             HttpServletResponse httpServletResponse, Object o) throws Exception {        // 获取参数信息        Enumeration<String> enumeration = httpServletRequest.getParameterNames();        while (enumeration.hasMoreElements()) {            String paraName = enumeration.nextElement();            logger.info("Param name: {}, value: {}", paraName, httpServletRequest.getParameter(paraName));        }        // 获取 body 信息        String body = StreamUtils.copyToString(httpServletRequest.getInputStream(), StandardCharsets.UTF_8);        logger.info("body: {}", body);        return true;    }    @Override    public void postHandle(HttpServletRequest httpServletRequest,                           HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {    }    @Override    public void afterCompletion(HttpServletRequest httpServletRequest,                                HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {    }}

非常的简单易懂,输出入参中的 parameter 参数和 body 信息。

然后指定一下生效的范围:

@Configurationpublic class SpringMvcConfig extends WebMvcConfigurerAdapter {    @Autowired    private LogHandlerInterceptor logHandlerInterceptor;    @Override    public void addInterceptors(InterceptorRegistry registry) {        registry.addInterceptor(logHandlerInterceptor)                .addPathPatterns("public class MyHttpServletRequestWrapper extends HttpServletRequestWrapper {    private byte[] requestBody = null;//用于将流保存下来    public MyHttpServletRequestWrapper(HttpServletRequest request) throws IOException {        super(request);        requestBody = StreamUtils.copyToByteArray(request.getInputStream());    }    @Override    public ServletInputStream getInputStream() {        final ByteArrayInputStream bais = new ByteArrayInputStream(requestBody);        return new ServletInputStream() {            @Override            public int read() {                return bais.read();  // 读取 requestBody 中的数据            }            @Override            public boolean isFinished() {                return false;            }            @Override            public boolean isReady() {                return false;            }            @Override            public void setReadListener(ReadListener readListener) {            }        };    }    @Override    public BufferedReader getReader() throws IOException {        return new BufferedReader(new InputStreamReader(getInputStream()));    }}

实现 Filter

我们上面重写的 MyHttpServletRequestWrapper 什么时候生效呢?

我们可以自己实现一个 Filter,对原有的请求进行替换:

import org.springframework.stereotype.Component;import javax.servlet.*;import javax.servlet.http.HttpServletRequest;import java.io.IOException;@Componentpublic 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 MyHttpServletRequestWrapper((HttpServletRequest) request);        }        if(requestWrapper == null) {            chain.doFilter(request, response);        } else {            chain.doFilter(requestWrapper, response);        }    }    @Override    public void init(FilterConfig arg0) throws ServletException {}}

然后就可以发现一切都好起来了。

以上是“springboot中inputStream神秘消失的示例分析”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注编程网行业资讯频道!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯