| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799 |
- // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
- //
- // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
- //
- // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
- namespace Admin.NET.Core.Service;
- /// <summary>
- /// 系统租户管理服务 🧩
- /// </summary>
- [ApiDescriptionSettings(Order = 390)]
- public class SysTenantService : IDynamicApiController, ITransient
- {
- private static readonly SysMenuService SysMenuService = App.GetService<SysMenuService>();
- private readonly SqlSugarRepository<SysUserExtOrg> _sysUserExtOrgRep;
- private readonly SqlSugarRepository<SysTenantMenu> _sysTenantMenuRep;
- private readonly SqlSugarRepository<SysRoleMenu> _sysRoleMenuRep;
- private readonly SqlSugarRepository<SysUserRole> _userRoleRep;
- private readonly SqlSugarRepository<SysTenant> _sysTenantRep;
- private readonly SqlSugarRepository<SysRole> _sysRoleRep;
- private readonly SqlSugarRepository<SysUser> _sysUserRep;
- private readonly SqlSugarRepository<SysOrg> _sysOrgRep;
- private readonly SqlSugarRepository<SysPos> _sysPosRep;
- private readonly SysRoleMenuService _sysRoleMenuService;
- private readonly SysConfigService _sysConfigService;
- private readonly SysCacheService _sysCacheService;
- private readonly UploadOptions _uploadOptions;
- public SysTenantService(
- SqlSugarRepository<SysUserExtOrg> sysUserExtOrgRep,
- SqlSugarRepository<SysTenantMenu> sysTenantMenuRep,
- SqlSugarRepository<SysRoleMenu> sysRoleMenuRep,
- SqlSugarRepository<SysUserRole> userRoleRep,
- SqlSugarRepository<SysTenant> sysTenantRep,
- SqlSugarRepository<SysUser> sysUserRep,
- SqlSugarRepository<SysRole> sysRoleRep,
- SqlSugarRepository<SysOrg> sysOrgRep,
- SqlSugarRepository<SysPos> sysPosRep,
- IOptions<UploadOptions> uploadOptions,
- SysRoleMenuService sysRoleMenuService,
- SysConfigService sysConfigService,
- SysCacheService sysCacheService)
- {
- _sysTenantRep = sysTenantRep;
- _sysOrgRep = sysOrgRep;
- _sysRoleRep = sysRoleRep;
- _sysPosRep = sysPosRep;
- _sysUserRep = sysUserRep;
- _userRoleRep = userRoleRep;
- _sysRoleMenuRep = sysRoleMenuRep;
- _sysCacheService = sysCacheService;
- _uploadOptions = uploadOptions.Value;
- _sysConfigService = sysConfigService;
- _sysTenantMenuRep = sysTenantMenuRep;
- _sysUserExtOrgRep = sysUserExtOrgRep;
- _sysRoleMenuService = sysRoleMenuService;
- }
- /// <summary>
- /// 获取租户分页列表 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("获取租户分页列表")]
- public async Task<SqlSugarPagedList<TenantOutput>> Page(PageTenantInput input)
- {
- return await _sysTenantRep.AsQueryable()
- .LeftJoin<SysUser>((u, a) => u.UserId == a.Id).ClearFilter()
- .LeftJoin<SysOrg>((u, a, b) => u.OrgId == b.Id).ClearFilter()
- .WhereIF(!string.IsNullOrWhiteSpace(input.Phone), (u, a) => a.Phone.Contains(input.Phone.Trim()))
- .WhereIF(!string.IsNullOrWhiteSpace(input.Name), (u, a, b) => b.Name.Contains(input.Name.Trim()))
- .OrderBy(u => new { u.OrderNo, u.Id })
- .Select((u, a, b) => new TenantOutput
- {
- Id = u.Id,
- OrgId = b.Id,
- Name = b.Name,
- UserId = a.Id,
- AdminAccount = a.Account,
- Phone = a.Phone,
- Host = u.Host,
- Email = a.Email,
- TenantType = u.TenantType,
- DbType = u.DbType,
- Connection = u.Connection,
- ConfigId = u.ConfigId,
- OrderNo = u.OrderNo,
- Remark = u.Remark,
- Status = u.Status,
- CreateTime = u.CreateTime,
- CreateUserName = u.CreateUserName,
- UpdateTime = u.UpdateTime,
- UpdateUserName = u.UpdateUserName,
- }, true)
- .ToPagedListAsync(input.Page, input.PageSize);
- }
- /// <summary>
- /// 获取租户列表
- /// </summary>
- /// <returns></returns>
- [AllowAnonymous]
- [DisplayName("获取租户列表"), HttpGet]
- public async Task<dynamic> GetList()
- {
- return await _sysTenantRep.AsQueryable()
- .LeftJoin<SysOrg>((u, a) => u.OrgId == a.Id).ClearFilter()
- .Where(u => u.Status == StatusEnum.Enable)
- .Select((u, a) => new
- {
- Label = SqlFunc.HasValue(u.Title) ? $"{u.Title}-{a.Name}" : a.Name,
- Host = u.Host.ToLower(),
- Value = u.Id,
- }).ToListAsync();
- }
- /// <summary>
- /// 获取当前租户系统信息
- /// </summary>
- /// <returns></returns>
- [NonAction]
- public async Task<SysTenant> GetCurrentTenantSysInfo()
- {
- var tenantId = long.Parse(App.User?.FindFirst(ClaimConst.TenantId)?.Value ?? "0");
- var host = App.HttpContext.Request.Host.Host.ToLower();
- var tenant = await _sysTenantRep.AsQueryable()
- .WhereIF(tenantId > 0, u => u.Id == tenantId && SqlFunc.ToLower(u.Host).Contains(host))
- .WhereIF(!(tenantId > 0), u => SqlFunc.ToLower(u.Host).Contains(host))
- .FirstAsync();
- tenant ??= await _sysTenantRep.GetFirstAsync(u => u.Id == SqlSugarConst.DefaultTenantId);
- _ = tenant ?? throw Oops.Oh(ErrorCodeEnum.D1002);
- return tenant;
- }
- /// <summary>
- /// 获取库隔离的租户列表
- /// </summary>
- /// <returns></returns>
- [NonAction]
- public async Task<List<SysTenant>> GetTenantDbList()
- {
- return await _sysTenantRep.GetListAsync(u => u.TenantType == TenantTypeEnum.Db && u.Status == StatusEnum.Enable);
- }
- /// <summary>
- /// 增加租户 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [UnitOfWork]
- [ApiDescriptionSettings(Name = "Add"), HttpPost]
- [DisplayName("增加租户")]
- public async Task AddTenant(AddTenantInput input)
- {
- var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
- input.Host = input.Host?.ToLower();
- isExist = await _sysTenantRep.IsAnyAsync(u => !string.IsNullOrWhiteSpace(u.Host) && u.Host == input.Host);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
- isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.Account == input.AdminAccount);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
- // 从库配置判断
- if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections)) throw Oops.Oh(ErrorCodeEnum.D1302);
- switch (input.TenantType)
- {
- // Id隔离时设置与主库一致
- case TenantTypeEnum.Id:
- var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
- input.DbType = config.DbType;
- input.Connection = config.ConnectionString;
- break;
- case TenantTypeEnum.Db:
- if (string.IsNullOrWhiteSpace(input.Connection))
- throw Oops.Oh(ErrorCodeEnum.Z1004);
- break;
- default:
- throw Oops.Oh(ErrorCodeEnum.D3004);
- }
- if (input.EnableReg == YesNoEnum.N) input.RegWayId = null;
- var tenant = input.Adapt<TenantOutput>();
- // 设置logo
- await SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
- tenant.Id = _sysTenantRep.InsertReturnEntity(tenant).Id;
- await InitNewTenant(tenant);
- await CacheTenant();
- }
- /// <summary>
- /// 设置logo
- /// </summary>
- /// <param name="tenant"></param>
- /// <param name="logoBase64"></param>
- /// <param name="logoFileName"></param>
- [NonAction]
- public async Task SetLogoUrl(SysTenant tenant, string logoBase64, string logoFileName)
- {
- if (tenant == null) return;
- if (string.IsNullOrWhiteSpace(logoBase64)) return;
- // 旧图标文件相对路径
- var oldSysLogoRelativeFilePath = tenant.Logo ?? "";
- var oldSysLogoAbsoluteFilePath = Path.Combine(App.WebHostEnvironment.WebRootPath, oldSysLogoRelativeFilePath.TrimStart('/'));
- var groups = Regex.Match(logoBase64, @"data:image/(?<type>.+?);base64,(?<data>.+)").Groups;
- //var type = groups["type"].Value;
- var base64Data = groups["data"].Value;
- var binData = Convert.FromBase64String(base64Data);
- // 根据文件名取扩展名
- var ext = string.IsNullOrWhiteSpace(logoFileName) ? ".png" : Path.GetExtension(logoFileName);
- // 本地图标保存路径
- var fileName = $"{tenant.ViceTitle}-logo{ext}".ToLower();
- var path = _uploadOptions.Path.Replace("/{yyyy}/{MM}/{dd}", "");
- path = path.StartsWith("/") || Regex.IsMatch(path, "^[A-Z|a-z]:") ? path : Path.Combine(App.WebHostEnvironment.WebRootPath, path);
- var absoluteFilePath = Path.Combine(path, fileName);
- // 删除已存在文件
- if (File.Exists(oldSysLogoAbsoluteFilePath)) File.Delete(oldSysLogoAbsoluteFilePath);
- // 创建文件夹
- var absoluteFileDir = Path.GetDirectoryName(absoluteFilePath);
- if (!Directory.Exists(absoluteFileDir)) Directory.CreateDirectory(absoluteFileDir);
- // 保存图标文件
- await File.WriteAllBytesAsync(absoluteFilePath, binData);
- // 保存图标配置
- tenant.Logo = $"/upload/{fileName}";
- }
- /// <summary>
- /// 设置租户状态 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("设置租户状态")]
- public async Task<int> SetStatus(TenantInput input)
- {
- var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id);
- if (tenant == null || tenant.ConfigId == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.Z1001);
- if (!Enum.IsDefined(typeof(StatusEnum), input.Status)) throw Oops.Oh(ErrorCodeEnum.D3005);
- tenant.Status = input.Status;
- return await _sysTenantRep.AsUpdateable(tenant).UpdateColumns(u => new { u.Status }).ExecuteCommandAsync();
- }
- /// <summary>
- /// 新增租户初始化
- /// </summary>
- /// <param name="tenant"></param>
- private async Task InitNewTenant(TenantOutput tenant)
- {
- var tenantId = tenant.Id;
- var tenantName = tenant.Name;
- // 初始化机构
- var newOrg = new SysOrg { TenantId = tenantId, Pid = 0, Name = tenantName, Code = tenantName, Remark = tenantName, };
- await _sysOrgRep.InsertAsync(newOrg);
- // 初始化默认角色
- var newRole = new SysRole { TenantId = tenantId, Name = CommonConst.DefaultBaseRoleName, Code = CommonConst.DefaultBaseRoleCode, DataScope = DataScopeEnum.Self, Remark = "此角色为系统自动创建角色" };
- var baseRole = await _sysRoleRep.InsertReturnEntityAsync(newRole);
- var baseRoleMenuIdList = GetBaseRoleMenuIdList().ToList();
- await _sysRoleMenuService.GrantRoleMenu(new RoleMenuInput { Id = baseRole.Id, MenuIdList = baseRoleMenuIdList.Select(u => u.MenuId).ToList() });
- // 初始化职位
- var newPos = new SysPos { TenantId = tenantId, Name = "管理员-" + tenantName, Code = tenantName, Remark = tenantName };
- await _sysPosRep.InsertAsync(newPos);
- // 初始化租户管理员账号
- var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
- var newUser = new SysUser
- {
- TenantId = tenantId,
- Account = tenant.AdminAccount,
- Password = CryptogramUtil.Encrypt(password),
- NickName = "系统管理员",
- Email = tenant.Email,
- Phone = tenant.Phone,
- AccountType = AccountTypeEnum.SysAdmin,
- OrgId = newOrg.Id,
- PosId = newPos.Id,
- Birthday = DateTime.Parse("2000-01-01"),
- RealName = "系统管理员",
- Remark = "系统管理员" + tenantName,
- };
- await _sysUserRep.InsertAsync(newUser);
- // 关联租户组织机构和管理员用户
- await _sysTenantRep.UpdateAsync(u => new SysTenant { UserId = newUser.Id, OrgId = newOrg.Id }, u => u.Id == tenantId);
- // 默认租户管理员角色菜单集合
- var menuList = GetTenantDefaultMenuList().ToList();
- await GrantMenu(new TenantMenuInput { Id = tenantId, MenuIdList = menuList.Select(u => u.MenuId).ToList() });
- }
- /// <summary>
- /// 聚合所有程序集内 SysMenu 种子(扁平列表,供角色授权、子树裁剪等复用)
- /// </summary>
- [NonAction]
- public List<SysMenu> GetAllSeedMenusFlat()
- {
- var defaultConfig = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault();
- if (defaultConfig == null) return new List<SysMenu>();
- var menuSeedDataTypeList = GetSeedDataTypes(defaultConfig, nameof(SysMenuSeedData));
- var allMenuList = new List<SysMenu>();
- foreach (var menu in menuSeedDataTypeList)
- {
- var menuSeedDataList = ((IEnumerable)menu.GetMethod("HasData")?.Invoke(Activator.CreateInstance(menu), null))?.Cast<SysMenu>();
- if (menuSeedDataList != null)
- allMenuList.AddRange(menuSeedDataList);
- }
- return allMenuList.DistinctBy(u => u.Id).ToList();
- }
- /// <summary>
- /// 获取租户默认菜单
- /// </summary>
- /// <param name="ignoreHome">如果某租户需要定制主页,可以忽略</param>
- /// <returns></returns>
- [NonAction]
- public IEnumerable<SysTenantMenu> GetTenantDefaultMenuList(bool ignoreHome = false)
- {
- var menuList = new List<SysMenu>();
- var allMenuList = GetAllSeedMenusFlat();
- if (allMenuList.Count == 0) return Enumerable.Empty<SysTenantMenu>();
- //实现三个层级的菜单
- var topMenuList = allMenuList.Where(u => u.Pid == 0 && u.Type == MenuTypeEnum.Dir).ToList();
- menuList.AddRange(topMenuList);
- var childMenuList = allMenuList.ToChildList(u => u.Id, u => u.Pid, u => topMenuList.Select(p => p.Id).Contains(u.Pid));
- menuList.AddRange(childMenuList);
- var endMenuList = allMenuList.ToChildList(u => u.Id, u => u.Pid, u => childMenuList.Select(p => p.Id).Contains(u.Pid));
- if (endMenuList != null)
- {
- menuList.AddRange(endMenuList);
- }
- //是否需要排除首页菜单
- if (ignoreHome) menuList = menuList.Where(u => !(u.Type == MenuTypeEnum.Menu && u.Name == "home")).ToList();
- menuList = menuList.Distinct().ToList();
- return menuList.Select(u => new SysTenantMenu
- {
- Id = CommonUtil.GetFixedHashCode("" + SqlSugarConst.DefaultTenantId + u.Id, 1300000000000),
- TenantId = SqlSugarConst.DefaultTenantId,
- MenuId = u.Id
- });
- }
- /// <summary>
- /// 获取种子数据类型
- /// </summary>
- /// <param name="config">数据库连接配置</param>
- /// <param name="typeName"></param>
- /// <returns>种子数据类型列表</returns>
- [NonAction]
- private List<Type> GetSeedDataTypes(DbConnectionConfig config, string typeName)
- {
- return App.EffectiveTypes
- .Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass && u.Name == typeName && u.GetInterfaces().Any(i => i.HasImplementedRawGeneric(typeof(ISqlSugarEntitySeedData<>))))
- .WhereIF(config.SeedSettings.EnableIncreSeed, u => u.IsDefined(typeof(IncreSeedAttribute), false))
- .OrderBy(u => u.GetCustomAttributes(typeof(SeedDataAttribute), false).Length > 0 ? ((SeedDataAttribute)u.GetCustomAttributes(typeof(SeedDataAttribute), false)[0]).Order : 0)
- .ToList();
- }
- /// <summary>
- /// 获取租户默认菜单
- /// </summary>
- /// <returns></returns>
- [NonAction]
- public IEnumerable<SysTenantMenu> GetBaseRoleMenuIdList()
- {
- var menuList = new List<SysMenu>();
- var allMenuList = (new SysMenuSeedData().HasData() ?? Enumerable.Empty<SysMenu>()).ToList();
- if (!allMenuList.Any()) return Enumerable.Empty<SysTenantMenu>();
- var dashboardMenu = allMenuList.FirstOrDefault(u => u.Type == MenuTypeEnum.Dir && u.Title == "工作台");
- if (dashboardMenu != null)
- menuList.AddRange(allMenuList.ToChildList(u => u.Id, u => u.Pid, dashboardMenu.Id));
- var systemMenu = allMenuList.FirstOrDefault(u => u.Type == MenuTypeEnum.Dir && u.Title == "系统管理");
- if (systemMenu != null)
- {
- menuList.Add(systemMenu);
- menuList.AddRange(allMenuList.ToChildList(u => u.Id, u => u.Pid, u => u.Pid == systemMenu.Id && new[] { "机构管理", "个人中心" }.Contains(u.Title)));
- }
- // 允许租户管理员在多租户场景下使用“切换租户”能力(仅按钮权限,不暴露整页租户管理)
- var switchTenantBtn = allMenuList.FirstOrDefault(u => u.Type == MenuTypeEnum.Btn && u.Permission == "sysTenant:changeTenant");
- if (switchTenantBtn != null) menuList.Add(switchTenantBtn);
- menuList = menuList.Where(u => !new[] { "增加", "编辑", "删除" }.Contains(u.Title)).ToList();
- return menuList.Select(u => new SysTenantMenu
- {
- Id = CommonUtil.GetFixedHashCode("" + SqlSugarConst.DefaultTenantId + u.Id, 1300000000000),
- TenantId = SqlSugarConst.DefaultTenantId,
- MenuId = u.Id
- });
- }
- /// <summary>
- /// 删除租户 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [UnitOfWork]
- [ApiDescriptionSettings(Name = "Delete"), HttpPost]
- [DisplayName("删除租户")]
- public async Task DeleteTenant(DeleteTenantInput input)
- {
- // 禁止删除默认租户
- if (input.Id.ToString() == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.D1023);
- // 若账号为开放接口绑定租户则禁止删除
- var isOpenAccessTenant = await _sysTenantRep.ChangeRepository<SqlSugarRepository<SysOpenAccess>>().IsAnyAsync(u => u.BindTenantId == input.Id);
- if (isOpenAccessTenant) throw Oops.Oh(ErrorCodeEnum.D1031);
- await _sysTenantRep.DeleteAsync(u => u.Id == input.Id);
- await CacheTenant(input.Id);
- // 删除与租户相关的表数据
- await _sysTenantMenuRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
- await _sysTenantRep.Context.Deleteable<SysTenantConfigData>().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
- var users = await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.Id).ToListAsync();
- var userIds = users.Select(u => u.Id).ToList();
- await _sysUserRep.AsDeleteable().Where(u => userIds.Contains(u.Id)).ExecuteCommandAsync();
- await _userRoleRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
- await _sysUserExtOrgRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
- await _sysTenantRep.Context.Deleteable<SysUserMenu>().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
- await _sysTenantRep.Context.Deleteable<SysUserConfigData>().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
- var roleIds = await _sysRoleRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.Id).Select(u => u.Id).ToListAsync();
- await _sysRoleRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
- await _sysRoleMenuRep.AsDeleteable().Where(u => roleIds.Contains(u.RoleId)).ExecuteCommandAsync();
- await _sysTenantRep.Context.Deleteable<SysRoleOrg>().Where(u => roleIds.Contains(u.RoleId)).ExecuteCommandAsync();
- await _sysOrgRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
- await _sysPosRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
- }
- /// <summary>
- /// 更新租户 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [ApiDescriptionSettings(Name = "Update"), HttpPost]
- [DisplayName("更新租户")]
- public async Task UpdateTenant(UpdateTenantInput input)
- {
- var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name && u.Id != input.OrgId);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
- input.Host = input.Host?.ToLower();
- isExist = await _sysTenantRep.IsAnyAsync(u => !string.IsNullOrWhiteSpace(u.Host) && u.Host == input.Host && u.Id != input.Id);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
- isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.Account == input.AdminAccount && u.Id != input.UserId);
- if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
- // Id隔离时设置与主库一致
- switch (input.TenantType)
- {
- case TenantTypeEnum.Id:
- var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
- input.DbType = config.DbType;
- input.Connection = config.ConnectionString;
- break;
- case TenantTypeEnum.Db:
- if (string.IsNullOrWhiteSpace(input.Connection))
- throw Oops.Oh(ErrorCodeEnum.Z1004);
- break;
- default:
- throw Oops.Oh(ErrorCodeEnum.D3004);
- }
- // 从库配置判断
- if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections))
- throw Oops.Oh(ErrorCodeEnum.D1302);
- // 设置logo
- var tenant = input.Adapt<SysTenant>();
- if (!string.IsNullOrWhiteSpace(input.LogoBase64)) await SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
- // 更新租户信息
- await _sysTenantRep.AsUpdateable(tenant).IgnoreColumns(true).ExecuteCommandAsync();
- // 更新系统机构
- await _sysOrgRep.UpdateAsync(u => new SysOrg() { Name = input.Name }, u => u.Id == input.OrgId);
- // 更新系统用户
- await _sysUserRep.UpdateAsync(u => new SysUser() { Account = input.AdminAccount, Phone = input.Phone, Email = input.Email }, u => u.Id == input.UserId);
- await CacheTenant(input.Id);
- }
- /// <summary>
- /// 授权租户菜单 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [UnitOfWork]
- [DisplayName("授权租户菜单")]
- public async Task GrantMenu(TenantMenuInput input)
- {
- // 获取需要授权的菜单列表
- var menuList = await _sysTenantRep.Context.Queryable<SysMenu>()
- .Where(u => input.MenuIdList.Contains(u.Id))
- .InnerJoin<SysTenantMenu>((u, t) => t.TenantId == input.Id && u.Id == t.MenuId)
- .ToListAsync();
- // 检查是否存在重复菜单
- if (menuList.Where(u => u.Type != MenuTypeEnum.Btn).GroupBy(u => new { u.Pid, u.Title }).Any(u => u.Count() > 1) ||
- menuList.Where(u => u.Type == MenuTypeEnum.Btn).GroupBy(u => u.Permission).Any(u => u.Count() > 1))
- throw Oops.Oh(ErrorCodeEnum.D1304);
- // 检查路由是否重复
- if (menuList.Where(u => !string.IsNullOrWhiteSpace(u.Name)).GroupBy(u => u.Name).Any(u => u.Count() > 1))
- throw Oops.Oh(ErrorCodeEnum.D4009);
- //获取默认租户授权菜单,种子数据主键ID保持不变,防止重复
- var tenantMenuList = input.Id == SqlSugarConst.DefaultTenantId ? await _sysTenantMenuRep.AsQueryable().Where(u => u.TenantId == input.Id).ToListAsync() : null;
- List<long> tenantIdList = [input.Id];
- if (input.TenantIdList?.Count > 0) tenantIdList.AddRange(input.TenantIdList);
- // 删除旧记录
- await _sysTenantMenuRep.AsDeleteable().Where(u => tenantIdList.Contains(u.TenantId)).ExecuteCommandAsync();
- // 追加父级菜单
- var allIdList = await _sysTenantRep.Context.Queryable<SysMenu>().Select(u => new { u.Id, u.Pid }).ToListAsync();
- var pIdList = allIdList.ToChildList(u => u.Pid, u => u.Id, u => input.MenuIdList.Contains(u.Id)).Select(u => u.Pid).Distinct().ToList();
- input.MenuIdList = input.MenuIdList.Concat(pIdList).Distinct().Where(u => u != 0).ToList();
- // 保存租户菜单
- List<SysTenantMenu> sysTenantMenuList = new();
- tenantIdList.ForEach(tenantId =>
- {
- sysTenantMenuList.AddRange(input.MenuIdList.Select(menuId => new SysTenantMenu { TenantId = tenantId, MenuId = menuId }));
- });
- //默认租户授权菜单主键ID不变
- if (tenantMenuList?.Count > 0)
- {
- foreach (var item in sysTenantMenuList)
- {
- var tenantMenu = tenantMenuList.FirstOrDefault(u => u.TenantId == item.TenantId && u.MenuId == item.MenuId);
- if (tenantMenu != null) item.Id = tenantMenu.Id;
- }
- }
- await _sysTenantMenuRep.InsertRangeAsync(sysTenantMenuList);
- // 清除菜单权限缓存
- SysMenuService.DeleteMenuCache();
- }
- /// <summary>
- /// 获取租户菜单Id集合 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("获取租户菜单Id集合")]
- public async Task<List<long>> GetTenantMenuList([FromQuery] BaseIdInput input)
- {
- var menuIds = await _sysTenantMenuRep.AsQueryable().Where(u => u.TenantId == input.Id).Select(u => u.MenuId).ToListAsync();
- return await SysMenuService.ExcludeParentMenuOfFullySelected(menuIds);
- }
- /// <summary>
- /// 重置租户管理员密码 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("重置租户管理员密码")]
- public async Task<string> ResetPwd(TenantUserInput input)
- {
- var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
- var encryptPassword = CryptogramUtil.Encrypt(password);
- await _sysUserRep.UpdateAsync(u => new SysUser { Password = encryptPassword }, u => u.Id == input.UserId);
- return password;
- }
- /// <summary>
- /// 切换租户 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [UnitOfWork]
- [DisplayName("切换租户")]
- public async Task<LoginOutput> ChangeTenant(BaseIdInput input)
- {
- var userId = (App.HttpContext?.User.FindFirst(ClaimConst.UserId)?.Value)?.ToLong();
- _ = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
- var user = await _sysUserRep.GetFirstAsync(u => u.Id == userId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
- user.TenantId = input.Id;
- return await GetAccessTokenInNotSingleLogin(user);
- }
- /// <summary>
- /// 进入租管端 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("进入租管端")]
- public async Task<LoginOutput> GoTenant(BaseIdInput input)
- {
- var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
- var user = await _sysUserRep.GetFirstAsync(u => u.Id == tenant.UserId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
- return await GetAccessTokenInNotSingleLogin(user);
- }
- /// <summary>
- /// 同步授权菜单(用于版本更新后,同步授权数据) 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [UnitOfWork]
- [DisplayName("同步授权菜单")]
- public async Task SyncGrantMenu(BaseIdInput input)
- {
- var menuIdList = input.Id == SqlSugarConst.DefaultTenantId
- ? new SysMenuSeedData().HasData().Select(u => u.Id).ToList()
- : await _sysRoleRep.AsQueryable().ClearFilter()
- .InnerJoin<SysTenant>((u, t) => t.Id == input.Id && u.TenantId == t.Id)
- .InnerJoin<SysRoleMenu>((u, t, rm) => u.Id == rm.RoleId)
- .Select((u, t, rm) => rm.MenuId)
- .Distinct()
- .ToListAsync() ?? throw Oops.Oh(ErrorCodeEnum.D1019);
- // 兼容历史租户:确保“切换租户”按钮权限存在
- var switchTenantMenuId = await _sysRoleRep.Context.Queryable<SysMenu>()
- .Where(u => u.Type == MenuTypeEnum.Btn && u.Permission == "sysTenant:changeTenant")
- .Select(u => u.Id)
- .FirstAsync();
- if (switchTenantMenuId > 0 && !menuIdList.Contains(switchTenantMenuId))
- menuIdList.Add(switchTenantMenuId);
- var adminRole = await _sysRoleRep.AsQueryable().ClearFilter().FirstAsync(u => u.TenantId == input.Id && u.Code == "sys_admin");
- if (adminRole != null)
- {
- await _sysRoleRep.Context.Deleteable<SysUserRole>().Where(u => u.RoleId == adminRole.Id).ExecuteCommandAsync();
- await App.GetService<SysRoleService>().DeleteRole(new DeleteRoleInput { Id = adminRole.Id });
- }
- await GrantMenu(new TenantMenuInput { Id = input.Id, MenuIdList = menuIdList });
- }
- /// <summary>
- /// 在非单用户登录模式下获取登录令牌
- /// </summary>
- /// <param name="user"></param>
- /// <returns></returns>
- [NonAction]
- public async Task<LoginOutput> GetAccessTokenInNotSingleLogin(SysUser user)
- {
- // 使用非单用户模式登录
- var singleLogin = _sysCacheService.Get<bool>($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}");
- try
- {
- _sysCacheService.Set($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}", false);
- return await App.GetService<SysAuthService>().CreateToken(user);
- }
- finally
- {
- // 恢复单用户登录参数
- if (singleLogin) _sysCacheService.Set($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}", true);
- }
- }
- /// <summary>
- /// 缓存所有租户
- /// </summary>
- /// <param name="tenantId"></param>
- /// <returns></returns>
- [NonAction]
- public async Task CacheTenant(long tenantId = 0)
- {
- // 移除 ISqlSugarClient 中的库连接并排除默认主库
- if (tenantId > 0 && tenantId.ToString() != SqlSugarConst.MainConfigId)
- _sysTenantRep.AsTenant().RemoveConnection(tenantId);
- var tenantList = await _sysTenantRep.GetListAsync();
- // 对租户库连接进行SM2加密
- foreach (var tenant in tenantList.Where(tenant => !string.IsNullOrWhiteSpace(tenant.Connection)))
- tenant.Connection = CryptogramUtil.SM2Encrypt(tenant.Connection);
- _sysCacheService.Set(CacheConst.KeyTenant, tenantList);
- }
- /// <summary>
- /// 创建租户数据库 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [ApiDescriptionSettings(Name = "CreateDb"), HttpPost]
- [DisplayName("创建租户数据库")]
- public async Task CreateDb(TenantInput input)
- {
- var tenant = await _sysTenantRep.GetSingleAsync(u => u.Id == input.Id);
- if (tenant == null) return;
- if (tenant.DbType == SqlSugar.DbType.Oracle)
- throw Oops.Oh(ErrorCodeEnum.Z1002);
- if (string.IsNullOrWhiteSpace(tenant.Connection) || tenant.Connection.Length < 10)
- throw Oops.Oh(ErrorCodeEnum.Z1004);
- // 默认数据库配置
- var defaultConfig = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault();
- var config = new DbConnectionConfig
- {
- ConfigId = tenant.Id.ToString(),
- DbType = tenant.DbType,
- ConnectionString = tenant.Connection,
- DbSettings = new DbSettings()
- {
- EnableInitDb = true,
- EnableDiffLog = false,
- EnableUnderLine = defaultConfig!.DbSettings.EnableUnderLine,
- }
- };
- SqlSugarSetup.InitTenantDatabase(App.GetRequiredService<ISqlSugarClient>().AsTenant(), config);
- }
- /// <summary>
- /// 获取租户下的用户列表 🔖
- /// </summary>
- /// <param name="input"></param>
- /// <returns></returns>
- [DisplayName("获取租户下的用户列表")]
- public async Task<List<SysUser>> UserList(TenantIdInput input)
- {
- return await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.TenantId).ToListAsync();
- }
- /// <summary>
- /// 获取租户数据库连接
- /// </summary>
- /// <returns></returns>
- [NonAction]
- public SqlSugarScopeProvider GetTenantDbConnectionScope(long tenantId)
- {
- var iTenant = _sysTenantRep.AsTenant();
- // 若已存在租户库连接,则直接返回
- if (iTenant.IsAnyConnection(tenantId.ToString())) return iTenant.GetConnectionScope(tenantId.ToString());
- lock (iTenant)
- {
- // 从缓存里面获取租户信息
- var tenant = _sysCacheService.Get<List<SysTenant>>(CacheConst.KeyTenant)?.FirstOrDefault(u => u.Id == tenantId);
- if (tenant == null || tenant.TenantType == TenantTypeEnum.Id) return null;
- // 获取默认库连接配置
- var dbOptions = App.GetOptions<DbConnectionOptions>();
- var mainConnConfig = dbOptions.ConnectionConfigs.First(u => u.ConfigId.ToString() == SqlSugarConst.MainConfigId);
- // 设置租户库连接配置
- var tenantConnConfig = new DbConnectionConfig
- {
- ConfigId = tenant.Id.ToString(),
- DbType = tenant.DbType,
- TenantType = tenant.TenantType,
- IsAutoCloseConnection = true,
- ConnectionString = CryptogramUtil.SM2Decrypt(tenant.Connection), // 对租户库连接进行SM2解密
- DbSettings = new DbSettings()
- {
- EnableUnderLine = mainConnConfig.DbSettings.EnableUnderLine,
- },
- SlaveConnectionConfigs = JSON.IsValid(tenant.SlaveConnections) ? JSON.Deserialize<List<SlaveConnectionConfig>>(tenant.SlaveConnections) : null // 从库连接配置
- };
- iTenant.AddConnection(tenantConnConfig);
- var sqlSugarScopeProvider = iTenant.GetConnectionScope(tenantId.ToString());
- SqlSugarSetup.SetDbConfig(tenantConnConfig);
- SqlSugarSetup.SetDbAop(sqlSugarScopeProvider, dbOptions.EnableConsoleSql, dbOptions.SuperAdminIgnoreIDeletedFilter);
- return sqlSugarScopeProvider;
- }
- }
- }
|