From 02ce9a186f1a905410279ed76cdc014828d78dd5 Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Fri, 2 Sep 2022 09:51:47 +0800
Subject: [PATCH] Implement MongoDB modules.

---
 .../IFeatureDefinitionRecordRepository.cs     |  6 +++-
 ...EfCoreFeatureDefinitionRecordRepository.cs | 11 +++++++
 .../AbpFeatureManagementMongoDbModule.cs      |  2 ++
 .../FeatureManagementMongoDbContext.cs        |  2 ++
 ...atureManagementMongoDbContextExtensions.cs | 10 +++++++
 .../IFeatureManagementMongoDbContext.cs       |  4 +++
 .../MongoFeatureDefinitionRecordRepository.cs | 30 +++++++++++++++++++
 ...oFeatureGroupDefinitionRecordRepository.cs | 16 ++++++++++
 .../MongoDB/MongoFeatureValueRepository.cs    |  4 ++-
 9 files changed, 83 insertions(+), 2 deletions(-)
 create mode 100644 modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureDefinitionRecordRepository.cs
 create mode 100644 modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureGroupDefinitionRecordRepository.cs

diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.Domain/Volo/Abp/FeatureManagement/IFeatureDefinitionRecordRepository.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.Domain/Volo/Abp/FeatureManagement/IFeatureDefinitionRecordRepository.cs
index 35144df2f0..d8718903c9 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.Domain/Volo/Abp/FeatureManagement/IFeatureDefinitionRecordRepository.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.Domain/Volo/Abp/FeatureManagement/IFeatureDefinitionRecordRepository.cs
@@ -1,9 +1,13 @@
 using System;
+using System.Threading;
+using System.Threading.Tasks;
 using Volo.Abp.Domain.Repositories;
 
 namespace Volo.Abp.FeatureManagement;
 
 public interface IFeatureDefinitionRecordRepository : IBasicRepository<FeatureDefinitionRecord, Guid>
 {
-
+    Task<FeatureDefinitionRecord> FindByNameAsync(
+        string name,
+        CancellationToken cancellationToken = default);
 }
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.EntityFrameworkCore/Volo/Abp/FeatureManagement/EntityFrameworkCore/EfCoreFeatureDefinitionRecordRepository.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.EntityFrameworkCore/Volo/Abp/FeatureManagement/EntityFrameworkCore/EfCoreFeatureDefinitionRecordRepository.cs
index 91648af33b..3547b9c009 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.EntityFrameworkCore/Volo/Abp/FeatureManagement/EntityFrameworkCore/EfCoreFeatureDefinitionRecordRepository.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.EntityFrameworkCore/Volo/Abp/FeatureManagement/EntityFrameworkCore/EfCoreFeatureDefinitionRecordRepository.cs
@@ -1,4 +1,8 @@
 using System;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
 using Volo.Abp.Domain.Repositories.EntityFrameworkCore;
 using Volo.Abp.EntityFrameworkCore;
 
@@ -13,4 +17,11 @@ public class EfCoreFeatureDefinitionRecordRepository :
         : base(dbContextProvider)
     {
     }
+
+    public virtual async Task<FeatureDefinitionRecord> FindByNameAsync(string name, CancellationToken cancellationToken = default)
+    {
+        return await (await GetDbSetAsync())
+            .OrderBy(x => x.Id)
+            .FirstOrDefaultAsync(r => r.Name == name, cancellationToken);
+    }
 }
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/AbpFeatureManagementMongoDbModule.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/AbpFeatureManagementMongoDbModule.cs
index 45d3d37271..cf1f917115 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/AbpFeatureManagementMongoDbModule.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/AbpFeatureManagementMongoDbModule.cs
@@ -16,6 +16,8 @@ public class AbpFeatureManagementMongoDbModule : AbpModule
         {
             options.AddDefaultRepositories<IFeatureManagementMongoDbContext>();
 
+            options.AddRepository<FeatureGroupDefinitionRecord, MongoFeatureGroupDefinitionRecordRepository>();
+            options.AddRepository<FeatureDefinitionRecord, MongoFeatureDefinitionRecordRepository>();
             options.AddRepository<FeatureValue, MongoFeatureValueRepository>();
         });
     }
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContext.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContext.cs
index 37273a3dcc..7709276746 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContext.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContext.cs
@@ -9,6 +9,8 @@ namespace Volo.Abp.FeatureManagement.MongoDB;
 [ConnectionStringName(AbpFeatureManagementDbProperties.ConnectionStringName)]
 public class FeatureManagementMongoDbContext : AbpMongoDbContext, IFeatureManagementMongoDbContext
 {
+    public IMongoCollection<FeatureGroupDefinitionRecord> FeatureGroups => Collection<FeatureGroupDefinitionRecord>();
+    public IMongoCollection<FeatureDefinitionRecord> Features => Collection<FeatureDefinitionRecord>();
     public IMongoCollection<FeatureValue> FeatureValues => Collection<FeatureValue>();
 
     protected override void CreateModel(IMongoModelBuilder modelBuilder)
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContextExtensions.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContextExtensions.cs
index 8d263bb71e..bb1e035659 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContextExtensions.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/FeatureManagementMongoDbContextExtensions.cs
@@ -9,6 +9,16 @@ public static class FeatureManagementMongoDbContextExtensions
     {
         Check.NotNull(builder, nameof(builder));
 
+        builder.Entity<FeatureGroupDefinitionRecord>(b =>
+        {
+            b.CollectionName = AbpFeatureManagementDbProperties.DbTablePrefix + "FeatureGroups";
+        });
+
+        builder.Entity<FeatureDefinitionRecord>(b =>
+        {
+            b.CollectionName = AbpFeatureManagementDbProperties.DbTablePrefix + "Features";
+        });
+
         builder.Entity<FeatureValue>(b =>
         {
             b.CollectionName = AbpFeatureManagementDbProperties.DbTablePrefix + "FeatureValues";
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/IFeatureManagementMongoDbContext.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/IFeatureManagementMongoDbContext.cs
index 1fefaf67cf..65004f1d7e 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/IFeatureManagementMongoDbContext.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/IFeatureManagementMongoDbContext.cs
@@ -9,5 +9,9 @@ namespace Volo.Abp.FeatureManagement.MongoDB;
 [ConnectionStringName(AbpFeatureManagementDbProperties.ConnectionStringName)]
 public interface IFeatureManagementMongoDbContext : IAbpMongoDbContext
 {
+    IMongoCollection<FeatureGroupDefinitionRecord> FeatureGroups { get; }
+
+    IMongoCollection<FeatureDefinitionRecord> Features { get; }
+
     IMongoCollection<FeatureValue> FeatureValues { get; }
 }
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureDefinitionRecordRepository.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureDefinitionRecordRepository.cs
new file mode 100644
index 0000000000..cb7f48aeb4
--- /dev/null
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureDefinitionRecordRepository.cs
@@ -0,0 +1,30 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+using MongoDB.Driver.Linq;
+using Volo.Abp.Domain.Repositories.MongoDB;
+using Volo.Abp.MongoDB;
+
+namespace Volo.Abp.FeatureManagement.MongoDB;
+
+public class MongoFeatureDefinitionRecordRepository :
+    MongoDbRepository<IFeatureManagementMongoDbContext, FeatureDefinitionRecord, Guid>,
+    IFeatureDefinitionRecordRepository
+{
+    public MongoFeatureDefinitionRecordRepository(
+        IMongoDbContextProvider<IFeatureManagementMongoDbContext> dbContextProvider)
+        : base(dbContextProvider)
+    {
+    }
+
+    public virtual async Task<FeatureDefinitionRecord> FindByNameAsync(string name, CancellationToken cancellationToken = default)
+    {
+        cancellationToken = GetCancellationToken(cancellationToken);
+        return await (await GetMongoQueryableAsync(cancellationToken))
+            .OrderBy(x => x.Id)
+            .FirstOrDefaultAsync(
+                s => s.Name == name,
+                cancellationToken
+            );
+    }
+}
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureGroupDefinitionRecordRepository.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureGroupDefinitionRecordRepository.cs
new file mode 100644
index 0000000000..279479dcfa
--- /dev/null
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureGroupDefinitionRecordRepository.cs
@@ -0,0 +1,16 @@
+using System;
+using Volo.Abp.Domain.Repositories.MongoDB;
+using Volo.Abp.MongoDB;
+
+namespace Volo.Abp.FeatureManagement.MongoDB;
+
+public class MongoFeatureGroupDefinitionRecordRepository :
+    MongoDbRepository<IFeatureManagementMongoDbContext, FeatureGroupDefinitionRecord, Guid>,
+    IFeatureGroupDefinitionRecordRepository
+{
+    public MongoFeatureGroupDefinitionRecordRepository(
+        IMongoDbContextProvider<IFeatureManagementMongoDbContext> dbContextProvider)
+        : base(dbContextProvider)
+    {
+    }
+}
diff --git a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureValueRepository.cs b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureValueRepository.cs
index 61d59b893b..9d9071ea1a 100644
--- a/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureValueRepository.cs
+++ b/modules/feature-management/src/Volo.Abp.FeatureManagement.MongoDB/Volo/Abp/FeatureManagement/MongoDB/MongoFeatureValueRepository.cs
@@ -9,7 +9,9 @@ using Volo.Abp.MongoDB;
 
 namespace Volo.Abp.FeatureManagement.MongoDB;
 
-public class MongoFeatureValueRepository : MongoDbRepository<IFeatureManagementMongoDbContext, FeatureValue, Guid>, IFeatureValueRepository
+public class MongoFeatureValueRepository :
+    MongoDbRepository<IFeatureManagementMongoDbContext, FeatureValue, Guid>,
+    IFeatureValueRepository
 {
     public MongoFeatureValueRepository(IMongoDbContextProvider<IFeatureManagementMongoDbContext> dbContextProvider)
         : base(dbContextProvider)