fixes

App-vNext · May 23, 2023 · dacf762 · dacf762
1 parent 6231fb3
commit dacf762
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/src/Polly.Core.Benchmarks/CircuitBreakerOpenedBenchmark.cs b/src/Polly.Core.Benchmarks/CircuitBreakerOpenedBenchmark.cs
@@ -1,8 +1,7 @@
 using System.Threading.Tasks;
 using BenchmarkDotNet.Attributes;
-using Polly.Core.Benchmarks;
 
-namespace Polly.Benchmarks;
+namespace Polly.Core.Benchmarks;
 
 public class CircuitBreakerOpenedBenchmark
 {

diff --git a/src/Polly.Core.Benchmarks/ResilienceStrategyBenchmark.cs b/src/Polly.Core.Benchmarks/ResilienceStrategyBenchmark.cs
@@ -12,10 +12,18 @@ public class ResilienceStrategyBenchmark
     private readonly ResilienceStrategy<string> _genericStrategy = NullResilienceStrategy<string>.Instance;
 
     [Benchmark(Baseline = true)]
+    public async ValueTask ExecuteOutcomeAsync()
+    {
+        var context = ResilienceContext.Get();
+        await _strategy.ExecuteOutcomeAsync((_, _) => new ValueTask<Outcome<string>>(new Outcome<string>("dummy")), context, "state").ConfigureAwait(false);
+        ResilienceContext.Return(context);
+    }
+
+    [Benchmark]
     public async ValueTask ExecuteAsync_ResilienceContextAndState()
     {
         var context = ResilienceContext.Get();
-        await _strategy.ExecuteAsync((_, _) => new ValueTask<Outcome<string>>(new Outcome<string>("dummy")), context, "state").ConfigureAwait(false);
+        await _strategy.ExecuteAsync((_, _) => new ValueTask<string>(new string("dummy")), context, "state").ConfigureAwait(false);
         ResilienceContext.Return(context);
     }
 
@@ -33,8 +41,13 @@ public async ValueTask ExecuteAsync_GenericStrategy_CancellationToken()
 
     private class DummyResilienceStrategy : ResilienceStrategy
     {
-        protected override ValueTask<TResult> ExecuteCoreAsync<TResult, TState>(
-            Func<ResilienceContext, TState, ValueTask<TResult>> callback,
+        public ValueTask<Outcome<TResult>> ExecuteOutcomeAsync<TResult, TState>(
+            Func<ResilienceContext, TState, ValueTask<Outcome<TResult>>> callback,
+            ResilienceContext context,
+            TState state) => ExecuteCoreAsync(callback, context, state);
+
+        protected override ValueTask<Outcome<TResult>> ExecuteCoreAsync<TResult, TState>(
+            Func<ResilienceContext, TState, ValueTask<Outcome<TResult>>> callback,
             ResilienceContext context,
             TState state) => callback(context, state);
     }

diff --git a/src/Polly.Core.Benchmarks/Utils/EmptyResilienceStrategy.cs b/src/Polly.Core.Benchmarks/Utils/EmptyResilienceStrategy.cs
@@ -1,3 +1,5 @@
+using Polly.Strategy;
+
 namespace Polly.Core.Benchmarks.Utils;
 
 internal class EmptyResilienceStrategy : ResilienceStrategy