Post-migration actions and post-migration database views refreshing

src/Kros.KORM.csproj

@@ -2,7 +2,7 @@
 
   <PropertyGroup>
     <TargetFrameworks>net8.0</TargetFrameworks>
-    <Version>7.0.0</Version>
+    <Version>7.1.0</Version>
     <Authors>KROS a. s.</Authors>
     <Company>KROS a. s.</Company>
     <Description>KORM is fast, easy to use, micro ORM tool (Kros Object Relation Mapper).</Description>

src/Migrations/MigrationOptions.cs

@@ -1,7 +1,10 @@
 using Kros.KORM.Migrations.Providers;
 using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Reflection;
+using System.Text;
+using System.Threading.Tasks;
 
 namespace Kros.KORM.Migrations
 {
@@ -11,14 +14,23 @@ namespace Kros.KORM.Migrations
     public class MigrationOptions
     {
         private const int DefaultTimeoutInSeconds = 30;
+        private const string DefaultResourceNamespace = "Resources";
+        private const string DefaultRefreshViewsScriptName = "RefreshViews.sql";
 
-        private List<IMigrationScriptsProvider> _providers = new List<IMigrationScriptsProvider>();
+        private List<IMigrationScriptsProvider> _providers = [];
+        private List<Func<IDatabase, long, Task>> _actions = [];
 
         /// <summary>
         /// List of <see cref="IMigrationScriptsProvider"/>.
         /// </summary>
         public IEnumerable<IMigrationScriptsProvider> Providers => _providers;
 
+
+        /// <summary>
+        /// List of actions to be executed on database after migration scripts are executed.
+        /// </summary>
+        public IEnumerable<Func<IDatabase, long, Task>> Actions => _actions;
+
         /// <summary>
         /// Timeout for the migration script command.
         /// If not set, default value 30s will be used.
@@ -46,5 +58,31 @@ public void AddAssemblyScriptsProvider(Assembly assembly, string resourceNamespa
         /// <param name="folderPath">Path to folder where migration scripts are stored.</param>
         public void AddFileScriptsProvider(string folderPath)
             => AddScriptsProvider(new FileMigrationScriptsProvider(folderPath));
+
+        /// <summary>
+        /// Add action to be executed on database after migration scripts are executed.
+        /// </summary>
+        /// <param name="actionToExecute"></param>
+        public void AddAfterMigrationAction(Func<IDatabase, long, Task> actionToExecute)
+        {
+            _actions.Add(actionToExecute);
+        }
+
+        /// <summary>
+        /// Add action of refreshing all database views.
+        /// </summary>
+        /// <param name="scriptName">Name of the script containing query for refreshing all views. Default one is Kros.KORM\Resources\RefreshViews.sql </param>
+        public void AddRefreshViewsAction(string scriptName = DefaultRefreshViewsScriptName)
+        {
+            var assembly = Assembly.GetExecutingAssembly();
+            var resourceName = $"{assembly.GetName().Name}.{DefaultResourceNamespace}.{scriptName}";
+            AddAfterMigrationAction(async (database, _) =>
+            {
+                Stream resourceStream = assembly.GetManifestResourceStream(resourceName);
+                using var reader = new StreamReader(resourceStream, Encoding.UTF8);
+                string script = await reader.ReadToEndAsync();
+                await database.ExecuteNonQueryAsync(script);
+            });
+        }
     }
 }

src/Migrations/MigrationsRunner.cs

@@ -82,6 +82,12 @@ public async Task MigrateAsync()
                 if (migrationScripts.Any())
                 {
                     await ExecuteMigrationScripts(helper.Database, migrationScripts);
+                    long maxScriptId = migrationScripts.Max(s => s.Id);
+
+                    foreach (var action in _migrationOptions.Actions)
+                    {
+                        await action(helper.Database, maxScriptId);
+                    }
                 }
             }
         }

src/Resources/RefreshViews.sql

@@ -0,0 +1,8 @@
+DECLARE @sql NVARCHAR(MAX)
+
+SET @sql = (
+    SELECT STRING_AGG('EXEC sp_refreshview ' + QUOTENAME(name), '; ') 
+    FROM sys.views
+)
+
+EXEC sp_executesql @sql