feat: Add async capabilities for the trace function

seasonedcc · Oct 24, 2023 · fd090e9 · fd090e9
1 parent 383b3ca
commit fd090e9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -313,9 +313,9 @@ traceToConsole(someOtherDomainFunction)()
 It would also be simple to create a function that will send the errors to some error tracking service under certain conditions:
 
 ```ts
-const trackErrors = trace(({ input, output, result }) => {
+const trackErrors = trace(async ({ input, output, result }) => {
   if(!result.success && someOtherConditions(result)) {
-    sendToExternalService({ input, output, result })
+    await sendToExternalService({ input, output, result })
   }
 })
 ```

diff --git a/src/domain-functions.ts b/src/domain-functions.ts
@@ -339,11 +339,15 @@ type TraceData<T> = {
  * //    ^? DomainFunction<number>
  */
 function trace<D extends DomainFunction = DomainFunction<unknown>>(
-  traceFn: ({ input, environment, result }: TraceData<UnpackResult<D>>) => void,
+  traceFn: ({
+    input,
+    environment,
+    result,
+  }: TraceData<UnpackResult<D>>) => Promise<void> | void,
 ): <T>(fn: DomainFunction<T>) => DomainFunction<T> {
   return (fn) => async (input, environment) => {
     const result = await fn(input, environment)
-    traceFn({ input, environment, result } as TraceData<UnpackResult<D>>)
+    await traceFn({ input, environment, result } as TraceData<UnpackResult<D>>)
     return result
   }
 }