-
Notifications
You must be signed in to change notification settings - Fork 0
/
SiloBuilderExtensions.cs
39 lines (35 loc) · 1.38 KB
/
SiloBuilderExtensions.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
using System;
using JetBrains.Annotations;
using Microsoft.Extensions.DependencyInjection;
using OrleansContrib.ActivationShedding;
// ReSharper disable once CheckNamespace
namespace Orleans.Hosting
{
[UsedImplicitly]
public static class SiloBuilderExtensions
{
/// <summary>
/// Enable grain activation shedding (rebalancing)
/// </summary>
public static ISiloBuilder UseActivationShedding(this ISiloBuilder siloBuilder)
{
return UseActivationShedding(siloBuilder, _ => { });
}
/// <summary>
/// Enable grain activation shedding (rebalancing)
/// </summary>
public static ISiloBuilder UseActivationShedding(this ISiloBuilder siloBuilder, Action<ActivationSheddingOptions> options)
{
siloBuilder.ConfigureServices(((context, collection) =>
{
collection.AddOptions<ActivationSheddingOptions>()
.Bind(context.Configuration.GetSection("ActivationShedding"))
// ReSharper disable once ConvertClosureToMethodGroup
.PostConfigure(sheddingOptions => options(sheddingOptions))
.ValidateDataAnnotations();
}));
siloBuilder.AddIncomingGrainCallFilter<ActivationSheddingFilter>();
return siloBuilder;
}
}
}