From 5168b9d3755153b11cf9f94eac48e51b6fedd3cf Mon Sep 17 00:00:00 2001
From: colin <colin.in@foxmail.com>
Date: Fri, 16 Jan 2026 11:03:34 +0800
Subject: [PATCH] feat: Add Workspace EFCore Repository Impl

---
 .../Workspaces/IWorkspaceRepository.cs        |  2 +-
 ...bpAIManagementEntityFrameworkCoreModule.cs |  5 +++-
 .../EfCoreWorkspaceRepository.cs              | 26 +++++++++++++++++++
 3 files changed, 31 insertions(+), 2 deletions(-)
 create mode 100644 aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/EfCoreWorkspaceRepository.cs

diff --git a/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.Domain/LINGYUN/Abp/AIManagement/Workspaces/IWorkspaceRepository.cs b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.Domain/LINGYUN/Abp/AIManagement/Workspaces/IWorkspaceRepository.cs
index 5405b0b0a..db729e5a5 100644
--- a/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.Domain/LINGYUN/Abp/AIManagement/Workspaces/IWorkspaceRepository.cs
+++ b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.Domain/LINGYUN/Abp/AIManagement/Workspaces/IWorkspaceRepository.cs
@@ -6,7 +6,7 @@ using Volo.Abp.Domain.Repositories;
 namespace LINGYUN.Abp.AIManagement.Workspaces;
 public interface IWorkspaceRepository : IBasicRepository<Workspace, Guid>
 {
-    Task<Workspace> FindByNameAsync(
+    Task<Workspace?> FindByNameAsync(
         string name,
         CancellationToken cancellationToken = default);
 }
diff --git a/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/AbpAIManagementEntityFrameworkCoreModule.cs b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/AbpAIManagementEntityFrameworkCoreModule.cs
index adbf7146d..566db1fc1 100644
--- a/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/AbpAIManagementEntityFrameworkCoreModule.cs
+++ b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/AbpAIManagementEntityFrameworkCoreModule.cs
@@ -1,4 +1,5 @@
-using Microsoft.Extensions.DependencyInjection;
+using LINGYUN.Abp.AIManagement.Workspaces;
+using Microsoft.Extensions.DependencyInjection;
 using Volo.Abp.EntityFrameworkCore;
 using Volo.Abp.Modularity;
 
@@ -14,6 +15,8 @@ public class AbpAIManagementEntityFrameworkCoreModule : AbpModule
         context.Services.AddAbpDbContext<AIManagementDbContext>(options =>
         {
             options.AddDefaultRepositories<IAIManagementDbContext>();
+
+            options.AddRepository<Workspace, EfCoreWorkspaceRepository>();
         });
     }
 }
diff --git a/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/EfCoreWorkspaceRepository.cs b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/EfCoreWorkspaceRepository.cs
new file mode 100644
index 000000000..c0fcc2cb9
--- /dev/null
+++ b/aspnet-core/modules/ai/LINGYUN.Abp.AIManagement.EntityFrameworkCore/LINGYUN/Abp/AIManagement/EntityFrameworkCore/EfCoreWorkspaceRepository.cs
@@ -0,0 +1,26 @@
+using LINGYUN.Abp.AIManagement.Workspaces;
+using Microsoft.EntityFrameworkCore;
+using System;
+using System.Linq;
+using System.Linq.Dynamic.Core;
+using System.Threading;
+using System.Threading.Tasks;
+using Volo.Abp.Domain.Repositories.EntityFrameworkCore;
+using Volo.Abp.EntityFrameworkCore;
+
+namespace LINGYUN.Abp.AIManagement.EntityFrameworkCore;
+public class EfCoreWorkspaceRepository : EfCoreRepository<IAIManagementDbContext, Workspace, Guid>, IWorkspaceRepository
+{
+    public EfCoreWorkspaceRepository(
+        IDbContextProvider<IAIManagementDbContext> dbContextProvider) 
+        : base(dbContextProvider)
+    {
+    }
+
+    public async virtual Task<Workspace?> FindByNameAsync(string name, CancellationToken cancellationToken = default)
+    {
+        return await (await GetQueryableAsync())
+            .Where(x => x.Name == name)
+            .FirstOrDefaultAsync(GetCancellationToken(cancellationToken)); ;
+    }
+}