-
Notifications
You must be signed in to change notification settings - Fork 344
/
AzureRedisExtensions.cs
110 lines (92 loc) · 5.71 KB
/
AzureRedisExtensions.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
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
#pragma warning disable AZPROVISION001
using System.Diagnostics.CodeAnalysis;
using Aspire.Hosting.ApplicationModel;
using Aspire.Hosting.Azure;
using Azure.Provisioning;
using Azure.Provisioning.KeyVaults;
using Azure.Provisioning.Redis;
namespace Aspire.Hosting;
/// <summary>
/// Provides extension methods for adding the Azure Redis resources to the application model.
/// </summary>
public static class AzureRedisExtensions
{
/// <summary>
/// Configures the resource to be published as Azure Cache for Redis when deployed via Azure Developer CLI.
/// </summary>
/// <param name="builder">The <see cref="IResourceBuilder{RedisResource}"/> builder.</param>
/// <returns>A reference to the <see cref="IResourceBuilder{RedisResource}"/> builder.</returns>
public static IResourceBuilder<RedisResource> PublishAsAzureRedis(this IResourceBuilder<RedisResource> builder)
{
return builder.PublishAsAzureRedis(null);
}
/// <summary>
/// Configures the resource to be published as Azure Cache for Redis when deployed via Azure Developer CLI.
/// </summary>
/// <param name="builder">The <see cref="IResourceBuilder{RedisResource}"/> builder.</param>
/// <param name="configureResource">Callback to configure the underlying <see cref="global::Azure.Provisioning.Redis.RedisCache"/> resource.</param>
/// <returns>A reference to the <see cref="IResourceBuilder{RedisResource}"/> builder.</returns>
[Experimental("AZPROVISION001", UrlFormat = "https://aka.ms/dotnet/aspire/diagnostics#{0}")]
public static IResourceBuilder<RedisResource> PublishAsAzureRedis(this IResourceBuilder<RedisResource> builder, Action<IResourceBuilder<AzureRedisResource>, ResourceModuleConstruct, RedisCache>? configureResource)
{
return builder.PublishAsAzureRedisInternal(configureResource);
}
internal static IResourceBuilder<RedisResource> PublishAsAzureRedisInternal(this IResourceBuilder<RedisResource> builder, Action<IResourceBuilder<AzureRedisResource>, ResourceModuleConstruct, RedisCache>? configureResource, bool useProvisioner = false)
{
builder.ApplicationBuilder.AddAzureProvisioning();
var configureConstruct = (ResourceModuleConstruct construct) =>
{
var redisCache = new RedisCache(construct, name: builder.Resource.Name);
redisCache.Properties.Tags["aspire-resource-name"] = construct.Resource.Name;
var vaultNameParameter = new Parameter("keyVaultName");
construct.AddParameter(vaultNameParameter);
var keyVault = KeyVault.FromExisting(construct, "keyVaultName");
var keyVaultSecret = new KeyVaultSecret(construct, keyVault, "connectionString");
keyVaultSecret.AssignProperty(
x => x.Properties.Value,
$$"""'${{{redisCache.Name}}.properties.hostName},ssl=true,password=${{{redisCache.Name}}.listKeys({{redisCache.Name}}.apiVersion).primaryKey}'"""
);
var resource = (AzureRedisResource)construct.Resource;
var resourceBuilder = builder.ApplicationBuilder.CreateResourceBuilder(resource);
configureResource?.Invoke(resourceBuilder, construct, redisCache);
};
var resource = new AzureRedisResource(builder.Resource, configureConstruct);
var resourceBuilder = builder.ApplicationBuilder.CreateResourceBuilder(resource)
.WithParameter(AzureBicepResource.KnownParameters.KeyVaultName)
.WithManifestPublishingCallback(resource.WriteToManifest);
if (useProvisioner)
{
// Used to hold a reference to the azure surrogate for use with the provisioner.
builder.WithAnnotation(new AzureBicepResourceAnnotation(resource));
builder.WithConnectionStringRedirection(resource);
// Remove the container annotation so that DCP doesn't do anything with it.
if (builder.Resource.Annotations.OfType<ContainerImageAnnotation>().SingleOrDefault() is { } containerAnnotation)
{
builder.Resource.Annotations.Remove(containerAnnotation);
}
}
return builder;
}
/// <summary>
/// Configures resource to use Azure for local development and when doing a deployment via the Azure Developer CLI.
/// </summary>
/// <param name="builder">The <see cref="IResourceBuilder{RedisResource}"/> builder.</param>
/// <returns>A reference to the <see cref="IResourceBuilder{RedisResource}"/> builder.</returns>
public static IResourceBuilder<RedisResource> AsAzureRedis(this IResourceBuilder<RedisResource> builder)
{
return builder.AsAzureRedis(null);
}
/// <summary>
/// Configures resource to use Azure for local development and when doing a deployment via the Azure Developer CLI.
/// </summary>
/// <param name="builder">The <see cref="IResourceBuilder{RedisResource}"/> builder.</param>
/// <param name="configureResource">Callback to configure the underlying <see cref="global::Azure.Provisioning.Redis.RedisCache"/> resource.</param>
/// <returns>A reference to the <see cref="IResourceBuilder{RedisResource}"/> builder.</returns>
[Experimental("AZPROVISION001", UrlFormat = "https://aka.ms/dotnet/aspire/diagnostics#{0}")]
public static IResourceBuilder<RedisResource> AsAzureRedis(this IResourceBuilder<RedisResource> builder, Action<IResourceBuilder<AzureRedisResource>, ResourceModuleConstruct, RedisCache>? configureResource)
{
return builder.PublishAsAzureRedisInternal(configureResource, useProvisioner: true);
}
}