package com.yk.replace;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.RandomAccessFile;

/**
 * RandomAccessFile向指定文件，指定位置插入內(nèi)容
 * 1.先使用File的createTempFile創(chuàng)建一個(gè)臨時(shí)文件
 * 2.將插入以后的文件內(nèi)容（末尾節(jié)點(diǎn)開(kāi)始后的內(nèi)容）保存到臨時(shí)文件中
 * 3.程序重新定位到插入點(diǎn)，將要插入的內(nèi)容保存到節(jié)點(diǎn)后
 * 4.這樣就相當(dāng)于完成了文件的拼接
 * 
 * @author yangkai
 */
public class InsertContent {
	public static void insert(String fileName, String startContext, String endContext, String insertContent)
			throws IOException {

		RandomAccessFile raf = null;
		// 創(chuàng)建臨時(shí)文件保存插入點(diǎn)后數(shù)據(jù)
		File tmp = File.createTempFile("tmp", null);
		FileOutputStream tmpout = null;
		FileInputStream tmpin = null;

		// 定位節(jié)點(diǎn)使用，readline方法不能識(shí)別\r\n,因此需要自己添加進(jìn)去，最開(kāi)始因?yàn)檫@個(gè)檢索節(jié)點(diǎn)始終不準(zhǔn)耽誤了很長(zhǎng)時(shí)間
		StringBuffer buffer = new StringBuffer();
		BufferedReader bf = new BufferedReader(new FileReader(fileName));
		String s = null;
		while ((s = bf.readLine()) != null) {// 使用readLine方法，一次讀一行
			buffer.append(s + "\r\n");
		}

		String xml = buffer.toString();

		long posStatr = xml.indexOf(startContext) + startContext.length() + 2;
		long posEnd = xml.indexOf(endContext);

		tmp.deleteOnExit();// jvm退出的時(shí)候刪除臨時(shí)文件
		try {
			raf = new RandomAccessFile(fileName, "rw");
			tmpout = new FileOutputStream(tmp);
			tmpin = new FileInputStream(tmp);
			raf.seek(posEnd);
			// 將插入點(diǎn)后的內(nèi)容讀入臨時(shí)文件中保存
			byte[] bbuf = new byte[64];
			int hasRead = 0;
			while ((hasRead = raf.read(bbuf)) > 0) {
				tmpout.write(bbuf, 0, hasRead);
			}
			// 把文件記錄指針定位到pos
			raf.seek(posStatr);
			raf.write(insertContent.getBytes());
			while ((hasRead = tmpin.read(bbuf)) > 0) {
				raf.write(bbuf, 0, hasRead);
			}

		} finally {
			raf.close();
		}

	}

	public static void main(String[] args) throws IOException {
		insert("filePath","posStart","posEnd", "replaceContext\r\n");
	}

}