|
15 | 15 | import org.elasticsearch.cluster.metadata.ProjectId; |
16 | 16 | import org.elasticsearch.cluster.metadata.ProjectMetadata; |
17 | 17 | import org.elasticsearch.cluster.project.ProjectStateRegistry; |
18 | | -import org.elasticsearch.common.settings.Settings; |
19 | 18 | import org.elasticsearch.gateway.GatewayService; |
20 | 19 | import org.elasticsearch.reservedstate.ReservedProjectStateHandler; |
21 | 20 | import org.elasticsearch.reservedstate.TransformState; |
@@ -73,14 +72,14 @@ protected ClusterState execute(ClusterState currentState) { |
73 | 72 | } |
74 | 73 |
|
75 | 74 | ClusterState updatedClusterState = result.v1(); |
76 | | - Settings updatedSettings = ProjectStateRegistry.getProjectSettings(projectId, updatedClusterState); |
77 | | - ProjectMetadata updatedProject = updatedClusterState.getMetadata().getProject(projectId); |
| 75 | + ProjectStateRegistry updatedProjectStateRegistry = updatedClusterState.custom( |
| 76 | + ProjectStateRegistry.TYPE, |
| 77 | + ProjectStateRegistry.EMPTY |
| 78 | + ); |
| 79 | + ProjectMetadata updatedProjectMetadata = updatedClusterState.getMetadata().getProject(projectId); |
78 | 80 | return ClusterState.builder(currentState) |
79 | | - .putCustom( |
80 | | - ProjectStateRegistry.TYPE, |
81 | | - ProjectStateRegistry.builder(currentState).putProjectSettings(projectId, updatedSettings).build() |
82 | | - ) |
83 | | - .putProjectMetadata(ProjectMetadata.builder(updatedProject).put(result.v2())) |
| 81 | + .putCustom(ProjectStateRegistry.TYPE, ProjectStateRegistry.builder(updatedProjectStateRegistry).build()) |
| 82 | + .putProjectMetadata(ProjectMetadata.builder(updatedProjectMetadata).put(result.v2())) |
84 | 83 | .build(); |
85 | 84 | } |
86 | 85 | } |
0 commit comments