Bring back transforming = ANY / <> ANY into IN / NOT IN

PeerDB-io · Nov 6, 2023 · e3d7b2d · e3d7b2d
1 parent f838b14
commit e3d7b2d
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/nexus/peer-bigquery/src/ast.rs b/nexus/peer-bigquery/src/ast.rs
@@ -239,6 +239,27 @@ impl BigqueryAst {
             ControlFlow::<()>::Continue(())
         });
 
+        // flatten ANY to IN operation overall.
+        visit_expressions_mut(query, |node| {
+            if let Expr::AnyOp { left, compare_op, right } = node {
+                if matches!(compare_op, BinaryOperator::Eq | BinaryOperator::NotEq) {
+                    let negated = matches!(compare_op, BinaryOperator::NotEq);
+                    let list = self
+                        .flatten_expr_to_in_list(right)
+                        .expect("failed to flatten");
+                    *node = Expr::InList {
+                        expr: left.clone(),
+                        list,
+                        negated,
+                    };
+                }
+            }
+
+            ControlFlow::<()>::Continue(())
+        });
+
+
+
         Ok(())
     }