快捷導(dǎo)航

Java中InputStream重復(fù)使用問(wèn)題的幾種解決方案

更新時(shí)間：2025年08月09日 14:06:42 作者：ecexy78041

在Java開(kāi)發(fā)中,InputStream是用于讀取字節(jié)流的類,在許多場(chǎng)景下,我們可能需要重復(fù)讀取InputStream中的數(shù)據(jù),這篇文章主要介紹了Java中InputStream重復(fù)使用問(wèn)題的幾種解決方案,需要的朋友可以參考下

前言

在Java中，InputStream通常設(shè)計(jì)為"一次性"使用，讀取后流會(huì)到達(dá)末尾，再次讀取時(shí)會(huì)返回-1或拋出異常。以下是幾種實(shí)現(xiàn)InputStream重復(fù)使用的方法：

1. 使用mark()和reset()方法（適用于支持標(biāo)記的流）

java

import java.io.*;

public class InputStreamReuser {
    public static void main(String[] args) throws IOException {
        String data = "Hello World";
        InputStream inputStream = new ByteArrayInputStream(data.getBytes());
        
        // 第一次讀取前標(biāo)記
        if (inputStream.markSupported()) {
            inputStream.mark(0); // 參數(shù)是readlimit，0表示不限制
        }
        
        // 第一次讀取
        System.out.println("第一次讀取:");
        readAndPrint(inputStream);
        
        // 重置流
        if (inputStream.markSupported()) {
            inputStream.reset();
        }
        
        // 第二次讀取
        System.out.println("第二次讀取:");
        readAndPrint(inputStream);
    }
    
    private static void readAndPrint(InputStream is) throws IOException {
        int content;
        while ((content = is.read()) != -1) {
            System.out.print((char) content);
        }
        System.out.println();
    }
}

2. 將流內(nèi)容緩存到字節(jié)數(shù)組（最可靠方法）

java

import java.io.*;

public class InputStreamCacher {
    public static void main(String[] args) throws IOException {
        InputStream originalStream = new FileInputStream("example.txt");
        
        // 將流內(nèi)容緩存到字節(jié)數(shù)組
        byte[] bytes = readAllBytes(originalStream);
        originalStream.close();
        
        // 創(chuàng)建可重復(fù)使用的流
        InputStream reusableStream1 = new ByteArrayInputStream(bytes);
        InputStream reusableStream2 = new ByteArrayInputStream(bytes);
        
        // 使用流1
        System.out.println("第一次使用:");
        readAndPrint(reusableStream1);
        
        // 使用流2
        System.out.println("第二次使用:");
        readAndPrint(reusableStream2);
    }
    
    private static byte[] readAllBytes(InputStream is) throws IOException {
        ByteArrayOutputStream buffer = new ByteArrayOutputStream();
        byte[] data = new byte[1024];
        int nRead;
        while ((nRead = is.read(data, 0, data.length)) != -1) {
            buffer.write(data, 0, nRead);
        }
        return buffer.toByteArray();
    }
    
    private static void readAndPrint(InputStream is) throws IOException {
        int content;
        while ((content = is.read()) != -1) {
            System.out.print((char) content);
        }
        System.out.println();
    }
}

3. 使用Apache Commons IO的IOUtils（簡(jiǎn)化代碼）

java

import org.apache.commons.io.IOUtils;
import java.io.*;

public class CommonsIOExample {
    public static void main(String[] args) throws IOException {
        InputStream originalStream = new FileInputStream("example.txt");
        
        // 將流轉(zhuǎn)換為字節(jié)數(shù)組
        byte[] bytes = IOUtils.toByteArray(originalStream);
        originalStream.close();
        
        // 創(chuàng)建可重復(fù)使用的流
        InputStream reusableStream1 = new ByteArrayInputStream(bytes);
        InputStream reusableStream2 = new ByteArrayInputStream(bytes);
        
        // 使用流
        System.out.println("第一次使用: " + IOUtils.toString(reusableStream1, "UTF-8"));
        System.out.println("第二次使用: " + IOUtils.toString(reusableStream2, "UTF-8"));
    }
}

4. 自定義可重置的InputStream包裝器

java

import java.io.*;

public class ResettableInputStream extends InputStream {
    private final InputStream original;
    private InputStream current;
    private byte[] buffer;
    private boolean buffered = false;
    
    public ResettableInputStream(InputStream original) {
        this.original = original;
        this.current = original;
    }
    
    @Override
    public int read() throws IOException {
        return current.read();
    }
    
    public void resetStream() throws IOException {
        if (!buffered) {
            // 第一次重置時(shí)緩存內(nèi)容
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            byte[] data = new byte[1024];
            int nRead;
            while ((nRead = original.read(data, 0, data.length)) != -1) {
                baos.write(data, 0, nRead);
            }
            buffer = baos.toByteArray();
            buffered = true;
        }
        current = new ByteArrayInputStream(buffer);
    }
    
    @Override
    public void close() throws IOException {
        original.close();
        if (buffered && current != null) {
            current.close();
        }
    }
}

// 使用示例
public class Main {
    public static void main(String[] args) throws IOException {
        ResettableInputStream ris = new ResettableInputStream(new FileInputStream("example.txt"));
        
        // 第一次讀取
        System.out.println("第一次讀取:");
        readAndPrint(ris);
        
        // 重置
        ris.resetStream();
        
        // 第二次讀取
        System.out.println("第二次讀取:");
        readAndPrint(ris);
        
        ris.close();
    }
    
    private static void readAndPrint(InputStream is) throws IOException {
        int content;
        while ((content = is.read()) != -1) {
            System.out.print((char) content);
        }
        System.out.println();
    }
}

注意事項(xiàng)

內(nèi)存考慮：緩存整個(gè)流到內(nèi)存中可能不適合處理非常大的文件
資源釋放：確保在所有使用完成后關(guān)閉流
性能影響：重復(fù)讀取相同的流內(nèi)容會(huì)有額外的內(nèi)存開(kāi)銷
網(wǎng)絡(luò)流：對(duì)于網(wǎng)絡(luò)流，通常無(wú)法重置，必須先緩存內(nèi)容
標(biāo)記支持：不是所有InputStream實(shí)現(xiàn)都支持mark()和reset()

總結(jié)

到此這篇關(guān)于Java中InputStream重復(fù)使用問(wèn)題的幾種解決方案的文章就介紹到這了,更多相關(guān)Java中InputStream重復(fù)使用內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: