/
KubernetesProbesExtensions.cs
105 lines (91 loc) · 4.25 KB
/
KubernetesProbesExtensions.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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Diagnostics.Probes;
using Microsoft.Shared.Diagnostics;
namespace Microsoft.Extensions.DependencyInjection;
/// <summary>
/// Extensions for setting up probes for Kubernetes.
/// </summary>
public static class KubernetesProbesExtensions
{
/// <summary>
/// Registers liveness, startup and readiness probes using the default options.
/// </summary>
/// <param name="services">The dependency injection container to add the probe to.</param>
/// <returns>The value of <paramref name="services"/>.</returns>
public static IServiceCollection AddKubernetesProbes(this IServiceCollection services)
=> services.AddKubernetesProbes((_) => { });
/// <summary>
/// Registers liveness, startup and readiness probes using the configured options.
/// </summary>
/// <param name="services">The dependency injection container to add the probe to.</param>
/// <param name="section">Configuration for <see cref="KubernetesProbesOptions"/>.</param>
/// <returns>The value of <paramref name="services"/>.</returns>
public static IServiceCollection AddKubernetesProbes(this IServiceCollection services, IConfigurationSection section)
{
_ = Throw.IfNull(services);
_ = Throw.IfNull(section);
return services.AddKubernetesProbes(o => section.Bind(o));
}
/// <summary>
/// Registers liveness, startup and readiness probes using the configured options.
/// </summary>
/// <param name="services">The dependency injection container to add the probe to.</param>
/// <param name="configure">Configure action for <see cref="KubernetesProbesOptions"/>.</param>
/// <returns>The value of <paramref name="services"/>.</returns>
public static IServiceCollection AddKubernetesProbes(this IServiceCollection services, Action<KubernetesProbesOptions> configure)
{
_ = Throw.IfNull(services);
_ = Throw.IfNull(configure);
var wrapperOptions = new KubernetesProbesOptions();
return services
.AddTcpEndpointProbe(ProbeTags.Liveness, options =>
{
options.TcpPort = wrapperOptions.LivenessProbe.TcpPort;
wrapperOptions.LivenessProbe = options;
configure(wrapperOptions);
var originalPredicate = options.FilterChecks;
if (originalPredicate == null)
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Liveness);
}
else
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Liveness) && originalPredicate(check);
}
})
.AddTcpEndpointProbe(ProbeTags.Startup, options =>
{
options.TcpPort = wrapperOptions.StartupProbe.TcpPort;
wrapperOptions.StartupProbe = options;
configure(wrapperOptions);
var originalPredicate = options.FilterChecks;
if (originalPredicate == null)
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Startup);
}
else
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Startup) && originalPredicate(check);
}
})
.AddTcpEndpointProbe(ProbeTags.Readiness, (options) =>
{
options.TcpPort = wrapperOptions.ReadinessProbe.TcpPort;
wrapperOptions.ReadinessProbe = options;
configure(wrapperOptions);
var originalPredicate = options.FilterChecks;
if (originalPredicate == null)
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Readiness);
}
else
{
options.FilterChecks = (check) => check.Tags.Contains(ProbeTags.Readiness) && originalPredicate(check);
}
});
}
}