Lucene_demo09_txt文件索引

Lucene_demo09_txt文件索引

/**
 * txt文件索引
 */
public class IndexFile {

	private Directory directory;

	private String indexPath = "D://lucene/index"; // 建立索引文件的目录

	private String dirPath = "D://lucene/data"; // txt资源目录

	private IndexWriter indexWriter;

	/*
	 * 获得所有txt文件
	 */
	public List<File> getFileList(String dirPath) {
		File[] files = new File(dirPath).listFiles();
		List<File> fileList = new ArrayList<File>();
		for (File file : files) {
			if (isTxtFile(file.getName())) {
				fileList.add(file);
			}
		}
		return fileList;
	}

	/*
	 * 判断是否是txt文件
	 */
	public boolean isTxtFile(String fileName) {
		if (fileName.lastIndexOf(".txt") > 0) {
			return true;
		}
		return false;
	}

	/*
	 * 将文件转换成Document对象
	 */
	public Document fileToDocument(File file) throws Exception {
		Document document = new Document();
		document.add(new Field("filename", file.getName(), Store.YES, Index.ANALYZED));
		document.add(new Field("content", getFileContent(file), Store.YES, Index.ANALYZED));
		document.add(new Field("size", String.valueOf(file.getTotalSpace()), Store.YES, Index.ANALYZED));
		return document;
	}

	/*
	 * 获得indexwriter对象
	 */
	public IndexWriter getIndexWriter(Directory dir) throws Exception {
		IndexWriter indexWriter = new IndexWriter(dir, LuceneUtils.analyzer, MaxFieldLength.LIMITED);
		return indexWriter;
	}

	/*
	 * 关闭indexwriter对象
	 */
	public void closeWriter() throws Exception {
		if (indexWriter != null) {
			indexWriter.close();
		}
	}

	/*
	 * 读取文件内容
	 */
	public String getFileContent(File file) throws Exception {
		Reader reader = new InputStreamReader(new FileInputStream(file), "UTF-8");
		BufferedReader br = new BufferedReader(reader);
		String result = "";
		while (br.readLine() != null) {
			result = result + "\n" + br.readLine();
		}
		br.close();
		reader.close();
		return result;
	}

	/**
	 * 启动初始化
	 */
	@Before
	public void init() {
		try {
			directory = FSDirectory.open(new File(indexPath));
			indexWriter = getIndexWriter(directory);
		}
		catch (Exception e) {
			System.out.println("索引打开异常！");
		}
	}

	/**
	 * 创建索引 Main方法
	 */
	@Test
	public void createIndex() throws Exception {
		List<File> fileList = getFileList(dirPath);
		Document document = null;
		for (File file : fileList) {
			document = fileToDocument(file);
			indexWriter.addDocument(document);
			System.out.println("filename:" + document.get("filename"));
			System.err.println("content:" + document.get("content"));
			indexWriter.commit();
		}
		closeWriter();
	}
}