refactor: Use branches for tokenizer states for 25% speed gains

inikulin · Jan 7, 2022 · 5d7a780 · 5d7a780
1 parent 71c0d8a
commit 5d7a780
Show file tree

Hide file tree

Showing 3 changed files with 579 additions and 166 deletions.
diff --git a/packages/parse5/lib/extensions/location-info/tokenizer-mixin.js b/packages/parse5/lib/extensions/location-info/tokenizer-mixin.js
@@ -127,9 +127,8 @@ export class LocationInfoTokenizerMixin extends Mixin {
         };
 
         //NOTE: patch initial states for each mode to obtain token start position
-        for (const modeName of Object.keys(Tokenizer.MODE)) {
-            const state = Tokenizer.MODE[modeName];
-
+        const statesToPatch = ['_stateData', '_stateRcdata', '_stateRawtext', '_stateScriptData', '_statePlaintext'];
+        for (const state of statesToPatch) {
             methods[state] = function (cp) {
                 mxn.ctLoc = mxn._getCurrentLocation();
                 orig[state].call(this, cp);