feat: 完成多db功能搭建

2026-04-03 07:36:36 +08:00 · 2024-11-19 11:53:57 +08:00
parent 9acb157fae
commit 5d2d269f11
10 changed files with 278 additions and 168 deletions
--- a/Yi.Abp.Net8/module/rbac/Yi.Framework.Rbac.SqlSugarCore/YiRbacDbContext.cs
+++ b/Yi.Abp.Net8/module/rbac/Yi.Framework.Rbac.SqlSugarCore/YiRbacDbContext.cs
@@ -0,0 +1,90 @@
+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;
+using Yi.Framework.Rbac.Domain.Shared.Consts;
+using Yi.Framework.Rbac.Domain.Shared.Enums;
+using Yi.Framework.SqlSugarCore;
+
+namespace Yi.Framework.Rbac.SqlSugarCore
+{
+    public class YiRbacDbContext : SqlSugarDbContext
+    {
+        protected IDataFilter DataFilter => LazyServiceProvider.LazyGetRequiredService<IDataFilter>();
+        protected ICurrentUser CurrentUser => LazyServiceProvider.GetRequiredService<ICurrentUser>();
+        protected override void CustomDataFilter(ISqlSugarClient sqlSugarClient)
+        {
+            if (DataFilter.IsEnabled<IDataPermission>())
+            {
+                DataPermissionFilter(sqlSugarClient);
+            }
+        }
+
+
+        /// <summary>
+        /// 数据权限过滤
+        /// </summary>
+        protected void DataPermissionFilter(ISqlSugarClient sqlSugarClient)
+        {
+            //获取当前用户的信息
+            if (CurrentUser.Id == null || CurrentUser.IsRefreshToken()) return;
+            //管理员不过滤
+            if (CurrentUser.UserName.Equals(UserConst.Admin) || CurrentUser.Roles.Any(f => f.Equals(UserConst.AdminRolesCode))) return;
+            var expUser = Expressionable.Create<UserAggregateRoot>();
+            var expRole = Expressionable.Create<RoleAggregateRoot>();
+
+
+            var roleInfo = CurrentUser.GetRoleInfo();
+
+            //如果无岗位，或者无角色，只能看自己的数据
+            if (/*CurrentUser.GetDeptId() is null ||*/ roleInfo is null)
+            {
+                expUser.Or(it => it.Id == CurrentUser.Id);
+                expRole.Or(it => roleInfo.Select(x => x.Id).Contains(it.Id));
+            }
+            else
+            {
+                foreach (var role in roleInfo.OrderBy(f => f.DataScope))
+                {
+                    var dataScope = role.DataScope;
+                    if (DataScopeEnum.ALL.Equals(dataScope))//所有权限
+                    {
+                        break;
+                    }
+                    else if (DataScopeEnum.CUSTOM.Equals(dataScope))//自定数据权限
+                    {
+                        //" OR {}.dept_id IN ( SELECT dept_id FROM sys_role_dept WHERE role_id = {} ) ", deptAlias, role.getRoleId()));
+
+                        expUser.Or(it => SqlFunc.Subqueryable<RoleDeptEntity>().Where(f => f.DeptId == it.DeptId && f.RoleId == role.Id).Any());
+                    }
+                    else if (DataScopeEnum.DEPT.Equals(dataScope))//本部门数据
+                    {
+                        expUser.Or(it => it.DeptId == CurrentUser.GetDeptId());
+                    }
+                    else if (DataScopeEnum.DEPT_FOLLOW.Equals(dataScope))//本部门及以下数据
+                    {
+                        //SQl  OR {}.dept_id IN ( SELECT dept_id FROM sys_dept WHERE dept_id = {} or find_in_set( {} , ancestors ) )
+                        var allChildDepts = sqlSugarClient.Queryable<DeptAggregateRoot>().ToChildList(it => it.ParentId, CurrentUser.GetDeptId());
+
+                        expUser.Or(it => allChildDepts.Select(f => f.Id).ToList().Contains(it.DeptId ?? Guid.Empty));
+                    }
+                    else if (DataScopeEnum.USER.Equals(dataScope))//仅本人数据
+                    {
+                        expUser.Or(it => it.Id == CurrentUser.Id);
+                        expRole.Or(it => roleInfo.Select(x => x.Id).Contains(it.Id));
+
+                    }
+                }
+
+            }
+
+
+
+            sqlSugarClient.QueryFilter.AddTableFilter(expUser.ToExpression());
+            sqlSugarClient.QueryFilter.AddTableFilter(expRole.ToExpression());
+        }
+    }
+}