using SamplingService.getSamplingConfiguration()

Author: masseyke

server/src/main/java/org/elasticsearch/action/admin/indices/sampling/TransportGetSampleAction.java

@@ -63,8 +63,11 @@ public TransportGetSampleAction(
     @Override
     protected Void createActionContext(Task task, Request request) {
         String indexName = request.indices()[0];
-        SamplingMetadata samplingMetadata = projectResolver.getProjectMetadata(clusterService.state()).custom(SamplingMetadata.TYPE);
-        if (samplingMetadata == null || samplingMetadata.getIndexToSamplingConfigMap().get(indexName) == null) {
+        SamplingConfiguration samplingConfiguration = samplingService.getSamplingConfiguration(
+            projectResolver.getProjectMetadata(clusterService.state()),
+            request.indices()[0]
+        );
+        if (samplingConfiguration == null) {
             throw new ResourceNotFoundException("No sampling configuration found for [" + indexName + "]");
         }
         return null;

server/src/main/java/org/elasticsearch/action/admin/indices/sampling/TransportGetSampleStatsAction.java

@@ -59,8 +59,11 @@ public TransportGetSampleStatsAction(
     @Override
     protected Void createActionContext(Task task, GetSampleStatsAction.Request request) {
         String indexName = request.indices()[0];
-        SamplingMetadata samplingMetadata = projectResolver.getProjectMetadata(clusterService.state()).custom(SamplingMetadata.TYPE);
-        if (samplingMetadata == null || samplingMetadata.getIndexToSamplingConfigMap().get(indexName) == null) {
+        SamplingConfiguration samplingConfiguration = samplingService.getSamplingConfiguration(
+            projectResolver.getProjectMetadata(clusterService.state()),
+            request.indices()[0]
+        );
+        if (samplingConfiguration == null) {
             throw new ResourceNotFoundException("No sampling configuration found for [" + indexName + "]");
         }
         return null;