1

iMES.Core/UserManager/DepartmentContext.cs

@@ -0,0 +1,116 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
+using System.Linq;
+using System.Text;
+using iMES.Core.CacheManager;
+using iMES.Core.DBManager;
+using iMES.Core.Extensions.AutofacManager;
+using iMES.Core.Utilities;
+using iMES.Entity.DomainModels;
+
+namespace iMES.Core.UserManager
+{
+   public static class DepartmentContext
+    {
+        static DepartmentContext()
+        {
+            GetAllDept();
+        }
+        private static object _deptObj = new object();
+
+
+        private static string _deptVersionn = "";
+
+        public const string _deptCacheKey = "inernalDept";
+        private static List<Dept> _depts { get; set; }
+        public static ICacheService CacheContext
+        {
+            get
+            {
+                return AutofacContainerModule.GetService<ICacheService>();
+            }
+        }
+
+        public static List<Dept> GetAllDept()
+        {
+            //每次比较缓存是否更新过，如果更新则重新获取数据
+            string _cacheVersion = CacheContext.Get(_deptCacheKey);
+            if (_deptVersionn != "" && _deptVersionn == _cacheVersion)
+            {
+                return _depts ?? new List<Dept>();
+            }
+            lock (_deptObj)
+            {
+                if (_deptVersionn != "" && _depts != null && _deptVersionn == _cacheVersion)
+                {
+                    return _depts;
+                }
+
+                _depts = DBServerProvider.DbContext.Set<Sys_Department>()
+                    .Select(s => new Dept()
+                    {
+                        id = s.DepartmentId,
+                        key = s.DepartmentId,
+                        value = s.DepartmentName,
+                        parentId = s.ParentId
+                    }).ToList();
+
+                string cacheVersion = CacheContext.Get(_deptCacheKey);
+                if (string.IsNullOrEmpty(cacheVersion))
+                {
+                    cacheVersion = DateTime.Now.ToString("yyyyMMddHHMMssfff");
+                    CacheContext.Add(_deptCacheKey, cacheVersion);
+                }
+                else
+                {
+                    _deptVersionn = cacheVersion;
+                }
+            }
+            return _depts;
+        }
+
+
+        public static List<Guid> GetAllChildrenIds([NotNull] List<Guid> ids)
+        {
+            ids = ids.Distinct().ToList();
+            if (ids.Count == 0)
+            {
+                return new List<Guid>() { Guid.NewGuid() };
+            }
+             
+            for (int i = 0; i < ids.Count(); i++)
+            {
+                Guid id = ids[i];
+                var list = _depts.Where(x => x.parentId == id && !ids.Contains(x.id)).Select(s => s.id).Distinct().ToList();
+                if (list.Count > 0)
+                { 
+                    ids.AddRange(list);
+                }
+            }
+
+            return ids;
+        }
+        public static List<Guid> GetAllChildrenIds(Guid id)
+        {
+            return GetAllChildrenIds(new List<Guid>() { id });
+        }
+
+        public static WebResponseContent Reload(this WebResponseContent webResponse)
+        {
+            if (webResponse.Status)
+            {
+                CacheContext.Remove(_deptCacheKey);
+            }
+            return webResponse;
+        }
+    }
+
+}
+public partial class Dept
+{
+    public Guid id { get; set; }
+    public Guid? parentId { get; set; }
+    public Guid key { get; set; }
+    public string value { get; set; }
+}

iMES.Core/UserManager/RoleContext.cs

@@ -0,0 +1,136 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using iMES.Core.CacheManager;
+using iMES.Core.DBManager;
+using iMES.Core.Extensions.AutofacManager;
+using iMES.Core.ManageUser;
+using iMES.Core.Services;
+using iMES.Entity.DomainModels;
+
+namespace iMES.Core.UserManager
+{
+    public static class RoleContext
+    {
+
+        private static object _RoleObj = new object();
+        private static string _RoleVersionn = "";
+        public const string Key = "inernalRole";
+
+        private static List<RoleNodes> _roles { get; set; }
+        public static List<RoleNodes> GetAllRoleId()
+        {
+            ICacheService cacheService = AutofacContainerModule.GetService<ICacheService>();
+            //每次比较缓存是否更新过，如果更新则重新获取数据
+            if (_roles != null && _RoleVersionn == cacheService.Get(Key))
+            {
+                return _roles;
+            }
+            lock (_RoleObj)
+            {
+                if (_RoleVersionn != "" && _roles != null && _RoleVersionn == cacheService.Get(Key)) return _roles;
+                _roles = DBServerProvider.DbContext
+                  .Set<Sys_Role>()
+                   .Where(x => x.Enable == 1)
+                   .Select(s => new RoleNodes() { Id = s.Role_Id, ParentId = s.ParentId, RoleName = s.RoleName })
+             .ToList();
+
+                string cacheVersion = cacheService.Get(Key);
+                if (string.IsNullOrEmpty(cacheVersion))
+                {
+                    cacheVersion = DateTime.Now.ToString("yyyyMMddHHMMssfff");
+                    cacheService.Add(Key, cacheVersion);
+                }
+                else
+                {
+                    _RoleVersionn = cacheVersion;
+                }
+            }
+            return _roles;
+        }
+
+        public static void Refresh()
+        {
+            AutofacContainerModule.GetService<ICacheService>().Remove(Key);
+        }
+        /// <summary>
+        /// 
+        /// 获取当前角色下的所有角色(不包括自己的角色)
+        /// </summary>
+        /// <param name="roleId"></param>
+        /// <returns></returns>
+        public static List<RoleNodes> GetAllChildren(int roleId)
+        {
+            if (roleId <= 0) return null;
+            var roles = GetAllRoleId();
+            if (UserContext.IsRoleIdSuperAdmin(roleId)) return roles;
+            Dictionary<int, bool> completedRoles = new Dictionary<int, bool>();
+            List<RoleNodes> rolesChildren = new List<RoleNodes>();
+            var list= GetChildren(roles, rolesChildren, roleId, completedRoles);
+            //2021.07.11增加无限递归异常数据移除当前节点
+            if (list.Any(x=>x.Id==roleId))
+            {
+                return list.Where(x => x.Id != roleId).ToList();
+            }
+            return list;
+        }
+        public static List<int> GetAllChildrenIds(int roleId)
+        {
+            return GetAllChildren(roleId)?.Select(x => x.Id)?.ToList();
+        }
+        /// <summary>
+        /// 递归获取所有子节点权限
+        /// </summary>
+        /// <param name="roleId"></param>
+        private static List<RoleNodes> GetChildren(List<RoleNodes> roles, List<RoleNodes> rolesChildren, int roleId, Dictionary<int, bool> completedRoles)
+        {
+            //2021.07.11修复不能获取三级以下角色的问题
+            roles.ForEach(x =>
+            {
+                if (x.ParentId == roleId)
+                {
+                    if (completedRoles.TryGetValue(x.Id, out bool isWrite))
+                    {
+                        if (!isWrite)
+                        {
+                            roles.Where(x => x.Id == roleId).FirstOrDefault().ParentId = 0;
+                            Logger.Error($"获取子角色异常RoleContext,角色id:{x.Id}");
+                            Console.WriteLine($"获取子角色异常RoleContext,角色id:{x.Id}");
+                            completedRoles[x.Id] = true;
+                        }
+                        return;
+                    }
+                    if (!rolesChildren.Any(c => c.Id == x.Id))
+                    {
+                        rolesChildren.Add(x);
+                    }
+                    completedRoles.Add(x.Id, false);
+                    GetChildren(roles, rolesChildren, x.Id, completedRoles);
+                }
+            });
+            return rolesChildren;
+        }
+        /// <summary>
+        /// 获取当前角色下的所有用户
+        /// </summary>
+        /// <returns></returns>
+        public static IQueryable<int> GetCurrentAllChildUser()
+        {
+            var roles = GetAllChildrenIds(UserContext.Current.RoleId);
+            if (roles == null)
+            {
+                throw new Exception("未获取到当前角色");
+            }
+            return DBServerProvider.DbContext
+                  .Set<Sys_User>()
+                  .Where(u => roles.Contains(u.Role_Id)).Select(s => s.User_Id);
+
+        }
+    }
+    public class RoleNodes
+    {
+        public int Id { get; set; }
+        public int ParentId { get; set; }
+        public string RoleName { get; set; }
+    }
+}

iMES.Core/UserManager/UserContext.cs

@@ -0,0 +1,395 @@
+using Microsoft.EntityFrameworkCore;
+using System;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.IdentityModel.Tokens.Jwt;
+using System.Linq;
+using System.Security.Claims;
+using System.Threading.Tasks;
+using iMES.Core.CacheManager;
+using iMES.Core.DBManager;
+using iMES.Core.Enums;
+using iMES.Core.Extensions;
+using iMES.Core.Extensions.AutofacManager;
+using iMES.Entity;
+using iMES.Entity.DomainModels;
+using iMES.Core.UserManager;
+
+namespace iMES.Core.ManageUser
+{
+    public class UserContext
+    {
+        /// <summary>
+        /// 为了尽量减少redis或Memory读取,保证执行效率,将UserContext注入到DI，
+        /// 每个UserContext的属性至多读取一次redis或Memory缓存从而提高查询效率
+        /// </summary>
+        public static UserContext Current
+        {
+            get
+            {
+                return Context.RequestServices.GetService(typeof(UserContext)) as UserContext;
+            }
+        }
+
+        private static Microsoft.AspNetCore.Http.HttpContext Context
+        {
+            get
+            {
+                return Utilities.HttpContext.Current;
+            }
+        }
+        private static ICacheService CacheService
+        {
+            get { return GetService<ICacheService>(); }
+        }
+
+        private static T GetService<T>() where T : class
+        {
+            return AutofacContainerModule.GetService<T>();
+        }
+
+        public UserInfo UserInfo
+        {
+            get
+            {
+                if (_userInfo != null)
+                {
+                    return _userInfo;
+                }
+                return GetUserInfo(UserId);
+            }
+        }
+
+        private UserInfo _userInfo { get; set; }
+
+        /// <summary>
+        /// 角色ID为1的默认为超级管理员
+        /// </summary>
+        public bool IsSuperAdmin
+        {
+            get { return IsRoleIdSuperAdmin(this.RoleId); }
+        }
+        /// <summary>
+        /// 角色ID为1的默认为超级管理员
+        /// </summary>
+        public static bool IsRoleIdSuperAdmin(int roleId)
+        {
+            return roleId == 1;
+        }
+
+        public UserInfo GetUserInfo(int userId)
+        {
+            if (_userInfo != null) return _userInfo;
+            if (userId <= 0)
+            {
+                _userInfo = new UserInfo();
+                return _userInfo;
+            }
+            string key = userId.GetUserIdKey();
+            _userInfo = CacheService.Get<UserInfo>(key);
+            if (_userInfo != null && _userInfo.User_Id > 0) return _userInfo;
+
+            _userInfo = DBServerProvider.DbContext.Set<Sys_User>()
+                .Where(x => x.User_Id == userId).Select(s => new
+                {
+                    User_Id = userId,
+                    Role_Id = s.Role_Id.GetInt(),
+                    RoleName = s.RoleName,
+                    //2022.08.15增加部门id
+                    DeptId = s.Dept_Id??0,
+                    Token = s.Token,
+                    UserName = s.UserName,
+                    UserTrueName = s.UserTrueName,
+                    Enable = s.Enable,
+                    DeptIds= s.DeptIds
+                }).ToList().Select(s => new UserInfo()
+                {
+                    User_Id = userId,
+                    Role_Id = s.Role_Id,
+                    Token = s.Token,
+                    UserName = s.UserName,
+                    UserTrueName = s.UserTrueName,
+                    Enable = 1,
+                    DeptIds = string.IsNullOrEmpty(s.DeptIds) ? new List<Guid>() : s.DeptIds.Split(",").Select(x => (Guid)x.GetGuid()).ToList(),
+                }).FirstOrDefault();
+
+            if (_userInfo != null && _userInfo.User_Id > 0)
+            {
+                CacheService.AddObject(key, _userInfo);
+            }
+            return _userInfo ?? new UserInfo();
+        }
+
+        /// <summary>
+        /// 获取角色权限时通过安全字典锁定的角色id
+        /// </summary>
+        private static ConcurrentDictionary<string, object> objKeyValue = new ConcurrentDictionary<string, object>();
+
+        /// <summary>
+        /// 角色权限的版本号
+        /// </summary>
+        private static readonly Dictionary<int, string> rolePermissionsVersion = new Dictionary<int, string>();
+
+        /// <summary>
+        /// 每个角色ID对应的菜单权限（已做静态化处理）
+        /// 每次获取权限时用当前服务器的版本号与redis/memory缓存的版本比较,如果不同会重新刷新缓存
+        /// </summary>
+        private static readonly Dictionary<int, List<Permissions>> rolePermissions = new Dictionary<int, List<Permissions>>();
+
+
+
+        /// <summary>
+        /// 获取用户所有的菜单权限
+        /// </summary>
+
+        public List<Permissions> Permissions
+        {
+            get
+            {
+                return GetPermissions(RoleId);
+            }
+        }
+
+        /// <summary>
+        /// 菜单按钮变更时，同时刷新权限缓存2022.05.23
+        /// </summary>
+        /// <param name="menuId"></param>
+        public void RefreshWithMenuActionChange(int menuId)
+        {
+            foreach (var roleId in rolePermissions.Where(c => c.Value.Any(x => x.Menu_Id == menuId)).Select(s => s.Key))
+            {
+                if (rolePermissionsVersion.ContainsKey(roleId))
+                {
+                    CacheService.Add(roleId.GetRoleIdKey(), DateTime.Now.ToString("yyyyMMddHHMMssfff"));
+                }
+            }
+
+        }
+
+        /// <summary>
+        /// 获取单个表的权限
+        /// </summary>
+        /// <param name="tableName"></param>
+        /// <returns></returns>
+        public Permissions GetPermissions(string tableName)
+        {
+            return GetPermissions(RoleId).Where(x => x.TableName == tableName).FirstOrDefault();
+        }
+        /// <summary>
+        /// 2022.03.26
+        /// 菜单类型1:移动端，0:PC端
+        /// </summary>
+        public static int MenuType
+        {
+            get
+            {
+                return Context.Request.Headers.ContainsKey("uapp") ? 1 : 0;
+            }
+        }
+        /// <summary>
+        /// 自定条件查询权限
+        /// </summary>
+        /// <param name="func"></param>
+        /// <returns></returns>
+        public Permissions GetPermissions(Func<Permissions, bool> func)
+        {
+            // 2022.03.26增移动端加菜单类型判断
+            return GetPermissions(RoleId).Where(func).Where(x => x.MenuType == MenuType).FirstOrDefault();
+        }
+
+        private List<Permissions> ActionToArray(List<Permissions> permissions)
+        {
+            permissions.ForEach(x =>
+            {
+                try
+                {
+                    var menuAuthArr = x.MenuAuth.DeserializeObject<List<Sys_Actions>>();
+                    x.UserAuthArr = string.IsNullOrEmpty(x.UserAuth)
+                    ? new string[0]
+                    : x.UserAuth.Split(",").Where(c => menuAuthArr.Any(m => m.Value == c)).ToArray();
+
+                }
+                catch { }
+                finally
+                {
+                    if (x.UserAuthArr == null)
+                    {
+                        x.UserAuthArr = new string[0];
+                    }
+                }
+            });
+            return permissions;
+        }
+        private List<Permissions> MenuActionToArray(List<Permissions> permissions)
+        {
+            permissions.ForEach(x =>
+            {
+                try
+                {
+                    x.UserAuthArr = string.IsNullOrEmpty(x.UserAuth)
+                    ? new string[0]
+                    : x.UserAuth.DeserializeObject<List<Sys_Actions>>().Select(s => s.Value).ToArray();
+                }
+                catch { }
+                finally
+                {
+                    if (x.UserAuthArr == null)
+                    {
+                        x.UserAuthArr = new string[0];
+                    }
+                }
+            });
+            return permissions;
+        }
+        public List<Permissions> GetPermissions(int roleId)
+        {
+            if (IsRoleIdSuperAdmin(roleId))
+            {
+                //2020.12.27增加菜单界面上不显示，但可以分配权限
+                var permissions = DBServerProvider.DbContext.Set<Sys_Menu>()
+                    .Where(x => x.Enable == 1 || x.Enable == 2)
+                    .Select(a => new Permissions
+                    {
+                        Menu_Id = a.Menu_Id,
+                        ParentId = a.ParentId,
+                        //2020.05.06增加默认将表名转换成小写，权限验证时不再转换
+                        TableName = (a.TableName ?? "").ToLower(),
+                        //MenuAuth = a.Auth,
+                        UserAuth = a.Auth,
+                        // 2022.03.26增移动端加菜单类型
+                        MenuType = a.MenuType ?? 0
+                    }).ToList();
+                return MenuActionToArray(permissions);
+            }
+            ICacheService cacheService = CacheService;
+            string roleKey = roleId.GetRoleIdKey();
+
+            //角色有缓存，并且当前服务器的角色版本号与redis/memory缓存角色的版本号相同直接返回静态对象角色权限
+            string currnetVeriosn = "";
+            if (rolePermissionsVersion.TryGetValue(roleId, out currnetVeriosn)
+                && currnetVeriosn == cacheService.Get(roleKey))
+            {
+                return rolePermissions.ContainsKey(roleId) ? rolePermissions[roleId] : new List<Permissions>();
+            }
+
+            //锁定每个角色，通过安全字典减少锁粒度，否则多个同时角色获取缓存会导致阻塞
+            object objId = objKeyValue.GetOrAdd(roleId.ToString(), new object());
+            //锁定每个角色
+            lock (objId)
+            {
+                if (rolePermissionsVersion.TryGetValue(roleId, out currnetVeriosn)
+                    && currnetVeriosn == cacheService.Get(roleKey))
+                {
+                    return rolePermissions.ContainsKey(roleId) ? rolePermissions[roleId] : new List<Permissions>();
+                }
+
+                //没有redis/memory缓存角色的版本号或与当前服务器的角色版本号不同时，刷新缓存
+                var dbContext = DBServerProvider.DbContext;
+                List<Permissions> _permissions = (from a in dbContext.Set<Sys_Menu>()
+                                                  join b in dbContext.Set<Sys_RoleAuth>()
+                                                  on a.Menu_Id equals b.Menu_Id
+                                                  where b.Role_Id == roleId //&& a.ParentId > 0
+                                                  && b.AuthValue != ""
+                                                  orderby a.ParentId
+                                                  select new Permissions
+                                                  {
+                                                      Menu_Id = a.Menu_Id,
+                                                      ParentId = a.ParentId,
+                                                      //2020.05.06增加默认将表名转换成小写，权限验证时不再转换
+                                                      TableName = (a.TableName ?? "").ToLower(),
+                                                      MenuAuth = a.Auth,
+                                                      UserAuth = b.AuthValue ?? "",
+                                                      // 2022.03.26增移动端加菜单类型
+                                                      MenuType = a.MenuType ?? 0
+                                                  }).ToList();
+                ActionToArray(_permissions);
+                string _version = cacheService.Get(roleKey);
+                //生成一个唯一版本号标识
+                if (_version == null)
+                {
+                    _version = DateTime.Now.ToString("yyyyMMddHHMMssfff");
+                    //将版本号写入缓存
+                    cacheService.Add(roleKey, _version);
+                }
+                //刷新当前服务器角色的权限
+                rolePermissions[roleId] = _permissions;
+
+                //写入当前服务器的角色最新版本号
+                rolePermissionsVersion[roleId] = _version;
+                return _permissions;
+            }
+
+        }
+
+        /// <summary>
+        /// 判断是否有权限
+        /// </summary>
+        /// <param name="tableName"></param>
+        /// <param name="authName"></param>
+        /// <param name="roleId"></param>
+        /// <returns></returns>
+        public bool ExistsPermissions(string tableName, string authName, int roleId = 0)
+        {
+            if (roleId <= 0) roleId = RoleId;
+            tableName = tableName.ToLower();
+            return GetPermissions(roleId).Any(x => x.TableName == tableName && x.UserAuthArr.Contains(authName));
+        }
+
+        /// <summary>
+        /// 判断是否有权限
+        /// </summary>
+        /// <param name="tableName"></param>
+        /// <param name="authName"></param>
+        /// <param name="roleId"></param>
+        /// <returns></returns>
+        public bool ExistsPermissions(string tableName, ActionPermissionOptions actionPermission, int roleId = 0)
+        {
+            return ExistsPermissions(tableName, actionPermission.ToString(), roleId);
+        }
+        public int UserId
+        {
+            get
+            {
+                return (Context.User.FindFirstValue(JwtRegisteredClaimNames.Jti)
+                    ?? Context.User.FindFirstValue(ClaimTypes.NameIdentifier)).GetInt();
+            }
+        }
+
+        public string UserName
+        {
+            get { return UserInfo.UserName; }
+        }
+
+        public string UserTrueName
+        {
+            get { return UserInfo.UserTrueName; }
+        }
+
+        public string Token
+        {
+            get { return UserInfo.Token; }
+        }
+
+        public int RoleId
+        {
+            get { return UserInfo.Role_Id; }
+        }
+        public List<Guid> DeptIds
+        {
+            get { return UserInfo.DeptIds; }
+        }
+        /// <summary>
+        /// 获取所有子部门
+        /// </summary>
+        /// <returns></returns>
+        public List<Guid> GetAllChildrenDeptIds()
+        {
+            return DepartmentContext.GetAllChildrenIds(DeptIds);
+        }
+
+        public void LogOut(int userId)
+        {
+            CacheService.Remove(userId.GetUserIdKey());
+        }
+    }
+}