InputStream数据读取不完整-解决方法

IO流的操作作为基本知识相信大家并不陌生,平时开发中也需要经常和各种流打交道。但是对于流的操作,看似简单,有时一不小心也会掉坑里。

记录

下面记录下最近遇到的一个小问题。对于 inputStream 流的操作,有时候为了方便,相信有不少人会像如下一样操作。

获取 inputStream 流 -> 获取数据长度 -> 开辟byte存储空间 -> 一次性读取数据到 byte[] -> 使用byte数据(写出或其他) -> 依次关闭流

以上操作如果是针对本地文件读取,那么没啥问题。但是如果放到网络环境,问题就出现了。且看如下代码:

    @RequestMapping("/demo.do")
    public void demo(HttpServletResponse resp) {
        InputStream is = null;
        OutputStream os = null;
        try {
            String url = "http://xxxxxxxxxxxxxxx";

            URL realUrl = new URL(url);
            HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();

            connection.connect();
            is = connection.getInputStream();

            int len = is.available();
            byte[] bytes = new byte[len];
            is.read(bytes);

            os = resp.getOutputStream();
            os.write(bytes);
            os.flush();

        } catch (Exception e) {
            logger.error("error", e);
        } finally {
            // close stream
        }
    }

可以看到,以上操作从一个网络链接读取数据然后写入 response,如果网络较好,而且文件较小的话,或许能侥幸读取完整数据。但是绝大部分情况下是获取不到完整数据的。因为网络和数据的原因,数据往往是‘分片’发送过来的,如果你做过测试的话你会发现

每次调用 is.available() 获取到的长度都不一样

所以会导致上述问题的发生。也就是 inputStream 数据获取不完整,对于网络文件,如图片等,直观的体现就是图片只加载了很小一部分。如果你使用上述代码下载文件,那么必然是残缺的。

解决方法

那么如何解决呢?方法就是好好使用read方法的返回值,下面提供一种公认的比较标准的写法,使用如下方法完美解决数据残缺问题,当然,如果你上网搜索的话,可以看到其他的一些写法,但是原理都一样,这里供大家参考。

    @RequestMapping("/demo.do")
    public void demo(HttpServletResponse resp) {
        InputStream is = null;
        OutputStream os = null;
        try {
            String url = "http://xxxxxxxxxxxxxxx";

            URL realUrl = new URL(url);
            HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();

            connection.connect();
            is = connection.getInputStream();
            byte[] bytes = new byte[102400];
            int len;
            os = resp.getOutputStream();
            while((len = is.read(bytes)) != -1){
                os.write(bytes, 0, len);
            }
            os.flush();
        } catch (Exception e) {
            logger.error("error", e);
        } finally {
            // close stream
        }
    }
It's
欢迎访问本站,欢迎留言、分享、点赞。愿您阅读愉快!
*转载请注明出处,严禁非法转载。
https://www.devsong.org
QQ留言 邮箱留言
头像
引用:
取消回复
提交
涂鸦
涂鸦
热门