Byte[]数据接收优化策略

最新推荐文章于 2024-07-01 17:20:46 发布

原创最新推荐文章于 2024-07-01 17:20:46 发布 · 397 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

该代码段展示了如何优化从InputStream读取字节数据到byte数组的方法，减少了ByteArrayOutputStream的锁竞争，并提高了内存利用率。通过动态调整byte数组大小，避免了不必要的内存分配，适合处理大量数据的传输场景。

//减少ByteArrayOutputStream write方法的锁结构，
//和增加byte数组空间的利用率，可以适用于数据量较大的传输
public static byte[] readByte(InputStream in) throws IOException {
	int off = 0;
	int length = 1024;
	byte[] data = new byte[length];

	while ((length = in.read(data, off, length)) > 0) {
		off = off + length;

		if (off < data.length){
			break;
		}else {
			length = in.available();
			data = Arrays.copyOf(data, off + (length = (length > 0 ? length + 1 : data.length << 1)));
		}
	}
	if (off == 0) return null;
	return  Arrays.copyOf(data, off);
}

优化项目中获取Http请求byte数据的ByteArrayOutputStream write方法。

希望有什么缺陷或者更加好的方式可以评论区提出，谢谢。