x

Author: forsaken628

src/query/sql/tests/it/optimizer/union_all.rs

@@ -14,6 +14,7 @@
 
 use std::collections::HashMap;
 use std::io::Write;
+use std::sync::Arc;
 
 use databend_common_catalog::BasicColumnStatistics;
 use databend_common_catalog::TableStatistics;
@@ -128,26 +129,22 @@ fn invalid_target_path(
     ))
 }
 
-fn table_statistics(rows: u64) -> TableStatistics {
-    TableStatistics {
-        num_rows: Some(rows),
-        data_size: Some(rows.saturating_mul(8)),
-        data_size_compressed: None,
-        index_size: None,
-        bloom_index_size: None,
-        ngram_index_size: None,
-        inverted_index_size: None,
-        vector_index_size: None,
-        virtual_column_size: None,
-        number_of_blocks: Some(1),
-        number_of_segments: Some(1),
-    }
-}
-
-async fn register_table(ctx: &std::sync::Arc<LiteTableContext>, input: TableInput) -> Result<()> {
+async fn register_table(ctx: &Arc<LiteTableContext>, input: TableInput) -> Result<()> {
     ctx.register_table_sql_with_stats(
         input.ddl,
-        Some(table_statistics(input.rows)),
+        Some(TableStatistics {
+            num_rows: Some(input.rows),
+            data_size: Some(input.rows.saturating_mul(8)),
+            data_size_compressed: None,
+            index_size: None,
+            bloom_index_size: None,
+            ngram_index_size: None,
+            inverted_index_size: None,
+            vector_index_size: None,
+            virtual_column_size: None,
+            number_of_blocks: Some(1),
+            number_of_segments: Some(1),
+        }),
         input.column_stats,
         input.histograms,
     )