Tree walking: Add missing NodeMut NullTest and TypeCast cases

pganalyze · May 31, 2023 · cdeae7c · cdeae7c
1 parent 9479dbd
commit cdeae7c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/node_enum.rs b/src/node_enum.rs
@@ -773,6 +773,14 @@ impl NodeEnum {
                         }
                     });
                 }
+                NodeMut::NullTest(e) => {
+                    let e = e.as_mut().unwrap();
+                    if let Some(n) = e.arg.as_mut() {
+                        if let Some(n) = n.node.as_mut() {
+                            iter.push((n.to_mut(), depth, context));
+                        }
+                    }
+                }
                 NodeMut::ResTarget(t) => {
                     let t = t.as_mut().unwrap();
                     if let Some(n) = t.val.as_mut() {
@@ -831,6 +839,14 @@ impl NodeEnum {
                         }
                     }
                 }
+                NodeMut::TypeCast(t) => {
+                    let t = t.as_mut().unwrap();
+                    if let Some(n) = t.arg.as_mut() {
+                        if let Some(n) = n.node.as_mut() {
+                            iter.push((n.to_mut(), depth, context));
+                        }
+                    }
+                }
                 //
                 // from-clause items
                 //