using Microsoft.EntityFrameworkCore; using System; using System.Collections.Generic; using System.Linq; using System.Linq.Expressions; using System.Text; using System.Threading.Tasks; using Yi.Framework.Interface; using Yi.Framework.Model; using Yi.Framework.Model.ModelFactory; using Yi.Framework.Model.Models; namespace Yi.Framework.Service { public partial class UserService : BaseService, IUserService { private IRoleService _roleService; public UserService(IDbContextFactory DbFactory, IRoleService roleService) : base(DbFactory) { _roleService = roleService; } public async Task EmailIsExsit(string emailAddress) { var userList = await GetAllEntitiesTrueAsync(); var is_email = userList.Where(u => u.email == emailAddress).FirstOrDefault(); if (is_email == null) { return true; } return false; } /// /// 大优化,直接传用户id即可 /// /// /// public async Task> GetMenusByUser(user _user) { var role_data = await GetRolesByUser(_user); var menuList = new List(); role_data.ForEach(u => { var menu_data = _roleService.GetMenusByRole(u); menuList = menuList.Concat(menu_data.GetAwaiter().GetResult()).ToList(); }); menuList.ForEach(u => u.roles = null); return menuList; } /// /// 大优化只用id即可,这个方法和GetMenusByUser几乎没有任何区别 /// /// /// public async Task> GetMouldByUser(user _user) { var menu = await GetMenusByUser(_user); var mouldList = menu.Select(u => u.mould).ToList(); return mouldList; } public async Task> GetRolesByUser(user _user) { var user_data = await _Db.Set().Include(u => u.roles) .Where(u => u.id == _user.id && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync(); var roleList = user_data.roles.Where(u => u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).ToList(); roleList.ForEach(u => u.users = null); return roleList; } public async Task Login(user _user) { var user_data = await _Db.Set().Include(u => u.roles).Where(u => u.username == _user.username && u.password == _user.password && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync(); return user_data; } public async Task Register(user _user) { var user_data = await GetEntity(u => u.username == _user.username); if (user_data != null) { return false; } return await AddAsync(_user); } /// /// 优化,如果是传的id,或者是传一个对象,不是要返回全部信息,那么可知直接认为该id一定存在 /// /// /// /// public async Task SetRolesByUser(List roleIds, List userIds) { var user_data = await _Db.Set().Include(u => u.roles).Where(u => userIds.Contains(u.id) ).ToListAsync(); if (user_data == null) { return false; } var roleList = await _Db.Set().Where(u => roleIds.Contains(u.id)).ToListAsync(); foreach (var item in user_data) { item.roles = roleList; } return await UpdateListAsync(user_data); } /// /// 有可简化的GetUserById /// /// /// public async Task> GetMenuByUser(user _user) { var user_data = await _Db.Set().Include(u => u.roles).ThenInclude(u => u.menus).Where(u => u.id == _user.id ).FirstOrDefaultAsync(); List menu_data = new(); foreach (var role in user_data.roles) { var menu = role.menus.ToList(); menu.ForEach(u => u.roles = null); menu_data = menu_data.Concat(menu).ToList(); } return menu_data; } public async Task GetUserInfoById(int user_id) { var user_data = await GetUserById(user_id); user_data.roles.ToList().ForEach(u => u.users = null); return user_data; } /// /// 这个方法可以被超级多个其他方法利用,这个是重要的 /// /// /// public async Task GetUserById(int user_id) { var user_data = await _Db.Set().Include(u => u.roles) .Where(u => u.id == user_id && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync(); return user_data; } /// /// 有可简化的GetUserById /// /// /// public async Task> GetMenuById(int user_id) { var user_data = await _Db.Set().Include(u => u.roles).Where(u => u.id == user_id).FirstOrDefaultAsync(); List menu_data = new(); foreach (var role in user_data.roles) { var menu = await _roleService.GetMenusByRole(role); menu.ForEach(u => u.roles = null); menu_data = menu_data.Union(menu).OrderByDescending(u => u.sort).ToList(); } //menu_data为角色所有的菜单,不是一个递归的啊 var allMenuIds = menu_data.Select(u => u.id).ToList(); var topMenu = menu_data.Where(u => u.is_top == (short)Common.Enum.ShowFlagEnum.Show); //现在要开始关联菜单了 List endMenu = new List(); foreach (var item in topMenu) { var p = await _Db.Set().Where(u => u.id == item.id).Include(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ToListAsync(); endMenu = endMenu.Union(p).ToList(); } return TopMenuBuild2(TopMenuBuild(endMenu, allMenuIds)); } /// /// 这种就命名的话,改成MenuIconAndShowBuild /// /// /// /// private List TopMenuBuild(List menu_data, List allMenuIds) { for (int i = menu_data.Count() - 1; i >= 0; i--) { if (menu_data[i].icon == null) { menu_data[i].icon = "Yi"; } if (!allMenuIds.Contains(menu_data[i].id) || menu_data[i].is_delete == (short)Common.Enum.DelFlagEnum.Deleted || menu_data[i].is_show == (short)Common.Enum.ShowFlagEnum.NoShow) { menu_data.Remove(menu_data[i]); } else if (menu_data[i].children != null) { menu_data[i].children = TopMenuBuild(menu_data[i].children.ToList(), allMenuIds); } } return menu_data; } /// /// 这种就命名的话,改成MenuChildrenBuild,这个方法应该可以提到core层,这是一个公用的方法 /// /// /// private List TopMenuBuild2(List menu_data) { for (int i = menu_data.Count() - 1; i >= 0; i--) { if (menu_data[i].children.Count() == 0) { menu_data[i].children = null;f } else if (menu_data[i].children != null) { menu_data[i].children = TopMenuBuild2(menu_data[i].children.ToList()); } } return menu_data; } public async Task> GetMenuByUserId(string router, int userId, List menuIds) { var user_data = await _Db.Set().Include(u => u.roles).ThenInclude(u => u.menus).ThenInclude(u => u.children) .ThenInclude(u => u.mould).Where(u => u.id == userId).FirstOrDefaultAsync(); var roleList = user_data.roles.ToList(); roleList.ForEach(u => u.users = null); List menu_data = new(); foreach (var item in roleList) { item.menus.ToList().ForEach(u => u.roles = null); var menuData = item.menus.Where(u => u.router.ToUpper() == router.ToUpper()).FirstOrDefault(); menu_data = menuData.children?.Where(u => menuIds.Contains(u.id) && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).ToList(); if (menu_data != null) { break; } } return menu_data; } } }