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
}
}