java io 字节替换_Java IO：使用mark/reset实现替换文件中的字符串

最近在补习Java基础，看到了输入和输出这块内容，之前对 InputStream 的 mark(int) 和 reset() 用法不熟悉，翻了一下书，总结一下。

首先看一下《Java核心技术》中怎么介绍这两个方法用法的：

void mark(int readlimit)

在输入流的当前位置打一个标记(并非所有的流都支持这个特性)。如果从输入流中已经读入的字节多于readlimit个，则这个流允许忽略这个标记。

void reset()

返回到最后一个标记，随后对read的调用将重新读入这些字节。如果当前没有任何标记，则这个流不被重置。

boolean markSupported()

如果这个流支持打标记，则返回true。

可能读第一遍不容易理解，解释一下：对当前输入流使用 mark(10) ，表示从当前位置到(当前位置+10)的区间内，我可以试着读完没关系，如果发现读的内容不是我想要的，再调用一下 reset() ，即可以还原刚才读的内容。

下面使用这两个方法来实现替换文档中的目标字符串，生成一个新的文件：

private static final void replaceString(InputStream is, OutputStream os, byte[] src, byte[] dst)

throws IOException {

if (src.length != dst.length)

throw new IllegalArgumentException();

BufferedInputStream bis = new BufferedInputStream(is);

int data;

starter: while ((data = bis.read()) != -1) {

if (data == src[0]) {// 开始匹配目标字符串

bis.mark(10);// 标记开始读取的位置

for (int j = 1; j < src.length; j++) {

int n = bis.read();

if ((n == -1) || (n != src[j])) {// 发现不相同的内容,还原重新匹配

bis.reset();// 重置刚才读过的内容

os.write(data);

continue starter;

}

}

// 匹配完成,找到了目标字符串,执行替换

os.write(dst);

} else {

os.write(data);

}

}

}

public static void main(String[] args) throws Exception {

File file = new File("/data/gitee/Java8Fundamental/src/replaceMe.txt");

InputStream input = new FileInputStream(file);

ByteArrayOutputStream baos = new ByteArrayOutputStream();

// 使用 "aaa" 替换 "AAA"

replaceString(input, baos, "AAA".getBytes(), "aaa".getBytes());

byte[] data = baos.toByteArray();

ByteArrayInputStream bais = new ByteArrayInputStream(data);

baos.reset();

// 使用 "bbb" 替换 "BBB"

replaceString(bais, baos, "BBB".getBytes(), "bbb".getBytes());

data = baos.toByteArray();

bais = new ByteArrayInputStream(data);

baos.reset();

// 使用 "ccc" 替换 "CCC"

replaceString(bais, baos, "CCC".getBytes(), "ccc".getBytes());

// 把替换后的流输出到文件

File outFile = new File("/data/gitee/Java8Fundamental/src/reaplaceMeCopy.txt");

FileOutputStream fos = new FileOutputStream(outFile);

fos.write(baos.toByteArray());

fos.close();

}

原文件里面的内容是这样：

AAA

BBB

CCC

执行之后新的文档内容如下：

aaa

bbb

ccc

使用场景很多，比如说结合JVM中的常量池等。