Lucene对本地文件多目录创建索引

jar包版本说明:
[color=violet]lucene[/color] 版本2.9(当时是为了和solr1.4一起用,不过换成Lucene3.0的jar包代码是不用修改的)
[color=violet]ik分词器[/color] 版本3.1.6(说起来还是solr,高版本的死活配不上去,汗···)
[color=violet]jsoup[/color] 1.4.1用来解析HTML的工具包很好用。(比htmlparser好用多了 呵呵个人意见)

[color=blue]参考范围,本文仅限对TXT,HTML,HTM文件的内容创建索引。[/color]

创建索引的代码如下:

public class IndexJob {
public static Date beginTime;

//读取TXT文件内容
private static String loadFileToString(File file) {
try {
InputStreamReader isr = new InputStreamReader(new FileInputStream(
file), "UTF-8");
BufferedReader br = new BufferedReader(isr);
StringBuffer sb = new StringBuffer();
String line = br.readLine();
while (line != null) {
sb.append(line);
line = br.readLine();
}
br.close();
return sb.toString();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}

//提取HTML文件的文本内容
private static String getDocument(File html) {
String text = "";
try {
//设置编码集
// org.jsoup.nodes.Document doc = Jsoup.parse(html, "UTF-8");
org.jsoup.nodes.Document doc = Jsoup.parse(html,"GBK");

//提取标题信息
Elements title = doc.select("title");
for (org.jsoup.nodes.Element link : title) {
text += link.text() + " ";
}

//提取table中的文本信息
Elements links = doc.select("table");
for (org.jsoup.nodes.Element link : links) {
text += link.text() + " ";
}

//提取div中的文本信息
Elements divs = doc.select("div[class=post]");
for (org.jsoup.nodes.Element link : divs) {
text += link.text() + " ";
}
} catch (IOException e) {
e.printStackTrace();
}

return text;
}

public static void createIndex(String inputDir) {
Directory directory = null;
IndexWriter writer = null;
IndexSearcher searcher = null;
// 索引创建开始时间
Date beginTime = new Date();
try {

//从配置文件中读取索引存放路径
String indexPath = IndexUtil.getValue(IndexUtil.INDEX_ROOT);
directory = FSDirectory.open(new File(indexPath));
File indexFile = new File(indexPath + "\\segments.gen");

//根据索引文件segments.gen是否存在判断是否是第一次创建索引
if (indexFile.exists()) {

//增量添加索引信息
writer = new IndexWriter(directory, new IKAnalyzer(), false,
IndexWriter.MaxFieldLength.LIMITED);
writer.setMergeFactor(1000);
writer.setMaxBufferedDocs(100);
writer.setMaxMergeDocs(Integer.MAX_VALUE);
} else {

//新建索引信息
writer = new IndexWriter(directory, new IKAnalyzer(), true,
IndexWriter.MaxFieldLength.LIMITED);
writer.setMergeFactor(1000);
writer.setMaxBufferedDocs(100);
writer.setMaxMergeDocs(Integer.MAX_VALUE);
}

File fileDir = new File(inputDir);
File[] files = fileDir.listFiles();
if (files.length > 1) {

//当前最新上传的文件夹名字(时间降序第二个,因为最新的可能还未上传完毕)
// String newLastFolder = files[files.length - 2].getName();

//将上次任务记录的文件夹名称取出
// String lastFolder = IndexUtil
// .getValue(IndexUtil.LAST_FOLDER_PATH);


//索引文件中存储的索引字段的定义
Field fieldName = new Field("id", "", Field.Store.YES,
Field.Index.UN_TOKENIZED);
Field fieldPath = new Field("path", "", Field.Store.YES,
Field.Index.NO);
Field fieldContent = new Field("content", "", Field.Store.COMPRESS,
Field.Index.ANALYZED);
Document doc = null;

//遍历文件根目录下所有子目录并创建索引
for (int i = 0; i < files.length - 1; i++) {
if (files[i].isDirectory()) {

//这里需要根据当前文件夹的命名规律和上次建索引后记录的文件夹名字比较
//避免出现对文件重复创建索引

File fileDirs = new File(files[i]
.getAbsolutePath());
File[] file = fileDirs.listFiles();
for (int j = 0; j < file.length; j++) {

String fileName = file[j].getName();

String lastName = "";
if (fileName.lastIndexOf(".") != -1) {
lastName = fileName
.substring(fileName
.lastIndexOf("."));
}
if (lastName.equals(".txt")) {
doc = new Document();
fieldName.setValue(fileName.substring(0, fileName.indexOf(".")));
doc.add(fieldName);
fieldPath.setValue(file[j]
.getAbsolutePath());
doc.add(fieldPath);
fieldContent
.setValue(loadFileToString(file[j]));
doc.add(fieldContent);
writer.addDocument(doc);
}
else if (lastName.equals(".html") ||lastName.equals(".htm")) {
doc = new Document();
String htmlCont = getDocument(file[j]);
fieldName.setValue(file[j].getName());
doc.add(fieldName);
fieldPath.setValue(file[j]
.getAbsolutePath());
doc.add(fieldPath);
fieldContent
.setValue(htmlCont);
doc.add(fieldContent);
writer.addDocument(doc);

}
}
Date endTime1 = new Date();
long timeOfSearch1 = endTime1.getTime()
- beginTime.getTime();
System.out.println("--->" + files[i]
+ "建立索引时间 " + timeOfSearch1 / 1000 / 60
+ " Minute" + "(" + timeOfSearch1
+ " ms )");
}
}

//创建索引完成后记录下本次创建索引的最后一个目录名称
// if (!lastFolder.equals(newLastFolder)) {
// String path = PropertyUtil.class.getResource(
// "/index.properties").toURI().getPath();
// PropertyUtil.updateValue(path, IndexUtil.LAST_FOLDER_PATH,
// LAST_FILE_PATH);
// IndexUtil.indexMap.put(IndexUtil.LAST_FOLDER_PATH,
// LAST_FILE_PATH);
// }
}

} catch (Exception e) {
//清空writer中的索引信息,否则writer在close时会将信息写入索引文件
writer = null;
e.printStackTrace();
} finally {

if (searcher != null) {
try {
searcher.close();
} catch (Exception e) {
e.printStackTrace();
}
}

if (writer != null) {
try {
//优化索引并合并索引文件
writer.optimize();
Date endTime1 = new Date();
long timeOfSearch1 = endTime1.getTime()
- beginTime.getTime();
System.out.println("--->"
+ "合并索引时间 " + timeOfSearch1 / 1000 / 60
+ " Minute" + "(" + timeOfSearch1
+ " ms )");
writer.close();
} catch (Exception e) {
e.printStackTrace();
}
}
if (directory != null) {
try {
directory.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}


}

/**
* 将小索引文件合并到大的索引文件中去
*
* @param from
* 将要合并到to文件的文件
* @param to
* 将from文件合并到该文件
*/
public static void mergeIndex() {
IndexWriter writer = null;
Directory toDirectory = null;
Directory fromDirectory = null;
try {
File from = new File(INDEX_THREAD1);
File to = new File(IndexUtil.getValue(IndexUtil.INDEX_ROOT));
toDirectory = FSDirectory.open(to);
fromDirectory = FSDirectory.open(from);
writer = new IndexWriter(toDirectory, new IKAnalyzer(), false,
IndexWriter.MaxFieldLength.LIMITED);
writer.setMergeFactor(100);
writer.setMaxBufferedDocs(100);
writer.setMaxMergeDocs(Integer.MAX_VALUE);
writer.addIndexes(IndexReader.open(fromDirectory));
writer.optimize();
writer.close();
} catch (Exception e) {
writer = null;
e.printStackTrace();
} finally {
try {
if (writer != null)
writer.close();
} catch (Exception e) {

}
if (toDirectory != null) {
try {
toDirectory.close();
} catch (Exception e) {
e.printStackTrace();
}
}

if (fromDirectory != null) {
try {
fromDirectory.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}

//索引合并完成时间
Date endTime = new Date();
//索引合并所耗时间
long timeOfSearch = endTime.getTime() - beginTime.getTime();
System.out.println("The total time For index creat is " + timeOfSearch
+ " ms");
}


//测试代码
public static void main(String[] args) {
IndexJob processor = new IndexJob();
processor.createIndex("g:\\file4");
}

}



对索引文件进行查询的代码如下:

public class Search {

public void indexSearch(String searchType, String searchKey) {
String INDEX_STORE_PATH = "g:\\index";
Directory directory = null;
IndexSearcher searcher = null;
// 搜索开始时间
Date beginTime = new Date();
try {
directory = FSDirectory.open(new File(INDEX_STORE_PATH));
searcher = new IndexSearcher(directory,true);
searcher.setDefaultFieldSortScoring(true, false);
searcher.setSimilarity(new IKSimilarity());
Query query = IKQueryParser.parse(searchType, searchKey);
System.out.println("查询条件为:" + query);

//索引排序条件
SortField[] sortfield = new SortField[] { SortField.FIELD_SCORE, new SortField(null, SortField.DOC, true) };
Sort sort = new Sort(sortfield);

TopDocs topDocs = searcher.search(query, null, 10, sort);
System.out.println("检索到总数:" + topDocs.totalHits);
ScoreDoc[] scoreDocs = topDocs.scoreDocs;

//设置高亮显示的颜色等样式
SimpleHTMLFormatter simpleHtmlFormatter = new SimpleHTMLFormatter("<FONT COLOR='RED'>", "</FONT>");
Highlighter highlighter = new Highlighter(simpleHtmlFormatter,new QueryScorer(query));
//设置高亮显示的字符串长度
highlighter.setTextFragmenter(new SimpleFragmenter(100));

Analyzer analyzer = new IKAnalyzer();
for (int i = 0; i < scoreDocs.length; i++) {
Document targetDoc = searcher.doc(scoreDocs[i].doc);
TokenStream tokenStream = analyzer.tokenStream("",new StringReader(targetDoc.get("content")));
//读取索引的高亮信息
String str = highlighter.getBestFragment(tokenStream, targetDoc.get("content"));
// System.out.println(targetDoc.get("id"));
// System.out.println(str);
}
} catch (Exception e) {
e.printStackTrace();
} finally {
if (searcher != null) {
try {
searcher.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (directory != null) {
try {
directory.close();
} catch (IOException e) {
e.printStackTrace();
}
}

// 搜索完成时间
Date endTime = new Date();
// 搜索所耗时间
long timeOfSearch = endTime.getTime() - beginTime.getTime();
System.out
.println("The time For indexsearch is " + timeOfSearch + " ms");
}
}

//测试代码
public static void main(String[] args) {
Search search = new Search();
search.indexSearch("content", "小贝");
}
}


[color=red]要注意的地方:[/color]
由于在windows系统中一个文件夹只能存放2W多个文件,在1W多个文件以后写入速度会急剧下降所以建议每个文件夹存放1W个文件。
本例只对2层目录结构中的第二层文件夹的文件创建索引,如需要读取其他位置的文件可自行修改代码,注意文件目录结构越多创建索引的速度就越慢,所以请谨慎决定。
由于创建索引时会影响查询,而且创建索引的时间会很长,所以建议将增量的索引创建到其他目录中,当创建完成后和主索引目录中的进行索引合并。IndexJob.Java中的mergeIndex()就是进行索引合并的方法。

[color=cyan]一些建议:[/color]
在对本地文件创建索引文件的时候,尽量不要使用多线程。因为本地磁盘的IO效率是一定的,多线程并不能提高效率而且会加重服务器CPU使用率。
创建索引的类IndexJob.Java可以配置成定时任务,根据相关数据量调整创建索引的周期。至于本地文件存放路径和创建索引的存放路径等等可以放在properties属性文件中设置,这样利于修改,不用去直接该代码。
还有一点是每次要记录下上次创建索引的文件夹位置,因为Lucene对增量支持不好,未避免重复创建索引出此下策。


我已经所有jar包打成压缩包放在下面了。
  • 0
    点赞
  • 0
    收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:大白 设计师:CSDN官方博客 返回首页
评论
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值