Skip to content

Commit a229c8d

Browse files
authored
Integrate project global blocks into existing checks for cluster blocks (Part 2) (#129570)
Relates #129467 Resolves ES-11209
1 parent f62fce9 commit a229c8d

File tree

39 files changed

+156
-63
lines changed

39 files changed

+156
-63
lines changed

modules/data-streams/src/main/java/org/elasticsearch/datastreams/action/TransportGetDataStreamSettingsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ public TransportGetDataStreamSettingsAction(
6363

6464
@Override
6565
protected ClusterBlockException checkBlock(GetDataStreamSettingsAction.Request request, ProjectState state) {
66-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
66+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
6767
}
6868

6969
@Override

modules/data-streams/src/main/java/org/elasticsearch/datastreams/action/TransportGetDataStreamsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,6 @@ static List<DataStream> getDataStreams(ProjectMetadata project, IndexNameExpress
410410

411411
@Override
412412
protected ClusterBlockException checkBlock(GetDataStreamAction.Request request, ProjectState state) {
413-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
413+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
414414
}
415415
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/action/TransportMigrateToDataStreamAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,6 @@ protected void masterOperation(
7777

7878
@Override
7979
protected ClusterBlockException checkBlock(MigrateToDataStreamAction.Request request, ProjectState state) {
80-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
80+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
8181
}
8282
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/action/TransportModifyDataStreamsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ protected void masterOperation(
6767

6868
@Override
6969
protected ClusterBlockException checkBlock(ModifyDataStreamsAction.Request request, ProjectState state) {
70-
ClusterBlockException globalBlock = state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
70+
ClusterBlockException globalBlock = state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
7171
if (globalBlock != null) {
7272
return globalBlock;
7373
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/action/TransportUpdateDataStreamSettingsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -339,6 +339,6 @@ private void updateSettingsOnSingleIndex(
339339

340340
@Override
341341
protected ClusterBlockException checkBlock(UpdateDataStreamSettingsAction.Request request, ClusterState state) {
342-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
342+
return state.blocks().globalBlockedException(projectResolver.getProjectId(), ClusterBlockLevel.METADATA_WRITE);
343343
}
344344
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/lifecycle/action/TransportGetDataStreamLifecycleStatsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,6 @@ GetDataStreamLifecycleStatsAction.Response collectStats(ProjectMetadata project)
105105

106106
@Override
107107
protected ClusterBlockException checkBlock(GetDataStreamLifecycleStatsAction.Request request, ProjectState state) {
108-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_READ);
108+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_READ);
109109
}
110110
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/lifecycle/action/TransportPutDataStreamLifecycleAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,6 @@ protected void masterOperation(
9494

9595
@Override
9696
protected ClusterBlockException checkBlock(PutDataStreamLifecycleAction.Request request, ProjectState state) {
97-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
97+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
9898
}
9999
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/options/action/TransportGetDataStreamOptionsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,6 @@ protected void localClusterStateOperation(
121121

122122
@Override
123123
protected ClusterBlockException checkBlock(GetDataStreamOptionsAction.Request request, ProjectState state) {
124-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_READ);
124+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_READ);
125125
}
126126
}

modules/data-streams/src/main/java/org/elasticsearch/datastreams/options/action/TransportPutDataStreamOptionsAction.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,6 @@ protected void masterOperation(
9393

9494
@Override
9595
protected ClusterBlockException checkBlock(PutDataStreamOptionsAction.Request request, ProjectState state) {
96-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_WRITE);
96+
return state.blocks().globalBlockedException(state.projectId(), ClusterBlockLevel.METADATA_WRITE);
9797
}
9898
}

server/src/main/java/org/elasticsearch/action/admin/cluster/storedscripts/TransportGetStoredScriptAction.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import org.elasticsearch.cluster.ClusterState;
1616
import org.elasticsearch.cluster.block.ClusterBlockException;
1717
import org.elasticsearch.cluster.block.ClusterBlockLevel;
18+
import org.elasticsearch.cluster.project.ProjectResolver;
1819
import org.elasticsearch.cluster.service.ClusterService;
1920
import org.elasticsearch.common.util.concurrent.EsExecutors;
2021
import org.elasticsearch.injection.guice.Inject;
@@ -25,12 +26,15 @@
2526

2627
public class TransportGetStoredScriptAction extends TransportMasterNodeReadAction<GetStoredScriptRequest, GetStoredScriptResponse> {
2728

29+
private final ProjectResolver projectResolver;
30+
2831
@Inject
2932
public TransportGetStoredScriptAction(
3033
TransportService transportService,
3134
ClusterService clusterService,
3235
ThreadPool threadPool,
33-
ActionFilters actionFilters
36+
ActionFilters actionFilters,
37+
ProjectResolver projectResolver
3438
) {
3539
super(
3640
GetStoredScriptAction.NAME,
@@ -42,6 +46,7 @@ public TransportGetStoredScriptAction(
4246
GetStoredScriptResponse::new,
4347
EsExecutors.DIRECT_EXECUTOR_SERVICE
4448
);
49+
this.projectResolver = projectResolver;
4550
}
4651

4752
@Override
@@ -56,7 +61,7 @@ protected void masterOperation(
5661

5762
@Override
5863
protected ClusterBlockException checkBlock(GetStoredScriptRequest request, ClusterState state) {
59-
return state.blocks().globalBlockedException(ClusterBlockLevel.METADATA_READ);
64+
return state.blocks().globalBlockedException(projectResolver.getProjectId(), ClusterBlockLevel.METADATA_READ);
6065
}
6166

6267
}

0 commit comments

Comments
 (0)