Update DbMigrationMiddleware.cs

abpframework · May 14, 2024 · 76651c4 · 76651c4
1 parent e6008a8
commit 76651c4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/CmsKitDemo/Data/DbMigrationMiddleware.cs b/src/CmsKitDemo/Data/DbMigrationMiddleware.cs
@@ -19,10 +19,11 @@ public async Task InvokeAsync(HttpContext context, RequestDelegate next)
     {
         var connString = await _connectionStringResolver.ResolveAsync();
         var dbFilePath = connString.Replace("Data Source=", "").Replace(";Cache=Shared", "");
-            
-        if (ShouldCreateUserDb(context, dbFilePath))
+
+        if (ShouldCreateUserDb(context, dbFilePath) || !context.Request.Cookies.ContainsKey("CMSKitDemoDbMigrated"))
         {
             await _dbMigrationService.MigrateAsync(connString);
+            context.Response.Cookies.Append("CMSKitDemoDbMigrated", "true");
         }
 
         await next(context);