Included MigrateDbContext method in HostExtensions
This commit is contained in:
		
							parent
							
								
									818978b549
								
							
						
					
					
						commit
						cc9af86f35
					
				| @ -61,6 +61,58 @@ public static class HostExtensions | |||||||
|         return host; |         return host; | ||||||
|     } |     } | ||||||
| 
 | 
 | ||||||
|  |     public static IWebHost MigrateDbContext<TContext>(this IWebHost host, Action<TContext, IServiceProvider> seeder) where TContext : DbContext | ||||||
|  |     { | ||||||
|  |         var underK8s = host.IsInKubernetes(); | ||||||
|  | 
 | ||||||
|  |         using (var scope = host.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 retry = Policy.Handle<SqlException>() | ||||||
|  |                             .WaitAndRetry(new TimeSpan[] | ||||||
|  |                             { | ||||||
|  |                             TimeSpan.FromSeconds(3), | ||||||
|  |                             TimeSpan.FromSeconds(5), | ||||||
|  |                             TimeSpan.FromSeconds(8), | ||||||
|  |                             }); | ||||||
|  | 
 | ||||||
|  |                     //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 host; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|     private static void InvokeSeeder<TContext>(Action<TContext, IServiceProvider> seeder, TContext context, IServiceProvider services) |     private static void InvokeSeeder<TContext>(Action<TContext, IServiceProvider> seeder, TContext context, IServiceProvider services) | ||||||
|         where TContext : DbContext |         where TContext : DbContext | ||||||
|     { |     { | ||||||
|  | |||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user