【Lucene】Apache Lucene全文检索引擎架构之入门实战

Lucene是一套用于全文检索和搜寻的开源程式库,由Apache软件基金会支持和提供。Lucene提供了一个简单却强大的应用程式接口,能够做全文索引和搜寻。在Java开发环境里Lucene是一个成熟的免费开源工具。就其本身而言,Lucene是当前以及最近几年最受欢迎的免费Java信息检索程序库。——《百度百科》

  这篇博文主要从两个方面出发,首先介绍一下Lucene中的全文搜索原理,其次通过程序示例来展现如何使用Lucene。关于全文搜索原理部分我上网搜索了一下,也看了好几篇文章,最后在写这篇文章的时候部分参考了其中两篇(地址我放在文章的末尾),感谢原文作者。

1. 全文检索

  何为全文检索?举个例子,比如现在要在一个文件中查找某个字符串,最直接的想法就是从头开始检索,查到了就OK,这种对于小数据量的文件来说,很实用,但是对于大数据量的文件来说,就有点呵呵了。或者说找包含某个字符串的文件,也是这样,如果在一个拥有几十个G的硬盘中找那效率可想而知,是很低的。
  文件中的数据是属于非结构化数据,也就是说它是没有什么结构可言的,要解决上面提到的效率问题,首先我们得即将非结构化数据中的一部分信息提取出来,重新组织,使其变得有一定结构,然后对此有一定结构的数据进行搜索,从而达到搜索相对较快的目的。这就叫全文搜索。即先建立索引,再对索引进行搜索的过程。
  那么lucene中是如何建立索引的呢?假设现在有两个文档,内容如下:

文章1的内容为:Tom lives in Guangzhou, I live in Guangzhou too.
文章2的内容为:He once lived in Shanghai.

  首先第一步是将文档传给分词组件(Tokenizer),分词组件会将文档分成一个个单词,并去除标点符号和停词。所谓的停词指的是没有特别意义的词,比如英文中的a,the,too等。经过分词后,得到词元(Token) 。如下:

文章1经过分词后的结果:[Tom] [lives] [Guangzhou] [I] [live] [Guangzhou]
文章2经过分词后的结果:[He] [lives] [Shanghai]

  然后将词元传给语言处理组件(Linguistic Processor),对于英语,语言处理组件一般会将字母变为小写,将单词缩减为词根形式,如”lives”到”live”等,将单词转变为词根形式,如”drove”到”drive”等。然后得到词(Term)。如下:

文章1经过处理后的结果:[tom] [live] [guangzhou] [i] [live] [guangzhou]
文章2经过处理后的结果:[he] [live] [shanghai]

  最后将得到的词传给索引组件(Indexer),索引组件经过处理,得到下面的索引结构:

关键词 文章号[出现频率] 出现位置
guangzhou 1[2] 3,6
he 2[1] 1
i 1[1] 4
live 1[2],2[1] 2,5,2
shanghai 2[1] 3
tom 1[1] 1

  以上就是lucene索引结构中最核心的部分。它的关键字是按字符顺序排列的,因此lucene可以用二元搜索算法快速定位关键词。实现时lucene将上面三列分别作为词典文件(Term Dictionary)、频率文件(frequencies)和位置文件(positions)保存。其中词典文件不仅保存有每个关键词,还保留了指向频率文件和位置文件的指针,通过指针可以找到该关键字的频率信息和位置信息。
  搜索的过程是先对词典二元查找、找到该词,通过指向频率文件的指针读出所有文章号,然后返回结果,然后就可以在具体的文章中根据出现位置找到该词了。所以lucene在第一次建立索引的时候可能会比较慢,但是以后就不需要每次都建立索引了,就快了。当然了,这是针对英文的检索,针对中文的规则会有不同,后面我再看看相关资料。

2. 示例代码

  根据上文的分析,全文检索有两个步骤,先建立索引,再检索。所以为了测试这个过程,我写了两个java类,一个是测试建立索引的,另一个是测试检索的。首先建立个maven工程,pom.xml如下:

 
  1. <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

  2. <modelVersion>4.0.0</modelVersion>

  3. <groupId>demo.lucene</groupId>

  4. <artifactId>Lucene01</artifactId>

  5. <version>0.0.1-SNAPSHOT</version>

  6. <build/>

  7.  
  8. <dependencies>

  9. <!-- lucene核心包 -->

  10. <dependency>

  11. <groupId>org.apache.lucene</groupId>

  12. <artifactId>lucene-core</artifactId>

  13. <version>5.3.1</version>

  14. </dependency>

  15. <!-- lucene查询解析包 -->

  16. <dependency>

  17. <groupId>org.apache.lucene</groupId>

  18. <artifactId>lucene-queryparser</artifactId>

  19. <version>5.3.1</version>

  20. </dependency>

  21. <!-- lucene解析器包 -->

  22. <dependency>

  23. <groupId>org.apache.lucene</groupId>

  24. <artifactId>lucene-analyzers-common</artifactId>

  25. <version>5.3.1</version>

  26. </dependency>

  27. </dependencies>

  28. </project>

  在写程序之前,首先得去弄一些文件,我随便找了一些英文的文档(中文的后面再研究),放到了D:\lucene\data\目录中,如下:
文档
文档里面都是密密麻麻的英文,我就不截图了。
接下来开始写建立索引的java程序:

 
  1. /**

  2. * 建立索引的类

  3. * @author Ni Shengwu

  4. *

  5. */

  6. public class Indexer {

  7.  
  8. private IndexWriter writer; //写索引实例

  9.  
  10. //构造方法,实例化IndexWriter

  11. public Indexer(String indexDir) throws Exception {

  12. Directory dir = FSDirectory.open(Paths.get(indexDir));

  13. Analyzer analyzer = new StandardAnalyzer(); //标准分词器,会自动去掉空格啊,is a the等单词

  14. IndexWriterConfig config = new IndexWriterConfig(analyzer); //将标准分词器配到写索引的配置中

  15. writer = new IndexWriter(dir, config); //实例化写索引对象

  16. }

  17.  
  18. //关闭写索引

  19. public void close() throws Exception {

  20. writer.close();

  21. }

  22.  
  23. //索引指定目录下的所有文件

  24. public int indexAll(String dataDir) throws Exception {

  25. File[] files = new File(dataDir).listFiles(); //获取该路径下的所有文件

  26. for(File file : files) {

  27. indexFile(file); //调用下面的indexFile方法,对每个文件进行索引

  28. }

  29. return writer.numDocs(); //返回索引的文件数

  30. }

  31.  
  32. //索引指定的文件

  33. private void indexFile(File file) throws Exception {

  34. System.out.println("索引文件的路径:" + file.getCanonicalPath());

  35. Document doc = getDocument(file); //获取该文件的document

  36. writer.addDocument(doc); //调用下面的getDocument方法,将doc添加到索引中

  37. }

  38.  
  39. //获取文档,文档里再设置每个字段,就类似于数据库中的一行记录

  40. private Document getDocument(File file) throws Exception{

  41. Document doc = new Document();

  42. //添加字段

  43. doc.add(new TextField("contents", new FileReader(file))); //添加内容

  44. doc.add(new TextField("fileName", file.getName(), Field.Store.YES)); //添加文件名,并把这个字段存到索引文件里

  45. doc.add(new TextField("fullPath", file.getCanonicalPath(), Field.Store.YES)); //添加文件路径

  46. return doc;

  47. }

  48.  
  49. public static void main(String[] args) {

  50. String indexDir = "D:\\lucene"; //将索引保存到的路径

  51. String dataDir = "D:\\lucene\\data"; //需要索引的文件数据存放的目录

  52. Indexer indexer = null;

  53. int indexedNum = 0;

  54. long startTime = System.currentTimeMillis(); //记录索引开始时间

  55. try {

  56. indexer = new Indexer(indexDir);

  57. indexedNum = indexer.indexAll(dataDir);

  58. } catch (Exception e) {

  59. e.printStackTrace();

  60. } finally {

  61. try {

  62. indexer.close();

  63. } catch (Exception e) {

  64. e.printStackTrace();

  65. }

  66. }

  67. long endTime = System.currentTimeMillis(); //记录索引结束时间

  68. System.out.println("索引耗时" + (endTime-startTime) + "毫秒");

  69. System.out.println("共索引了" + indexedNum + "个文件");

  70. }

  71. }

  我是按照建立索引的过程来写的程序,在注释中已经解释的很清楚了,这里就不再赘述了。然后运行一下main方法看一下结果,如下:
索引结果
  共索引了7个文件,耗时649毫秒,还是蛮快的,而且索引文件的路径也是对的,然后可以看一下D:\lucene\会生成一些文件,这些就是生成的索引。
索引
  现在有了索引了,我们可以检索想要查询的字符了,我随便打开了一个文件,在里面找了个比较丑的字符串“generate-maven-artifacts”来作为检索的对象。在检索之前先看一下检索的java代码:

 
  1. public class Searcher {

  2.  
  3. public static void search(String indexDir, String q) throws Exception {

  4.  
  5. Directory dir = FSDirectory.open(Paths.get(indexDir)); //获取要查询的路径,也就是索引所在的位置

  6. IndexReader reader = DirectoryReader.open(dir);

  7. IndexSearcher searcher = new IndexSearcher(reader);

  8. Analyzer analyzer = new StandardAnalyzer(); //标准分词器,会自动去掉空格啊,is a the等单词

  9. QueryParser parser = new QueryParser("contents", analyzer); //查询解析器

  10. Query query = parser.parse(q); //通过解析要查询的String,获取查询对象

  11.  
  12. long startTime = System.currentTimeMillis(); //记录索引开始时间

  13. TopDocs docs = searcher.search(query, 10);//开始查询,查询前10条数据,将记录保存在docs中

  14. long endTime = System.currentTimeMillis(); //记录索引结束时间

  15. System.out.println("匹配" + q + "共耗时" + (endTime-startTime) + "毫秒");

  16. System.out.println("查询到" + docs.totalHits + "条记录");

  17.  
  18. for(ScoreDoc scoreDoc : docs.scoreDocs) { //取出每条查询结果

  19. Document doc = searcher.doc(scoreDoc.doc); //scoreDoc.doc相当于docID,根据这个docID来获取文档

  20. System.out.println(doc.get("fullPath")); //fullPath是刚刚建立索引的时候我们定义的一个字段

  21. }

  22. reader.close();

  23. }

  24.  
  25. public static void main(String[] args) {

  26. String indexDir = "D:\\lucene";

  27. String q = "generate-maven-artifacts"; //查询这个字符串

  28. try {

  29. search(indexDir, q);

  30. } catch (Exception e) {

  31. e.printStackTrace();

  32. }

  33. }

  34. }

运行一下main方法,看一下结果:
检索
  Lucene已经正确的帮我们检索到了,然后我把中间的“-”去掉,它也能帮我们检索到,但是我把前面的字符都去掉,只留下“rtifacts”就检索不到了,这也能说明Lucene中建立索引是以单词来划分的,但是这个问题是可以解决的,我会在后续的文章中写到。

部分参考自:
http://blog.csdn.net/forfuture1978/article/details/4711308
http://www.cnblogs.com/dewin/archive/2009/11/24/1609905.html

猜你喜欢

转载自blog.csdn.net/keepfriend/article/details/113592047