Example usage for org.apache.lucene.search QueryRescorer.QueryRescoreContext query

List of usage examples for org.apache.lucene.search QueryRescorer.QueryRescoreContext query

Introduction

In this page you can find the example usage for org.apache.lucene.search QueryRescorer.QueryRescoreContext query.

Prototype

Query query

To view the source code for org.apache.lucene.search QueryRescorer.QueryRescoreContext query.

Click Source Link

Usage

From source file:com.o19s.es.ltr.logging.LoggingFetchSubPhase.java

License:Apache License

private Tuple<RankerQuery, HitLogConsumer> extractRescore(LoggingSearchExtBuilder.LogSpec logSpec,
        List<RescoreContext> contexts) {
    if (logSpec.getRescoreIndex() >= contexts.size()) {
        throw new IllegalArgumentException("rescore index [" + logSpec.getRescoreIndex()
                + "] is out of bounds, only " + "[" + contexts.size() + "] rescore context(s) are available");
    }/* www  .  j av a  2 s  .  co  m*/
    RescoreContext context = contexts.get(logSpec.getRescoreIndex());
    if (!(context instanceof QueryRescorer.QueryRescoreContext)) {
        throw new IllegalArgumentException("Expected a [QueryRescoreContext] but found a " + "["
                + context.getClass().getSimpleName() + "] " + "at index [" + logSpec.getRescoreIndex() + "]");
    }
    QueryRescorer.QueryRescoreContext qrescore = (QueryRescorer.QueryRescoreContext) context;
    return toLogger(logSpec,
            inspectQuery(qrescore.query()).orElseThrow(() -> new IllegalArgumentException(
                    "Expected a [sltr] query but found a " + "[" + qrescore.query().getClass().getSimpleName()
                            + "] " + "at index [" + logSpec.getRescoreIndex() + "]")));
}