Simplify OutcomeResilienceStrategy

App-vNext · Jul 27, 2023 · b3caa61 · b3caa61
1 parent 65df668
commit b3caa61
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/Polly.Core/Utils/OutcomeResilienceStrategy.cs b/src/Polly.Core/Utils/OutcomeResilienceStrategy.cs
@@ -16,13 +16,11 @@ protected internal sealed override ValueTask<Outcome<TResult>> ExecuteCore<TResu
         TState state)
     {
         // Check if we can cast directly, thus saving some cycles and improving the performance
-        if (context.ResultType == typeof(T))
+        if (callback is Func<ResilienceContext, TState, ValueTask<Outcome<T>>> casted)
         {
-            // cast is safe here, because TResult and T are the same type
-            var callbackCasted = (Func<ResilienceContext, TState, ValueTask<Outcome<T>>>)(object)callback;
-            var valueTask = ExecuteCore(callbackCasted, context, state);
-
-            return TaskHelper.ConvertValueTask<T, TResult>(valueTask, context);
+            return TaskHelper.ConvertValueTask<T, TResult>(
+                ExecuteCore(casted, context, state),
+                context);
         }
         else
         {