Flush the cache data on exit

kubagdynia · Dec 5, 2023 · c2d9230 · c2d9230
1 parent 9ccc66a
commit c2d9230
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 2 deletions.
diff --git a/CacheDrive.ExampleConsoleApp/appsettings.json b/CacheDrive.ExampleConsoleApp/appsettings.json
@@ -5,6 +5,7 @@
     "CacheExpirationType": "Minutes",
     "CacheExpiration": 60,
     "CacheType": "MemoryAndFile",
-    "InitializeOnStartup": true
+    "InitializeOnStartup": true,
+    "FlushOnExit": true
   }
 }
diff --git a/CacheDrive/Configuration/CacheSettings.cs b/CacheDrive/Configuration/CacheSettings.cs
@@ -39,4 +39,10 @@ public class CacheSettings
     /// Default value is true.
     /// </summary>
     public bool InitializeOnStartup { get; set; } = true;
+
+    /// <summary>
+    /// Before exit, flush the cache data to a files if necessary.
+    /// Default value is true.
+    /// </summary>
+    public bool FlushOnExit { get; set; } = true;
 }
diff --git a/CacheDrive/Extensions/ServiceCollectionExtensions.cs b/CacheDrive/Extensions/ServiceCollectionExtensions.cs
@@ -31,9 +31,12 @@ public static void RegisterCacheDrive(this IServiceCollection services, CacheSet
                 services.Configure<CacheSettings>(opt =>
                 {
                     opt.CacheEnabled = settings.CacheEnabled;
+                    opt.CacheFolderName = settings.CacheFolderName;
                     opt.CacheExpiration = settings.CacheExpiration;
                     opt.CacheExpirationType = settings.CacheExpirationType;
                     opt.CacheType = settings.CacheType;
+                    opt.InitializeOnStartup = settings.InitializeOnStartup;
+                    opt.FlushOnExit = settings.FlushOnExit;
                 });
             }
             else
@@ -48,9 +51,12 @@ public static void RegisterCacheDrive(this IServiceCollection services, CacheSet
                 services.Configure<CacheSettings>(opt =>
                 {
                     opt.CacheEnabled = settings.CacheEnabled;
+                    opt.CacheFolderName = settings.CacheFolderName;
                     opt.CacheExpiration = settings.CacheExpiration;
                     opt.CacheExpirationType = settings.CacheExpirationType;
                     opt.CacheType = settings.CacheType;
+                    opt.InitializeOnStartup = settings.InitializeOnStartup;
+                    opt.FlushOnExit = settings.FlushOnExit;
                 });
             }
             else

diff --git a/CacheDrive/Services/MemoryCacheFileStorageService.cs b/CacheDrive/Services/MemoryCacheFileStorageService.cs
@@ -175,6 +175,9 @@ private void DisposeManageResource()
 
     private void DisposeUnManageResource()
     {
-        Flush();
+        if (CacheSettings.FlushOnExit)
+        {
+            Flush();
+        }
     }
 }