Merge pull request #30878 from ggevay/typ-keys-with-input-keys

Eliminate unnecessary `.typ()` call from `keys_with_input_keys`
MaterializeInc · Jan 6, 2025 · 0498dcc · 0498dcc
2 parents 23c85ef + 55e0c3d
commit 0498dcc
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/src/expr/src/relation.rs b/src/expr/src/relation.rs
@@ -1107,13 +1107,18 @@ impl MirRelationExpr {
                                         {
                                             if first_id == second_id {
                                                 result.extend(
-                                                    inputs[0].typ().keys.drain(..).filter(|key| {
-                                                        key.iter().all(|c| {
-                                                            outputs.get(*c) == Some(c)
-                                                                && base_projection.get(*c)
-                                                                    == Some(c)
+                                                    input_keys
+                                                        .next()
+                                                        .unwrap()
+                                                        .into_iter()
+                                                        .filter(|key| {
+                                                            key.iter().all(|c| {
+                                                                outputs.get(*c) == Some(c)
+                                                                    && base_projection.get(*c)
+                                                                        == Some(c)
+                                                            })
                                                         })
-                                                    }),
+                                                        .cloned(),
                                                 );
                                             }
                                         }