Add SearchResponse proto fields for matchAll

Signed-off-by: Finn Carroll <[email protected]>
opensearch-project · Nov 23, 2024 · 1937f5f · 1937f5f
1 parent 2eb3cf3
commit 1937f5f
Show file tree

Hide file tree

Showing 2 changed files with 335 additions and 11 deletions.
diff --git a/server/src/main/java/org/opensearch/action/search/SearchResponse.java b/server/src/main/java/org/opensearch/action/search/SearchResponse.java
@@ -695,6 +695,14 @@ private PhaseTook(StreamInput in) throws IOException {
             this(in.readMap(StreamInput::readString, StreamInput::readLong));
         }
 
+        /**
+         * @param phaseName search phase
+         * @return phase took in ms, -1 if no record
+         */
+        public long getPhaseTook(SearchPhaseName phaseName) {
+            return phaseTookMap.getOrDefault(phaseName.getName(), -1L);
+        }
+
         @Override
         public void writeTo(StreamOutput out) throws IOException {
             out.writeMap(phaseTookMap, StreamOutput::writeString, StreamOutput::writeLong);