forked from elastic/elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'upstream/8.x' into 118585_backport_8x
- Loading branch information
Showing
6 changed files
with
142 additions
and
200 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
86 changes: 86 additions & 0 deletions
86
...nternalClusterTest/java/org/elasticsearch/xpack/esql/action/AbstractPauseFieldPlugin.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,86 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License | ||
| * 2.0; you may not use this file except in compliance with the Elastic License | ||
| * 2.0. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.esql.action; | ||
|
|
||
| import org.elasticsearch.common.settings.Settings; | ||
| import org.elasticsearch.index.mapper.OnScriptError; | ||
| import org.elasticsearch.plugins.Plugin; | ||
| import org.elasticsearch.plugins.ScriptPlugin; | ||
| import org.elasticsearch.script.LongFieldScript; | ||
| import org.elasticsearch.script.ScriptContext; | ||
| import org.elasticsearch.script.ScriptEngine; | ||
| import org.elasticsearch.search.lookup.SearchLookup; | ||
|
|
||
| import java.util.Collection; | ||
| import java.util.Map; | ||
| import java.util.Set; | ||
|
|
||
| import static org.junit.Assert.assertTrue; | ||
|
|
||
| /** | ||
| * A plugin that provides a script language "pause" that can be used to simulate slow running queries. | ||
| * See also {@link AbstractPausableIntegTestCase}. | ||
| */ | ||
| public abstract class AbstractPauseFieldPlugin extends Plugin implements ScriptPlugin { | ||
|
|
||
| // Called when the engine enters the execute() method. | ||
| protected void onStartExecute() {} | ||
|
|
||
| // Called when the engine needs to wait for further execution to be allowed. | ||
| protected abstract boolean onWait() throws InterruptedException; | ||
|
|
||
| @Override | ||
| public ScriptEngine getScriptEngine(Settings settings, Collection<ScriptContext<?>> contexts) { | ||
| return new ScriptEngine() { | ||
| @Override | ||
| public String getType() { | ||
| return "pause"; | ||
| } | ||
|
|
||
| @Override | ||
| @SuppressWarnings("unchecked") | ||
| public <FactoryType> FactoryType compile( | ||
| String name, | ||
| String code, | ||
| ScriptContext<FactoryType> context, | ||
| Map<String, String> params | ||
| ) { | ||
| if (context == LongFieldScript.CONTEXT) { | ||
| return (FactoryType) new LongFieldScript.Factory() { | ||
| @Override | ||
| public LongFieldScript.LeafFactory newFactory( | ||
| String fieldName, | ||
| Map<String, Object> params, | ||
| SearchLookup searchLookup, | ||
| OnScriptError onScriptError | ||
| ) { | ||
| return ctx -> new LongFieldScript(fieldName, params, searchLookup, onScriptError, ctx) { | ||
| @Override | ||
| public void execute() { | ||
| onStartExecute(); | ||
| try { | ||
| assertTrue(onWait()); | ||
| } catch (InterruptedException e) { | ||
| throw new AssertionError(e); | ||
| } | ||
| emit(1); | ||
| } | ||
| }; | ||
| } | ||
| }; | ||
| } | ||
| throw new IllegalStateException("unsupported type " + context); | ||
| } | ||
|
|
||
| @Override | ||
| public Set<ScriptContext<?>> getSupportedContexts() { | ||
| return Set.of(LongFieldScript.CONTEXT); | ||
| } | ||
| }; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.