Add SearchResponse proto fields for matchAll

Signed-off-by: Finn Carroll <carrofin@amazon.com>
opensearch-project · Nov 22, 2024 · da16c66 · da16c66
1 parent 853b206
commit da16c66
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);