-
Notifications
You must be signed in to change notification settings - Fork 2k
/
AzureQueueStreamOptions.cs
127 lines (112 loc) · 5.58 KB
/
AzureQueueStreamOptions.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Azure;
using Azure.Core;
using Azure.Storage;
using Azure.Storage.Queues;
using Orleans.AzureUtils;
using Orleans.Runtime;
namespace Orleans.Configuration
{
/// <summary>
/// Azure queue stream provider options.
/// </summary>
public class AzureQueueOptions
{
/// <summary>
/// Options to be used when configuring the queue storage client, or <see langword="null"/> to use the default options.
/// </summary>
public QueueClientOptions ClientOptions { get; set; } = new QueueClientOptions
{
Retry =
{
Mode = RetryMode.Fixed,
Delay = AzureQueueDefaultPolicies.PauseBetweenQueueOperationRetries,
MaxRetries = AzureQueueDefaultPolicies.MaxQueueOperationRetries,
NetworkTimeout = AzureQueueDefaultPolicies.QueueOperationTimeout,
}
};
/// <summary>
/// The optional delegate used to create a <see cref="QueueServiceClient"/> instance.
/// </summary>
internal Func<Task<QueueServiceClient>> CreateClient { get; private set; }
public TimeSpan? MessageVisibilityTimeout { get; set; }
public List<string> QueueNames { get; set; }
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using a connection string.
/// </summary>
public void ConfigureQueueServiceClient(string connectionString)
{
if (string.IsNullOrWhiteSpace(connectionString)) throw new ArgumentNullException(nameof(connectionString));
CreateClient = () => Task.FromResult(new QueueServiceClient(connectionString, ClientOptions));
}
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using an authenticated service URI.
/// </summary>
public void ConfigureQueueServiceClient(Uri serviceUri)
{
if (serviceUri is null) throw new ArgumentNullException(nameof(serviceUri));
CreateClient = () => Task.FromResult(new QueueServiceClient(serviceUri, ClientOptions));
}
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using the provided callback.
/// </summary>
public void ConfigureQueueServiceClient(Func<Task<QueueServiceClient>> createClientCallback)
{
CreateClient = createClientCallback ?? throw new ArgumentNullException(nameof(createClientCallback));
}
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using an authenticated service URI and a <see cref="Azure.Core.TokenCredential"/>.
/// </summary>
public void ConfigureQueueServiceClient(Uri serviceUri, TokenCredential tokenCredential)
{
if (serviceUri is null) throw new ArgumentNullException(nameof(serviceUri));
if (tokenCredential is null) throw new ArgumentNullException(nameof(tokenCredential));
CreateClient = () => Task.FromResult(new QueueServiceClient(serviceUri, tokenCredential, ClientOptions));
}
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using an authenticated service URI and a <see cref="Azure.AzureSasCredential"/>.
/// </summary>
public void ConfigureQueueServiceClient(Uri serviceUri, AzureSasCredential azureSasCredential)
{
if (serviceUri is null) throw new ArgumentNullException(nameof(serviceUri));
if (azureSasCredential is null) throw new ArgumentNullException(nameof(azureSasCredential));
CreateClient = () => Task.FromResult(new QueueServiceClient(serviceUri, azureSasCredential, ClientOptions));
}
/// <summary>
/// Configures the <see cref="QueueServiceClient"/> using an authenticated service URI and a <see cref="StorageSharedKeyCredential"/>.
/// </summary>
public void ConfigureQueueServiceClient(Uri serviceUri, StorageSharedKeyCredential sharedKeyCredential)
{
if (serviceUri is null) throw new ArgumentNullException(nameof(serviceUri));
if (sharedKeyCredential is null) throw new ArgumentNullException(nameof(sharedKeyCredential));
CreateClient = () => Task.FromResult(new QueueServiceClient(serviceUri, sharedKeyCredential, ClientOptions));
}
}
public class AzureQueueOptionsValidator : IConfigurationValidator
{
private readonly AzureQueueOptions options;
private readonly string name;
private AzureQueueOptionsValidator(AzureQueueOptions options, string name)
{
this.options = options;
this.name = name;
}
public void ValidateConfiguration()
{
if (this.options.CreateClient is null)
{
throw new OrleansConfigurationException($"No credentials specified. Use the {options.GetType().Name}.{nameof(AzureQueueOptions.ConfigureQueueServiceClient)} method to configure the Azure Queue Service client.");
}
if (options.QueueNames == null || options.QueueNames.Count == 0)
throw new OrleansConfigurationException(
$"{nameof(AzureQueueOptions)} on stream provider {this.name} is invalid. {nameof(AzureQueueOptions.QueueNames)} is invalid");
}
public static IConfigurationValidator Create(IServiceProvider services, string name)
{
AzureQueueOptions aqOptions = services.GetOptionsByName<AzureQueueOptions>(name);
return new AzureQueueOptionsValidator(aqOptions, name);
}
}
}