Request level coordinator slow logs

Signed-off-by: David Zane <[email protected]>
dzane17 · Sep 29, 2023 · 9cbc018 · 9cbc018
1 parent 7dc6683
commit 9cbc018
Show file tree

Hide file tree

Showing 6 changed files with 490 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/action/search/AbstractSearchAsyncAction.java b/server/src/main/java/org/opensearch/action/search/AbstractSearchAsyncAction.java
@@ -440,6 +440,10 @@ private void onPhaseStart(SearchPhase phase) {
         this.searchRequestOperationsListener.ifPresent(searchRequestOperations -> { searchRequestOperations.onPhaseStart(this); });
     }
 
+    private void onRequestEnd() {
+        this.searchRequestOperationsListener.ifPresent(searchRequestOperations -> { searchRequestOperations.onRequestEnd(this); });
+    }
+
     private void executePhase(SearchPhase phase) {
         try {
             onPhaseStart(phase);
@@ -697,6 +701,7 @@ public void sendSearchResponse(InternalSearchResponse internalSearchResponse, At
             listener.onResponse(buildSearchResponse(internalSearchResponse, failures, scrollId, searchContextId));
         }
         onPhaseEnd();
+        onRequestEnd();
         setCurrentPhase(null);
     }
 

diff --git a/server/src/main/java/org/opensearch/action/search/SearchRequestOperationsListener.java b/server/src/main/java/org/opensearch/action/search/SearchRequestOperationsListener.java
@@ -26,6 +26,10 @@ public interface SearchRequestOperationsListener {
 
     void onPhaseFailure(SearchPhaseContext context);
 
+    default void onRequestStart() {};
+
+    default void onRequestEnd(SearchPhaseContext context) {};
+
     /**
      * Holder of Composite Listeners
      *
@@ -73,5 +77,27 @@ public void onPhaseFailure(SearchPhaseContext context) {
                 }
             }
         }
+
+        @Override
+        public void onRequestStart() {
+            for (SearchRequestOperationsListener listener : listeners) {
+                try {
+                    listener.onRequestStart();
+                } catch (Exception e) {
+                    logger.warn(() -> new ParameterizedMessage("onRequestStart listener [{}] failed", listener), e);
+                }
+            }
+        }
+
+        @Override
+        public void onRequestEnd(SearchPhaseContext context) {
+            for (SearchRequestOperationsListener listener : listeners) {
+                try {
+                    listener.onRequestEnd(context);
+                } catch (Exception e) {
+                    logger.warn(() -> new ParameterizedMessage("onRequestEnd listener [{}] failed", listener), e);
+                }
+            }
+        }
     }
 }