lucene3搜索引擎,索引建立搜索排序分页高亮显示, IKAnalyzer分词

package com.zjr.service.impl;

 import java.io.File;
 import java.io.IOException;
 import java.io.StringReader;
 import java.lang.reflect.InvocationTargetException;
 import java.util.ArrayList;
 import java.util.List;

 import org.apache.commons.beanutils.BeanUtils;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.apache.lucene.analysis.Analyzer;
 import org.apache.lucene.analysis.TokenStream;
 import org.apache.lucene.document.Document;
 import org.apache.lucene.document.Field;
 import org.apache.lucene.document.Field.Index;
 import org.apache.lucene.document.Field.Store;
 import org.apache.lucene.index.CorruptIndexException;
 import org.apache.lucene.index.IndexReader;
 import org.apache.lucene.index.IndexWriter;
 import org.apache.lucene.index.Term;
 import org.apache.lucene.search.BooleanClause;
 import org.apache.lucene.search.IndexSearcher;
 import org.apache.lucene.search.Query;
 import org.apache.lucene.search.ScoreDoc;
 import org.apache.lucene.search.Sort;
 import org.apache.lucene.search.SortField;
 import org.apache.lucene.search.TopDocs;
 import org.apache.lucene.search.TopScoreDocCollector;
 import org.apache.lucene.search.highlight.Highlighter;
 import org.apache.lucene.search.highlight.InvalidTokenOffsetsException;
 import org.apache.lucene.search.highlight.QueryScorer;
 import org.apache.lucene.search.highlight.SimpleHTMLFormatter;
 import org.apache.lucene.store.Directory;
 import org.apache.lucene.store.FSDirectory;
 import org.wltea.analyzer.lucene.IKAnalyzer;
 import org.wltea.analyzer.lucene.IKQueryParser;
 import org.wltea.analyzer.lucene.IKSimilarity;

 import com.zjr.model.User;

 public class UserIndexService {

 private final Log logger = LogFactory.getLog(UserIndexService.class);
 private final String dirPath = "d:/temp/user";

 Analyzer analyzer = new IKAnalyzer();
 Directory directory = null;
 IndexWriter writer = null;
 IndexSearcher indexSearcher = null;

 private void confirmDirs() {
 File indexFile = new File(dirPath);
 if (!indexFile.exists()) {
 indexFile.mkdirs();
 }
 if (!indexFile.exists() || !indexFile.canWrite()) {
 if (logger.isDebugEnabled())
 logger.error("索引文件目录创建失败或不可写入！");
 }
 }

 public void init() {
 confirmDirs();
 try {
 File f = new File(dirPath);
 directory = FSDirectory.open(f);

 } catch (Exception e) {
 if (logger.isDebugEnabled()) {
 logger.error("解除索引文件锁定失败！" + e.getCause());
 }
 }
 }

 public void createIndex(List<User> userList) {
 init();
 try {

 // 第一个参数是存放索引目录有FSDirectory（存储到磁盘上）和RAMDirectory（存储到内存中），
 // 第二个参数是使用的分词器，第三个：true，建立全新的索引，false,建立增量索引，第四个是建立的索引的最大长度
 writer = new IndexWriter(directory, analyzer, true,IndexWriter.MaxFieldLength.LIMITED);
 writer.setMergeFactor(500);
 writer.setMaxBufferedDocs(155);
 writer.setMaxFieldLength(Integer.MAX_VALUE);
 writeIndex(writer, userList);
 writer.optimize();
 writer.close();
 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }
 }

 public List<User> search(String keyword) {

 File indexFile = new File(dirPath);
 if (!indexFile.exists()) {
 return null;
 }
 Directory dir;
 try {
 dir = FSDirectory.open(indexFile);
 indexSearcher = new IndexSearcher(dir);
 indexSearcher.setSimilarity(new IKSimilarity());
 // 单字段查询,单条件查询
 // Query query = IKQueryParser.parse("userInfo", keyword);

 // 多字段，单条件查询
 String[] fields = new String[] { "userInfo", "parameter1" };
 Query query = IKQueryParser.parseMultiField(fields, keyword);

 // 多字体，单条件，多BooleanClause.Occur[] flags ，查询条件的组合方式（Or/And）
 // BooleanClause.Occur[]数组,它表示多个条件之间的关系,
 // BooleanClause.Occur.MUST表示 and,
 // BooleanClause.Occur.MUST_NOT表示not,
 // BooleanClause.Occur.SHOULD表示or.
 // String[] fields =new String[]{"userInfo","parameter1"};
 // BooleanClause.Occur[] flags=new
 // BooleanClause.Occur[]{BooleanClause.Occur.MUST,BooleanClause.Occur.SHOULD};
 // Query query = IKQueryParser.parseMultiField(fields,
 // keyword,flags);

 // //多Field，多条件查询分析
 // String[] fields =new String[]{"userInfo","parameter1"};
 // String[] queries = new String[]{keyword,keyword};
 // Query query = IKQueryParser.parseMultiField(fields,queries);

 // 多Field,多条件,多Occur 查询
 // String[] fields =new String[]{"userInfo","parameter1"};
 // String[] queries = new String[]{keyword,keyword};
 // BooleanClause.Occur[] flags=new
 // BooleanClause.Occur[]{BooleanClause.Occur.MUST,BooleanClause.Occur.SHOULD};
 // Query query =
 // IKQueryParser.parseMultiField(fields,queries,flags);

 // 搜索相似度最高的20条记录
 TopDocs topDocs = indexSearcher.search(query, 20);
 ScoreDoc[] hits = topDocs.scoreDocs;
 return hitsToQuery(hits, query);

 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }

 return null;
 }

 private List<User> hitsToQuery(ScoreDoc[] hits, Query query) {
 List<User> list = new ArrayList<User>();
 try {
 for (int i = 0; i < hits.length; i++) {
 User u = new User();
 Document doc = indexSearcher.doc(hits[i].doc);
 u.setUserId(Integer.parseInt(doc.get("userId")));
 u.setUserName(doc.get("userName"));
 u.setUserAge(Integer.parseInt(doc.get("userAge")));
 // 高亮设置
 SimpleHTMLFormatter simpleHtmlFormatter = new SimpleHTMLFormatter(
 "", "");
 Highlighter highlighter = new Highlighter(simpleHtmlFormatter,
 new QueryScorer(query));
 TokenStream tokenStream = analyzer.tokenStream("text",
 new StringReader(doc.get("userInfo")));
 String userInfo = highlighter.getBestFragment(tokenStream, doc
 .get("userInfo"));
 if (userInfo != null) {
 u.setUserInfo(userInfo);
 } else {
 u.setUserInfo(doc.get("userInfo"));
 }

 SimpleHTMLFormatter simpleHtmlFormatter1 = new SimpleHTMLFormatter(
 "", "");
 Highlighter highlighter1 = new Highlighter(
 simpleHtmlFormatter1, new QueryScorer(query));
 TokenStream tokenStream1 = analyzer.tokenStream("text1",
 new StringReader(doc.get("parameter1")));
 String p1 = highlighter1.getBestFragment(tokenStream1, doc
 .get("parameter1"));
 if (p1 != null) {
 u.setParameter1(p1);
 } else {
 u.setParameter1(doc.get("parameter1"));
 }

 u.setParameter2(doc.get("parameter2"));
 u.setParameter3(doc.get("parameter3"));
 u.setParameter4(doc.get("parameter4"));
 list.add(u);
 }

 indexSearcher.close();
 return list;
 } catch (CorruptIndexException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 } catch (InvalidTokenOffsetsException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }
 return null;
 }

 public void writeIndex(IndexWriter writer, List<User> userList) {

 try {
 for (User u : userList) {
 Document doc = getDoc(u);
 writer.addDocument(doc);
 }
 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }

 }

 private Document getDoc(User user) {
 System.out.println("用户ID 为" + user.getUserId() + " 索引被创建");
 Document doc = new Document();
 addField2Doc(doc, user, "userId", Store.YES, Index.NOT_ANALYZED);
 addField2Doc(doc, user, "userName", Store.YES, Index.NOT_ANALYZED);// Index.NOT_ANALYZED
 // 不分词，但建立索引
 addField2Doc(doc, user, "userAge", Store.YES, Index.NOT_ANALYZED);// Index.ANALYZED
 // 分词并且建立索引
 addField2Doc(doc, user, "userInfo", Store.YES, Index.ANALYZED);
 addField2Doc(doc, user, "parameter1", Store.YES, Index.ANALYZED);
 addField2Doc(doc, user, "parameter2", Store.YES, Index.ANALYZED);
 addField2Doc(doc, user, "parameter3", Store.YES, Index.ANALYZED);
 addField2Doc(doc, user, "parameter4", Store.YES, Index.ANALYZED);
 return doc;
 }

 private void addField2Doc(Document doc, Object bean, String name, Store s,
 Index i) {
 String value;
 try {
 value = BeanUtils.getProperty(bean, name);
 if (value != null) {
 doc.add(new Field(name, value, s, i,
 Field.TermVector.WITH_POSITIONS_OFFSETS));
 }
 } catch (IllegalAccessException e) {
 logger.error("get bean property error", e);
 } catch (InvocationTargetException e) {
 logger.error("get bean property error", e);
 } catch (NoSuchMethodException e) {
 logger.error("get bean property error", e);
 }
 }

 /**
 * 没有排序，有高亮，有分页
 *
 * @param pageNo
 * @param pageSize
 * @param keyword
 * @return
 */
 public PageBean getPageQuery(int pageNo, int pageSize, String keyword) {
 List result = new ArrayList();
 File indexFile = new File(dirPath);
 if (!indexFile.exists()) {
 return null;
 }
 Directory dir;
 try {
 dir = FSDirectory.open(indexFile);
 indexSearcher = new IndexSearcher(dir);
 indexSearcher.setSimilarity(new IKSimilarity());

 String[] fields = new String[] { "userInfo", "parameter1" };
 BooleanClause.Occur[] flags = new BooleanClause.Occur[] {
 BooleanClause.Occur.MUST, BooleanClause.Occur.SHOULD };
 Query query = IKQueryParser.parseMultiField(fields, keyword, flags);

 TopScoreDocCollector topCollector = TopScoreDocCollector.create(
 indexSearcher.maxDoc(), true);
 indexSearcher.search(query, topCollector);
 // 查询当页的记录
 ScoreDoc[] docs = topCollector.topDocs((pageNo - 1) * pageSize,
 pageSize).scoreDocs;

 // String[] highlightCol = {"userInfo", "parameter1"};
 // 高亮设置
 SimpleHTMLFormatter simpleHtmlFormatter = new SimpleHTMLFormatter(
 "", "");
 Highlighter highlighter = new Highlighter(simpleHtmlFormatter,
 new QueryScorer(query));

 for (ScoreDoc scdoc : docs) {
 User u = new User();
 Document doc = indexSearcher.doc(scdoc.doc);
 //
 // for (Fieldable fa : doc.getFields()) {
 // System.out.println(fa.name());
 // String value = doc.get(fa.name());
 // for (String col : highlightCol) {
 // if(fa.name().equals(col)) {
 // //设置高显内容
 // TokenStream tokenStream = analyzer.tokenStream("text",new
 // StringReader(value));
 // value = highlighter.getBestFragment(tokenStream, value);
 // }
 // }
 //
 // }

 u.setUserId(Integer.parseInt(doc.get("userId")));
 u.setUserName(doc.get("userName"));
 u.setUserAge(Integer.parseInt(doc.get("userAge")));

 TokenStream tokenStream = analyzer.tokenStream("text",
 new StringReader(doc.get("userInfo")));
 String userInfo = highlighter.getBestFragment(tokenStream, doc
 .get("userInfo"));
 if (userInfo != null) {
 u.setUserInfo(userInfo);
 } else {
 u.setUserInfo(doc.get("userInfo"));
 }

 TokenStream tokenStream1 = analyzer.tokenStream("text1",
 new StringReader(doc.get("parameter1")));
 String p1 = highlighter.getBestFragment(tokenStream1, doc
 .get("parameter1"));
 if (p1 != null) {
 u.setParameter1(p1);
 } else {
 u.setParameter1(doc.get("parameter1"));
 }

 u.setParameter2(doc.get("parameter2"));
 u.setParameter3(doc.get("parameter3"));
 u.setParameter4(doc.get("parameter4"));
 result.add(u);

 }
 PageBean pb = new PageBean();
 pb.setCurrentPage(pageNo);// 当前页
 pb.setPageSize(pageSize);
 pb.setAllRow(topCollector.getTotalHits());// hit中的记录数目
 pb.setList(result);
 return pb;

 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 } catch (InvalidTokenOffsetsException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }

 return null;
 }

 /**
 * 排序，有高亮，有分页
 *
 * @param pageNo
 * @param pageSize
 * @param keyword
 * @return
 */
 public PageBean getPageQuery2(int pageNo, int pageSize, String keyword) {
 List result = new ArrayList();
 File indexFile = new File(dirPath);
 if (!indexFile.exists()) {
 return null;
 }
 Directory dir;
 try {
 dir = FSDirectory.open(indexFile);
 indexSearcher = new IndexSearcher(dir);
 indexSearcher.setSimilarity(new IKSimilarity());

 String[] fields = new String[] { "userInfo", "parameter1" };
 BooleanClause.Occur[] flags = new BooleanClause.Occur[] {
 BooleanClause.Occur.MUST, BooleanClause.Occur.SHOULD };
 Query query = IKQueryParser.parseMultiField(fields, keyword, flags);

 // 多字段排序，设置在前面的会优先排序
 SortField[] sortFields = new SortField[2];
 SortField sortField = new SortField("userId", SortField.INT, false);//false升序，true降序
 SortField FIELD_SEX = new SortField("userAge", SortField.INT, true);
 sortFields[0] = sortField;
 sortFields[1] = FIELD_SEX;
 Sort sort = new Sort(sortFields);

 TopDocs topDocs = indexSearcher.search(query, null, 50, sort);

 if (topDocs.totalHits != 0) {
 // for(ScoreDoc sd : topDocs.scoreDocs) {
 //
 // }
 // 高亮设置
 SimpleHTMLFormatter simpleHtmlFormatter = new SimpleHTMLFormatter("", "");
 Highlighter highlighter = new Highlighter(simpleHtmlFormatter,new QueryScorer(query));

 for (int i = (pageNo - 1) * pageSize; i < pageSize * pageNo; i++) {
 ScoreDoc scdoc = topDocs.scoreDocs[i];
 User u = new User();
 Document doc = indexSearcher.doc(scdoc.doc);
 u.setUserId(Integer.parseInt(doc.get("userId")));
 u.setUserName(doc.get("userName"));
 u.setUserAge(Integer.parseInt(doc.get("userAge")));
 TokenStream tokenStream = analyzer.tokenStream("text",new StringReader(doc.get("userInfo")));
 String userInfo = highlighter.getBestFragment(tokenStream,doc.get("userInfo"));
 if (userInfo != null) {
 u.setUserInfo(userInfo);
 } else {
 u.setUserInfo(doc.get("userInfo"));
 }

 TokenStream tokenStream1 = analyzer.tokenStream("text1",new StringReader(doc.get("parameter1")));
 String p1 = highlighter.getBestFragment(tokenStream1, doc.get("parameter1"));
 if (p1 != null) {
 u.setParameter1(p1);
 } else {
 u.setParameter1(doc.get("parameter1"));
 }

 u.setParameter2(doc.get("parameter2"));
 u.setParameter3(doc.get("parameter3"));
 u.setParameter4(doc.get("parameter4"));
 result.add(u);

 }
 PageBean pb = new PageBean();
 pb.setCurrentPage(pageNo);// 当前页
 pb.setPageSize(pageSize);
 pb.setAllRow(topDocs.totalHits);// hit中的记录数目
 pb.setList(result);
 return pb;

 }
 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 } catch (InvalidTokenOffsetsException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }

 return null;
 }

 /**
 * 删除索引
 * @param userId
 */
 public void deleIndex(String userId){

 try {
 File f = new File(dirPath);
 directory = FSDirectory.open(f);
 IndexReader reader = IndexReader.open(directory,false);
 Term term = new Term("userId", userId);
 reader.deleteDocuments(term);
 reader.close();
 } catch (IOException e) {
 // TODO Auto-generated catch block
 e.printStackTrace();
 }


 }

 }

lucene3搜索引擎,索引建立搜索排序分页高亮显示, IKAnalyzer分词

猜你喜欢