perf(search): reduce highlight fragments (#11349)

datahub-project · Sep 12, 2024 · 48ef215 · 48ef215
1 parent 2890b6b
commit 48ef215
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/.../com/linkedin/metadata/search/elasticsearch/query/request/AutocompleteRequestHandler.java b/.../com/linkedin/metadata/search/elasticsearch/query/request/AutocompleteRequestHandler.java
@@ -226,10 +226,12 @@ private static BoolQueryBuilder defaultQuery(
 
   // Get HighlightBuilder to highlight the matched field
   private HighlightBuilder getHighlights(@Nullable String field) {
-    HighlightBuilder highlightBuilder = new HighlightBuilder();
-    // Don't set tags to get the original field value
-    highlightBuilder.preTags("");
-    highlightBuilder.postTags("");
+    HighlightBuilder highlightBuilder =
+        new HighlightBuilder()
+            // Don't set tags to get the original field value
+            .preTags("")
+            .postTags("")
+            .numOfFragments(1);
     // Check for each field name and any subfields
     getAutocompleteFields(field)
         .forEach(

diff --git a/...n/java/com/linkedin/metadata/search/elasticsearch/query/request/SearchRequestHandler.java b/...n/java/com/linkedin/metadata/search/elasticsearch/query/request/SearchRequestHandler.java
@@ -366,11 +366,12 @@ public QueryBuilder getQuery(
 
   @VisibleForTesting
   public HighlightBuilder getHighlights() {
-    HighlightBuilder highlightBuilder = new HighlightBuilder();
-
-    // Don't set tags to get the original field value
-    highlightBuilder.preTags("");
-    highlightBuilder.postTags("");
+    HighlightBuilder highlightBuilder =
+        new HighlightBuilder()
+            // Don't set tags to get the original field value
+            .preTags("")
+            .postTags("")
+            .numOfFragments(1);
 
     // Check for each field name and any subfields
     defaultQueryFieldNames.stream()