open source Notesnook API

2026-07-10 12:48:34 +02:00 · 2022-12-28 16:20:25 +05:00
parent cf406454cd
commit d2217adce7
85 changed files with 4137 additions and 0 deletions
@@ -0,0 +1,90 @@
+using Microsoft.Extensions.Configuration;
+using MongoDB.Driver;
+using Streetwriters.Data.Interfaces;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace Streetwriters.Data.DbContexts
+{
+    public class MongoDbContext : IDbContext
+    {
+        private IMongoDatabase Database { get; set; }
+        private MongoClient MongoClient { get; set; }
+        private readonly List<Func<IClientSessionHandle, CancellationToken, Task>> _commands;
+        private IDbSettings DbSettings { get; set; }
+        public MongoDbContext(IDbSettings dbSettings)
+        {
+            DbSettings = dbSettings;
+            Configure();
+            // Every command will be stored and it'll be processed at SaveChanges
+            _commands = new List<Func<IClientSessionHandle, CancellationToken, Task>>();
+        }
+
+        public async Task<int> SaveChanges()
+        {
+            try
+            {
+                var count = _commands.Count;
+
+                using (IClientSessionHandle session = await MongoClient.StartSessionAsync())
+                {
+#if DEBUG
+                    await Task.WhenAll(_commands.Select(c => c(session, default(CancellationToken))));
+#else
+                    await session.WithTransactionAsync(async (handle, token) =>
+                    {
+                        await Task.WhenAll(_commands.Select(c => c(handle, token)));
+                        return true;
+                    });
+#endif
+
+                }
+                return count;
+            }
+            catch (Exception ex)
+            {
+                // TODO use Slogger here.
+                await Console.Error.WriteLineAsync(ex.ToString());
+                return 0;
+            }
+        }
+
+        private void Configure()
+        {
+            if (MongoClient != null)
+            {
+                return;
+            }
+            var settings = MongoClientSettings.FromConnectionString(DbSettings.ConnectionString);
+            settings.MaxConnectionPoolSize = 5000;
+            settings.MinConnectionPoolSize = 300;
+            MongoClient = new MongoClient(settings);
+        }
+
+        public IMongoCollection<T> GetCollection<T>(string databaseName, string collectionName)
+        {
+            return MongoClient.GetDatabase(databaseName).GetCollection<T>(collectionName, new MongoCollectionSettings()
+            {
+                AssignIdOnInsert = true,
+            });
+        }
+
+        public void AddCommand(Func<IClientSessionHandle, CancellationToken, Task> func)
+        {
+            _commands.Add(func);
+        }
+
+        public void Dispose()
+        {
+            GC.SuppressFinalize(this);
+        }
+
+        public Task DropDatabaseAsync()
+        {
+            return MongoClient.DropDatabaseAsync(DbSettings.DatabaseName);
+        }
+    }
+}