Package org.elasticsearch.search

Class MockSearchService

java.lang.Object

org.elasticsearch.common.component.AbstractLifecycleComponent

org.elasticsearch.search.SearchService

org.elasticsearch.search.MockSearchService

All Implemented Interfaces:

Closeable, AutoCloseable, LifecycleComponent, Releasable, IndexEventListener

public class MockSearchService
extends SearchService

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MockSearchService.TestPlugin	Marker plugin used by MockNode to enable MockSearchService.

Nested classes/interfaces inherited from class org.elasticsearch.search.SearchService

SearchService.ResultsType

Field Summary

Fields inherited from class org.elasticsearch.search.SearchService

ALLOW_EXPENSIVE_QUERIES, BATCHED_QUERY_PHASE, CCS_COLLECT_TELEMETRY, CCS_VERSION_CHECK_SETTING, DEFAULT_ALLOW_PARTIAL_SEARCH_RESULTS, DEFAULT_ALLOW_PARTIAL_SEARCH_RESULTS_SETTING, DEFAULT_FROM, DEFAULT_KEEPALIVE_SETTING, DEFAULT_SEARCH_TIMEOUT_SETTING, DEFAULT_SIZE, ENABLE_REWRITE_AGGS_TO_FILTER_BY_FILTER, FETCH_PHASE_CHUNKED_ENABLED, FETCH_PHASE_CHUNKED_TARGET_CHUNK_BYTES, FETCH_PHASE_MAX_IN_FLIGHT_CHUNKS, KEEPALIVE_INTERVAL_SETTING, LOW_LEVEL_CANCELLATION_SETTING, MAX_ASYNC_SEARCH_RESPONSE_SIZE_SETTING, MAX_KEEPALIVE_SETTING, MAX_OPEN_SCROLL_CONTEXT, MEMORY_ACCOUNTING_BUFFER_SIZE, MINIMUM_DOCS_PER_SLICE, NO_TIMEOUT, PIT_RELOCATION_ENABLED, PIT_RELOCATION_FEATURE_FLAG, PREWARMING_THRESHOLD_THREADPOOL_SIZE_FACTOR_POOL_SIZE, QUERY_PHASE_PARALLEL_COLLECTION_ENABLED, SEARCH_WORKER_THREADS_ENABLED

Fields inherited from class org.elasticsearch.common.component.AbstractLifecycleComponent

lifecycle

Constructor Summary

Constructors

Constructor	Description
MockSearchService(ClusterService clusterService, IndicesService indicesService, ThreadPool threadPool, ScriptService scriptService, BigArrays bigArrays, FetchPhase fetchPhase, CircuitBreakerService circuitBreakerService, ExecutorSelector executorSelector, Tracer tracer, OnlinePrewarmingService onlinePrewarmingService)

Method Summary

Modifier and Type	Method	Description
static void	assertNoInFlightContext()	Throw an AssertionError if there are still in-flight contexts.
protected void	checkCancelled(CancellableTask task)
protected SearchContext	createContext(ReaderContext readerContext, ShardSearchRequest request, CancellableTask task, SearchService.ResultsType resultsType, boolean includeAggregations)
SearchContext	createSearchContext(ShardSearchRequest request, TimeValue timeout)
protected void	putReaderContext(ReaderContext context)
protected void	putRelocatedReaderContext(Long mappingKey, PitReaderContext context)
protected ReaderContext	removeReaderContext(ShardSearchContextId id)
void	setOnCheckCancelled(Function<CancellableTask,CancellableTask> onCheckCancelled)
void	setOnCreateSearchContext(Consumer<SearchContext> onCreateSearchContext)
void	setOnPutContext(Consumer<ReaderContext> onPutContext)
void	setOnRemoveContext(Consumer<ReaderContext> onRemoveContext)

Methods inherited from class org.elasticsearch.search.SearchService

afterIndexRemoved, aggReduceContextBuilder, batchQueryPhase, beforeIndexShardCreated, buildAliasFilter, canMatch, canMatch, canRewriteToMatchNone, createAndPutRelocatedPitContext, defaultAllowPartialSearchResults, doClose, doStart, doStop, executeDfsPhase, executeFetchPhase, executeFetchPhase, executeQueryPhase, executeQueryPhase, executeQueryPhase, executeRankFeaturePhase, fetchPhaseChunked, freeAllScrollContexts, freeReaderContext, getActiveContexts, getActivePITContexts, getActivePITContexts, getBigArrays, getCircuitBreaker, getCoordinatorRewriteContextProvider, getDefaultKeepAliveInMillis, getIndicesService, getOpenScrollContexts, getRewriteContext, getRewriteContext, isPitRelocationEnabled, openReaderContext, openReaderContext, queryStillMatchesAfterRewrite

Methods inherited from class org.elasticsearch.common.component.AbstractLifecycleComponent

addLifecycleListener, close, lifecycleState, start, stop

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.elasticsearch.index.shard.IndexEventListener

afterFilesRestoredFromRepository, afterIndexCreated, afterIndexShardClosed, afterIndexShardClosing, afterIndexShardCreated, afterIndexShardDeleted, afterIndexShardRecovery, afterIndexShardStarted, beforeIndexAddedToCluster, beforeIndexCreated, beforeIndexRemoved, beforeIndexShardClosed, beforeIndexShardDeleted, beforeIndexShardRecovery, indexShardStateChanged, onStoreClosed, onStoreCreated, shardRoutingChanged

Constructor Details

MockSearchService

public MockSearchService(ClusterService clusterService,
 IndicesService indicesService,
 ThreadPool threadPool,
 ScriptService scriptService,
 BigArrays bigArrays,
 FetchPhase fetchPhase,
 CircuitBreakerService circuitBreakerService,
 ExecutorSelector executorSelector,
 Tracer tracer,
 OnlinePrewarmingService onlinePrewarmingService)

Method Details

assertNoInFlightContext

public static void assertNoInFlightContext()

Throw an AssertionError if there are still in-flight contexts.

Note: this assertion can spuriously trip when search.low_level_cancellation is false (a setting that is randomized by ESIntegTestCase) and a test stops the coordinating node while a search is in-flight. With low-level cancellation disabled, the search is not canceled when the associated connection drops, leaving the reader context active long enough to fail this check. Tests that deliberately stop coordinators mid-search should override the setting via their nodeSettings (e.g. search.low_level_cancellation: true) rather than relaxing this assertion, so that genuine context leaks remain detectable.

putReaderContext

protected void putReaderContext(ReaderContext context)

Overrides:

putReaderContext in class SearchService

putRelocatedReaderContext

protected void putRelocatedReaderContext(Long mappingKey,
 PitReaderContext context)

Overrides:

putRelocatedReaderContext in class SearchService

removeReaderContext

protected ReaderContext removeReaderContext(ShardSearchContextId id)

Overrides:

removeReaderContext in class SearchService

setOnPutContext

public void setOnPutContext(Consumer<ReaderContext> onPutContext)

setOnRemoveContext

public void setOnRemoveContext(Consumer<ReaderContext> onRemoveContext)

setOnCreateSearchContext

public void setOnCreateSearchContext(Consumer<SearchContext> onCreateSearchContext)

createContext

protected SearchContext createContext(ReaderContext readerContext,
 ShardSearchRequest request,
 CancellableTask task,
 SearchService.ResultsType resultsType,
 boolean includeAggregations)
                               throws IOException

Overrides:

createContext in class SearchService

Throws:

IOException

createSearchContext

public SearchContext createSearchContext(ShardSearchRequest request,
 TimeValue timeout)
                                  throws IOException

Overrides:

createSearchContext in class SearchService

Throws:

IOException

setOnCheckCancelled

public void setOnCheckCancelled(Function<CancellableTask,CancellableTask> onCheckCancelled)

checkCancelled

protected void checkCancelled(CancellableTask task)

Overrides:

checkCancelled in class SearchService