From bf461b3992b99648fa82619b3a25f3902d6432c5 Mon Sep 17 00:00:00 2001
From: overlookmotel <theoverlookmotel@gmail.com>
Date: Sun, 17 Nov 2024 01:19:34 +0000
Subject: [PATCH] perf(transformer/arrow-functions): move arguments transform
 checks to aid inlining

---
 .../src/common/arrow_function_converter.rs    | 29 +++++++++++++------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/crates/oxc_transformer/src/common/arrow_function_converter.rs b/crates/oxc_transformer/src/common/arrow_function_converter.rs
index e1b80ff69c4c5e..46cfe3d7a6a4e4 100644
--- a/crates/oxc_transformer/src/common/arrow_function_converter.rs
+++ b/crates/oxc_transformer/src/common/arrow_function_converter.rs
@@ -368,20 +368,34 @@ impl<'a> Traverse<'a> for ArrowFunctionConverter<'a> {
         }
     }
 
+    // `#[inline]` because this is a hot path
+    #[inline]
     fn enter_identifier_reference(
         &mut self,
         ident: &mut IdentifierReference<'a>,
         ctx: &mut TraverseCtx<'a>,
     ) {
-        self.transform_identifier_reference_for_arguments(ident, ctx);
+        // Do this check here rather than in `transform_identifier_reference_for_arguments`
+        // so that the fast path for "no transform required" doesn't require a function call
+        let arguments_needs_transform = *self.arguments_needs_transform_stack.last();
+        if arguments_needs_transform {
+            self.transform_identifier_reference_for_arguments(ident, ctx);
+        }
     }
 
+    // `#[inline]` because this is a hot path
+    #[inline]
     fn enter_binding_identifier(
         &mut self,
         ident: &mut BindingIdentifier<'a>,
         ctx: &mut TraverseCtx<'a>,
     ) {
-        self.transform_binding_identifier_for_arguments(ident, ctx);
+        // Do this check here rather than in `transform_binding_identifier_for_arguments`
+        // so that the fast path for "no transform required" doesn't require a function call
+        let arguments_needs_transform = *self.arguments_needs_transform_stack.last();
+        if arguments_needs_transform {
+            self.transform_binding_identifier_for_arguments(ident, ctx);
+        }
     }
 }
 
@@ -907,8 +921,7 @@ impl<'a> ArrowFunctionConverter<'a> {
         ident: &mut IdentifierReference<'a>,
         ctx: &mut TraverseCtx<'a>,
     ) {
-        let arguments_needs_transform = *self.arguments_needs_transform_stack.last();
-        if !arguments_needs_transform || &ident.name != "arguments" {
+        if &ident.name != "arguments" {
             return;
         }
 
@@ -952,11 +965,9 @@ impl<'a> ArrowFunctionConverter<'a> {
         ident: &mut BindingIdentifier<'a>,
         ctx: &mut TraverseCtx<'a>,
     ) {
-        let arguments_needs_transform = *self.arguments_needs_transform_stack.last();
-        if !arguments_needs_transform
-            || ctx.current_scope_flags().is_strict_mode() // `arguments` is not allowed to be defined in strict mode
-            || &ident.name != "arguments"
-        {
+        // `arguments` is not allowed to be defined in strict mode.
+        // Check if strict mode first to avoid the more expensive string comparison check if possible.
+        if ctx.current_scope_flags().is_strict_mode() || &ident.name != "arguments" {
             return;
         }