Refactored pushStack function

vercel · Mar 8, 2021 · 2eea19e · 2eea19e
1 parent 9e06f82
commit 2eea19e
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/packages/next/build/tracer.ts b/packages/next/build/tracer.ts
@@ -9,20 +9,14 @@ export function stackPush(compiler: any, spanName: string, attrs?: any): any {
   let stack = compilerStacks.get(compiler)
   let span
 
-  if (!stack) {
-    compilerStacks.set(compiler, (stack = []))
+  const fillSpan = () => {
     span = tracer.startSpan(spanName, attrs ? attrs() : undefined)
-  } else {
-    const parent = stack[stack.length - 1]
-    if (parent) {
-      tracer.withSpan(parent, () => {
-        span = tracer.startSpan(spanName, attrs ? attrs() : undefined)
-      })
-    } else {
-      span = tracer.startSpan(spanName, attrs ? attrs() : undefined)
-    }
   }
 
+  !stack && compilerStacks.set(compiler, (stack = []))
+  const parent = stack[stack.length - 1]
+  parent ? tracer.withSpan(parent, fillSpan) : fillSpan()
+
   stack.push(span)
   return span
 }