namespace Admin.NET.Core.Service; /// /// 系统菜单服务 /// [ApiDescriptionSettings(Name = "系统菜单", Order = 195)] public class SysMenuService : IDynamicApiController, ITransient { private readonly SqlSugarRepository _sysMenuRep; private readonly IUserManager _userManager; private readonly SysRoleMenuService _sysRoleMenuService; private readonly SysUserRoleService _sysUserRoleService; private readonly ISysCacheService _sysCacheService; public SysMenuService(SqlSugarRepository sysMenuRep, IUserManager userManager, SysRoleMenuService sysRoleMenuService, SysUserRoleService sysUserRoleService, ISysCacheService sysCacheService) { _sysMenuRep = sysMenuRep; _userManager = userManager; _sysRoleMenuService = sysRoleMenuService; _sysUserRoleService = sysUserRoleService; _sysCacheService = sysCacheService; } /// /// 获取登录菜单树 /// /// [HttpGet("/getMenuList")] public async Task> GetLoginMenuTree() { if (_userManager.SuperAdmin) { var menuList = await _sysMenuRep.AsQueryable() .Where(u => u.Type != MenuTypeEnum.Btn) .OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0); return menuList.Adapt>(); } else { var menuIdList = await GetMenuIdList(); var menuList = await _sysMenuRep.AsQueryable() .Where(u => u.Type != MenuTypeEnum.Btn) .Where(u => menuIdList.Contains(u.Id)) .OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0); return menuList.Adapt>(); } } /// /// 获取菜单列表 /// /// [HttpGet("/sysMenu/list")] public async Task> GetMenuList([FromQuery] MenuInput input) { var menuIdList = new List(); if (!_userManager.SuperAdmin) menuIdList = await GetMenuIdList(); if (!string.IsNullOrWhiteSpace(input.Title) || input.Type > 0) { return await _sysMenuRep.AsQueryable() .WhereIF(!string.IsNullOrWhiteSpace(input.Title), u => u.Title.Contains(input.Title)) .WhereIF(input.Type > 0, u => u.Type == (MenuTypeEnum)input.Type) .WhereIF(menuIdList.Count > 1, u => menuIdList.Contains(u.Id)) .OrderBy(u => u.OrderNo).ToListAsync(); } return await _sysMenuRep.AsQueryable() .WhereIF(menuIdList.Count > 1, u => menuIdList.Contains(u.Id)) .OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0); } /// /// 增加菜单 /// /// /// [HttpPost("/sysMenu/add")] public async Task AddMenu(AddMenuInput input) { var isExist = input.Type != 3 ? await _sysMenuRep.IsAnyAsync(u => u.Name == input.Name) : await _sysMenuRep.IsAnyAsync(u => u.Permission == input.Permission); if (isExist) throw Oops.Oh(ErrorCodeEnum.D4000); // 校验菜单参数 CheckMenuParam(input.Adapt()); var menu = input.Adapt(); await _sysMenuRep.InsertAsync(menu); // 清除缓存 DeleteMenuCache(); } /// /// 更新菜单 /// /// /// [HttpPost("/sysMenu/update"),] public async Task UpdateMenu(UpdateMenuInput input) { var isExist = await _sysMenuRep.IsAnyAsync(u => u.Name == input.Name && u.Id != input.Id); if (isExist) throw Oops.Oh(ErrorCodeEnum.D4000); // 校验菜单参数 CheckMenuParam(input.Adapt()); var menu = input.Adapt(); await _sysMenuRep.AsUpdateable(menu).IgnoreColumns(true).ExecuteCommandAsync(); // 清除缓存 DeleteMenuCache(); } /// /// 删除菜单 /// /// /// [HttpPost("/sysMenu/delete")] [UnitOfWork] public async Task DeleteMenu(DeleteMenuInput input) { var menuTreeList = await _sysMenuRep.AsQueryable().ToChildListAsync(u => u.Pid, input.Id); var menuIdList = menuTreeList.Select(u => u.Id).ToList(); await _sysMenuRep.DeleteAsync(u => menuIdList.Contains(u.Id)); // 级联删除角色菜单数据 await _sysRoleMenuService.DeleteRoleMenuByMenuIdList(menuIdList); // 清除缓存 DeleteMenuCache(); } /// /// 增加和编辑时检查菜单数据 /// /// private static void CheckMenuParam(MenuInput input) { var type = input.Type; //var component = input.Component; var permission = input.Permission; //if (type == (int)MenuTypeEnum.Dir || type == (int)MenuTypeEnum.Menu) //{ // if (string.IsNullOrEmpty(component)) // throw Oops.Oh(ErrorCodeEnum.D4001); //} //else if (type == (int)MenuTypeEnum.Btn) { if (string.IsNullOrEmpty(permission)) throw Oops.Oh(ErrorCodeEnum.D4003); if (!permission.Contains(':')) throw Oops.Oh(ErrorCodeEnum.D4004); } } /// /// 获取按钮权限列表(登录) /// /// [HttpGet("getPermCode")] public async Task> GetPermCodeList() { var userId = _userManager.UserId; var permissions = await _sysCacheService.GetPermission(userId); // 先从缓存里面读取 if (permissions == null || permissions.Count == 0) { if (_userManager.SuperAdmin) { permissions = await _sysMenuRep.AsQueryable() .Where(u => u.Type == MenuTypeEnum.Btn) .Select(u => u.Permission).ToListAsync(); } else { var menuIdList = await GetMenuIdList(); permissions = await _sysMenuRep.AsQueryable() .Where(u => u.Type == MenuTypeEnum.Btn) .Where(u => menuIdList.Contains(u.Id)) .Select(u => u.Permission).ToListAsync(); } await _sysCacheService.SetPermission(userId, permissions); // 缓存结果 } return permissions; } /// /// 获取所有按钮权限集合 /// /// [NonAction] public async Task> GetAllPermCodeList() { var permissions = await _sysCacheService.GetPermission(0); // 先从缓存里面读取 if (permissions == null || permissions.Count == 0) { permissions = await _sysMenuRep.AsQueryable() .Where(u => u.Type == MenuTypeEnum.Btn) .Select(u => u.Permission).ToListAsync(); await _sysCacheService.SetPermission(0, permissions); // 缓存结果 } return permissions; } /// /// 清除菜单和按钮缓存 /// private async void DeleteMenuCache() { await _sysCacheService.DelByPatternAsync(CacheConst.KeyMenu); await _sysCacheService.DelByPatternAsync(CacheConst.KeyPermission); } /// /// 获取当前用户菜单Id集合 /// /// private async Task> GetMenuIdList() { var roleIdList = await _sysUserRoleService.GetUserRoleIdList(_userManager.UserId); return await _sysRoleMenuService.GetRoleMenuIdList(roleIdList); } }