ericuss
5 years ago
2 changed files with 85 additions and 7 deletions
Unified View
Diff Options
-
+79 -0src/Services/Identity/Identity.API/IWebHostExtensions.cs
-
+6 -7src/Services/Identity/Identity.API/Identity.API.csproj
+ 79
- 0
src/Services/Identity/Identity.API/IWebHostExtensions.cs
View File
| @ -0,0 +1,79 @@ | |||||
| using Microsoft.EntityFrameworkCore; | |||||
| using Microsoft.Extensions.Configuration; | |||||
| using Microsoft.Extensions.DependencyInjection; | |||||
| using Microsoft.Extensions.Logging; | |||||
| using Polly; | |||||
| using System; | |||||
| using System.Data.SqlClient; | |||||
| namespace Microsoft.AspNetCore.Hosting | |||||
| { | |||||
| public static class IWebHostExtensions | |||||
| { | |||||
| public static bool IsInKubernetes(this IWebHost webHost) | |||||
| { | |||||
| var cfg = webHost.Services.GetService<IConfiguration>(); | |||||
| var orchestratorType = cfg.GetValue<string>("OrchestratorType"); | |||||
| return orchestratorType?.ToUpper() == "K8S"; | |||||
| } | |||||
| public static IWebHost MigrateDbContext<TContext>(this IWebHost webHost, Action<TContext, IServiceProvider> seeder) where TContext : DbContext | |||||
| { | |||||
| var underK8s = webHost.IsInKubernetes(); | |||||
| using (var scope = webHost.Services.CreateScope()) | |||||
| { | |||||
| var services = scope.ServiceProvider; | |||||
| var logger = services.GetRequiredService<ILogger<TContext>>(); | |||||
| var context = services.GetService<TContext>(); | |||||
| try | |||||
| { | |||||
| logger.LogInformation("Migrating database associated with context {DbContextName}", typeof(TContext).Name); | |||||
| if (underK8s) | |||||
| { | |||||
| InvokeSeeder(seeder, context, services); | |||||
| } | |||||
| else | |||||
| { | |||||
| var retries = 10; | |||||
| var retry = Policy.Handle<SqlException>() | |||||
| .WaitAndRetry( | |||||
| retryCount: retries, | |||||
| sleepDurationProvider: retryAttempt => TimeSpan.FromSeconds(Math.Pow(2, retryAttempt)), | |||||
| onRetry: (exception, timeSpan, retry, ctx) => | |||||
| { | |||||
| logger.LogWarning(exception, "[{prefix}] Exception {ExceptionType} with message {Message} detected on attempt {retry} of {retries}", nameof(TContext), exception.GetType().Name, exception.Message, retry, retries); | |||||
| }); | |||||
| //if the sql server container is not created on run docker compose this | |||||
| //migration can't fail for network related exception. The retry options for DbContext only | |||||
| //apply to transient exceptions | |||||
| // Note that this is NOT applied when running some orchestrators (let the orchestrator to recreate the failing service) | |||||
| retry.Execute(() => InvokeSeeder(seeder, context, services)); | |||||
| } | |||||
| logger.LogInformation("Migrated database associated with context {DbContextName}", typeof(TContext).Name); | |||||
| } | |||||
| catch (Exception ex) | |||||
| { | |||||
| logger.LogError(ex, "An error occurred while migrating the database used on context {DbContextName}", typeof(TContext).Name); | |||||
| if (underK8s) | |||||
| { | |||||
| throw; // Rethrow under k8s because we rely on k8s to re-run the pod | |||||
| } | |||||
| } | |||||
| } | |||||
| return webHost; | |||||
| } | |||||
| private static void InvokeSeeder<TContext>(Action<TContext, IServiceProvider> seeder, TContext context, IServiceProvider services) | |||||
| where TContext : DbContext | |||||
| { | |||||
| context.Database.Migrate(); | |||||
| seeder(context, services); | |||||
| } | |||||
| } | |||||
| } | |||||