Feature(compiler): Support break label in DoStatement

ovr · ovr · commit 2b9e868ae31e · 2018-12-08T01:01:55.000+09:00
diff --git a/src/backend/llvm/code-generation/do-statement.ts b/src/backend/llvm/code-generation/do-statement.ts
@@ -16,11 +16,15 @@ export class DoStatementGenerator implements NodeGenerateInterface<ts.DoStatemen
         const next = llvm.BasicBlock.create(ctx.llvmContext);
         ctx.scope.enclosureFunction.llvmFunction.addBasicBlock(next);
 
+        ctx.scope.breakBlock = next;
+
         builder.createBr(positiveBlock);
         builder.setInsertionPoint(positiveBlock);
 
         passStatement(<any>node.statement, ctx, builder);
 
+        ctx.scope.continueBlock = null;
+
         builder.createBr(conditionBlock);
         builder.setInsertionPoint(conditionBlock);
 
diff --git a/tests/snapshots/math/do.ts b/tests/snapshots/math/do.ts
@@ -12,6 +12,23 @@
         return total;
     }
 
+    function simpleBreak(): number {
+        let total: number = 0;
+        let i: number = 0;
+
+        do {
+            total = total + i;
+            i += 1;
+
+            if (i > 100) {
+                break;
+            }
+        } while (true);
+
+        return total;
+    }
+
     calculateTotalFromRange(1, 100);
     calculateTotalFromRange(50, 100);
+    simpleBreak();
 }
