Watcher: Ensure that execution triggers properly on initial setup (#33360)
This commit reverts most of #33157 as it introduces another race condition and breaks a common case of watcher, when the first watch is added to the system and the index does not exist yet. This means, that the index will be created, which triggers a reload, but during this time the put watch operation that triggered this is not yet indexed, so that both processes finish roughly add the same time and should not overwrite each other but act complementary. This commit reverts the logic of cleaning out the ticker engine watches on start up, as this is done already when the execution is paused - which also gets paused on the cluster state listener again, as we can be sure here, that the watches index has not yet been created. This also adds a new test, that starts a one node cluster and emulates the case of a non existing watches index and a watch being added, which should result in proper execution. Closes #33320
This commit is contained in:
parent
9c0316869b
commit
1de2a925ce
|
@ -635,7 +635,7 @@ public abstract class ESRestTestCase extends ESTestCase {
|
||||||
if (name.startsWith(".monitoring-")) {
|
if (name.startsWith(".monitoring-")) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
if (name.startsWith(".watch-history-")) {
|
if (name.startsWith(".watch") || name.startsWith(".triggered_watches")) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
if (name.startsWith(".ml-")) {
|
if (name.startsWith(".ml-")) {
|
||||||
|
|
|
@ -27,12 +27,12 @@ import org.elasticsearch.common.settings.Settings;
|
||||||
import org.elasticsearch.common.unit.TimeValue;
|
import org.elasticsearch.common.unit.TimeValue;
|
||||||
import org.elasticsearch.common.util.concurrent.AbstractRunnable;
|
import org.elasticsearch.common.util.concurrent.AbstractRunnable;
|
||||||
import org.elasticsearch.common.util.concurrent.EsExecutors;
|
import org.elasticsearch.common.util.concurrent.EsExecutors;
|
||||||
import org.elasticsearch.common.util.concurrent.ThreadContext;
|
|
||||||
import org.elasticsearch.common.xcontent.XContentType;
|
import org.elasticsearch.common.xcontent.XContentType;
|
||||||
import org.elasticsearch.search.SearchHit;
|
import org.elasticsearch.search.SearchHit;
|
||||||
import org.elasticsearch.search.builder.SearchSourceBuilder;
|
import org.elasticsearch.search.builder.SearchSourceBuilder;
|
||||||
import org.elasticsearch.search.sort.SortBuilders;
|
import org.elasticsearch.search.sort.SortBuilders;
|
||||||
import org.elasticsearch.threadpool.ThreadPool;
|
import org.elasticsearch.threadpool.ThreadPool;
|
||||||
|
import org.elasticsearch.xpack.core.ClientHelper;
|
||||||
import org.elasticsearch.xpack.core.upgrade.UpgradeField;
|
import org.elasticsearch.xpack.core.upgrade.UpgradeField;
|
||||||
import org.elasticsearch.xpack.core.watcher.execution.TriggeredWatchStoreField;
|
import org.elasticsearch.xpack.core.watcher.execution.TriggeredWatchStoreField;
|
||||||
import org.elasticsearch.xpack.core.watcher.watch.Watch;
|
import org.elasticsearch.xpack.core.watcher.watch.Watch;
|
||||||
|
@ -63,7 +63,6 @@ import static org.elasticsearch.cluster.routing.ShardRoutingState.RELOCATING;
|
||||||
import static org.elasticsearch.cluster.routing.ShardRoutingState.STARTED;
|
import static org.elasticsearch.cluster.routing.ShardRoutingState.STARTED;
|
||||||
import static org.elasticsearch.common.util.concurrent.EsExecutors.daemonThreadFactory;
|
import static org.elasticsearch.common.util.concurrent.EsExecutors.daemonThreadFactory;
|
||||||
import static org.elasticsearch.xpack.core.ClientHelper.WATCHER_ORIGIN;
|
import static org.elasticsearch.xpack.core.ClientHelper.WATCHER_ORIGIN;
|
||||||
import static org.elasticsearch.xpack.core.ClientHelper.stashWithOrigin;
|
|
||||||
import static org.elasticsearch.xpack.core.watcher.support.Exceptions.illegalState;
|
import static org.elasticsearch.xpack.core.watcher.support.Exceptions.illegalState;
|
||||||
import static org.elasticsearch.xpack.core.watcher.watch.Watch.INDEX;
|
import static org.elasticsearch.xpack.core.watcher.watch.Watch.INDEX;
|
||||||
|
|
||||||
|
@ -92,7 +91,7 @@ public class WatcherService extends AbstractComponent {
|
||||||
this.scrollSize = settings.getAsInt("xpack.watcher.watch.scroll.size", 100);
|
this.scrollSize = settings.getAsInt("xpack.watcher.watch.scroll.size", 100);
|
||||||
this.defaultSearchTimeout = settings.getAsTime("xpack.watcher.internal.ops.search.default_timeout", TimeValue.timeValueSeconds(30));
|
this.defaultSearchTimeout = settings.getAsTime("xpack.watcher.internal.ops.search.default_timeout", TimeValue.timeValueSeconds(30));
|
||||||
this.parser = parser;
|
this.parser = parser;
|
||||||
this.client = client;
|
this.client = ClientHelper.clientWithOrigin(client, WATCHER_ORIGIN);
|
||||||
this.executor = executor;
|
this.executor = executor;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -184,6 +183,10 @@ public class WatcherService extends AbstractComponent {
|
||||||
// changes
|
// changes
|
||||||
processedClusterStateVersion.set(state.getVersion());
|
processedClusterStateVersion.set(state.getVersion());
|
||||||
|
|
||||||
|
triggerService.pauseExecution();
|
||||||
|
int cancelledTaskCount = executionService.clearExecutionsAndQueue();
|
||||||
|
logger.info("reloading watcher, reason [{}], cancelled [{}] queued tasks", reason, cancelledTaskCount);
|
||||||
|
|
||||||
executor.execute(wrapWatcherService(() -> reloadInner(state, reason, false),
|
executor.execute(wrapWatcherService(() -> reloadInner(state, reason, false),
|
||||||
e -> logger.error("error reloading watcher", e)));
|
e -> logger.error("error reloading watcher", e)));
|
||||||
}
|
}
|
||||||
|
@ -232,10 +235,6 @@ public class WatcherService extends AbstractComponent {
|
||||||
// also this is the place where we pause the trigger service execution and clear the current execution service, so that we make sure
|
// also this is the place where we pause the trigger service execution and clear the current execution service, so that we make sure
|
||||||
// that existing executions finish, but no new ones are executed
|
// that existing executions finish, but no new ones are executed
|
||||||
if (processedClusterStateVersion.get() == state.getVersion()) {
|
if (processedClusterStateVersion.get() == state.getVersion()) {
|
||||||
triggerService.pauseExecution();
|
|
||||||
int cancelledTaskCount = executionService.clearExecutionsAndQueue();
|
|
||||||
logger.info("reloading watcher, reason [{}], cancelled [{}] queued tasks", reason, cancelledTaskCount);
|
|
||||||
|
|
||||||
executionService.unPause();
|
executionService.unPause();
|
||||||
triggerService.start(watches);
|
triggerService.start(watches);
|
||||||
if (triggeredWatches.isEmpty() == false) {
|
if (triggeredWatches.isEmpty() == false) {
|
||||||
|
@ -273,7 +272,7 @@ public class WatcherService extends AbstractComponent {
|
||||||
|
|
||||||
SearchResponse response = null;
|
SearchResponse response = null;
|
||||||
List<Watch> watches = new ArrayList<>();
|
List<Watch> watches = new ArrayList<>();
|
||||||
try (ThreadContext.StoredContext ignore = stashWithOrigin(client.threadPool().getThreadContext(), WATCHER_ORIGIN)) {
|
try {
|
||||||
RefreshResponse refreshResponse = client.admin().indices().refresh(new RefreshRequest(INDEX))
|
RefreshResponse refreshResponse = client.admin().indices().refresh(new RefreshRequest(INDEX))
|
||||||
.actionGet(TimeValue.timeValueSeconds(5));
|
.actionGet(TimeValue.timeValueSeconds(5));
|
||||||
if (refreshResponse.getSuccessfulShards() < indexMetaData.getNumberOfShards()) {
|
if (refreshResponse.getSuccessfulShards() < indexMetaData.getNumberOfShards()) {
|
||||||
|
@ -357,13 +356,11 @@ public class WatcherService extends AbstractComponent {
|
||||||
}
|
}
|
||||||
} finally {
|
} finally {
|
||||||
if (response != null) {
|
if (response != null) {
|
||||||
try (ThreadContext.StoredContext ignore = stashWithOrigin(client.threadPool().getThreadContext(), WATCHER_ORIGIN)) {
|
|
||||||
ClearScrollRequest clearScrollRequest = new ClearScrollRequest();
|
ClearScrollRequest clearScrollRequest = new ClearScrollRequest();
|
||||||
clearScrollRequest.addScrollId(response.getScrollId());
|
clearScrollRequest.addScrollId(response.getScrollId());
|
||||||
client.clearScroll(clearScrollRequest).actionGet(scrollTimeout);
|
client.clearScroll(clearScrollRequest).actionGet(scrollTimeout);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
logger.debug("Loaded [{}] watches for execution", watches.size());
|
logger.debug("Loaded [{}] watches for execution", watches.size());
|
||||||
|
|
||||||
|
|
|
@ -22,6 +22,7 @@ import org.joda.time.DateTime;
|
||||||
import java.time.Clock;
|
import java.time.Clock;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
@ -49,14 +50,23 @@ public class TickerScheduleTriggerEngine extends ScheduleTriggerEngine {
|
||||||
@Override
|
@Override
|
||||||
public synchronized void start(Collection<Watch> jobs) {
|
public synchronized void start(Collection<Watch> jobs) {
|
||||||
long startTime = clock.millis();
|
long startTime = clock.millis();
|
||||||
Map<String, ActiveSchedule> schedules = new ConcurrentHashMap<>();
|
Map<String, ActiveSchedule> schedules = new HashMap<>(jobs.size());
|
||||||
for (Watch job : jobs) {
|
for (Watch job : jobs) {
|
||||||
if (job.trigger() instanceof ScheduleTrigger) {
|
if (job.trigger() instanceof ScheduleTrigger) {
|
||||||
ScheduleTrigger trigger = (ScheduleTrigger) job.trigger();
|
ScheduleTrigger trigger = (ScheduleTrigger) job.trigger();
|
||||||
schedules.put(job.id(), new ActiveSchedule(job.id(), trigger.getSchedule(), startTime));
|
schedules.put(job.id(), new ActiveSchedule(job.id(), trigger.getSchedule(), startTime));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
this.schedules = schedules;
|
// why are we calling putAll() here instead of assigning a brand
|
||||||
|
// new concurrent hash map you may ask yourself over here
|
||||||
|
// This requires some explanation how TriggerEngine.start() is
|
||||||
|
// invoked, when a reload due to the cluster state listener is done
|
||||||
|
// If the watches index does not exist, and new document is stored,
|
||||||
|
// then the creation of that index will trigger a reload which calls
|
||||||
|
// this method. The index operation however will run at the same time
|
||||||
|
// as the reload, so if we clean out the old data structure here,
|
||||||
|
// that can lead to that one watch not being triggered
|
||||||
|
this.schedules.putAll(schedules);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -6,19 +6,21 @@
|
||||||
package org.elasticsearch.xpack.watcher;
|
package org.elasticsearch.xpack.watcher;
|
||||||
|
|
||||||
import org.elasticsearch.Version;
|
import org.elasticsearch.Version;
|
||||||
|
import org.elasticsearch.action.ActionListener;
|
||||||
|
import org.elasticsearch.action.admin.indices.refresh.RefreshAction;
|
||||||
import org.elasticsearch.action.admin.indices.refresh.RefreshRequest;
|
import org.elasticsearch.action.admin.indices.refresh.RefreshRequest;
|
||||||
import org.elasticsearch.action.admin.indices.refresh.RefreshResponse;
|
import org.elasticsearch.action.admin.indices.refresh.RefreshResponse;
|
||||||
|
import org.elasticsearch.action.search.ClearScrollAction;
|
||||||
import org.elasticsearch.action.search.ClearScrollRequest;
|
import org.elasticsearch.action.search.ClearScrollRequest;
|
||||||
import org.elasticsearch.action.search.ClearScrollResponse;
|
import org.elasticsearch.action.search.ClearScrollResponse;
|
||||||
|
import org.elasticsearch.action.search.SearchAction;
|
||||||
import org.elasticsearch.action.search.SearchRequest;
|
import org.elasticsearch.action.search.SearchRequest;
|
||||||
import org.elasticsearch.action.search.SearchResponse;
|
import org.elasticsearch.action.search.SearchResponse;
|
||||||
import org.elasticsearch.action.search.SearchResponseSections;
|
import org.elasticsearch.action.search.SearchResponseSections;
|
||||||
|
import org.elasticsearch.action.search.SearchScrollAction;
|
||||||
import org.elasticsearch.action.search.SearchScrollRequest;
|
import org.elasticsearch.action.search.SearchScrollRequest;
|
||||||
import org.elasticsearch.action.search.ShardSearchFailure;
|
import org.elasticsearch.action.search.ShardSearchFailure;
|
||||||
import org.elasticsearch.action.support.PlainActionFuture;
|
|
||||||
import org.elasticsearch.client.AdminClient;
|
|
||||||
import org.elasticsearch.client.Client;
|
import org.elasticsearch.client.Client;
|
||||||
import org.elasticsearch.client.IndicesAdminClient;
|
|
||||||
import org.elasticsearch.cluster.ClusterName;
|
import org.elasticsearch.cluster.ClusterName;
|
||||||
import org.elasticsearch.cluster.ClusterState;
|
import org.elasticsearch.cluster.ClusterState;
|
||||||
import org.elasticsearch.cluster.metadata.IndexMetaData;
|
import org.elasticsearch.cluster.metadata.IndexMetaData;
|
||||||
|
@ -42,7 +44,6 @@ import org.elasticsearch.search.SearchHits;
|
||||||
import org.elasticsearch.search.SearchShardTarget;
|
import org.elasticsearch.search.SearchShardTarget;
|
||||||
import org.elasticsearch.test.ESTestCase;
|
import org.elasticsearch.test.ESTestCase;
|
||||||
import org.elasticsearch.threadpool.ThreadPool;
|
import org.elasticsearch.threadpool.ThreadPool;
|
||||||
import org.elasticsearch.xpack.core.XPackSettings;
|
|
||||||
import org.elasticsearch.xpack.core.watcher.trigger.Trigger;
|
import org.elasticsearch.xpack.core.watcher.trigger.Trigger;
|
||||||
import org.elasticsearch.xpack.core.watcher.watch.Watch;
|
import org.elasticsearch.xpack.core.watcher.watch.Watch;
|
||||||
import org.elasticsearch.xpack.core.watcher.watch.WatchStatus;
|
import org.elasticsearch.xpack.core.watcher.watch.WatchStatus;
|
||||||
|
@ -55,6 +56,7 @@ import org.elasticsearch.xpack.watcher.trigger.TriggerService;
|
||||||
import org.elasticsearch.xpack.watcher.watch.WatchParser;
|
import org.elasticsearch.xpack.watcher.watch.WatchParser;
|
||||||
import org.joda.time.DateTime;
|
import org.joda.time.DateTime;
|
||||||
import org.joda.time.DateTimeZone;
|
import org.joda.time.DateTimeZone;
|
||||||
|
import org.junit.Before;
|
||||||
import org.mockito.ArgumentCaptor;
|
import org.mockito.ArgumentCaptor;
|
||||||
|
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
@ -67,6 +69,7 @@ import static org.hamcrest.Matchers.hasSize;
|
||||||
import static org.hamcrest.Matchers.is;
|
import static org.hamcrest.Matchers.is;
|
||||||
import static org.mockito.Matchers.any;
|
import static org.mockito.Matchers.any;
|
||||||
import static org.mockito.Matchers.eq;
|
import static org.mockito.Matchers.eq;
|
||||||
|
import static org.mockito.Mockito.doAnswer;
|
||||||
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.mock;
|
||||||
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.never;
|
||||||
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.verify;
|
||||||
|
@ -76,6 +79,16 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
|
|
||||||
private final ExecutorService executorService = EsExecutors.newDirectExecutorService();
|
private final ExecutorService executorService = EsExecutors.newDirectExecutorService();
|
||||||
|
|
||||||
|
private final Client client = mock(Client.class);
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void configureMockClient() {
|
||||||
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
||||||
|
ThreadPool threadPool = mock(ThreadPool.class);
|
||||||
|
when(client.threadPool()).thenReturn(threadPool);
|
||||||
|
when(threadPool.getThreadContext()).thenReturn(new ThreadContext(Settings.EMPTY));
|
||||||
|
}
|
||||||
|
|
||||||
public void testValidateStartWithClosedIndex() {
|
public void testValidateStartWithClosedIndex() {
|
||||||
TriggerService triggerService = mock(TriggerService.class);
|
TriggerService triggerService = mock(TriggerService.class);
|
||||||
TriggeredWatchStore triggeredWatchStore = mock(TriggeredWatchStore.class);
|
TriggeredWatchStore triggeredWatchStore = mock(TriggeredWatchStore.class);
|
||||||
|
@ -83,7 +96,7 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
WatchParser parser = mock(WatchParser.class);
|
WatchParser parser = mock(WatchParser.class);
|
||||||
|
|
||||||
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, triggeredWatchStore,
|
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, triggeredWatchStore,
|
||||||
executionService, parser, mock(Client.class), executorService) {
|
executionService, parser, client, executorService) {
|
||||||
@Override
|
@Override
|
||||||
void stopExecutor() {
|
void stopExecutor() {
|
||||||
}
|
}
|
||||||
|
@ -102,18 +115,11 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
}
|
}
|
||||||
|
|
||||||
public void testLoadOnlyActiveWatches() throws Exception {
|
public void testLoadOnlyActiveWatches() throws Exception {
|
||||||
// this is just, so we dont have to add any mocking to the threadpool
|
|
||||||
Settings settings = Settings.builder().put(XPackSettings.SECURITY_ENABLED.getKey(), false).build();
|
|
||||||
|
|
||||||
TriggerService triggerService = mock(TriggerService.class);
|
TriggerService triggerService = mock(TriggerService.class);
|
||||||
TriggeredWatchStore triggeredWatchStore = mock(TriggeredWatchStore.class);
|
TriggeredWatchStore triggeredWatchStore = mock(TriggeredWatchStore.class);
|
||||||
ExecutionService executionService = mock(ExecutionService.class);
|
ExecutionService executionService = mock(ExecutionService.class);
|
||||||
WatchParser parser = mock(WatchParser.class);
|
WatchParser parser = mock(WatchParser.class);
|
||||||
Client client = mock(Client.class);
|
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, triggeredWatchStore,
|
||||||
ThreadPool threadPool = mock(ThreadPool.class);
|
|
||||||
when(client.threadPool()).thenReturn(threadPool);
|
|
||||||
when(threadPool.getThreadContext()).thenReturn(new ThreadContext(Settings.EMPTY));
|
|
||||||
WatcherService service = new WatcherService(settings, triggerService, triggeredWatchStore,
|
|
||||||
executionService, parser, client, executorService) {
|
executionService, parser, client, executorService) {
|
||||||
@Override
|
@Override
|
||||||
void stopExecutor() {
|
void stopExecutor() {
|
||||||
|
@ -150,21 +156,21 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
RefreshResponse refreshResponse = mock(RefreshResponse.class);
|
RefreshResponse refreshResponse = mock(RefreshResponse.class);
|
||||||
when(refreshResponse.getSuccessfulShards())
|
when(refreshResponse.getSuccessfulShards())
|
||||||
.thenReturn(clusterState.getMetaData().getIndices().get(Watch.INDEX).getNumberOfShards());
|
.thenReturn(clusterState.getMetaData().getIndices().get(Watch.INDEX).getNumberOfShards());
|
||||||
AdminClient adminClient = mock(AdminClient.class);
|
doAnswer(invocation -> {
|
||||||
IndicesAdminClient indicesAdminClient = mock(IndicesAdminClient.class);
|
ActionListener<RefreshResponse> listener = (ActionListener<RefreshResponse>) invocation.getArguments()[2];
|
||||||
when(client.admin()).thenReturn(adminClient);
|
listener.onResponse(refreshResponse);
|
||||||
when(adminClient.indices()).thenReturn(indicesAdminClient);
|
return null;
|
||||||
PlainActionFuture<RefreshResponse> refreshFuture = new PlainActionFuture<>();
|
}).when(client).execute(eq(RefreshAction.INSTANCE), any(RefreshRequest.class), any(ActionListener.class));
|
||||||
when(indicesAdminClient.refresh(any(RefreshRequest.class))).thenReturn(refreshFuture);
|
|
||||||
refreshFuture.onResponse(refreshResponse);
|
|
||||||
|
|
||||||
// empty scroll response, no further scrolling needed
|
// empty scroll response, no further scrolling needed
|
||||||
SearchResponseSections scrollSearchSections = new SearchResponseSections(SearchHits.empty(), null, null, false, false, null, 1);
|
SearchResponseSections scrollSearchSections = new SearchResponseSections(SearchHits.empty(), null, null, false, false, null, 1);
|
||||||
SearchResponse scrollSearchResponse = new SearchResponse(scrollSearchSections, "scrollId", 1, 1, 0, 10,
|
SearchResponse scrollSearchResponse = new SearchResponse(scrollSearchSections, "scrollId", 1, 1, 0, 10,
|
||||||
ShardSearchFailure.EMPTY_ARRAY, SearchResponse.Clusters.EMPTY);
|
ShardSearchFailure.EMPTY_ARRAY, SearchResponse.Clusters.EMPTY);
|
||||||
PlainActionFuture<SearchResponse> searchScrollResponseFuture = new PlainActionFuture<>();
|
doAnswer(invocation -> {
|
||||||
when(client.searchScroll(any(SearchScrollRequest.class))).thenReturn(searchScrollResponseFuture);
|
ActionListener<SearchResponse> listener = (ActionListener<SearchResponse>) invocation.getArguments()[2];
|
||||||
searchScrollResponseFuture.onResponse(scrollSearchResponse);
|
listener.onResponse(scrollSearchResponse);
|
||||||
|
return null;
|
||||||
|
}).when(client).execute(eq(SearchScrollAction.INSTANCE), any(SearchScrollRequest.class), any(ActionListener.class));
|
||||||
|
|
||||||
// one search response containing active and inactive watches
|
// one search response containing active and inactive watches
|
||||||
int count = randomIntBetween(2, 200);
|
int count = randomIntBetween(2, 200);
|
||||||
|
@ -192,13 +198,17 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
SearchResponseSections sections = new SearchResponseSections(searchHits, null, null, false, false, null, 1);
|
SearchResponseSections sections = new SearchResponseSections(searchHits, null, null, false, false, null, 1);
|
||||||
SearchResponse searchResponse = new SearchResponse(sections, "scrollId", 1, 1, 0, 10, ShardSearchFailure.EMPTY_ARRAY,
|
SearchResponse searchResponse = new SearchResponse(sections, "scrollId", 1, 1, 0, 10, ShardSearchFailure.EMPTY_ARRAY,
|
||||||
SearchResponse.Clusters.EMPTY);
|
SearchResponse.Clusters.EMPTY);
|
||||||
PlainActionFuture<SearchResponse> searchResponseFuture = new PlainActionFuture<>();
|
doAnswer(invocation -> {
|
||||||
when(client.search(any(SearchRequest.class))).thenReturn(searchResponseFuture);
|
ActionListener<SearchResponse> listener = (ActionListener<SearchResponse>) invocation.getArguments()[2];
|
||||||
searchResponseFuture.onResponse(searchResponse);
|
listener.onResponse(searchResponse);
|
||||||
|
return null;
|
||||||
|
}).when(client).execute(eq(SearchAction.INSTANCE), any(SearchRequest.class), any(ActionListener.class));
|
||||||
|
|
||||||
PlainActionFuture<ClearScrollResponse> clearScrollFuture = new PlainActionFuture<>();
|
doAnswer(invocation -> {
|
||||||
when(client.clearScroll(any(ClearScrollRequest.class))).thenReturn(clearScrollFuture);
|
ActionListener<ClearScrollResponse> listener = (ActionListener<ClearScrollResponse>) invocation.getArguments()[2];
|
||||||
clearScrollFuture.onResponse(new ClearScrollResponse(true, 1));
|
listener.onResponse(new ClearScrollResponse(true, 1));
|
||||||
|
return null;
|
||||||
|
}).when(client).execute(eq(ClearScrollAction.INSTANCE), any(ClearScrollRequest.class), any(ActionListener.class));
|
||||||
|
|
||||||
service.start(clusterState, () -> {});
|
service.start(clusterState, () -> {});
|
||||||
|
|
||||||
|
@ -228,7 +238,7 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
assertThat(triggerService.count(), is(1L));
|
assertThat(triggerService.count(), is(1L));
|
||||||
|
|
||||||
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, mock(TriggeredWatchStore.class),
|
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, mock(TriggeredWatchStore.class),
|
||||||
mock(ExecutionService.class), mock(WatchParser.class), mock(Client.class), executorService) {
|
mock(ExecutionService.class), mock(WatchParser.class), client, executorService) {
|
||||||
@Override
|
@Override
|
||||||
void stopExecutor() {
|
void stopExecutor() {
|
||||||
}
|
}
|
||||||
|
@ -245,7 +255,7 @@ public class WatcherServiceTests extends ESTestCase {
|
||||||
ExecutionService executionService = mock(ExecutionService.class);
|
ExecutionService executionService = mock(ExecutionService.class);
|
||||||
TriggerService triggerService = mock(TriggerService.class);
|
TriggerService triggerService = mock(TriggerService.class);
|
||||||
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, mock(TriggeredWatchStore.class),
|
WatcherService service = new WatcherService(Settings.EMPTY, triggerService, mock(TriggeredWatchStore.class),
|
||||||
executionService, mock(WatchParser.class), mock(Client.class), executorService) {
|
executionService, mock(WatchParser.class), client, executorService) {
|
||||||
@Override
|
@Override
|
||||||
void stopExecutor() {
|
void stopExecutor() {
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,66 @@
|
||||||
|
/*
|
||||||
|
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
|
||||||
|
* or more contributor license agreements. Licensed under the Elastic License;
|
||||||
|
* you may not use this file except in compliance with the Elastic License.
|
||||||
|
*/
|
||||||
|
package org.elasticsearch.xpack.watcher.test.integration;
|
||||||
|
|
||||||
|
import org.elasticsearch.action.admin.cluster.state.ClusterStateResponse;
|
||||||
|
import org.elasticsearch.action.search.SearchResponse;
|
||||||
|
import org.elasticsearch.cluster.metadata.IndexMetaData;
|
||||||
|
import org.elasticsearch.protocol.xpack.watcher.PutWatchResponse;
|
||||||
|
import org.elasticsearch.test.ESIntegTestCase.ClusterScope;
|
||||||
|
import org.elasticsearch.xpack.core.watcher.watch.Watch;
|
||||||
|
import org.elasticsearch.xpack.watcher.test.AbstractWatcherIntegrationTestCase;
|
||||||
|
import org.elasticsearch.xpack.watcher.trigger.schedule.IntervalSchedule;
|
||||||
|
import org.elasticsearch.xpack.watcher.watch.WatchStoreUtils;
|
||||||
|
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
|
import static org.elasticsearch.test.ESIntegTestCase.Scope.SUITE;
|
||||||
|
import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
|
||||||
|
import static org.elasticsearch.xpack.watcher.actions.ActionBuilders.loggingAction;
|
||||||
|
import static org.elasticsearch.xpack.watcher.client.WatchSourceBuilders.watchBuilder;
|
||||||
|
import static org.elasticsearch.xpack.watcher.input.InputBuilders.simpleInput;
|
||||||
|
import static org.elasticsearch.xpack.watcher.trigger.TriggerBuilders.schedule;
|
||||||
|
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.interval;
|
||||||
|
import static org.hamcrest.Matchers.greaterThanOrEqualTo;
|
||||||
|
import static org.hamcrest.Matchers.is;
|
||||||
|
|
||||||
|
@ClusterScope(scope = SUITE, numClientNodes = 0, transportClientRatio = 0, maxNumDataNodes = 1, supportsDedicatedMasters = false)
|
||||||
|
public class SingleNodeTests extends AbstractWatcherIntegrationTestCase {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected boolean timeWarped() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// this is the standard setup when starting watcher in a regular cluster
|
||||||
|
// the index does not exist, a watch gets added
|
||||||
|
// the watch should be executed properly, despite the index being created and the cluster state listener being reloaded
|
||||||
|
public void testThatLoadingWithNonExistingIndexWorks() throws Exception {
|
||||||
|
stopWatcher();
|
||||||
|
ClusterStateResponse clusterStateResponse = client().admin().cluster().prepareState().get();
|
||||||
|
IndexMetaData metaData = WatchStoreUtils.getConcreteIndex(Watch.INDEX, clusterStateResponse.getState().metaData());
|
||||||
|
String watchIndexName = metaData.getIndex().getName();
|
||||||
|
assertAcked(client().admin().indices().prepareDelete(watchIndexName));
|
||||||
|
startWatcher();
|
||||||
|
|
||||||
|
String watchId = randomAlphaOfLength(20);
|
||||||
|
// now we start with an empty set up, store a watch and expected it to be executed
|
||||||
|
PutWatchResponse putWatchResponse = watcherClient().preparePutWatch(watchId)
|
||||||
|
.setSource(watchBuilder()
|
||||||
|
.trigger(schedule(interval(1, IntervalSchedule.Interval.Unit.SECONDS)))
|
||||||
|
.input(simpleInput())
|
||||||
|
.addAction("_logger", loggingAction("logging of watch _name")))
|
||||||
|
.get();
|
||||||
|
assertThat(putWatchResponse.isCreated(), is(true));
|
||||||
|
|
||||||
|
assertBusy(() -> {
|
||||||
|
client().admin().indices().prepareRefresh(".watcher-history*");
|
||||||
|
SearchResponse searchResponse = client().prepareSearch(".watcher-history*").setSize(0).get();
|
||||||
|
assertThat(searchResponse.getHits().getTotalHits(), is(greaterThanOrEqualTo(1L)));
|
||||||
|
}, 5, TimeUnit.SECONDS);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -35,9 +35,7 @@ import java.util.function.Consumer;
|
||||||
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.daily;
|
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.daily;
|
||||||
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.interval;
|
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.interval;
|
||||||
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.weekly;
|
import static org.elasticsearch.xpack.watcher.trigger.schedule.Schedules.weekly;
|
||||||
import static org.hamcrest.Matchers.everyItem;
|
|
||||||
import static org.hamcrest.Matchers.is;
|
import static org.hamcrest.Matchers.is;
|
||||||
import static org.hamcrest.Matchers.startsWith;
|
|
||||||
import static org.joda.time.DateTimeZone.UTC;
|
import static org.joda.time.DateTimeZone.UTC;
|
||||||
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.mock;
|
||||||
|
|
||||||
|
@ -110,40 +108,6 @@ public class TickerScheduleEngineTests extends ESTestCase {
|
||||||
assertThat(bits.cardinality(), is(count));
|
assertThat(bits.cardinality(), is(count));
|
||||||
}
|
}
|
||||||
|
|
||||||
public void testStartClearsExistingSchedules() throws Exception {
|
|
||||||
final CountDownLatch latch = new CountDownLatch(1);
|
|
||||||
List<String> firedWatchIds = new ArrayList<>();
|
|
||||||
engine.register(new Consumer<Iterable<TriggerEvent>>() {
|
|
||||||
@Override
|
|
||||||
public void accept(Iterable<TriggerEvent> events) {
|
|
||||||
for (TriggerEvent event : events) {
|
|
||||||
firedWatchIds.add(event.jobName());
|
|
||||||
}
|
|
||||||
latch.countDown();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
int count = randomIntBetween(2, 5);
|
|
||||||
List<Watch> watches = new ArrayList<>();
|
|
||||||
for (int i = 0; i < count; i++) {
|
|
||||||
watches.add(createWatch(String.valueOf(i), interval("1s")));
|
|
||||||
}
|
|
||||||
engine.start(watches);
|
|
||||||
|
|
||||||
watches.clear();
|
|
||||||
for (int i = 0; i < count; i++) {
|
|
||||||
watches.add(createWatch("another_id" + i, interval("1s")));
|
|
||||||
}
|
|
||||||
engine.start(watches);
|
|
||||||
|
|
||||||
advanceClockIfNeeded(new DateTime(clock.millis(), UTC).plusMillis(1100));
|
|
||||||
if (!latch.await(3 * count, TimeUnit.SECONDS)) {
|
|
||||||
fail("waiting too long for all watches to be triggered");
|
|
||||||
}
|
|
||||||
|
|
||||||
assertThat(firedWatchIds, everyItem(startsWith("another_id")));
|
|
||||||
}
|
|
||||||
|
|
||||||
public void testAddHourly() throws Exception {
|
public void testAddHourly() throws Exception {
|
||||||
final String name = "job_name";
|
final String name = "job_name";
|
||||||
final CountDownLatch latch = new CountDownLatch(1);
|
final CountDownLatch latch = new CountDownLatch(1);
|
||||||
|
|
|
@ -61,8 +61,8 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
switch (state) {
|
switch (state) {
|
||||||
case "stopped":
|
case "stopped":
|
||||||
Response startResponse = adminClient().performRequest(new Request("POST", "/_xpack/watcher/_start"));
|
Response startResponse = adminClient().performRequest(new Request("POST", "/_xpack/watcher/_start"));
|
||||||
String body = EntityUtils.toString(startResponse.getEntity());
|
Map<String, Object> responseMap = entityAsMap(startResponse);
|
||||||
assertThat(body, containsString("\"acknowledged\":true"));
|
assertThat(responseMap, hasEntry("acknowledged", true));
|
||||||
break;
|
break;
|
||||||
case "stopping":
|
case "stopping":
|
||||||
throw new AssertionError("waiting until stopping state reached stopped state to start again");
|
throw new AssertionError("waiting until stopping state reached stopped state to start again");
|
||||||
|
@ -135,7 +135,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/33320")
|
|
||||||
public void testSearchInputHasPermissions() throws Exception {
|
public void testSearchInputHasPermissions() throws Exception {
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
builder.startObject();
|
builder.startObject();
|
||||||
|
@ -159,7 +158,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
assertThat(conditionMet, is(true));
|
assertThat(conditionMet, is(true));
|
||||||
}
|
}
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/29893")
|
|
||||||
public void testSearchInputWithInsufficientPrivileges() throws Exception {
|
public void testSearchInputWithInsufficientPrivileges() throws Exception {
|
||||||
String indexName = "index_not_allowed_to_read";
|
String indexName = "index_not_allowed_to_read";
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
|
@ -186,7 +184,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
assertThat(conditionMet, is(false));
|
assertThat(conditionMet, is(false));
|
||||||
}
|
}
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/33320")
|
|
||||||
public void testSearchTransformHasPermissions() throws Exception {
|
public void testSearchTransformHasPermissions() throws Exception {
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
builder.startObject();
|
builder.startObject();
|
||||||
|
@ -216,7 +213,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
assertThat(value, is("15"));
|
assertThat(value, is("15"));
|
||||||
}
|
}
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/33291")
|
|
||||||
public void testSearchTransformInsufficientPermissions() throws Exception {
|
public void testSearchTransformInsufficientPermissions() throws Exception {
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
builder.startObject();
|
builder.startObject();
|
||||||
|
@ -244,7 +240,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
assertThat(response.getStatusLine().getStatusCode(), is(404));
|
assertThat(response.getStatusLine().getStatusCode(), is(404));
|
||||||
}
|
}
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/30777")
|
|
||||||
public void testIndexActionHasPermissions() throws Exception {
|
public void testIndexActionHasPermissions() throws Exception {
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
builder.startObject();
|
builder.startObject();
|
||||||
|
@ -269,7 +264,6 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
assertThat(spam, is("eggs"));
|
assertThat(spam, is("eggs"));
|
||||||
}
|
}
|
||||||
|
|
||||||
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/33320")
|
|
||||||
public void testIndexActionInsufficientPrivileges() throws Exception {
|
public void testIndexActionInsufficientPrivileges() throws Exception {
|
||||||
try (XContentBuilder builder = jsonBuilder()) {
|
try (XContentBuilder builder = jsonBuilder()) {
|
||||||
builder.startObject();
|
builder.startObject();
|
||||||
|
@ -299,6 +293,8 @@ public class SmokeTestWatcherWithSecurityIT extends ESRestTestCase {
|
||||||
Response response = client().performRequest(request);
|
Response response = client().performRequest(request);
|
||||||
Map<String, Object> responseMap = entityAsMap(response);
|
Map<String, Object> responseMap = entityAsMap(response);
|
||||||
assertThat(responseMap, hasEntry("_id", watchId));
|
assertThat(responseMap, hasEntry("_id", watchId));
|
||||||
|
assertThat(responseMap, hasEntry("created", true));
|
||||||
|
assertThat(responseMap, hasEntry("_version", 1));
|
||||||
}
|
}
|
||||||
|
|
||||||
private ObjectPath getWatchHistoryEntry(String watchId) throws Exception {
|
private ObjectPath getWatchHistoryEntry(String watchId) throws Exception {
|
||||||
|
|
Loading…
Reference in New Issue