Time for build and pull phases (#18)

Firehed · Aug 17, 2021 · 2b1b59f · 2b1b59f
1 parent 8967204
commit 2b1b59f
Show file tree

Hide file tree

Showing 3 changed files with 107 additions and 43 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/helpers.ts b/src/helpers.ts
@@ -110,3 +110,34 @@ export async function runDockerCommand(command: DockerCommand, ...args: string[]
     stdout,
   }
 }
+
+
+export async function time<T>(name: string, timedFunction: () => Promise<T>): Promise<T> {
+  const start = Date.now()
+  try {
+    return await timedFunction()
+  } finally {
+    const durationMs = Date.now() - start
+
+    const formattedDuration = formatMs(durationMs)
+
+    core.info(`${name} completed in ${formattedDuration}`)
+  }
+}
+
+function formatMs(ms: number): string {
+  if (ms < 1000) {
+    return `${ms}ms`
+  }
+  const seconds = Math.floor(ms / 1000)
+  if (seconds < 60) {
+    return `${ms / 1000}s`
+  }
+  const minutes = Math.floor(seconds / 60)
+  if (minutes < 60) {
+    return `${minutes}m ${seconds % 60}s`
+  }
+
+  const hours = Math.floor(minutes / 60)
+  return `${hours}h ${minutes % 60}m`
+}
diff --git a/src/index.ts b/src/index.ts
@@ -8,12 +8,15 @@ import {
   getAllStages,
   getTaggedImageForStage,
   runDockerCommand,
+  time,
 } from './helpers'
 
 async function run(): Promise<void> {
   try {
-    await core.group('Pull images for layer cache', pull)
-    await core.group('Build', build)
+    await time('Pull images', () =>
+      core.group('Pull images for layer cache', pull)
+    )
+    await time('Full Build', build)
   } catch (error) {
     core.setFailed(error.message)
   }
@@ -84,34 +87,36 @@ async function build(): Promise<void> {
  * a tag specific to the ref/branch that the action is run on.
  */
 async function buildStage(stage: string, extraTags: string[]): Promise<string> {
-  core.startGroup(`Building stage: ${stage}`)
+  return time(`Build ${stage}`, async () => {
+    core.startGroup(`Building stage: ${stage}`)
 
-  const dockerfile = core.getInput('dockerfile')
+    const dockerfile = core.getInput('dockerfile')
 
-  const targetTag = getTaggedImageForStage(stage, getTagForRun())
+    const targetTag = getTaggedImageForStage(stage, getTagForRun())
 
-  const cacheFromArg = getAllPossibleCacheTargets()
+    const cacheFromArg = getAllPossibleCacheTargets()
     .flatMap(target => ['--cache-from', target])
 
-  const result = await runDockerCommand(
-    'build',
-    // '--build-arg', 'BUILDKIT_INLINE_CACHE="1"',
-    ...cacheFromArg,
-    '--file', dockerfile,
-    '--tag', targetTag,
-    '--target', stage,
-    '.'
-  )
-  if (result.exitCode > 0) {
-    throw 'Docker build failed'
-  }
-  await dockerPush(targetTag)
+    const result = await runDockerCommand(
+      'build',
+      // '--build-arg', 'BUILDKIT_INLINE_CACHE="1"',
+      ...cacheFromArg,
+      '--file', dockerfile,
+      '--tag', targetTag,
+      '--target', stage,
+      '.'
+    )
+    if (result.exitCode > 0) {
+      throw 'Docker build failed'
+    }
+    await dockerPush(targetTag)
 
-  for (const extraTag of extraTags) {
-    await addTagAndPush(targetTag, stage, extraTag)
-  }
-  core.endGroup()
-  return targetTag
+    for (const extraTag of extraTags) {
+      await addTagAndPush(targetTag, stage, extraTag)
+    }
+    core.endGroup()
+    return targetTag
+  })
 }
 
 async function dockerPush(taggedImage: string): Promise<void> {