问题描述:
在项目中用有个上传功能,会把文件转换成inputstream然后转换成byte[] 数组
转换代码如下:
publicstaticbyte[] convertstream(InputStream inStream)throwsIOException {
ByteArrayOutputStream swapStream = newByteArrayOutputStream();
byte[] buff =newbyte[10240];
intrc =0;
while((rc = inStream.read(buff,0,10240)) >0) {
swapStream.write(buff, 0, rc);
}
byte[] buf = swapStream.toByteArray();
swapStream.close();
returnbuf;
}根据测试,上传一个7M的文件, 但是执行swapStream.write(buff, 0, rc); 这段代码的时候,一共消耗了20M的内存,不应该啊,内存多么重要的东西。
问题原因:
查看 ByteArrayOutputStream 的API得知
此类实现了一个输出流,其中的数据被写入一个字节数组。缓冲区会随着数据的不断写入而自动增长。可使用 toByteArray() 和 toString() 检索数据。
得知一个关键信息,就是缓冲区,查看ByteArrayOutputStream 源代码如下:
/**
* Creates a new byte array output stream. The buffer capacity is
* initially 32 bytes, though its size increases if necessary.
*/
publicByteArrayOutputStream() {
this(32);
}我使用ByteArrayOutputStream的时候并没有在构造函数制定缓冲区的大小,所以默认是初始化32bytes,但是对于7M大的文件来说,32bytes 远远不够,所以
ByteArrayOutputStream 缓冲区会自动增长,这个自动增长的过程,根据源码来看会有copyArray动作,所以整个write过程,消耗内存成倍增长。
源码如下:
publicsynchronizedvoidwrite(byteb[],intoff,intlen) {
if((off <0) || (off > b.length) || (len <0) ||
((off + len) > b.length) || ((off + len)
thrownewIndexOutOfBoundsException();
} elseif(len ==0) {
return;
}
intnewcount = count + len;
if(newcount > buf.length) {
buf = Arrays.copyOf(buf, Math.max(buf.length <
}
System.arraycopy(b, off, buf, count, len);
count = newcount;
}
解决办法:
我是这么解决的,不知道还有没有更好的办法。
我们可以得到上传文件File对象的length,在创建 ByteArrayOutputStream 对象的时候把length属性放在构造函数里面:
publicstaticbyte[] convertstream(InputStream inStream,longlength)throwsIOException {
length = length > Integer.MAX_VALUE ? Integer.MAX_VALUE : length;
ByteArrayOutputStream swapStream = newByteArrayOutputStream((int)length);
byte[] buff =newbyte[10240];
intrc =0;
while((rc = inStream.read(buff,0,10240)) >0) {
swapStream.write(buff, 0, rc);
}
byte[] buf = swapStream.toByteArray();
swapStream.close();
returnbuf;
}最后测试结果
swapStream.write(buff,0, rc); 一共消耗7M内存若有更好的解决方案,请告知。