1. 程式人生 > >Java實現inputstream流的複製

Java實現inputstream流的複製

獲取到一個inputstream後,可能要多次利用它進行read的操作。由於流讀過一次就不能再讀了,而InputStream物件本身不能複製,而且它也沒有實現Cloneable介面,所以得想點辦法。

實現思路:

1、先把InputStream轉化成ByteArrayOutputStream

2、後面要使用InputStream物件時,再從ByteArrayOutputStream轉化回來

程式碼實現如下:

package com.test;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

public class StreamOperateUtil {
	public static void main(String[] args) throws FileNotFoundException {
		 
		InputStream input =  new FileInputStream("c:\\test.txt"); 
		//InputStream input =  httpconn.getInputStream(); //這裡可以寫你獲取到的流
		
		ByteArrayOutputStream baos = cloneInputStream(input);
		
		// 開啟兩個新的輸入流  
		InputStream stream1 = new ByteArrayInputStream(baos.toByteArray());  
		InputStream stream2 = new ByteArrayInputStream(baos.toByteArray());
		
	}

	private static ByteArrayOutputStream cloneInputStream(InputStream input) {
		try {
			ByteArrayOutputStream baos = new ByteArrayOutputStream();
			byte[] buffer = new byte[1024];
			int len;
			while ((len = input.read(buffer)) > -1) {
				baos.write(buffer, 0, len);
			}
			baos.flush();
			return baos;
		} catch (IOException e) {
			e.printStackTrace();
			return null;
		}
	}

}

這種適用於一些不是很大的流,因為快取流是會消耗記憶體的。

備註:關於InputStream為什麼不能被重複讀取?