Fix comments

Signed-off-by: David Zane <davizane@amazon.com>

Author: dzane17

server/src/main/java/org/opensearch/action/search/SearchRequest.java

@@ -649,13 +649,6 @@ public int getMaxConcurrentShardRequests() {
         return maxConcurrentShardRequests == 0 ? 5 : maxConcurrentShardRequests;
     }
 
-    /**
-     * Returns the raw internal value of maxConcurrentShardRequests without applying the default.
-     */
-    public int getMaxConcurrentShardRequestsRaw() {
-        return maxConcurrentShardRequests;
-    }
-
     /**
      * Sets the number of shard requests that should be executed concurrently on a single node. This value should be used as a
      * protection mechanism to reduce the number of shard requests fired per high level search request. Searches that hit the entire

server/src/main/java/org/opensearch/cluster/metadata/WorkloadGroup.java

@@ -25,7 +25,6 @@
 
 import java.io.IOException;
 import java.util.HashMap;
-import java.util.Locale;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;

server/src/main/java/org/opensearch/node/Node.java

@@ -1140,8 +1140,7 @@ protected Node(final Environment initialEnvironment, Collection<PluginInfo> clas
 
             final WorkloadGroupRequestOperationListener workloadGroupRequestOperationListener = new WorkloadGroupRequestOperationListener(
                 workloadGroupService,
-                threadPool,
-                clusterService
+                threadPool
             );
 
             // register all standard SearchRequestOperationsCompositeListenerFactory to the SearchRequestOperationsCompositeListenerFactory

server/src/main/java/org/opensearch/wlm/WorkloadGroupSearchSettings.java

@@ -56,7 +56,7 @@ public String getSettingName() {
         void validate(String value) {
             String error = validator.apply(value);
             if (error != null) {
-                throw new IllegalArgumentException("Invalid value for " + settingName + ": " + error);
+                throw new IllegalArgumentException("Invalid value '" + value + "' for " + settingName + ": " + error);
             }
         }
 

server/src/main/java/org/opensearch/wlm/WorkloadGroupService.java

@@ -325,6 +325,15 @@ public Set<WorkloadGroup> getActiveWorkloadGroups() {
         return activeWorkloadGroups;
     }
 
+    /**
+     * Returns the workload group with the given ID, or null if not found.
+     * @param workloadGroupId the workload group identifier
+     * @return the WorkloadGroup or null
+     */
+    public WorkloadGroup getWorkloadGroupById(String workloadGroupId) {
+        return clusterService.state().metadata().workloadGroups().get(workloadGroupId);
+    }
+
     public Set<WorkloadGroup> getDeletedWorkloadGroups() {
         return deletedWorkloadGroups;
     }

server/src/main/java/org/opensearch/wlm/listeners/WorkloadGroupRequestOperationListener.java

@@ -15,8 +15,6 @@
 import org.opensearch.action.search.SearchRequestContext;
 import org.opensearch.action.search.SearchRequestOperationsListener;
 import org.opensearch.cluster.metadata.WorkloadGroup;
-import org.opensearch.cluster.metadata.WorkloadGroupMetadata;
-import org.opensearch.cluster.service.ClusterService;
 import org.opensearch.threadpool.ThreadPool;
 import org.opensearch.wlm.WorkloadGroupSearchSettings;
 import org.opensearch.wlm.WorkloadGroupService;
@@ -32,16 +30,10 @@ public class WorkloadGroupRequestOperationListener extends SearchRequestOperatio
     private static final Logger logger = LogManager.getLogger(WorkloadGroupRequestOperationListener.class);
     private final WorkloadGroupService workloadGroupService;
     private final ThreadPool threadPool;
-    private final ClusterService clusterService;
 
-    public WorkloadGroupRequestOperationListener(
-        WorkloadGroupService workloadGroupService,
-        ThreadPool threadPool,
-        ClusterService clusterService
-    ) {
+    public WorkloadGroupRequestOperationListener(WorkloadGroupService workloadGroupService, ThreadPool threadPool) {
         this.workloadGroupService = workloadGroupService;
         this.threadPool = threadPool;
-        this.clusterService = clusterService;
     }
 
     /**
@@ -69,18 +61,13 @@ protected void onRequestFailure(SearchPhaseContext context, SearchRequestContext
      * @param searchRequest the search request to modify
      */
     private void applyWorkloadGroupSearchSettings(String workloadGroupId, SearchRequest searchRequest) {
-        // Skip if no workload group ID (default group is added later)
         if (workloadGroupId == null) {
+            // Return if request contains no WLM group assignment (default group is added later)
             return;
         }
 
-        WorkloadGroupMetadata metadata = clusterService.state().metadata().custom(WorkloadGroupMetadata.TYPE);
-        if (metadata == null) {
-            return;
-        }
+        WorkloadGroup workloadGroup = workloadGroupService.getWorkloadGroupById(workloadGroupId);
 
-        // Get the workload group object by ID
-        WorkloadGroup workloadGroup = metadata.workloadGroups().get(workloadGroupId);
         if (workloadGroup == null) {
             return;
         }

server/src/test/java/org/opensearch/wlm/listeners/WorkloadGroupRequestOperationListenerTests.java

@@ -13,7 +13,6 @@
 import org.opensearch.cluster.ClusterState;
 import org.opensearch.cluster.metadata.Metadata;
 import org.opensearch.cluster.metadata.WorkloadGroup;
-import org.opensearch.cluster.metadata.WorkloadGroupMetadata;
 import org.opensearch.cluster.service.ClusterService;
 import org.opensearch.common.util.concurrent.ThreadContext;
 import org.opensearch.core.concurrency.OpenSearchRejectedExecutionException;
@@ -69,7 +68,7 @@ public void setUp() throws Exception {
         Metadata mockMetaData = mock(Metadata.class);
         when(mockClusterState.metadata()).thenReturn(mockMetaData);
         workloadGroupService = mock(WorkloadGroupService.class);
-        sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool, mockClusterService);
+        sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool);
         mockSearchRequest = new SearchRequest();
         mockSearchRequestContext = mock(SearchRequestContext.class);
         when(mockSearchRequestContext.getRequest()).thenReturn(mockSearchRequest);
@@ -147,7 +146,7 @@ public void testMultiThreadedValidWorkloadGroupRequestFailures() {
             Collections.emptySet()
         );
 
-        sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool, mockClusterService);
+        sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool);
 
         List<Thread> threads = new ArrayList<>();
         for (int i = 0; i < ITERATIONS; i++) {
@@ -265,7 +264,7 @@ private void assertSuccess(
                 Collections.emptySet(),
                 Collections.emptySet()
             );
-            sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool, mockClusterService);
+            sut = new WorkloadGroupRequestOperationListener(workloadGroupService, testThreadPool);
             sut.onRequestFailure(null, null);
 
             HashSet<String> set = new HashSet<>();
@@ -296,11 +295,7 @@ public void testApplySearchSettings_NullWorkloadGroupId() {
 
     public void testApplySearchSettings_NullMetadata() {
         testThreadPool.getThreadContext().putHeader(WorkloadGroupTask.WORKLOAD_GROUP_ID_HEADER, "test-wg-id");
-        ClusterState state = mock(ClusterState.class);
-        Metadata metadata = mock(Metadata.class);
-        when(mockClusterService.state()).thenReturn(state);
-        when(state.metadata()).thenReturn(metadata);
-        when(metadata.custom(WorkloadGroupMetadata.TYPE)).thenReturn(null);
+        when(workloadGroupService.getWorkloadGroupById("test-wg-id")).thenReturn(null);
 
         sut.onRequestStart(mockSearchRequestContext);
 
@@ -309,7 +304,7 @@ public void testApplySearchSettings_NullMetadata() {
 
     public void testApplySearchSettings_WorkloadGroupNotFound() {
         testThreadPool.getThreadContext().putHeader(WorkloadGroupTask.WORKLOAD_GROUP_ID_HEADER, "non-existent-id");
-        setupWorkloadGroupMetadata(Map.of());
+        when(workloadGroupService.getWorkloadGroupById("non-existent-id")).thenReturn(null);
 
         sut.onRequestStart(mockSearchRequestContext);
 
@@ -321,24 +316,14 @@ public void testApplySearchSettings_PhaseTook() {
 
         String wgId = "test-wg";
         WorkloadGroup wg = createWorkloadGroup(wgId, Map.of("phase_took", "true"));
-        setupWorkloadGroupMetadata(Map.of(wgId, wg));
+        when(workloadGroupService.getWorkloadGroupById(wgId)).thenReturn(wg);
         testThreadPool.getThreadContext().putHeader(WorkloadGroupTask.WORKLOAD_GROUP_ID_HEADER, wgId);
 
         sut.onRequestStart(mockSearchRequestContext);
 
         assertTrue(mockSearchRequest.isPhaseTook());
     }
 
-    private void setupWorkloadGroupMetadata(Map<String, WorkloadGroup> workloadGroups) {
-        ClusterState state = mock(ClusterState.class);
-        Metadata metadata = mock(Metadata.class);
-        WorkloadGroupMetadata wgMetadata = new WorkloadGroupMetadata(workloadGroups);
-        when(mockClusterService.state()).thenReturn(state);
-        when(state.metadata()).thenReturn(metadata);
-        when(metadata.custom(WorkloadGroupMetadata.TYPE)).thenReturn(wgMetadata);
-        when(metadata.workloadGroups()).thenReturn(workloadGroups);
-    }
-
     private WorkloadGroup createWorkloadGroup(String id, Map<String, String> searchSettings) {
         return new WorkloadGroup(
             "test-name",