|
7 | 7 | using System.Threading; |
8 | 8 | using System.Threading.Tasks; |
9 | 9 | using Autofac; |
| 10 | +using Azure.Identity; |
10 | 11 | using Azure.Storage.Blobs; |
11 | 12 | using GitHubVulnerabilities2v3.Configuration; |
12 | 13 | using GitHubVulnerabilities2v3.Extensions; |
|
16 | 17 | using Microsoft.Extensions.Logging; |
17 | 18 | using Microsoft.Extensions.Options; |
18 | 19 | using NuGet.Jobs; |
| 20 | +using NuGet.Services.Configuration; |
19 | 21 | using NuGet.Services.Cursor; |
20 | 22 | using NuGet.Services.GitHub.Collector; |
21 | 23 | using NuGet.Services.GitHub.Configuration; |
@@ -68,7 +70,7 @@ protected override void ConfigureAutofacServices(ContainerBuilder containerBuild |
68 | 70 |
|
69 | 71 | ConfigureQueryServices(containerBuilder); |
70 | 72 | ConfigureIngestionServices(containerBuilder); |
71 | | - ConfigureCollectorServices(containerBuilder); |
| 73 | + ConfigureCollectorServices(containerBuilder, configurationRoot); |
72 | 74 | } |
73 | 75 |
|
74 | 76 | protected void ConfigureIngestionServices(ContainerBuilder containerBuilder) |
@@ -111,14 +113,14 @@ protected void ConfigureQueryServices(ContainerBuilder containerBuilder) |
111 | 113 | .As<IAdvisoryQueryService>(); |
112 | 114 | } |
113 | 115 |
|
114 | | - protected void ConfigureCollectorServices(ContainerBuilder containerBuilder) |
| 116 | + protected void ConfigureCollectorServices(ContainerBuilder containerBuilder, IConfigurationRoot configurationRoot) |
115 | 117 | { |
116 | 118 | containerBuilder |
117 | 119 | .Register(ctx => |
118 | 120 | { |
119 | 121 | var config = ctx.Resolve<GitHubVulnerabilities2v3Configuration>(); |
120 | | - var connectionString = AzureStorageFactory.PrepareConnectionString(config.StorageConnectionString); |
121 | | - return new BlobServiceClient(connectionString); |
| 122 | + var credential = new ManagedIdentityCredential(configurationRoot[Constants.ManagedIdentityClientIdKey]); |
| 123 | + return new BlobServiceClient(new Uri(config.StorageConnectionString), credential); |
122 | 124 | }) |
123 | 125 | .As<BlobServiceClient>(); |
124 | 126 |
|
|
0 commit comments