From adbe59d889ab35b5bf9ce6a930df8b4620269ea5 Mon Sep 17 00:00:00 2001
From: IceS2 <pjt1991@gmail.com>
Date: Thu, 12 Sep 2024 16:50:04 +0200
Subject: [PATCH] Hotfix to the Term Aggregation size on Data Insights (#17822)

---
 .../service/jdbi3/DataInsightSystemChartRepository.java         | 2 +-
 .../ElasticSearchLineChartAggregator.java                       | 2 +-
 .../dataInsightAggregator/OpenSearchLineChartAggregator.java    | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DataInsightSystemChartRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DataInsightSystemChartRepository.java
index e2d826c653d..5ff383b7c18 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DataInsightSystemChartRepository.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DataInsightSystemChartRepository.java
@@ -21,7 +21,7 @@ public class DataInsightSystemChartRepository extends EntityRepository<DataInsig
   public static final String DI_SEARCH_INDEX = "di-data-assets-*";
 
   public static final String FORMULA_FUNC_REGEX =
-      "\\b(count|sum|min|max|avg)+\\((k='([^']*)')?,?\\s*(q='([^']*)')?\\)?";
+      "\\b(count|sum|min|max|avg|unique)+\\((k='([^']*)')?,?\\s*(q='([^']*)')?\\)?";
 
   public static final String NUMERIC_VALIDATION_REGEX = "[\\d\\.+-\\/\\*\\(\\)\s]+";
 
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/search/elasticsearch/dataInsightAggregators/ElasticSearchLineChartAggregator.java b/openmetadata-service/src/main/java/org/openmetadata/service/search/elasticsearch/dataInsightAggregators/ElasticSearchLineChartAggregator.java
index 2d933a90f22..811e2f36058 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/search/elasticsearch/dataInsightAggregators/ElasticSearchLineChartAggregator.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/search/elasticsearch/dataInsightAggregators/ElasticSearchLineChartAggregator.java
@@ -60,7 +60,7 @@ public class ElasticSearchLineChartAggregator
         excludeArr = lineChart.getExcludeGroups().toArray(new String[0]);
       }
       TermsAggregationBuilder termsAggregationBuilder =
-          AggregationBuilders.terms("0").field(lineChart.getGroupBy()).size(20);
+          AggregationBuilders.terms("0").field(lineChart.getGroupBy()).size(1000);
       termsAggregationBuilder.subAggregation(dateHistogramAggregationBuilder);
       if (includeArr != null || excludeArr != null) {
         IncludeExclude includeExclude = new IncludeExclude(includeArr, excludeArr);
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/search/opensearch/dataInsightAggregator/OpenSearchLineChartAggregator.java b/openmetadata-service/src/main/java/org/openmetadata/service/search/opensearch/dataInsightAggregator/OpenSearchLineChartAggregator.java
index c9f71ea0150..fc21c3a8828 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/search/opensearch/dataInsightAggregator/OpenSearchLineChartAggregator.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/search/opensearch/dataInsightAggregator/OpenSearchLineChartAggregator.java
@@ -59,7 +59,7 @@ public class OpenSearchLineChartAggregator implements OpenSearchDynamicChartAggr
         excludeArr = lineChart.getExcludeGroups().toArray(new String[0]);
       }
       TermsAggregationBuilder termsAggregationBuilder =
-          AggregationBuilders.terms("0").field(lineChart.getGroupBy()).size(20);
+          AggregationBuilders.terms("0").field(lineChart.getGroupBy()).size(1000);
       termsAggregationBuilder.subAggregation(dateHistogramAggregationBuilder);
       if (includeArr != null || excludeArr != null) {
         IncludeExclude includeExclude = new IncludeExclude(includeArr, excludeArr);