Lucence

  1 package com.itheima.lucene;
  2 
  3 import java.io.File;
  4 import java.util.ArrayList;
  5 import java.util.List;
  6 
  7 import org.apache.lucene.analysis.Analyzer;
  8 import org.apache.lucene.analysis.standard.StandardAnalyzer;
  9 import org.apache.lucene.document.Document;
 10 import org.apache.lucene.document.Field.Store;
 11 import org.apache.lucene.document.TextField;
 12 import org.apache.lucene.index.DirectoryReader;
 13 import org.apache.lucene.index.IndexReader;
 14 import org.apache.lucene.index.IndexWriter;
 15 import org.apache.lucene.index.IndexWriterConfig;
 16 import org.apache.lucene.queryparser.classic.QueryParser;
 17 import org.apache.lucene.search.IndexSearcher;
 18 import org.apache.lucene.search.Query;
 19 import org.apache.lucene.search.ScoreDoc;
 20 import org.apache.lucene.search.TopDocs;
 21 import org.apache.lucene.store.Directory;
 22 import org.apache.lucene.store.FSDirectory;
 23 import org.apache.lucene.util.Version;
 24 import org.junit.Test;
 25 import org.wltea.analyzer.lucene.IKAnalyzer;
 26 
 27 import com.itheima.dao.BookDao;
 28 import com.itheima.dao.impl.BookDaoImpl;
 29 import com.itheima.pojo.Book;
 30 
 31 public class CreateIndexTest {
 32     //分词
 33     @Test
 34     public void testCreateIndex() throws Exception{
 35     //    1. 采集数据
 36         BookDao bookDao = new BookDaoImpl();
 37         List<Book> listBook = bookDao.queryBookList();
 38         
 39     //    2. 创建Document文档对象
 40         List<Document> documents = new ArrayList<>();
 41         for (Book bk : listBook) {
 42             
 43             Document doc = new Document();
 44             doc.add(new TextField("id", String.valueOf(bk.getId()), Store.YES));// Store.YES:表示存储到文档域中
 45             doc.add(new TextField("name", bk.getName(), Store.YES));
 46             doc.add(new TextField("price", String.valueOf(bk.getPrice()), Store.YES));
 47             doc.add(new TextField("pic", bk.getPic(), Store.YES));
 48             doc.add(new TextField("desc", bk.getDesc(), Store.YES));
 49             
 50             // 把Document放到list中
 51             documents.add(doc);
 52         }
 53         
 54     //    3. 创建分析器（分词器）
 55         //Analyzer analyzer = new StandardAnalyzer();
 56         //中文  IK
 57         Analyzer analyzer = new IKAnalyzer();
 58         
 59     //    4. 创建IndexWriterConfig配置信息类
 60         IndexWriterConfig config = new IndexWriterConfig(Version.LUCENE_4_10_3, analyzer);
 61         
 62     //    5. 创建Directory对象，声明索引库存储位置
 63         Directory directory = FSDirectory.open(new File("H:\\temp"));
 64         
 65     //    6. 创建IndexWriter写入对象
 66         IndexWriter writer = new IndexWriter(directory, config);
 67         
 68     //    7. 把Document写入到索引库中
 69         for (Document doc : documents) {
 70             writer.addDocument(doc);
 71         }
 72         
 73     //    8. 释放资源
 74         writer.close();
 75     }
 76 

         //查
 77     @Test
 78     public void serachIndex() throws Exception{
 79         //创建分词器   必须和检索时的分析器一致
 80         Analyzer analyzer = new StandardAnalyzer();
 81         // 创建搜索解析器，第一个参数：默认Field域，第二个参数：分词器
 82         QueryParser queryParser = new QueryParser("desc", analyzer);
 83         
 84         // 1. 创建Query搜索对象
 85         Query query = queryParser.parse("desc:java AND lucene");
 86             
 87         // 2. 创建Directory流对象,声明索引库位置
 88         Directory directory = FSDirectory.open(new File("H:\\temp"));
 89         
 90         // 3. 创建索引读取对象IndexReader
 91         IndexReader indexReader = DirectoryReader.open(directory);
 92         
 93         // 4. 创建索引搜索对象IndexSearcher
 94         IndexSearcher indexSearcher = new IndexSearcher(indexReader);
 95         
 96         // 5. 使用索引搜索对象，执行搜索，返回结果集TopDocs
 97         // 第一个参数：搜索对象，第二个参数：返回的数据条数，指定查询结果最顶部的n条数据返回
 98         TopDocs topDocs  = indexSearcher.search(query, 10);
 99         System.out.println("查询到的数据总条数是：" + topDocs.totalHits);
100         //获得结果集
101         ScoreDoc[] docs = topDocs.scoreDocs;
102         
103         // 6. 解析结果集
104         for (ScoreDoc scoreDoc : docs) {
105             //获得文档
106             int docID = scoreDoc.doc;
107             Document doc = indexSearcher.doc(docID);
108             
109             System.out.println("docID:"+docID);
110             System.out.println("bookid:"+doc.get("id"));
111             System.out.println("pic:"+doc.get("pic"));
112             System.out.println("name:"+doc.get("name"));
113             System.out.println("desc:"+doc.get("desc"));
114             System.out.println("price:"+doc.get("price"));
115         }
116         
117         // 7. 释放资源
118         indexReader.close();
119     }
120 }