feat: add UseExtendedSqlServerMigrations()

Informatievlaanderen · Aug 28, 2019 · e6e7cff · e6e7cff
1 parent d439301
commit e6e7cff
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/...ctionHandling.Runner/MigrationExtensions/ExtendedSqlServerMigrationsAnnotationProvider.cs b/...ctionHandling.Runner/MigrationExtensions/ExtendedSqlServerMigrationsAnnotationProvider.cs
@@ -1,4 +1,4 @@
-namespace Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner.MigrationExtensions
+namespace Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner.MigrationExtensions
 {
     using System.Collections.Generic;
     using System.Linq;
@@ -20,4 +20,4 @@ public override IEnumerable<IAnnotation> For(IIndex index)
             return baseAnnotations.Concat(customAnnotations);
         }
     }
-}
+}
diff --git a/...rs.ProjectionHandling.Runner/MigrationExtensions/ExtendedSqlServerMigrationsExtensions.cs b/...rs.ProjectionHandling.Runner/MigrationExtensions/ExtendedSqlServerMigrationsExtensions.cs
@@ -0,0 +1,18 @@
+namespace Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner.MigrationExtensions
+{
+    using Microsoft.EntityFrameworkCore;
+    using Microsoft.EntityFrameworkCore.Migrations;
+
+    public static class ExtendedSqlServerMigrationsExtensions
+    {
+        public static DbContextOptionsBuilder UseExtendedSqlServerMigrations(this DbContextOptionsBuilder optionsBuilder) =>
+            optionsBuilder
+                .ReplaceService<IMigrationsSqlGenerator, ExtendedSqlServerMigrationsSqlGenerator>()
+                .ReplaceService<IMigrationsAnnotationProvider, ExtendedSqlServerMigrationsAnnotationProvider>();
+
+        public static DbContextOptionsBuilder<T> UseExtendedSqlServerMigrations<T>(this DbContextOptionsBuilder<T> optionsBuilder) where T : DbContext =>
+            optionsBuilder
+                .ReplaceService<IMigrationsSqlGenerator, ExtendedSqlServerMigrationsSqlGenerator>()
+                .ReplaceService<IMigrationsAnnotationProvider, ExtendedSqlServerMigrationsAnnotationProvider>();
+    }
+}
diff --git a/...Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner/RunnerDbContextMigrationFactory.cs b/...Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner/RunnerDbContextMigrationFactory.cs
@@ -7,6 +7,7 @@ namespace Be.Vlaanderen.Basisregisters.ProjectionHandling.Runner
     using Microsoft.EntityFrameworkCore.Infrastructure;
     using Microsoft.Extensions.Configuration;
     using Microsoft.Extensions.Logging;
+    using MigrationExtensions;
 
     public interface IRunnerDbContextMigratorFactory
     {
@@ -75,7 +76,8 @@ private DbContextOptionsBuilder<TContext> CreateOptionsBuilder(IConfiguration co
 
                         ConfigureSqlServerOptions(sqlServerOptions);
                     }
-                );
+                )
+                .UseExtendedSqlServerMigrations();
 
             ConfigureOptionsBuilder(optionsBuilder);