前面完成了基础管理的相关API，接下来就得做一个菜单管理了，用于对接管理后台前端界面。
设计菜单结构
菜单是一个多级结构，所以我们得设计一个树形的。包含自己上级和下级的属性。同时预留Permission用于做可选的权限限制。
namespace Wheel.Domain.Menus { /// <summary> /// 菜单 /// </summary> public class Menu : Entity<Guid> { /// <summary> /// 名称 /// </summary> public string Name { get; set; } /// <summary> /// 显示名称 /// </summary> public string DisplayName { get; set; } /// <summary> /// 菜单类型 /// </summary> public MenuType MenuType { get; set; } /// <summary> /// 菜单路径 /// </summary> public string? Path { get; set; } /// <summary> /// 权限名称 /// </summary> public string? Permission { get; set; } /// <summary> /// 图标 /// </summary> public string? Icon { get; set; } /// <summary> /// 排序 /// </summary> public int Sort { get; set; } /// <summary> /// 上级菜单Id /// </summary> public virtual Guid? ParentId { get; set; } /// <summary> /// 上级菜单 /// </summary> public virtual Menu? Parent { get; set; } /// <summary> /// 子菜单 /// </summary> public virtual List<Menu> Children { get; set; } } } 然后菜单和角色关联。创建RoleMenu表。
namespace Wheel.Domain.Menus { public class RoleMenu { public virtual string RoleId { get; set; } public virtual Role Role { get; set; } public virtual Guid MenuId { get; set; } public virtual Menu Menu { get; set; } } } 修改DbContext
接下来还是老套路，修改WheelDbContext
添加代码：
#region Menu public DbSet<Menu> Menus { get; set; } public DbSet<RoleMenu> RoleMenus { get; set; } #endregion protected override void OnModelCreating(ModelBuilder builder) { base.OnModelCreating(builder); ConfigureIdentity(builder); ConfigureLocalization(builder); ConfigurePermissionGrants(builder); ConfigureMenus(builder); } void ConfigureMenus(ModelBuilder builder) { builder.Entity<Menu>(b => { b.HasKey(o => o.Id); b.Property(o => o.Permission).HasMaxLength(128); b.Property(o => o.Path).HasMaxLength(128); b.Property(o => o.Name).HasMaxLength(128); b.Property(o => o.Icon).HasMaxLength(128); b.Property(o => o.DisplayName).HasMaxLength(128); b.HasMany(o => o.Children).WithOne(o => o.Parent); b.HasIndex(o => o.ParentId); }); builder.Entity<RoleMenu>(b => { b.HasKey(o => new { o.MenuId, o.RoleId }); b.Property(o => o.RoleId).HasMaxLength(36); }); } 实现菜单管理
实现MenuAppService
IMenuAppService
namespace Wheel.Services.Menus { public interface IMenuAppService : ITransientDependency { Task<R> Create(CreateOrUpdateMenuDto dto); Task<R> Update(Guid id, CreateOrUpdateMenuDto dto); Task<R> Delete(Guid id); Task<R<MenuDto>> GetById(Guid id); Task<R<List<MenuDto>>> GetList(); Task<R<List<MenuDto>>> GetRoleMenuList(string roleId); Task<R<List<AntdMenuDto>>> GetCurrentMenu(); Task<R> UpdateRoleMenu(string roleId, UpdateRoleMenuDto dto); } } MenuAppService
namespace Wheel.Services.Menus { public class MenuAppService : WheelServiceBase, IMenuAppService { private readonly IBasicRepository<Menu, Guid> _menuRepository; private readonly IBasicRepository<Role, string> _roleRepository; private readonly IBasicRepository<RoleMenu> _roleMenuRepository; public MenuAppService(IBasicRepository<Menu, Guid> menuRepository) { _menuRepository = menuRepository; } public async Task<R> Create(CreateOrUpdateMenuDto dto) { var menu = Mapper.Map<Menu>(dto); menu.Id = GuidGenerator.Create(); await _menuRepository.InsertAsync(menu, true); return new R(); } public async Task<R> Update(Guid id,CreateOrUpdateMenuDto dto) { var menu = await _menuRepository.FindAsync(id); if(menu != null) { Mapper.Map(dto, menu); await _menuRepository.UpdateAsync(menu, true); } return new R(); } public async Task<R> Delete(Guid id) { await _menuRepository.DeleteAsync(id, true); return new R(); } public async Task<R<MenuDto>> GetById(Guid id) { var menu = await _menuRepository.FindAsync(id); var dto = Mapper.Map<MenuDto>(menu); return new R<MenuDto>(dto); } public async Task<R<List<MenuDto>>> GetList() { var items = await _menuRepository.GetListAsync( a => a.ParentId == null, propertySelectors: a=>a.Children ); items.ForEach(a => a.Children = a.Children.OrderBy(b => b.Sort).ToList()); items = items.OrderBy(a => a.Sort).ToList(); var resultItems = Mapper.Map<List<MenuDto>>(items); return new R<List<MenuDto>>(resultItems); } public async Task<R> UpdateRoleMenu(string roleId, UpdateRoleMenuDto dto) { using (var uow = await UnitOfWork.BeginTransactionAsync()) { if (await _roleMenuRepository.AnyAsync(a => a.RoleId == roleId)) { await _roleMenuRepository.DeleteAsync(a => a.RoleId == roleId); } if(dto.MenuIds.Any()) { var roleMenus = dto.MenuIds.Select(a => new RoleMenu { RoleId = roleId, MenuId = a }); await _roleMenuRepository.InsertManyAsync(roleMenus.ToList()); } await uow.CommitAsync(); } return new R(); } public async Task<R<List<MenuDto>>> GetRoleMenuList(string roleId) { var items = await _roleMenuRepository.SelectListAsync(a => a.RoleId == roleId && a.Menu.ParentId == null, a => a.Menu, propertySelectors: a => a.Menu.Children); items.ForEach(a => a.Children = a.Children.OrderBy(b => b.Sort).ToList()); items = items.OrderBy(a => a.Sort).ToList(); var resultItems = Mapper.Map<List<MenuDto>>(items); return new R<List<MenuDto>>(resultItems); } public async Task<R<List<AntdMenuDto>>> GetCurrentMenu() { if (CurrentUser.IsInRoles("admin")) { var menus = await _menuRepository.GetListAsync(a => a.ParentId == null); return new R<List<AntdMenuDto>>(MaptoAntdMenu(menus)); } else { var roleIds = await _roleRepository.SelectListAsync(a => CurrentUser.Roles.Contains(a.Name), a => a.Id); var menus = await _roleMenuRepository.SelectListAsync(a => roleIds.Contains(a.RoleId) && a.Menu.ParentId == null, a => a.Menu, propertySelectors: a => a.Menu.Children); return new R<List<AntdMenuDto>>(MaptoAntdMenu(menus.DistinctBy(a=>a.Id).ToList())); } } private List<AntdMenuDto> MaptoAntdMenu(List<Menu> menus) { return menus.OrderBy(m => m.Sort).Select(m => { var result = new AntdMenuDto { Name = m.Name, Icon = m.Icon, Path = m.Path, Access = m.Permission }; if(m.Children != null && m.Children.Count > 0) { result.Children = MaptoAntdMenu(m.Children); } return result; }).ToList(); } } } 实现MenuController
namespace Wheel.Controllers { /// <summary> /// 菜单管理 /// </summary> [Route("api/[controller]")] [ApiController] public class MenuController : WheelControllerBase { private readonly IMenuAppService _menuAppService; public MenuController(IMenuAppService menuAppService) { _menuAppService = menuAppService; } /// <summary> /// 新增菜单 /// </summary> /// <param name="dto"></param> /// <returns></returns> [HttpPost()] public Task<R> Create(CreateOrUpdateMenuDto dto) { return _menuAppService.Create(dto); } /// <summary> /// 删除菜单 /// </summary> /// <param name="id"></param> /// <returns></returns> [HttpDelete("{id}")] public Task<R> Delete(Guid id) { return _menuAppService.Delete(id); } /// <summary> /// 获取单个菜单详情 /// </summary> /// <param name="id"></param> /// <returns></returns> [HttpGet("{id}")] public Task<R<MenuDto>> GetById(Guid id) { return _menuAppService.GetById(id); } /// <summary> /// 查询菜单列表 /// </summary> /// <returns></returns> [HttpGet] public Task<R<List<MenuDto>>> GetList() { return _menuAppService.GetList(); } /// <summary> /// 修改菜单 /// </summary> /// <param name="id"></param> /// <param name="dto"></param> /// <returns></returns> [HttpPut("{id}")] public Task<R> Update(Guid id, CreateOrUpdateMenuDto dto) { return _menuAppService.Update(id, dto); } /// <summary> /// 修改角色菜单 /// </summary> /// <param name="roleId"></param> /// <param name="dto"></param> /// <returns></returns> [HttpPut("role/{roleId}")] public Task<R> UpdateRoleMenu(string roleId, UpdateRoleMenuDto dto) { return _menuAppService.UpdateRoleMenu(roleId, dto); } /// <summary> /// 获取角色菜单列表 /// </summary> /// <param name="roleId"></param> /// <returns></returns> [HttpGet("role/{roleId}")] public Task<R<List<MenuDto>>> GetRoleMenuList(string roleId) { return _menuAppService.GetRoleMenuList(roleId); } } } 就这样我们就完成了菜单管理相关的API功能，包含菜单的增删查改和角色菜单绑定功能。
到这里我们最基础的后台管理功能API基本开发完成。
轮子仓库地址https://github.com/Wheel-Framework/Wheel
欢迎进群催更。
![洛谷 P11345 [KTSC 2023 R2] 基地简化 题解](http://www.itfaba.com/wp-content/themes/kemi/timthumb.php?src=http://www.itfaba.com/wp-content/themes/kemi/img/random/3.jpg&w=218&h=124&zc=1)
