fix(cubesql): Allow any aggregation for number measure as it can be a wildcard

Author: paveltiunov

rust/cubesql/cubesql/src/compile/mod.rs

@@ -1707,6 +1707,28 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_select_number() {
+        let query_plan = convert_select_to_query_plan(
+            "SELECT MEASURE(someNumber) as s1, SUM(someNumber) as s2, MIN(someNumber) as s3, MAX(someNumber) as s4, COUNT(someNumber) as s5 FROM NumberCube".to_string(),
+            DatabaseProtocol::PostgreSQL).await;
+
+        let logical_plan = query_plan.as_logical_plan();
+        assert_eq!(
+            logical_plan.find_cube_scan().request,
+            V1LoadRequestQuery {
+                measures: Some(vec!["NumberCube.someNumber".to_string(),]),
+                segments: Some(vec![]),
+                dimensions: Some(vec![]),
+                time_dimensions: None,
+                order: None,
+                limit: None,
+                offset: None,
+                filters: None
+            }
+        );
+    }
+
     #[tokio::test]
     async fn test_select_null_if_measure_diff() {
         let query_plan = convert_select_to_query_plan(
@@ -3748,10 +3770,6 @@ ORDER BY \"COUNT(count)\" DESC"
                 "SELECT AVG(maxPrice) FROM KibanaSampleDataEcommerce".to_string(),
                 CompilationError::user("Error during rewrite: Measure aggregation type doesn't match. The aggregation type for 'maxPrice' is 'MAX()' but 'AVG()' was provided. Please check logs for additional information.".to_string()),
             ),
-            (
-                "SELECT AVG(someNumber) FROM NumberCube".to_string(),
-                CompilationError::user("Error during rewrite: Measure aggregation type doesn't match. The aggregation type for 'someNumber' is 'MEASURE()' but 'AVG()' was provided. Please check logs for additional information.".to_string()),
-            ),
         ];
 
         for (input_query, expected_error) in variants.iter() {

rust/cubesql/cubesql/src/transport/ext.rs

@@ -25,6 +25,7 @@ impl V1CubeMetaMeasureExt for V1CubeMetaMeasure {
 
                 agg_type.eq(&"countDistinct".to_string())
                     || agg_type.eq(&"countDistinctApprox".to_string())
+                    || agg_type.eq(&"number".to_string())
             } else if expect_agg_type.eq(&"sum".to_string()) {
                 let agg_type = self.agg_type.as_ref().unwrap();
 
@@ -42,7 +43,9 @@ impl V1CubeMetaMeasureExt for V1CubeMetaMeasure {
                     || agg_type.eq(&"boolean".to_string())
                     || agg_type.eq(expect_agg_type)
             } else {
-                self.agg_type.as_ref().unwrap().eq(expect_agg_type)
+                let agg_type = self.agg_type.as_ref().unwrap();
+
+                agg_type.eq(&"number".to_string()) || agg_type.eq(expect_agg_type)
             }
         } else {
             false