Lucene自定义评分查询

  1 package com.lucene.score;
  2 
  3 import java.io.IOException;
  4 import java.sql.Date;
  5 import java.text.SimpleDateFormat;
  6 
  7 import org.apache.lucene.document.Document;
  8 import org.apache.lucene.index.CorruptIndexException;
  9 import org.apache.lucene.index.IndexReader;
 10 import org.apache.lucene.index.Term;
 11 import org.apache.lucene.search.FieldCache;
 12 import org.apache.lucene.search.IndexSearcher;
 13 import org.apache.lucene.search.Query;
 14 import org.apache.lucene.search.ScoreDoc;
 15 import org.apache.lucene.search.TermQuery;
 16 import org.apache.lucene.search.TopDocs;
 17 import org.apache.lucene.search.function.CustomScoreProvider;
 18 import org.apache.lucene.search.function.CustomScoreQuery;
 19 import org.apache.lucene.search.function.FieldScoreQuery;
 20 import org.apache.lucene.search.function.FieldScoreQuery.Type;
 21 import org.apache.lucene.search.function.ValueSourceQuery;
 22 
 23 import com.lucene.search.IndexUtils;
 24 
 25 public class MyCustomScore {
 26     public void searchByScoreQuery() {
 27         try {
 28             IndexSearcher searcher = new IndexSearcher(
 29                     IndexReader.open(IndexUtils.getDirectory()));
 30             Query q = new TermQuery(new Term("content", "1"));
 31             // 1.创建评分域
 32             FieldScoreQuery fieldScoreQuery = new FieldScoreQuery("size",
 33                     Type.INT);
 34             // 2.根据评分域和原有的query创建自定义的query对象
 35             // 2.1 创建MyCustomScoreQuery继承CustomScoreQuery
 36             // 2.2重写getCustomScoreProvider方法
 37             // 2.3创建MyCustomScoreProvider继承CustomScoreProvider
 38             // 2.4重写customScore方法,自定义评分算法
 39             MyCustomScoreQuery query = new MyCustomScoreQuery(q,
 40                     fieldScoreQuery);
 41             TopDocs docs=searcher.search(query, 30);
 42             // 输出信息
 43             ScoreDoc[] sds = docs.scoreDocs;
 44             Document d = null;
 45             SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
 46             for (ScoreDoc s : sds) {
 47                 d = searcher.doc(s.doc);
 48                 System.out.println(s.doc + "->" + s.score + "->"
 49                         + d.get("filename") + "->" + d.get("size") + "->"
 50                         + sdf.format(new Date(Long.valueOf(d.get("date")))));
 51             }
 52 
 53         } catch (CorruptIndexException e) {
 54             e.printStackTrace();
 55         } catch (IOException e) {
 56             e.printStackTrace();
 57         }
 58     }
 59 }
 60 
 61 @SuppressWarnings("serial")
 62 class MyCustomScoreQuery extends CustomScoreQuery {
 63 
 64     public MyCustomScoreQuery(Query subQuery, ValueSourceQuery valSrcQuery) {
 65         super(subQuery, valSrcQuery);
 66     }
 67 
 68     @Override
 69     protected CustomScoreProvider getCustomScoreProvider(IndexReader reader)
 70             throws IOException {
 71         // 默认情况评分是根据原有的评分*传入进来的评分
 72         //return super.getCustomScoreProvider(reader);
 73         
 74         return new MyCustomScoreProvider(reader);
 75     }
 76 }
 77 
 78 class MyCustomScoreProvider extends CustomScoreProvider {
 79 
 80     String[] filenames = null;
 81 
 82     public MyCustomScoreProvider(IndexReader reader) {
 83         super(reader);
 84         try {
 85             filenames = FieldCache.DEFAULT.getStrings(reader, "filename");
 86         } catch (IOException e) {
 87             e.printStackTrace();
 88         }
 89     }
 90 
 91     @Override
 92     public float customScore(int doc, float subQueryScore, float valSrcScore)
 93             throws IOException {
 94         // 根据doc获取文件名
 95         String filename = filenames[doc];
 96         //判断文件后缀是否为txt结尾,是把评分*10  否则/10
 97         if (filename.endsWith(".txt")) {
 98             return subQueryScore * 10;
 99         } else {
100             return subQueryScore / 10;
101         }
102     }
103 }

    @Test
    public void test01(){
        MyCustomScore score=new MyCustomScore();
        score.searchByScoreQuery();
    }