Yi.Framework.Net6/Yi.Framework.WebCore/DbExtend/DbFiterExtend.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using Microsoft.Extensions.DependencyInjection;
using SqlSugar;
using System;
using System.Linq;
using Yi.Framework.Common.Const;
using Yi.Framework.Common.Enum;
using Yi.Framework.Common.Models;
using Yi.Framework.DTOModel;
using Yi.Framework.Model.Models;
using Yi.Framework.WebCore;

namespace Yi.Framework.Core
{
    public class DbFiterExtend
    {
        public static void Data(SqlSugarClient db)
        {
            //非请求情况
            if (!ServiceLocator.GetHttp(out var httpContext))
            {
                return;
            }

            //无需授权情况
            var userName = httpContext?.GetUserIdInfo();
            if (userName is null)
            {
                return;
            }

            //超级管理员直接放行
            if (SystemConst.Admin.Equals(userName))
            {
                return;
            }

            var userId = httpContext?.GetUserIdInfo();

            //根据缓存获取全部用户信息
            var userRoleMenu = ServiceLocator.Instance?.GetService<CacheInvoker>()?.Get<UserRoleMenuDto>($"Yi:UserInfo:{userId}");


            var roles = userRoleMenu?.Roles;
            if (roles.IsNull())
            {
                roles = new();
            }

            long deptId = userRoleMenu?.User.DeptId ?? -1;
            //根据角色的数据范围，来添加相对于的数据权限
            if (roles is not null)
            {
                foreach (var role in roles)
                {
                    //默认为全部
                    DataScopeEnum dataScope = (DataScopeEnum)(role.DataScope ?? DataScopeEnum.ALL.GetHashCode());
                    switch (dataScope)
                    {
                        case DataScopeEnum.ALL:
                            //直接放行