From 52fd34a68424bb91ddb14ee3fbb4c6bb57a6ca56 Mon Sep 17 00:00:00 2001
From: Christopher Schleiden <cschleiden@github.com>
Date: Fri, 17 Feb 2023 16:16:16 -0800
Subject: [PATCH] Await operation

---
 actions-languageserver/src/utils/timer.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/actions-languageserver/src/utils/timer.ts b/actions-languageserver/src/utils/timer.ts
index 0278a11..f9dc71e 100644
--- a/actions-languageserver/src/utils/timer.ts
+++ b/actions-languageserver/src/utils/timer.ts
@@ -1,8 +1,12 @@
 import {log} from "@github/actions-languageservice/log";
 
-export function timeOperation<T>(name: string, f: () => T): T {
+export async function timeOperation<T>(name: string, f: () => T): Promise<T> {
   const start = Date.now();
   const result = f();
+  if (result instanceof Promise) {
+    await result;
+  }
+
   const end = Date.now();
 
   log(`${name} took ${end - start}ms`);