From 8b8c750dc08553dbee1c1c138bbd87d66534271e Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Tue, 7 Mar 2023 10:46:57 +0800
Subject: [PATCH] Add `GetListByIdsAsync` method to `IIdentityUserRepository`.

---
 .../Volo/Abp/Identity/IIdentityUserRepository.cs          | 6 ++++++
 .../EfCoreIdentityLinkUserRepository.cs                   | 6 ++++--
 .../EntityFrameworkCore/EfCoreIdentityUserRepository.cs   | 8 ++++++++
 .../Abp/Identity/MongoDB/MongoIdentityUserRepository.cs   | 7 +++++++
 4 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IIdentityUserRepository.cs b/modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IIdentityUserRepository.cs
index d2a049dd84..d8b3543703 100644
--- a/modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IIdentityUserRepository.cs
+++ b/modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IIdentityUserRepository.cs
@@ -125,4 +125,10 @@ public interface IIdentityUserRepository : IBasicRepository<IdentityUser, Guid>
         bool includeDetails = true,
         CancellationToken cancellationToken = default
     );
+
+    Task<List<IdentityUser>> GetListByIdsAsync(
+        IEnumerable<Guid> ids,
+        bool includeDetails = false,
+        CancellationToken cancellationToken = default
+    );
 }
diff --git a/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityLinkUserRepository.cs b/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityLinkUserRepository.cs
index 6a0245fa4f..035904120c 100644
--- a/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityLinkUserRepository.cs
+++ b/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityLinkUserRepository.cs
@@ -20,6 +20,7 @@ public class EfCoreIdentityLinkUserRepository : EfCoreRepository<IIdentityDbCont
     public virtual async Task<IdentityLinkUser> FindAsync(IdentityLinkUserInfo sourceLinkUserInfo, IdentityLinkUserInfo targetLinkUserInfo, CancellationToken cancellationToken = default)
     {
         return await (await GetDbSetAsync())
+            .AsNoTracking()
             .OrderBy(x => x.Id).FirstOrDefaultAsync(x =>
                 x.SourceUserId == sourceLinkUserInfo.UserId && x.SourceTenantId == sourceLinkUserInfo.TenantId &&
                 x.TargetUserId == targetLinkUserInfo.UserId && x.TargetTenantId == targetLinkUserInfo.TenantId ||
@@ -31,7 +32,8 @@ public class EfCoreIdentityLinkUserRepository : EfCoreRepository<IIdentityDbCont
     public virtual async Task<List<IdentityLinkUser>> GetListAsync(IdentityLinkUserInfo linkUserInfo, List<IdentityLinkUserInfo> excludes = null,
         CancellationToken cancellationToken = default)
     {
-        IQueryable<IdentityLinkUser> query = (await GetDbSetAsync())
+        var query = (await GetDbSetAsync())
+            .AsNoTracking()
             .Where(x =>
                 x.SourceUserId == linkUserInfo.UserId && x.SourceTenantId == linkUserInfo.TenantId ||
                 x.TargetUserId == linkUserInfo.UserId && x.TargetTenantId == linkUserInfo.TenantId);
@@ -51,7 +53,7 @@ public class EfCoreIdentityLinkUserRepository : EfCoreRepository<IIdentityDbCont
 
     public virtual async Task DeleteAsync(IdentityLinkUserInfo linkUserInfo, CancellationToken cancellationToken = default)
     {
-        var linkUsers = await (await GetDbSetAsync()).Where(x =>
+        var linkUsers = await (await GetDbSetAsync()).AsNoTracking().Where(x =>
                 x.SourceUserId == linkUserInfo.UserId && x.SourceTenantId == linkUserInfo.TenantId ||
                 x.TargetUserId == linkUserInfo.UserId && x.TargetTenantId == linkUserInfo.TenantId)
             .ToListAsync(cancellationToken: GetCancellationToken(cancellationToken));
diff --git a/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityUserRepository.cs b/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityUserRepository.cs
index e2527555b2..60abbb3054 100644
--- a/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityUserRepository.cs
+++ b/modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreIdentityUserRepository.cs
@@ -348,4 +348,12 @@ public class EfCoreIdentityUserRepository : EfCoreRepository<IIdentityDbContext,
                 GetCancellationToken(cancellationToken)
             );
     }
+
+    public virtual async Task<List<IdentityUser>> GetListByIdsAsync(IEnumerable<Guid> ids, bool includeDetails = false, CancellationToken cancellationToken = default)
+    {
+        return await (await GetDbSetAsync())
+            .IncludeDetails(includeDetails)
+            .Where(x => ids.Contains(x.Id))
+            .ToListAsync(GetCancellationToken(cancellationToken));
+    }
 }
diff --git a/modules/identity/src/Volo.Abp.Identity.MongoDB/Volo/Abp/Identity/MongoDB/MongoIdentityUserRepository.cs b/modules/identity/src/Volo.Abp.Identity.MongoDB/Volo/Abp/Identity/MongoDB/MongoIdentityUserRepository.cs
index ac2ab63809..ae805d9a03 100644
--- a/modules/identity/src/Volo.Abp.Identity.MongoDB/Volo/Abp/Identity/MongoDB/MongoIdentityUserRepository.cs
+++ b/modules/identity/src/Volo.Abp.Identity.MongoDB/Volo/Abp/Identity/MongoDB/MongoIdentityUserRepository.cs
@@ -316,4 +316,11 @@ public class MongoIdentityUserRepository : MongoDbRepository<IAbpIdentityMongoDb
                 GetCancellationToken(cancellationToken)
             );
     }
+
+    public virtual async Task<List<IdentityUser>> GetListByIdsAsync(IEnumerable<Guid> ids, bool includeDetails = false, CancellationToken cancellationToken = default)
+    {
+        return await (await GetMongoQueryableAsync(cancellationToken))
+            .Where(x => ids.Contains(x.Id))
+            .ToListAsync(GetCancellationToken(cancellationToken));
+    }
 }