Saved space by combining conditionals within for loop in both files

Author: Victoria Wei

src/demo-app/tree/dynamic-tree-demo/dynamic-database.ts

@@ -100,22 +100,16 @@ export class DynamicDataSource {
     }
     node.isLoading = true;
 
-    let count = 0;
-
     setTimeout(() => {
       if (expand) {
         const nodes = children.map(name =>
           new DynamicFlatNode(name, node.level + 1, this.database.isExpandable(name)));
         this.data.splice(index + 1, 0, ...nodes);
       } else {
-        for (let i = index + 1; i < this.data.length; i++) {
-          if (this.data[i].level > node.level) {
-              count++;
-          } else {
-            break;
-          }
-        }
-        this.data.splice(index + 1, count);
+          let count = 0;
+          for (let i = index + 1; i < this.data.length
+          && this.data[i].level > node.level; i++, count++) {}
+          this.data.splice(index + 1, count);
       }
 
       // notify the change

src/material-examples/tree-dynamic/tree-dynamic-example.ts

@@ -94,21 +94,15 @@ export class DynamicDataSource {
 
     node.isLoading = true;
 
-    let count = 0;
-
     setTimeout(() => {
       if (expand) {
         const nodes = children.map(name =>
           new DynamicFlatNode(name, node.level + 1, this.database.isExpandable(name)));
         this.data.splice(index + 1, 0, ...nodes);
       } else {
-        for (let i = index + 1; i < this.data.length; i++) {
-          if (this.data[i].level > node.level) {
-            count++;
-          } else {
-            break;
-          }
-        }
+        let count = 0;
+        for (let i = index + 1; i < this.data.length
+            && this.data[i].level > node.level; i++, count++) {}
         this.data.splice(index + 1, count);
       }