fix(cubesql): Ignore __user IS NOT NULL filter

Author: MazterQyou

rust/cubesql/cubesql/src/compile/rewrite/rules/filters.rs

@@ -362,6 +362,22 @@ impl RewriteRules for FilterRules {
                 change_user_member("?user"),
                 self.transform_change_user_eq("?column", "?literal", "?user"),
             ),
+            transforming_rewrite(
+                "user-is-not-null-filter",
+                filter_replacer(
+                    is_not_null_expr(column_expr("?column")),
+                    "?alias_to_cube",
+                    "?members",
+                    "?filter_aliases",
+                ),
+                filter_replacer(
+                    literal_bool(true),
+                    "?alias_to_cube",
+                    "?members",
+                    "?filter_aliases",
+                ),
+                self.transform_user_is_not_null("?column"),
+            ),
             transforming_rewrite(
                 "join-field-filter-eq",
                 filter_replacer(
@@ -3338,6 +3354,22 @@ impl FilterRules {
         }
     }
 
+    fn transform_user_is_not_null(
+        &self,
+        column_var: &'static str,
+    ) -> impl Fn(&mut EGraph<LogicalPlanLanguage, LogicalPlanAnalysis>, &mut Subst) -> bool {
+        let column_var = var!(column_var);
+
+        move |egraph, subst| {
+            for column in var_iter!(egraph[subst[column_var]], ColumnExprColumn).cloned() {
+                if column.name.eq_ignore_ascii_case("__user") {
+                    return true;
+                }
+            }
+            false
+        }
+    }
+
     // Transform ?expr IN (?literal) to ?expr = ?literal
     fn transform_filter_in_to_equal(
         &self,

rust/cubesql/cubesql/src/compile/test/test_user_change.rs

@@ -110,6 +110,36 @@ async fn test_change_user_via_in_filter_thoughtspot() {
     assert_eq!(cube_scan.request, expected_request);
 }
 
+#[tokio::test]
+async fn test_change_user_via_filter_powerbi() {
+    init_testing_logger();
+
+    let query_plan = convert_select_to_query_plan(
+        "SELECT COUNT(*) as cnt FROM KibanaSampleDataEcommerce WHERE NOT __user IS NULL AND __user = 'gopher'".to_string(),
+        DatabaseProtocol::PostgreSQL,
+    )
+    .await;
+
+    let cube_scan = query_plan.as_logical_plan().find_cube_scan();
+
+    assert_eq!(cube_scan.options.change_user, Some("gopher".to_string()));
+
+    assert_eq!(
+        cube_scan.request,
+        V1LoadRequestQuery {
+            measures: Some(vec!["KibanaSampleDataEcommerce.count".to_string(),]),
+            segments: Some(vec![]),
+            dimensions: Some(vec![]),
+            time_dimensions: None,
+            order: Some(vec![]),
+            limit: None,
+            offset: None,
+            filters: None,
+            ungrouped: None,
+        }
+    )
+}
+
 #[tokio::test]
 async fn test_change_user_via_filter_and() {
     let query_plan = convert_select_to_query_plan(