feat(instrumentation): allow error of safeExecuteInTheMiddleAsync to be async

Author: JPeer264

experimental/packages/opentelemetry-instrumentation/src/utils.ts

@@ -50,7 +50,10 @@ export function safeExecuteInTheMiddle<T>(
  */
 export async function safeExecuteInTheMiddleAsync<T>(
   execute: () => T,
-  onFinish: (e: Error | undefined, result: T | undefined) => void,
+  onFinish: (
+    e: Error | undefined,
+    result: T | undefined
+  ) => Promise<void> | void,
   preventThrowingError?: boolean
 ): Promise<T> {
   let error: Error | undefined;
@@ -60,7 +63,7 @@ export async function safeExecuteInTheMiddleAsync<T>(
   } catch (e) {
     error = e;
   } finally {
-    onFinish(error, result);
+    await onFinish(error, result);
     if (error && !preventThrowingError) {
       // eslint-disable-next-line no-unsafe-finally
       throw error;

experimental/packages/opentelemetry-instrumentation/test/common/utils.test.ts

@@ -89,14 +89,15 @@ describe('safeExecuteInTheMiddle', function () {
 });
 
 describe('safeExecuteInTheMiddleAsync', function () {
-  it('should not throw error', function () {
+  it('should not throw error', function (done) {
     safeExecuteInTheMiddleAsync(
       async () => {
-        await setTimeout(() => {}, 1);
+        await new Promise(res => setTimeout(res, 1));
         return 'foo';
       },
       err => {
         assert.deepStrictEqual(err, undefined);
+        done();
       },
       true
     );
@@ -106,7 +107,7 @@ describe('safeExecuteInTheMiddleAsync', function () {
     try {
       await safeExecuteInTheMiddleAsync(
         async () => {
-          await setTimeout(() => {}, 1);
+          await new Promise(res => setTimeout(res, 1));
           throw error;
         },
         err => {
@@ -120,7 +121,7 @@ describe('safeExecuteInTheMiddleAsync', function () {
   it('should return result', async function () {
     const result = await safeExecuteInTheMiddleAsync(
       async () => {
-        await setTimeout(() => {}, 1);
+        await new Promise(res => setTimeout(res, 1));
         return 1;
       },
       (err, result) => {
@@ -130,4 +131,17 @@ describe('safeExecuteInTheMiddleAsync', function () {
     );
     assert.deepStrictEqual(result, 1);
   });
+  it('should wait for the error', async function () {
+    const result = await Promise.race([
+      safeExecuteInTheMiddleAsync(
+        () => 1,
+        async () => {
+          await new Promise(res => setTimeout(res, 100));
+        }
+      ),
+      new Promise(res => setTimeout(() => res('waited'), 10)),
+    ]);
+
+    assert.deepStrictEqual(result, 'waited');
+  });
 });