feat: show transform time

vitest-dev · Aug 16, 2022 · b54a13e · b54a13e
1 parent c663f39
commit b54a13e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/packages/vite-node/src/server.ts b/packages/vite-node/src/server.ts
@@ -18,6 +18,7 @@ export class ViteNodeServer {
   private transformPromiseMap = new Map<string, Promise<TransformResult | null | undefined>>()
 
   fetchCache = new Map<string, {
+    duration?: number
     timestamp: number
     result: FetchResult
   }>()
@@ -125,16 +126,20 @@ export class ViteNodeServer {
       return cache.result
 
     const externalize = await this.shouldExternalize(filePath)
+    let duration: number | undefined
     if (externalize) {
       result = { externalize }
       this.debugger?.recordExternalize(id, externalize)
     }
     else {
+      const start = performance.now()
       const r = await this._transformRequest(id)
+      duration = performance.now() - start
       result = { code: r?.code, map: r?.map as unknown as RawSourceMap }
     }
 
     this.fetchCache.set(filePath, {
+      duration,
       timestamp,
       result,
     })

diff --git a/packages/vitest/src/node/reporters/base.ts b/packages/vitest/src/node/reporters/base.ts
@@ -218,6 +218,7 @@ export abstract class BaseReporter implements Reporter {
     const collectTime = files.reduce((acc, test) => acc + Math.max(0, test.collectDuration || 0), 0)
     const setupTime = files.reduce((acc, test) => acc + Math.max(0, test.setupDuration || 0), 0)
     const testsTime = files.reduce((acc, test) => acc + Math.max(0, test.result?.duration || 0), 0)
+    const transformTime = Array.from(this.ctx.vitenode.fetchCache.values()).reduce((a, b) => a + (b?.duration || 0), 0)
     const threadTime = collectTime + testsTime + setupTime
 
     const padTitle = (str: string) => c.dim(`${str.padStart(10)} `)
@@ -227,6 +228,14 @@ export abstract class BaseReporter implements Reporter {
       return `${Math.round(time)}ms`
     }
 
+    // show top 10 costly transform module
+    // console.log(Array.from(this.ctx.vitenode.fetchCache.entries()).filter(i => i[1].duration)
+    //   .sort((a, b) => b[1].duration! - a[1].duration!)
+    //   .map(i => `${time(i[1].duration!)} ${i[0]}`)
+    //   .slice(0, 10)
+    //   .join('\n'),
+    // )
+
     const snapshotOutput = renderSnapshotSummary(this.ctx.config.root, this.ctx.snapshot.summary)
     if (snapshotOutput.length) {
       logger.log(snapshotOutput.map((t, i) => i === 0
@@ -243,7 +252,7 @@ export abstract class BaseReporter implements Reporter {
     if (this.watchFilters)
       logger.log(padTitle('Duration'), time(threadTime))
     else
-      logger.log(padTitle('Duration'), time(executionTime) + c.gray(` (setup ${time(setupTime)}, collect ${time(collectTime)}, tests ${time(testsTime)})`))
+      logger.log(padTitle('Duration'), time(executionTime) + c.dim(` (transform ${time(transformTime)}, setup ${time(setupTime)}, collect ${time(collectTime)}, tests ${time(testsTime)})`))
 
     logger.log()
   }