feat: 完成多db功能搭建

This commit is contained in:
chenchun
2024-11-19 11:53:57 +08:00
parent 9acb157fae
commit 5d2d269f11
10 changed files with 278 additions and 168 deletions

View File

@@ -15,7 +15,7 @@ namespace Yi.Framework.Rbac.SqlSugarCore
{
public override void ConfigureServices(ServiceConfigurationContext context)
{
context.Services.TryAddYiDbContext<YiRbacDbContextFactory>();
context.Services.AddYiDbContext<YiRbacDbContext>();
}
}
}

View File

@@ -1,5 +1,7 @@
using SqlSugar;
using Volo.Abp.DependencyInjection;
using Microsoft.Extensions.DependencyInjection;
using SqlSugar;
using Volo.Abp.Data;
using Volo.Abp.Users;
using Yi.Framework.Rbac.Domain.Authorization;
using Yi.Framework.Rbac.Domain.Entities;
using Yi.Framework.Rbac.Domain.Extensions;
@@ -9,21 +11,16 @@ using Yi.Framework.SqlSugarCore;
namespace Yi.Framework.Rbac.SqlSugarCore
{
public class YiRbacDbContextFactory : SqlSugarDbContextFactory
public class YiRbacDbContext : SqlSugarDbContext
{
public YiRbacDbContextFactory(IAbpLazyServiceProvider lazyServiceProvider) : base(lazyServiceProvider)
{
}
protected IDataFilter DataFilter => LazyServiceProvider.LazyGetRequiredService<IDataFilter>();
protected ICurrentUser CurrentUser => LazyServiceProvider.GetRequiredService<ICurrentUser>();
protected override void CustomDataFilter(ISqlSugarClient sqlSugarClient)
{
if (DataFilter.IsEnabled<IDataPermission>())
{
DataPermissionFilter(sqlSugarClient);
}
base.CustomDataFilter(sqlSugarClient);
}