Add support for singleton based LiteDB implementation

sangeethnandakumar · Jan 21, 2024 · b4b88b0 · b4b88b0
1 parent f6f89d1
commit b4b88b0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Twileloop.UOW.LiteDB/Support/Extensions.cs b/Twileloop.UOW.LiteDB/Support/Extensions.cs
@@ -9,6 +9,22 @@ namespace Twileloop.UOW.LiteDB.Support
 
     public static class Extensions
     {
+        public static void AddSingletonUnitOfWork(this IServiceCollection services, Action<Options>? options = null)
+        {
+            var uowOptions = new Options();
+            options?.Invoke(uowOptions);
+
+            var contexts = new ConcurrentDictionary<string, LiteDBContext>();
+            foreach (var conn in uowOptions.Connections)
+            {
+                var db = new LiteDatabase(conn.ConnectionString);
+                contexts.TryAdd(conn.Name, new LiteDBContext(conn.Name, db));
+            }
+
+            services.AddSingleton(contexts);
+            services.AddSingleton<UnitOfWork>();
+        }
+
         public static void AddUnitOfWork(this IServiceCollection services, Action<Options>? options = null)
         {
             var uowOptions = new Options();