fix: InplaceAggregate for boolean columns (#127)

cube-js · Dec 1, 2023 · 2dfe4e1 · 2dfe4e1
1 parent 5c5924c
commit 2dfe4e1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/datafusion/src/physical_plan/filter.rs b/datafusion/src/physical_plan/filter.rs
@@ -38,7 +38,7 @@ use async_trait::async_trait;
 
 use crate::logical_plan::Operator;
 use crate::physical_plan::expressions::{
-    BinaryExpr, CastExpr, Column, Literal, TryCastExpr,
+    BinaryExpr, CastExpr, Column, Literal, NotExpr, TryCastExpr,
 };
 use futures::stream::{Stream, StreamExt};
 
@@ -198,6 +198,13 @@ fn extract_single_value_columns_impl<'a>(
             }
             _ => {}
         }
+    } else if predicate.is::<Column>() {
+        out.push(predicate.downcast_ref::<Column>().unwrap());
+    } else if predicate.is::<NotExpr>() {
+        let inner = predicate.downcast_ref::<NotExpr>().unwrap().arg();
+        if inner.as_any().is::<Column>() {
+            out.push(inner.as_any().downcast_ref::<Column>().unwrap());
+        }
     }
 }