一、使用ByteArrayOutputStream的方式
在Java语言中,InputStream是读取数据的基本方式。但是,有时候我们需要将InputStream转化为Byte数组。而使用ByteArrayOutputStream
是最常见的方法之一。下面是代码示例:
public byte[] readBytesFromInputStream(InputStream input) throws IOException { ByteArrayOutputStream output = new ByteArrayOutputStream(); byte[] buffer = new byte[4096]; int n = 0; while (-1 != (n = input.read(buffer))) { output.write(buffer, 0, n); } return output.toByteArray(); }
首先,我们创建了ByteArrayOutputStream
对象output
,并定义了一个byte
类型的buffer
数组。接下来,我们使用input.read(buffer)
读取input
中的数据到buffer
中,并通过output.write(buffer, 0, n)
将buffer
中的数据写入到output
中。最后,调用output.toByteArray()
将output
转换成byte
数组。该方法的缺点是需要在内存中保持所有数据。
二、使用ByteArrayInputStream的方式
除了使用ByteArrayOutputStream
实现InputStream转换成byte数组,还可以使用ByteArrayInputStream
。下面是代码示例:
public byte[] readBytesFromInputStream(InputStream input) { try { return org.apache.commons.io.IOUtils.toByteArray(input); } catch (IOException e) { e.printStackTrace(); } return null; }
在这个方法中,我们使用了Apache Commons IO库,调用org.apache.commons.io.IOUtils
的toByteArray
方法将input
转换为byte
数组。这种方法不需要担心内存限制,因为只有需要数据时才会从input
对象中读取。
三、使用BufferedInputStream的方式
Java标准库中,还提供了BufferedInputStream
,可以实现输入流的缓存。下面是代码示例:
public byte[] readBytesFromInputStream(InputStream input) { try { BufferedInputStream bufferedInput = new BufferedInputStream(input); byte[] result = new byte[bufferedInput.available()]; bufferedInput.read(result, 0, result.length); return result; } catch (IOException e) { e.printStackTrace(); } return null; }
在这个方法中,我们首先创建了BufferedInputStream
对象bufferedInput
,然后调用available()
获取输入流的字节数,创建相应大小的byte
数组result
。最后,调用bufferedInput.read(result, 0, result.length)
从输入流中读取数据。
四、InputStream转ByteArray的性能比较
下面的代码演示了三个方法的性能比较:
public static void main(String[] args) { int count = 100; long start1 = System.currentTimeMillis(); for (int i = 0; i < count; i++) { InputStream input = new ByteArrayInputStream("Hello World!".getBytes()); readBytesFromInputStream(input); } long end1 = System.currentTimeMillis(); long start2 = System.currentTimeMillis(); for (int i = 0; i < count; i++) { InputStream input = new ByteArrayInputStream("Hello World!".getBytes()); readBytesFromInputStream2(input); } long end2 = System.currentTimeMillis(); long start3 = System.currentTimeMillis(); for (int i = 0; i < count; i++) { InputStream input = new ByteArrayInputStream("Hello World!".getBytes()); readBytesFromInputStream3(input); } long end3 = System.currentTimeMillis(); System.out.println("ByteArrayOutputStream 方法耗时:" + (end1 - start1) + " ms"); System.out.println("IOUtils.toByteArray 方法耗时:" + (end2 - start2) + " ms"); System.out.println("BufferedInputStream 方法耗时:" + (end3 - start3) + " ms"); } private static byte[] readBytesFromInputStream(InputStream input) { try { ByteArrayOutputStream output = new ByteArrayOutputStream(); byte[] buffer = new byte[4096]; int n = 0; while (-1 != (n = input.read(buffer))) { output.write(buffer, 0, n); } return output.toByteArray(); } catch (IOException e) { e.printStackTrace(); } return null; } private static byte[] readBytesFromInputStream2(InputStream input) { try { return org.apache.commons.io.IOUtils.toByteArray(input); } catch (IOException e) { e.printStackTrace(); } return null; } private static byte[] readBytesFromInputStream3(InputStream input) { try { BufferedInputStream bufferedInput = new BufferedInputStream(input); byte[] result = new byte[bufferedInput.available()]; bufferedInput.read(result, 0, result.length); return result; } catch (IOException e) { e.printStackTrace(); } return null; }
可以看到,使用ByteArrayOutputStream
的方式是最慢的,使用BufferedInputStream
的方式较快,而使用IOUtils.toByteArray
方法的方式最快。在实际情况中,需要根据自己的需求选择合适的方法。
五、总结
本文介绍了三种将InputStream
转化为Byte
数组的方式,分别是使用ByteArrayOutputStream
、ByteArrayInputStream
以及BufferedInputStream
。此外,我们还对三种方法进行了性能比较。在实际使用中需要根据自己的需求选择最适合的方法。