-
-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
Program.cs
77 lines (67 loc) · 3.17 KB
/
Program.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
using Chaos;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Http.Resilience;
using Polly;
using Polly.Simmy;
using Polly.Simmy.Fault;
using Polly.Simmy.Latency;
using Polly.Simmy.Outcomes;
var builder = WebApplication.CreateBuilder(args);
var services = builder.Services;
services.TryAddSingleton<IChaosManager, ChaosManager>();
services.AddHttpContextAccessor();
var httpClientBuilder = services.AddHttpClient<TodosClient>(client => client.BaseAddress = new Uri("https://jsonplaceholder.typicode.com"));
// Configure the standard resilience handler
httpClientBuilder
.AddStandardResilienceHandler()
.Configure(options =>
{
// Update attempt timeout to 1 second
options.AttemptTimeout.Timeout = TimeSpan.FromSeconds(1);
// Update circuit breaker to handle transient errors and InvalidOperationException
options.CircuitBreaker.ShouldHandle = args => args.Outcome switch
{
{} outcome when HttpClientResiliencePredicates.IsTransient(outcome) => PredicateResult.True(),
{ Exception: InvalidOperationException } => PredicateResult.True(),
_ => PredicateResult.False()
};
// Update retry strategy to handle transient errors and InvalidOperationException
options.Retry.ShouldHandle = args => args.Outcome switch
{
{} outcome when HttpClientResiliencePredicates.IsTransient(outcome) => PredicateResult.True(),
{ Exception: InvalidOperationException } => PredicateResult.True(),
_ => PredicateResult.False()
};
});
// Configure the chaos injection
httpClientBuilder.AddResilienceHandler("chaos", (builder, context) =>
{
// Get IChaosManager from dependency injection
var chaosManager = context.ServiceProvider.GetRequiredService<IChaosManager>();
builder
.AddChaosLatency(new ChaosLatencyStrategyOptions
{
EnabledGenerator = args => chaosManager.IsChaosEnabledAsync(args.Context),
InjectionRateGenerator = args => chaosManager.GetInjectionRateAsync(args.Context),
Latency = TimeSpan.FromSeconds(5)
})
.AddChaosFault(new ChaosFaultStrategyOptions
{
EnabledGenerator = args => chaosManager.IsChaosEnabledAsync(args.Context),
InjectionRateGenerator = args => chaosManager.GetInjectionRateAsync(args.Context),
FaultGenerator = new FaultGenerator().AddException(() => new InvalidOperationException("Chaos strategy injection!"))
})
.AddChaosOutcome(new ChaosOutcomeStrategyOptions<HttpResponseMessage>
{
EnabledGenerator = args => chaosManager.IsChaosEnabledAsync(args.Context),
InjectionRateGenerator = args => chaosManager.GetInjectionRateAsync(args.Context),
OutcomeGenerator = new OutcomeGenerator<HttpResponseMessage>().AddResult(() => new HttpResponseMessage(System.Net.HttpStatusCode.InternalServerError))
});
});
// Run the app
var app = builder.Build();
app.MapGet("/", async (TodosClient client, HttpContext httpContext, CancellationToken cancellationToken) =>
{
return await client.GetTodosAsync(cancellationToken);
});
app.Run();