[Enhancement] modify histogram_statistic cache expire-update logical

Signed-off-by: stephen <[email protected]>

Author: stephen-shelby

fe/fe-core/src/main/java/com/starrocks/sql/optimizer/statistics/CachedStatisticStorage.java

@@ -161,6 +161,36 @@ public void refreshColumnStatistics(Table table, List<String> columns, boolean i
         }
     }
 
+    @Override
+    public void refreshHistogramStatistics(Table table, List<String> columns, boolean isSync) {
+        Preconditions.checkState(table != null);
+
+        if (StatisticUtils.statisticTableBlackListCheck(table.getId()) ||
+                !StatisticUtils.checkStatisticTableStateNormal()) {
+            return;
+        }
+
+        List<ColumnStatsCacheKey> cacheKeys = new ArrayList<>();
+        long tableId = table.getId();
+        for (String column : columns) {
+            cacheKeys.add(new ColumnStatsCacheKey(tableId, column));
+        }
+
+        try {
+            ColumnHistogramStatsCacheLoader loader = new ColumnHistogramStatsCacheLoader();
+            CompletableFuture<Map<ColumnStatsCacheKey, Optional<Histogram>>> future =
+                    loader.asyncLoadAll(cacheKeys, statsCacheRefresherExecutor);
+            if (isSync) {
+                Map<ColumnStatsCacheKey, Optional<Histogram>> result = future.get();
+                histogramCache.synchronous().putAll(result);
+            } else {
+                future.whenComplete((res, e) -> histogramCache.synchronous().putAll(res));
+            }
+        } catch (Exception e) {
+            LOG.warn("Failed to refresh histogram", e);
+        }
+    }
+
     @Override
     public List<ConnectorTableColumnStats> getConnectorTableStatistics(Table table, List<String> columns) {
         Preconditions.checkState(table != null);

fe/fe-core/src/main/java/com/starrocks/sql/optimizer/statistics/StatisticStorage.java

@@ -41,6 +41,9 @@ default void refreshColumnStatistics(Table table, List<String> columns, boolean
     default void refreshMultiColumnStatistics(Long tableId) {
     }
 
+    default void refreshHistogramStatistics(Table table, List<String> columns, boolean isSync) {
+    }
+
     /**
      * Overwrite the statistics of `targetPartition` with `sourcePartition`
      */

fe/fe-core/src/main/java/com/starrocks/statistic/AnalyzeMgr.java

@@ -381,12 +381,7 @@ public void refreshHistogramStatisticsCache(Long dbId, Long tableId, List<String
             return;
         }
 
-        GlobalStateMgr.getCurrentState().getStatisticStorage().expireHistogramStatistics(table.getId(), columns);
-        if (async) {
-            GlobalStateMgr.getCurrentState().getStatisticStorage().getHistogramStatistics(table, columns);
-        } else {
-            GlobalStateMgr.getCurrentState().getStatisticStorage().getHistogramStatisticsSync(table, columns);
-        }
+        GlobalStateMgr.getCurrentState().getStatisticStorage().refreshHistogramStatistics(table, columns, !async);
     }
 
     public void replayRemoveHistogramStatsMeta(HistogramStatsMeta histogramStatsMeta) {