Add try-with for compensation

Author: gusty

src/FSharpPlus/Extensions/Task.fs

@@ -327,6 +327,9 @@ module Task =
 
     [<ObsoleteAttribute("Swap parameters")>]
     let rec tryWith (body: unit -> Task<'T>) (compensation: exn -> Task<'T>) : Task<'T> =
+        let runCompensation exn =
+            try compensation exn
+            with e -> Task.FromException<'T> e
         let unwrapException (agg: AggregateException) =
             if agg.InnerExceptions.Count = 1 then agg.InnerExceptions.[0]
             else agg :> Exception
@@ -336,11 +339,11 @@ module Task =
             if task.IsCompleted then
                 match task with
                 | Succeeded _ -> task
-                | Faulted aex -> compensation (unwrapException aex)
+                | Faulted aex -> runCompensation (unwrapException aex)
                 | Canceled    -> canceled
             else
                 task.ContinueWith(fun (x: Task<'T>) -> tryWith (fun () -> x) compensation).Unwrap ()
-        | Error exn -> compensation exn
+        | Error exn -> runCompensation exn
 
     [<ObsoleteAttribute("Swap parameters")>]
     let rec tryFinally (body: unit -> Task<'T>) (compensation : unit -> unit) : Task<'T> =

src/FSharpPlus/Extensions/ValueTask.fs

@@ -301,6 +301,9 @@ module ValueTask =
 
     /// Used to de-sugar try .. with .. blocks in Computation Expressions.
     let inline tryWith ([<InlineIfLambda>]compensation: exn -> ValueTask<'T>) ([<InlineIfLambda>]body: unit -> ValueTask<'T>) : ValueTask<'T> =
+        let runCompensation exn =
+            try compensation exn
+            with e -> ValueTask.FromException<'T> e
         let unwrapException (agg: AggregateException) =
             if agg.InnerExceptions.Count = 1 then agg.InnerExceptions.[0]
             else agg :> Exception
@@ -310,11 +313,11 @@ module ValueTask =
             if task.IsCompleted then
                 match task with
                 | Succeeded _ -> task
-                | Faulted aex -> compensation (unwrapException aex)
+                | Faulted aex -> runCompensation (unwrapException aex)
                 | Canceled    -> canceled
             else
                 task.AsTask().ContinueWith(fun (x: Task<'T>) -> Task.tryWith (compensation >> fun x -> x.AsTask()) (fun () -> x)).Unwrap () |> ValueTask<'T>
-        | Error exn -> compensation exn
+        | Error exn -> runCompensation exn
 
 
     /// Used to de-sugar try .. finally .. blocks in Computation Expressions.