Expose mongodb transactions

Backend.Tests/Controllers/MergeControllerTests.cs

@@ -11,6 +11,7 @@ namespace Backend.Tests.Controllers
 {
     public class MergeControllerTests : IDisposable
     {
+        private IMongoDbContext _mongoDbContext = null!;
         private IMergeBlacklistRepository _mergeBlacklistRepo = null!;
         private IMergeGraylistRepository _mergeGraylistRepo = null!;
         private IWordRepository _wordRepo = null!;
@@ -38,11 +39,13 @@ protected virtual void Dispose(bool disposing)
         [SetUp]
         public void Setup()
         {
+            _mongoDbContext = new MongoDbContextMock();
             _mergeBlacklistRepo = new MergeBlacklistRepositoryMock();
             _mergeGraylistRepo = new MergeGraylistRepositoryMock();
             _wordRepo = new WordRepositoryMock();
             _wordService = new WordService(_wordRepo);
-            _mergeService = new MergeService(_mergeBlacklistRepo, _mergeGraylistRepo, _wordRepo, _wordService);
+            _mergeService = new MergeService(
+                _mongoDbContext, _mergeBlacklistRepo, _mergeGraylistRepo, _wordRepo, _wordService);
             _permissionService = new PermissionServiceMock();
             _mergeController = new MergeController(_mergeService, _permissionService);
         }

Backend.Tests/Mocks/MongoDbContextMock.cs

@@ -0,0 +1,32 @@
+using System;
+using System.Threading.Tasks;
+using BackendFramework.Interfaces;
+using MongoDB.Driver;
+
+namespace Backend.Tests.Mocks;
+
+public class MongoDbContextMock : IMongoDbContext
+{
+    public IMongoDatabase Db => throw new NotSupportedException();
+    public Task<IMongoTransaction> BeginTransaction()
+    {
+        return Task.FromResult<IMongoTransaction>(new MongoTransactionMock());
+    }
+
+    private sealed class MongoTransactionMock : IMongoTransaction
+    {
+        public Task CommitTransactionAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        public Task AbortTransactionAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        public void Dispose()
+        {
+        }
+    }
+}

Backend.Tests/Services/MergeServiceTests.cs

@@ -10,6 +10,7 @@ namespace Backend.Tests.Services
 {
     public class MergeServiceTests
     {
+        private IMongoDbContext _mongoDbContext = null!;
         private IMergeBlacklistRepository _mergeBlacklistRepo = null!;
         private IMergeGraylistRepository _mergeGraylistRepo = null!;
         private IWordRepository _wordRepo = null!;
@@ -22,11 +23,13 @@ public class MergeServiceTests
         [SetUp]
         public void Setup()
         {
+            _mongoDbContext = new MongoDbContextMock();
             _mergeBlacklistRepo = new MergeBlacklistRepositoryMock();
             _mergeGraylistRepo = new MergeGraylistRepositoryMock();
             _wordRepo = new WordRepositoryMock();
             _wordService = new WordService(_wordRepo);
-            _mergeService = new MergeService(_mergeBlacklistRepo, _mergeGraylistRepo, _wordRepo, _wordService);
+            _mergeService = new MergeService(
+                _mongoDbContext, _mergeBlacklistRepo, _mergeGraylistRepo, _wordRepo, _wordService);
         }
 
         [Test]

Backend/Contexts/BannerContext.cs

@@ -1,22 +1,20 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class BannerContext : IBannerContext
     {
-        private readonly IMongoDatabase _db;
+        private readonly IMongoDbContext _mongoDbContext;
 
-        public BannerContext(IOptions<Startup.Settings> options)
+        public BannerContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<Banner> Banners => _db.GetCollection<Banner>("BannerCollection");
+        public IMongoCollection<Banner> Banners => _mongoDbContext.Db.GetCollection<Banner>("BannerCollection");
     }
 }

Backend/Contexts/MergeBlacklistContext.cs

@@ -1,23 +1,20 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class MergeBlacklistContext : IMergeBlacklistContext
     {
-        private readonly IMongoDatabase _db;
-
-        public MergeBlacklistContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public MergeBlacklistContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<MergeWordSet> MergeBlacklist => _db.GetCollection<MergeWordSet>(
+        public IMongoCollection<MergeWordSet> MergeBlacklist => _mongoDbContext.Db.GetCollection<MergeWordSet>(
             "MergeBlacklistCollection");
     }
 }

Backend/Contexts/MergeGraylistContext.cs

@@ -1,23 +1,20 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class MergeGraylistContext : IMergeGraylistContext
     {
-        private readonly IMongoDatabase _db;
-
-        public MergeGraylistContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public MergeGraylistContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<MergeWordSet> MergeGraylist => _db.GetCollection<MergeWordSet>(
+        public IMongoCollection<MergeWordSet> MergeGraylist => _mongoDbContext.Db.GetCollection<MergeWordSet>(
             "MergeGraylistCollection");
     }
 }

Backend/Contexts/MongoDbContext.cs

@@ -0,0 +1,49 @@
+using System.Threading.Tasks;
+using BackendFramework.Interfaces;
+using Microsoft.Extensions.Options;
+using MongoDB.Driver;
+
+namespace BackendFramework.Contexts;
+
+public class MongoDbContext : IMongoDbContext
+{
+    public IMongoDatabase Db { get; }
+
+    public MongoDbContext(IOptions<Startup.Settings> options)
+    {
+        var client = new MongoClient(options.Value.ConnectionString);
+        Db = client.GetDatabase(options.Value.CombineDatabase);
+    }
+
+    public async Task<IMongoTransaction> BeginTransaction()
+    {
+        var session = await Db.Client.StartSessionAsync();
+        session.StartTransaction();
+        return new MongoTransactionWrapper(session);
+    }
+
+    private class MongoTransactionWrapper : IMongoTransaction
+    {
+        private readonly IClientSessionHandle _session;
+
+        public MongoTransactionWrapper(IClientSessionHandle session)
+        {
+            _session = session;
+        }
+
+        public Task CommitTransactionAsync()
+        {
+            return _session.CommitTransactionAsync();
+        }
+
+        public Task AbortTransactionAsync()
+        {
+            return _session.AbortTransactionAsync();
+        }
+
+        public void Dispose()
+        {
+            _session.Dispose();
+        }
+    }
+}

Backend/Contexts/PasswordResetContext.cs

@@ -10,17 +10,16 @@ namespace BackendFramework.Contexts
     [ExcludeFromCodeCoverage]
     public class PasswordResetContext : IPasswordResetContext
     {
-        private readonly IMongoDatabase _db;
+        private readonly IMongoDbContext _mongoDbContext;
         public int ExpireTime { get; }
 
-        public PasswordResetContext(IOptions<Startup.Settings> options)
+        public PasswordResetContext(IOptions<Startup.Settings> options, IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
             ExpireTime = options.Value.PassResetExpireTime;
         }
 
-        private IMongoCollection<PasswordReset> PasswordResets => _db.GetCollection<PasswordReset>(
+        private IMongoCollection<PasswordReset> PasswordResets => _mongoDbContext.Db.GetCollection<PasswordReset>(
             "PasswordResetCollection");
 
         public Task ClearAll(string email)

Backend/Contexts/ProjectContext.cs

@@ -1,22 +1,19 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class ProjectContext : IProjectContext
     {
-        private readonly IMongoDatabase _db;
-
-        public ProjectContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public ProjectContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<Project> Projects => _db.GetCollection<Project>("ProjectsCollection");
+        public IMongoCollection<Project> Projects => _mongoDbContext.Db.GetCollection<Project>("ProjectsCollection");
     }
 }

Backend/Contexts/SemanticDomainContext.cs

@@ -1,23 +1,20 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class SemanticDomainContext : ISemanticDomainContext
     {
-        private readonly IMongoDatabase _db;
-
-        public SemanticDomainContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public SemanticDomainContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<SemanticDomainTreeNode> SemanticDomains => _db.GetCollection<SemanticDomainTreeNode>("SemanticDomainTree");
-        public IMongoCollection<SemanticDomainFull> FullSemanticDomains => _db.GetCollection<SemanticDomainFull>("SemanticDomains");
+        public IMongoCollection<SemanticDomainTreeNode> SemanticDomains => _mongoDbContext.Db.GetCollection<SemanticDomainTreeNode>("SemanticDomainTree");
+        public IMongoCollection<SemanticDomainFull> FullSemanticDomains => _mongoDbContext.Db.GetCollection<SemanticDomainFull>("SemanticDomains");
     }
 }

Backend/Contexts/SpeakerContext.cs

@@ -1,22 +1,20 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class SpeakerContext : ISpeakerContext
     {
-        private readonly IMongoDatabase _db;
+        private readonly IMongoDbContext _mongoDbContext;
 
-        public SpeakerContext(IOptions<Startup.Settings> options)
+        public SpeakerContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<Speaker> Speakers => _db.GetCollection<Speaker>("SpeakersCollection");
+        public IMongoCollection<Speaker> Speakers => _mongoDbContext.Db.GetCollection<Speaker>("SpeakersCollection");
     }
 }

Backend/Contexts/UserContext.cs

@@ -1,22 +1,19 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class UserContext : IUserContext
     {
-        private readonly IMongoDatabase _db;
-
-        public UserContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public UserContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<User> Users => _db.GetCollection<User>("UsersCollection");
+        public IMongoCollection<User> Users => _mongoDbContext.Db.GetCollection<User>("UsersCollection");
     }
 }

Backend/Contexts/UserEditContext.cs

@@ -1,22 +1,19 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class UserEditContext : IUserEditContext
     {
-        private readonly IMongoDatabase _db;
-
-        public UserEditContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public UserEditContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<UserEdit> UserEdits => _db.GetCollection<UserEdit>("UserEditsCollection");
+        public IMongoCollection<UserEdit> UserEdits => _mongoDbContext.Db.GetCollection<UserEdit>("UserEditsCollection");
     }
 }

Backend/Contexts/UserRoleContext.cs

@@ -1,22 +1,19 @@
 using System.Diagnostics.CodeAnalysis;
 using BackendFramework.Interfaces;
 using BackendFramework.Models;
-using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 
 namespace BackendFramework.Contexts
 {
     [ExcludeFromCodeCoverage]
     public class UserRoleContext : IUserRoleContext
     {
-        private readonly IMongoDatabase _db;
-
-        public UserRoleContext(IOptions<Startup.Settings> options)
+        private readonly IMongoDbContext _mongoDbContext;
+        public UserRoleContext(IMongoDbContext mongoDbContext)
         {
-            var client = new MongoClient(options.Value.ConnectionString);
-            _db = client.GetDatabase(options.Value.CombineDatabase);
+            _mongoDbContext = mongoDbContext;
         }
 
-        public IMongoCollection<UserRole> UserRoles => _db.GetCollection<UserRole>("UserRolesCollection");
+        public IMongoCollection<UserRole> UserRoles => _mongoDbContext.Db.GetCollection<UserRole>("UserRolesCollection");
     }
 }