10、自定义QueryParser(lucene笔记)

一、自定义QueryParser

  • 1、有些时候我们需要自定义一些QueryParser,一般原因为两个:
    • (1)对于某些QueryParser在查询时会使得性能降低,所以考虑将这些查询取消,比如FuzzyQueryWildcardQuery
    • (2)在具体的查询时,很有可能有这样一种需求:需要获取的是一个数字的范围查询。所以必须扩展原有的QueryParser才能进行查询。

说明:就是有些默认的QueryParser查询性能很低,比如上面说到的FuzzyQueryWildcardQuery,所以我们需要将这样的查询禁用。

  • 2、步骤(工程lucene_analyzer05
    自定义QueryParser
    CustomParser.java
package cn.itcast.util;
import java.text.SimpleDateFormat;
import java.util.regex.Pattern;
import jdk.nashorn.internal.runtime.ParserException;
import org.apache.lucene.search.NumericRangeQuery;
import org.apache.lucene.search.Query;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.util.Version;

public class CustomParser extends QueryParser {

    public CustomParser(Version matchVersion, String f, Analyzer a) {
        super(matchVersion, f, a);
    }

    @Override
    protected Query getWildcardQuery(String field, String termStr)
            throws ParseException {
        throw new ParserException("由于性能原因,这种通配符查询已经被禁用");
    }

    @Override
    protected Query getFuzzyQuery(String field, String termStr,
            float minSimilarity) throws ParseException {
        throw new ParserException("由于性能原因,这种模糊查询已经被禁用");
    }
}

说明:这里当进行模糊查询和通配符查询时我们让其抛出异常。

CustomParserUtil.java

package cn.itcast.util;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.util.Version;

public class CustomParserUtil {
    public void searchByQuery(String value){
        try {
            IndexSearcher searcher = new IndexSearcher(IndexReader.open(FileIndexUtil.getDirectory()));
            TopDocs tds = null;
            
            CustomParser parser = new CustomParser(Version.LUCENE_35, "content", new StandardAnalyzer(Version.LUCENE_35));
            Query query = parser.parse(value);
            
            tds = searcher.search(query, 50);
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            for (ScoreDoc sd : tds.scoreDocs) {
                Document doc = searcher.doc(sd.doc);
                System.out.println("id:" + sd.doc + ",评分:" + sd.score + ",名称:"
                        + doc.get("filename") + ",路径:" + doc.get("path")
                        + ",文件大小:" + doc.get("size") + ",日期:"
                        + sdf.format(new Date(Long.valueOf(doc.get("date")))));
            }
            searcher.close();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException e) {
            System.err.println(e.getMessage());
        }
    }
}

说明:这是一个工具类,接收一个查询的字符串,然后我们在其中进行查询。

测试:
TestCustomParser.java

@Test
public void test01(){
    CustomParserUtil cp = new CustomParserUtil();
    //cp.searchByQuery("ja?va");
    cp.searchByQuery("java~");
}

说明:这里的两个测试都会抛出异常。

二、扩展基于数字和日期的查询

同样我们也是在自定义QueryParser中进行定义:

@Override
protected Query getRangeQuery(String field,
        String part1, String part2, boolean inclusive) throws ParseException {
    if(field.equals("size")){
        return NumericRangeQuery.newIntRange(field, Integer.parseInt(part1), Integer.parseInt(part2), inclusive, inclusive);
    }else if(field.equals("date")){
        String dateType = "yyyy-MM-dd";
        Pattern pattern = Pattern.compile("\\d{4}-\\d{2}-\\d{2}");
        if(pattern.matcher(part1).matches() && pattern.matcher(part2).matches()){
            SimpleDateFormat sdf = new SimpleDateFormat(dateType);
            try {
                long start = sdf.parse(part1).getTime();
                long end = sdf.parse(part2).getTime();
                return NumericRangeQuery.newLongRange(field, start, end, inclusive, inclusive);
            } catch (java.text.ParseException e) {
                e.printStackTrace();
            }
        }else{
            throw new ParseException("要检索的日期格式不正确,请使用" + dateType + "这种格式");
        }
    }
    return super.newRangeQuery(field, part1, part2, inclusive);
}

说明:这里我们判断传递进来的参数是不是符合我们要求的格式,如果不是则抛出异常,否则让其进行查询。

测试:

@Test
public void test02(){
    CustomParserUtil cp = new CustomParserUtil();
    //cp.searchByQuery("size:[30 TO 800]");
    //cp.searchByQuery("content:[a TO j]");
    cp.searchByQuery("date:[2006-12-04 TO 2016-12-04]");
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,923评论 18 139
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 173,288评论 25 708
  • 一、爱的语言,爱的箱子 爱的箱子,说的就是我们每个人的基本需求,当我们的需求没有被满足的时候,我们就会很难静下心来...
    张俊豪阅读 840评论 1 1
  • 现在是凌晨4:17 刚刚他说他想要拉屎 于是匆忙挂断视频 离上一次推送真的过去了好久好久 从在泰国那次我就一直想写...
    贰八还行阅读 312评论 0 0