perf: cache rbtree end iterator

eslint · Aug 31, 2022 · 946f3d8 · 946f3d8
1 parent 4093492
commit 946f3d8
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/lib/rules/indent.js b/lib/rules/indent.js
@@ -136,6 +136,7 @@ class BinarySearchTree {
      */
     constructor() {
         this._rbTree = new OrderedMap();
+        this._rbTreeEnd = this._rbTree.end();
     }
 
     /**
@@ -172,10 +173,19 @@ class BinarySearchTree {
             return;
         }
         const iterator = this._rbTree.lowerBound(start);
-        const endIterator = this._rbTree.end();
 
-        while (!iterator.equals(endIterator) && iterator.pointer[0] < end) {
-            this._rbTree.eraseElementByIterator(iterator);
+        if (iterator.equals(this._rbTreeEnd)) {
+            return;
+        }
+
+        if (end > this._rbTree.back()[0]) {
+            while (!iterator.equals(this._rbTreeEnd)) {
+                this._rbTree.eraseElementByIterator(iterator);
+            }
+        } else {
+            while (iterator.pointer[0] < end) {
+                this._rbTree.eraseElementByIterator(iterator);
+            }
         }
     }
 }