Skip profile queries (#298) (#304) (#319)

(cherry picked from commit dc1c3f2)





(cherry picked from commit 38cb816)

Signed-off-by: Du Tran <quangdutran809@gmail.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Du Tran <du.tran@array.world>

Author: 3 people

src/main/java/org/opensearch/plugin/insights/core/listener/QueryInsightsListener.java

@@ -221,6 +221,11 @@ public void onRequestFailure(final SearchPhaseContext context, final SearchReque
     }
 
     private void constructSearchQueryRecord(final SearchPhaseContext context, final SearchRequestContext searchRequestContext) {
+        // Skip profile queries
+        if (searchRequestContext.getRequest().source().profile()) {
+            return;
+        }
+
         SearchTask searchTask = context.getTask();
         List<TaskResourceInfo> tasksResourceUsages = searchRequestContext.getPhaseResourceUsage();
         tasksResourceUsages.add(

src/test/java/org/opensearch/plugin/insights/core/listener/QueryInsightsListenerTests.java

@@ -79,7 +79,7 @@ public void setup() {
         ClusterServiceUtils.setState(clusterService, state);
         when(queryInsightsService.isCollectionEnabled(MetricType.LATENCY)).thenReturn(true);
         when(queryInsightsService.getTopQueriesService(MetricType.LATENCY)).thenReturn(topQueriesService);
-
+        when(searchRequestContext.getRequest()).thenReturn(searchRequest);
         ThreadContext threadContext = new ThreadContext(Settings.EMPTY);
         threadPool.getThreadContext().setHeaders(new Tuple<>(Collections.singletonMap(Task.X_OPAQUE_ID, "userLabel"), new HashMap<>()));
     }
@@ -345,4 +345,12 @@ private QueryInsightsListener testFeatureEnableDisableSetup(
 
         return queryInsightsListener;
     }
+
+    public void testSkipProfileQuery() {
+        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder().profile(true);
+        when(searchRequest.source()).thenReturn(searchSourceBuilder);
+        QueryInsightsListener queryInsightsListener = new QueryInsightsListener(clusterService, queryInsightsService);
+        queryInsightsListener.onRequestEnd(searchPhaseContext, searchRequestContext);
+        verify(queryInsightsService, times(0)).addRecord(any());
+    }
 }