Java SearchContext.scrollContext方法代码示例

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
public QuerySearchResultProvider executeQueryPhase(ShardSearchRequest request, SearchTask task) throws IOException {
    final SearchContext context = createAndPutContext(request);
    final SearchOperationListener operationListener = context.indexShard().getSearchOperationListener();
    context.incRef();
    try {
        context.setTask(task);
        operationListener.onPreQueryPhase(context);
        long time = System.nanoTime();
        contextProcessing(context);

        loadOrExecuteQueryPhase(request, context);

        if (context.queryResult().hasHits() == false && context.scrollContext() == null) {
            freeContext(context.id());
        } else {
            contextProcessedSuccessfully(context);
        }
        final long afterQueryTime = System.nanoTime();
        operationListener.onQueryPhase(context, afterQueryTime - time);
        if (request.numberOfShards() == 1) {
            return executeFetchPhase(context, operationListener, afterQueryTime);
        }
        return context.queryResult();
    } catch (Exception e) {
        // execution exception can happen while loading the cache, strip it
        if (e instanceof ExecutionException) {
            e = (e.getCause() == null || e.getCause() instanceof Exception) ?
                (Exception) e.getCause() : new ElasticsearchException(e.getCause());
        }
        operationListener.onFailedQueryPhase(context);
        logger.trace("Query phase failed", e);
        processFailure(context, e);
        throw ExceptionsHelper.convertToRuntime(e);
    } finally {
        cleanContext(context);
    }
}
 

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
private boolean fetchPhaseShouldFreeContext(SearchContext context) {
    if (context.scrollContext() == null) {
        // simple search, no scroll
        return true;
    } else {
        // scroll request, but the scroll was not extended
        return context.scrollContext().scroll == null;
    }
}
 

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
public FetchSearchResult executeFetchPhase(ShardFetchRequest request, SearchTask task) {
    final SearchContext context = findContext(request.id());
    final SearchOperationListener operationListener = context.indexShard().getSearchOperationListener();
    context.incRef();
    try {
        context.setTask(task);
        contextProcessing(context);
        if (request.lastEmittedDoc() != null) {
            context.scrollContext().lastEmittedDoc = request.lastEmittedDoc();
        }
        context.docIdsToLoad(request.docIds(), 0, request.docIdsSize());
        operationListener.onPreFetchPhase(context);
        long time = System.nanoTime();
        fetchPhase.execute(context);
        if (fetchPhaseShouldFreeContext(context)) {
            freeContext(request.id());
        } else {
            contextProcessedSuccessfully(context);
        }
        operationListener.onFetchPhase(context, System.nanoTime() - time);
        return context.fetchResult();
    } catch (Exception e) {
        operationListener.onFailedFetchPhase(context);
        logger.trace("Fetch phase failed", e);
        processFailure(context, e);
        throw ExceptionsHelper.convertToRuntime(e);
    } finally {
        cleanContext(context);
    }
}
 

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
public void freeAllScrollContexts() {
    for (SearchContext searchContext : activeContexts.values()) {
        if (searchContext.scrollContext() != null) {
            freeContext(searchContext.id());
        }
    }
}
 

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
private void processScroll(InternalScrollSearchRequest request, SearchContext context) {
    // process scroll
    context.from(context.from() + context.size());
    context.scrollContext().scroll = request.scroll();
    // update the context keep alive based on the new scroll value
    if (request.scroll() != null && request.scroll().keepAlive() != null) {
        context.keepAlive(request.scroll().keepAlive().millis());
    }
}
 

import org.elasticsearch.search.internal.SearchContext; //导入方法依赖的package包/类
public CollapseContext build(SearchContext context) {
    if (context.scrollContext() != null) {
        throw new SearchContextException(context, "cannot use `collapse` in a scroll context");
    }
    if (context.searchAfter() != null) {
        throw new SearchContextException(context, "cannot use `collapse` in conjunction with `search_after`");
    }
    if (context.rescore() != null && context.rescore().isEmpty() == false) {
        throw new SearchContextException(context, "cannot use `collapse` in conjunction with `rescore`");
    }

    MappedFieldType fieldType = context.getQueryShardContext().fieldMapper(field);
    if (fieldType == null) {
        throw new SearchContextException(context, "no mapping found for `" + field + "` in order to collapse on");
    }
    if (fieldType instanceof KeywordFieldMapper.KeywordFieldType == false &&
        fieldType instanceof NumberFieldMapper.NumberFieldType == false) {
        throw new SearchContextException(context, "unknown type for collapse field `" + field +
            "`, only keywords and numbers are accepted");
    }

    if (fieldType.hasDocValues() == false) {
        throw new SearchContextException(context, "cannot collapse on field `" + field + "` without `doc_values`");
    }
    if (fieldType.indexOptions() == IndexOptions.NONE && innerHit != null) {
        throw new SearchContextException(context, "cannot expand `inner_hits` for collapse field `"
            + field + "`, " + "only indexed field can retrieve `inner_hits`");
    }
    return new CollapseContext(fieldType, innerHit);
}