253 lines
9.5 KiB
C#
253 lines
9.5 KiB
C#
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.Models;
|
||
|
||
namespace Yi.Framework.Service
|
||
{
|
||
public partial class UserService : BaseService<user>, IUserService
|
||
{
|
||
private IRoleService _roleService;
|
||
public UserService(DbContext Db, IRoleService roleService) : base(Db)
|
||
{
|
||
_roleService = roleService;
|
||
}
|
||
|
||
public async Task<bool> 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;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 大优化，直接传用户id即可
|
||
/// </summary>
|
||
/// <param name="_user"></param>
|
||
/// <returns></returns>
|
||
public async Task<List<menu>> GetMenusByUser(user _user)
|
||
{
|
||
var role_data = await GetRolesByUser(_user);
|
||
var menuList = new List<menu>();
|
||
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;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 大优化只用id即可，这个方法和GetMenusByUser几乎没有任何区别
|
||
/// </summary>
|
||
/// <param name="_user"></param>
|
||
/// <returns></returns>
|
||
public async Task<List<mould>> GetMouldByUser(user _user)
|
||
{
|
||
var menu = await GetMenusByUser(_user);
|
||
var mouldList = menu.Select(u => u.mould).ToList();
|
||
return mouldList;
|
||
}
|
||
|
||
|
||
public async Task<List<role>> GetRolesByUser(user _user)
|
||
{
|
||
var user_data = await _Db.Set<user>().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<user> Login(user _user)
|
||
{
|
||
var user_data = await _Db.Set<user>().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<bool> Register(user _user)
|
||
{
|
||
var user_data = await GetEntity(u => u.username == _user.username);
|
||
if (user_data != null)
|
||
{
|
||
return false;
|
||
}
|
||
return await AddAsync(_user);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 优化，如果是传的id，或者是传一个对象，不是要返回全部信息，那么可知直接认为该id一定存在
|
||
/// </summary>
|
||
/// <param name="roleIds"></param>
|
||
/// <param name="userIds"></param>
|
||
/// <returns></returns>
|
||
public async Task<bool> SetRolesByUser(List<int> roleIds, List<int> userIds)
|
||
{
|
||
var user_data = await _Db.Set<user>().Include(u => u.roles).Where(u => userIds.Contains(u.id) ).ToListAsync();
|
||
if (user_data == null)
|
||
{
|
||
return false;
|
||
}
|
||
var roleList = await _Db.Set<role>().Where(u => roleIds.Contains(u.id)).ToListAsync();
|
||
foreach (var item in user_data)
|
||
{
|
||
item.roles = roleList;
|
||
}
|
||
return await UpdateListAsync(user_data);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 有可简化的GetUserById
|
||
/// </summary>
|
||
/// <param name="_user"></param>
|
||
/// <returns></returns>
|
||
public async Task<List<menu>> GetMenuByUser(user _user)
|
||
{
|
||
var user_data = await _Db.Set<user>().Include(u => u.roles).ThenInclude(u => u.menus).Where(u => u.id == _user.id ).FirstOrDefaultAsync();
|
||
List<menu> 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<user> GetUserInfoById(int user_id)
|
||
{
|
||
var user_data = await GetUserById(user_id);
|
||
user_data.roles.ToList().ForEach(u => u.users = null);
|
||
return user_data;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 这个方法可以被超级多个其他方法利用，这个是重要的
|
||
/// </summary>
|
||
/// <param name="user_id"></param>
|
||
/// <returns></returns>
|
||
public async Task<user> GetUserById(int user_id)
|
||
{
|
||
var user_data = await _Db.Set<user>().Include(u => u.roles)
|
||
.Where(u => u.id == user_id && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync();
|
||
return user_data;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 有可简化的GetUserById
|
||
/// </summary>
|
||
/// <param name="user_id"></param>
|
||
/// <returns></returns>
|
||
public async Task<List<menu>> GetMenuById(int user_id)
|
||
{
|
||
var user_data = await _Db.Set<user>().Include(u => u.roles).Where(u => u.id == user_id).FirstOrDefaultAsync();
|
||
List<menu> 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<menu> endMenu = new List<menu>();
|
||
foreach (var item in topMenu)
|
||
{
|
||
var p = await _Db.Set<menu>().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));
|
||
}
|
||
/// <summary>
|
||
/// 这种就命名的话，改成MenuIconAndShowBuild
|
||
/// </summary>
|
||
/// <param name="menu_data"></param>
|
||
/// <param name="allMenuIds"></param>
|
||
/// <returns></returns>
|
||
private List<menu> TopMenuBuild(List<menu> menu_data, List<int> 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;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 这种就命名的话，改成MenuChildrenBuild,这个方法应该可以提到core层，这是一个公用的方法
|
||
/// </summary>
|
||
/// <param name="menu_data"></param>
|
||
/// <returns></returns>
|
||
private List<menu> TopMenuBuild2(List<menu> menu_data)
|
||
{
|
||
|
||
for (int i = menu_data.Count() - 1; i >= 0; i--)
|
||
{
|
||
if (menu_data[i].children.Count() == 0)
|
||
{
|
||
menu_data[i].children = null;
|
||
}
|
||
else if (menu_data[i].children != null)
|
||
{
|
||
menu_data[i].children = TopMenuBuild2(menu_data[i].children.ToList());
|
||
}
|
||
}
|
||
return menu_data;
|
||
}
|
||
|
||
|
||
|
||
public async Task<List<menu>> GetMenuByUserId(string router, int userId, List<int> menuIds)
|
||
{
|
||
var user_data = await _Db.Set<user>().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> 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;
|
||
}
|
||
}
|
||
}
|