ericuss
5 years ago
2 changed files with 85 additions and 7 deletions
Split 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); | |||
| } | |||
| } | |||
| } | |||