Loop instead of recursion

gluon-lang · Aug 31, 2019 · 045f2ed · 045f2ed
1 parent 7bf25e8
commit 045f2ed
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -677,12 +677,15 @@ impl Compiler {
         let original_expr = mem::replace(expr, prelude_expr);
 
         // Replace the 0 in the prelude with the actual expression
-        fn assign_last_body(l: &mut SpannedExpr<Symbol>, original_expr: SpannedExpr<Symbol>) {
-            match l.value {
-                ast::Expr::LetBindings(_, ref mut e) => {
-                    assign_last_body(e, original_expr);
+        fn assign_last_body(mut l: &mut SpannedExpr<Symbol>, original_expr: SpannedExpr<Symbol>) {
+            loop {
+                match l.value {
+                    ast::Expr::LetBindings(_, ref mut e) => l = e,
+                    _ => {
+                        *l = original_expr;
+                        return;
+                    }
                 }
-                _ => *l = original_expr,
             }
         }
         assign_last_body(expr, original_expr);