首页 > 代码库 > lucene自定义过滤器

lucene自定义过滤器

先介绍下查询与过滤的区别和联系,其实查询(各种Query)和过滤(各种Filter)之间非常相似,可以这样说只要用Query能完成的事,用过滤也都可以完成,它们之间可以相互转换,最大的区别就是使用过滤返回的结果集不带评分操作,而使用Query返回的结果都是带相关性评分的,所以当我们如果有一些跟评分操作没有关系的业务,优先使用Filter操作,将会获取更好的性能,其实这也是Solr里面的q参数跟fq参数的区别。 


下面,开始进入正题,在这之前,散仙还是喜欢老生常谈的先来了解一下Lucene里面有关于Filter的整体知识   

技术分享 

下面,我们来看下具体的在代码里怎么实现,先来看下我们的测试数据   

Java代码   技术分享
  1. id      score       bookname    ename       type            price       date  
  2. 1       1       飘渺之旅        pmzl        小说      52.23       201005        
  3. 2       1       三国演义        sgyy        小说      36.13       201207        
  4. 3       1       数据库实战       sjksz       技术      77.13       200811        
  5. 4       1       编程宝典        bcbd        技术      100.3       200501        
  6. 5       1       职场关系论       zcgxl       职场      36.59       200501        
  7. 6       1       健康生活        jksh        生活      20.47       200008        
  8. 7       1       看清本质        kqbz        社会      10.37       201004        
  9. 8       1       编程,编程       bcbc        社会      10.37       201004    


核心代码   

Java代码   技术分享
  1. //使用过滤器   最后一个为true时包含边界部分,为false时不包含边界部分  
  2.            //倒数第二个为true时,包含查询边界,为false时不包含  
  3.            TermRangeFilter filter=new TermRangeFilter("ename", new BytesRef("h"), new BytesRef("n"), true, true);  
  4.            TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);//默认无排序方式  


输出结果   

Java代码   技术分享
  1. 6       1       健康生活        jksh        生活      20.47       200008        
  2. 7       1       看清本质        kqbz        社会      10.37       201004    


核心代码   

Java代码   技术分享
  1. NumericRangeFilter<Double> filter=NumericRangeFilter.newDoubleRange("price", 10D, 40D, true, false);  
  2.   TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);//默认无排序方式  


输出结果   

Java代码   技术分享
  1. 2       1       三国演义        sgyy        小说      36.13       201207        
  2. 5       1       职场关系论       zcgxl       职场      36.59       200501        
  3. 6       1       健康生活        jksh        生活      20.47       200008        
  4. 7       1       看清本质        kqbz        社会      10.37       201004        
  5. 8       1       编程,编程       bcbc        社会      10.37       201004  



核心代码   

Java代码   技术分享
  1. //使用缓存过滤  
  2.     Filter filter=FieldCacheRangeFilter.newDoubleRange("price", 20D, 50D, true, true);  
  3.     TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);//默认无排序方式  


输出结果   

Java代码   技术分享
  1. 2       1       三国演义        sgyy        小说      36.13       201207        
  2. 5       1       职场关系论       zcgxl       职场      36.59       200501        
  3. 6       1       健康生活        jksh        生活      20.47       200008    


核心代码   

Java代码   技术分享
  1. // 缓存域过滤特定的类别  
  2.  Filter filter=new FieldCacheTermsFilter("type", new String[]{"技术","社会"});  
  3.  TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);//默认无排序方式  


输出结果   

Java代码   技术分享
  1. 3       1       数据库实战       sjksz       技术      77.13       200811        
  2. 4       1       编程宝典        bcbd        技术      100.3       200501        
  3. 7       1       看清本质        kqbz        社会      10.37       201004        
  4. 8       1       编程,编程       bcbc        社会      10.37       201004    


核心代码   

Java代码   技术分享
  1. //使用QueryWrapperFilter类包装一个Query  
  2.          QueryWrapperFilter  filter=new QueryWrapperFilter(new TermQuery(new Term("type", "技术")));  
  3.          TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);//默认无排序方式  


输出结果   

Java代码   技术分享
  1. 3       1       数据库实战       sjksz       技术      77.13       200811        
  2. 4       1       编程宝典        bcbd        技术      100.3       200501    



最后我来看下,如何继承Filter基类,来定制我们自己的filter,自定义的Filter,虽然某些时候,功能很强大灵活,但是有几个缺点,我们的了解1,保证是内容不重复的字段,例如主键,如果重复,默认返回第一个作为结果集显示2,保证不能被分词的内容,如果是分词的字段,则可能会出现一些不正确的结果。   
自定义Filter类   

Java代码   技术分享
  1. package com.sanjiesanxian.test;  
  2.   
  3. import java.io.IOException;  
  4. import java.util.BitSet;  
  5.   
  6. import org.apache.lucene.analysis.tokenattributes.CharTermAttribute;  
  7. import org.apache.lucene.index.AtomicReaderContext;  
  8. import org.apache.lucene.index.DocsEnum;  
  9. import org.apache.lucene.index.Term;  
  10. import org.apache.lucene.search.DocIdSet;  
  11. import org.apache.lucene.search.Filter;  
  12. import org.apache.lucene.util.AttributeSource;  
  13. import org.apache.lucene.util.Bits;  
  14. import org.apache.lucene.util.DocIdBitSet;  
  15. import org.apache.lucene.util.FixedBitSet;  
  16. import org.apache.lucene.util.OpenBitSet;  
  17.   
  18. /*** 
  19.  *^_^  ^_^  ^_^ 
  20.  * 自定义过滤器 
  21.  * @author 
  22.  * */  
  23. public class MyCustomFilter extends Filter{  
  24.       
  25.     public MyCustomFilter() {  
  26.         // TODO Auto-generated constructor stub  
  27.     }  
  28.       
  29.     private String[] terms;//限制返回的数据字典  
  30.     public MyCustomFilter(String ...terms) {  
  31.         // TODO Auto-generated constructor stub  
  32.         this.terms=terms;  
  33.     }  
  34.     @Override  
  35.     public DocIdSet getDocIdSet(AtomicReaderContext arg0, Bits arg1)  
  36.             throws IOException {  
  37.         FixedBitSet bits=new FixedBitSet(arg0.reader().maxDoc())  ;//获取没有所有的docid包括未删除的  
  38.          int base=arg0.docBase;//段的相对基数,保证多个段时相对位置正确  
  39.          //int limit=base+arg0.reader().maxDoc();//计算最大限制值  
  40.         for(String s:terms){  
  41.               DocsEnum doc=arg0.reader().termDocsEnum(new Term("id", s));//必须是唯一的不重复  
  42.               //保证是单个不重复的term,如果重复的话,默认会取第一个作为返回结果集,分词后的term也不适用自定义term  
  43.               if(doc.nextDoc()!=-1){   
  44.                 bits.set(doc.docID());//对付符合条件约束的docid循环添加到bits里面  
  45.                 }  
  46.               }  
  47.         return bits;  
  48.     }  
  49.       
  50.       
  51.       
  52.       
  53.   
  54.       
  55.       
  56.       
  57. }  


测试查询代码   

Java代码   技术分享
  1. MyCustomFilter filter=new MyCustomFilter("3","5","2");//随意指定1之多个需要过滤的项  
  2. TopDocs topDocs=searcher.search(new MatchAllDocsQuery(),filter,10000);  


输出结果   

Java代码   技术分享
  1. 2       1       三国演义        sgyy        小说      36.13       201207        
  2. 3       1       数据库实战       sjksz       技术      77.13       200811        
  3. 5       1       职场关系论       zcgxl       职场      36.59       200501    



自定义过滤器虽然有缺点,但是某些场景下却能发挥很灵活的作用,特别是对没有分词的字段进行过滤操作。   


                

lucene自定义过滤器