Java IO：使用mark/reset实现替换文件中的字符串

最新推荐文章于 2022-01-03 21:14:53 发布

weixin_34364071

最新推荐文章于 2022-01-03 21:14:53 发布

阅读量123

点赞数

文章标签： java python

原文链接：https://my.oschina.net/javamaster/blog/2988294

版权

2019独角兽企业重金招聘Python工程师标准>>>

最近在补习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();
	}

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