记一次OutputStream转InputStream

最新推荐文章于 2024-04-16 14:28:02 发布

weixin_34015566

最新推荐文章于 2024-04-16 14:28:02 发布

阅读量347

点赞数

文章标签： python java

原文链接：https://my.oschina.net/liuyuantao/blog/793347

版权

2019独角兽企业重金招聘Python工程师标准>>>

写这篇博客的原因是做的项目需要下载文件，后台从数据库读取数据，然后数据已经写入到OutputStream，现在需要使用ServletOutputStream进行处理，所以需要将OutputStream转成InputStream再转成ServletOutputStream。

首先要先理解什么是流！Stream的概念有点类似于pipe管道，就是你只能看到一段数据，
必须顺序读取，
从开始到结束，
才能访问到全部内容
而且你事先是不知道长度的。
所以上述方法，不管哪一种，都是要先把流从inputStream里读取，然后放到outStream里
这个“取”，“放”的动作可以同步，也可以异步。

标准的做法：

in = openInStream();
    out = openOutStream()';
    byte[] buff = new byte[1024];
    while (in.read(buff) != -1) {
          out.write(buff);
    }

所以就想到使用ByteArrayOutStream

ByteArrayOutStreamout = new ByteArrayOutStream();

把数据写入到ByteArrayOutStream()这个字节数组中

InputStream in = new ByteArrayInputStream(out.toByteArray())

创建输入流，构造函数中传递有数据的字节数组。

全部的源代码：

后台读取数据并写入数据的工具方法部分代码：

ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
InputStream in = null;
try{
    doc.save(byteArrayOutputStream,SaveFormat.PDF);
    in = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
}catch(Exception e){
    e.printStackTrace();
}
return in;

前台控制器SpringMVC部分代码：

InputStream in = writeMistake2Local.getExportIdList(Arrays.asList(splitId),Arrays.asList(splitItem));
String format = userName + String.valueOf(1 + Math.random()*10);
response.setContentType("multipart/form-data");
response.setHeader("Content-Disposition","attachment;fileName=" + format + ".pdf");
ServletOutputStream out = null;
try{
    out = response.getOutputStream();
    int b = -1;
    byte[] buffer = new byte[1024];
    while((b = in.read(buffer)) != -1){
         out.write(buffer,0,b);
    }
    in.close();
    out.close();
    out.flush();
}catch(IOException e){
    response.setHeader("Content-Disposition","attachment;fileName=导出错题出错！.pdf");
}

转载于:https://my.oschina.net/liuyuantao/blog/793347