Added async wrapper for TryWith and TryFinally in CE builder

demystifyfp · Jun 21, 2023 · ae08b58 · ae08b58
1 parent 521ee3c
commit ae08b58
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/src/FsToolkit.ErrorHandling/AsyncValidationCE.fs b/src/FsToolkit.ErrorHandling/AsyncValidationCE.fs
@@ -44,20 +44,26 @@ module AsyncValidationCE =
                 [<InlineIfLambda>] generator: unit -> AsyncValidation<'ok, 'error>,
                 [<InlineIfLambda>] handler: exn -> AsyncValidation<'ok, 'error>
             ) : AsyncValidation<'ok, 'error> =
-            try
-                this.Run generator
-            with e ->
-                handler e
+            async {
+                return!
+                    try
+                        this.Run generator
+                    with e ->
+                        handler e
+            }
 
         member inline this.TryFinally
             (
                 [<InlineIfLambda>] generator: unit -> AsyncValidation<'ok, 'error>,
                 [<InlineIfLambda>] compensation: unit -> unit
             ) : AsyncValidation<'ok, 'error> =
-            try
-                this.Run generator
-            finally
-                compensation ()
+            async {
+                return!
+                    try
+                        this.Run generator
+                    finally
+                        compensation ()
+            }
 
         member inline this.Using
             (