|
11 | 11 |
|
12 | 12 | import org.apache.logging.log4j.LogManager; |
13 | 13 | import org.apache.logging.log4j.Logger; |
| 14 | +import org.elasticsearch.ElasticsearchStatusException; |
14 | 15 | import org.elasticsearch.action.ActionListener; |
15 | 16 | import org.elasticsearch.action.support.ActionFilters; |
16 | 17 | import org.elasticsearch.action.support.master.AcknowledgedRequest; |
|
22 | 23 | import org.elasticsearch.cluster.block.ClusterBlockException; |
23 | 24 | import org.elasticsearch.cluster.block.ClusterBlockLevel; |
24 | 25 | import org.elasticsearch.cluster.metadata.ProjectId; |
| 26 | +import org.elasticsearch.cluster.metadata.ProjectMetadata; |
25 | 27 | import org.elasticsearch.cluster.metadata.StreamsMetadata; |
26 | 28 | import org.elasticsearch.cluster.project.ProjectResolver; |
27 | 29 | import org.elasticsearch.cluster.service.ClusterService; |
28 | 30 | import org.elasticsearch.cluster.service.MasterServiceTaskQueue; |
29 | 31 | import org.elasticsearch.common.Priority; |
30 | 32 | import org.elasticsearch.injection.guice.Inject; |
| 33 | +import org.elasticsearch.rest.RestStatus; |
31 | 34 | import org.elasticsearch.tasks.Task; |
32 | 35 | import org.elasticsearch.threadpool.ThreadPool; |
33 | 36 | import org.elasticsearch.transport.TransportService; |
34 | 37 |
|
| 38 | +import java.util.Arrays; |
35 | 39 | import java.util.Locale; |
36 | 40 |
|
37 | 41 | /** |
@@ -77,17 +81,34 @@ protected void masterOperation( |
77 | 81 | ActionListener<AcknowledgedResponse> listener |
78 | 82 | ) throws Exception { |
79 | 83 | ProjectId projectId = projectResolver.getProjectId(); |
80 | | - StreamsMetadata streamsState = state.metadata().getProject(projectId).custom(StreamsMetadata.TYPE, StreamsMetadata.EMPTY); |
| 84 | + ProjectMetadata projectMetadata = state.metadata().getProject(projectId); |
| 85 | + StreamsMetadata streamsState = projectMetadata.custom(StreamsMetadata.TYPE, StreamsMetadata.EMPTY); |
81 | 86 | boolean currentlyEnabled = streamsState.isLogsEnabled(); |
82 | 87 | boolean shouldEnable = request.shouldEnable(); |
83 | | - if (shouldEnable != currentlyEnabled) { |
84 | | - StreamsMetadataUpdateTask updateTask = new StreamsMetadataUpdateTask(request, listener, projectId, shouldEnable); |
85 | | - String taskName = String.format(Locale.ROOT, "enable-streams-logs-[%s]", shouldEnable ? "enable" : "disable"); |
86 | | - taskQueue.submitTask(taskName, updateTask, updateTask.timeout()); |
87 | | - } else { |
| 88 | + |
| 89 | + if (shouldEnable == currentlyEnabled) { |
88 | 90 | logger.debug("Logs streams are already in the requested state: {}", shouldEnable); |
89 | 91 | listener.onResponse(AcknowledgedResponse.TRUE); |
| 92 | + return; |
| 93 | + } |
| 94 | + |
| 95 | + if (shouldEnable && logsIndexExists(projectMetadata)) { |
| 96 | + listener.onFailure( |
| 97 | + new ElasticsearchStatusException( |
| 98 | + "Cannot enable logs streams: indices named 'logs' or starting with 'logs.' already exist.", |
| 99 | + RestStatus.CONFLICT |
| 100 | + ) |
| 101 | + ); |
| 102 | + return; |
90 | 103 | } |
| 104 | + |
| 105 | + StreamsMetadataUpdateTask updateTask = new StreamsMetadataUpdateTask(request, listener, projectId, shouldEnable); |
| 106 | + String taskName = String.format(Locale.ROOT, "enable-streams-logs-[%s]", shouldEnable ? "enable" : "disable"); |
| 107 | + taskQueue.submitTask(taskName, updateTask, updateTask.timeout()); |
| 108 | + } |
| 109 | + |
| 110 | + private boolean logsIndexExists(ProjectMetadata projectMetadata) { |
| 111 | + return Arrays.stream(projectMetadata.getConcreteAllIndices()).anyMatch(name -> name.equals("logs") || name.startsWith("logs.")); |
91 | 112 | } |
92 | 113 |
|
93 | 114 | @Override |
|
0 commit comments