SysTenantService.cs 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750
  1. // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
  2. //
  3. // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
  4. //
  5. // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
  6. namespace Admin.NET.Core.Service;
  7. /// <summary>
  8. /// 系统租户管理服务 🧩
  9. /// </summary>
  10. [ApiDescriptionSettings(Order = 390)]
  11. public class SysTenantService : IDynamicApiController, ITransient
  12. {
  13. private static readonly SysMenuService SysMenuService = App.GetService<SysMenuService>();
  14. private readonly SqlSugarRepository<SysUserExtOrg> _sysUserExtOrgRep;
  15. private readonly SqlSugarRepository<SysTenantMenu> _sysTenantMenuRep;
  16. private readonly SqlSugarRepository<SysRoleMenu> _sysRoleMenuRep;
  17. private readonly SqlSugarRepository<SysUserRole> _userRoleRep;
  18. private readonly SqlSugarRepository<SysTenant> _sysTenantRep;
  19. private readonly SqlSugarRepository<SysRole> _sysRoleRep;
  20. private readonly SqlSugarRepository<SysUser> _sysUserRep;
  21. private readonly SqlSugarRepository<SysOrg> _sysOrgRep;
  22. private readonly SqlSugarRepository<SysPos> _sysPosRep;
  23. private readonly SysRoleMenuService _sysRoleMenuService;
  24. private readonly SysConfigService _sysConfigService;
  25. private readonly SysCacheService _sysCacheService;
  26. private readonly UploadOptions _uploadOptions;
  27. public SysTenantService(
  28. SqlSugarRepository<SysUserExtOrg> sysUserExtOrgRep,
  29. SqlSugarRepository<SysTenantMenu> sysTenantMenuRep,
  30. SqlSugarRepository<SysRoleMenu> sysRoleMenuRep,
  31. SqlSugarRepository<SysUserRole> userRoleRep,
  32. SqlSugarRepository<SysTenant> sysTenantRep,
  33. SqlSugarRepository<SysUser> sysUserRep,
  34. SqlSugarRepository<SysRole> sysRoleRep,
  35. SqlSugarRepository<SysOrg> sysOrgRep,
  36. SqlSugarRepository<SysPos> sysPosRep,
  37. IOptions<UploadOptions> uploadOptions,
  38. SysRoleMenuService sysRoleMenuService,
  39. SysConfigService sysConfigService,
  40. SysCacheService sysCacheService)
  41. {
  42. _sysTenantRep = sysTenantRep;
  43. _sysOrgRep = sysOrgRep;
  44. _sysRoleRep = sysRoleRep;
  45. _sysPosRep = sysPosRep;
  46. _sysUserRep = sysUserRep;
  47. _userRoleRep = userRoleRep;
  48. _sysRoleMenuRep = sysRoleMenuRep;
  49. _sysCacheService = sysCacheService;
  50. _uploadOptions = uploadOptions.Value;
  51. _sysConfigService = sysConfigService;
  52. _sysTenantMenuRep = sysTenantMenuRep;
  53. _sysUserExtOrgRep = sysUserExtOrgRep;
  54. _sysRoleMenuService = sysRoleMenuService;
  55. }
  56. /// <summary>
  57. /// 获取租户分页列表 🔖
  58. /// </summary>
  59. /// <param name="input"></param>
  60. /// <returns></returns>
  61. [DisplayName("获取租户分页列表")]
  62. public async Task<SqlSugarPagedList<TenantOutput>> Page(PageTenantInput input)
  63. {
  64. return await _sysTenantRep.AsQueryable()
  65. .LeftJoin<SysUser>((u, a) => u.UserId == a.Id)
  66. .LeftJoin<SysOrg>((u, a, b) => u.OrgId == b.Id)
  67. .WhereIF(!string.IsNullOrWhiteSpace(input.Phone), (u, a) => a.Phone.Contains(input.Phone.Trim()))
  68. .WhereIF(!string.IsNullOrWhiteSpace(input.Name), (u, a, b) => b.Name.Contains(input.Name.Trim()))
  69. .OrderBy(u => new { u.OrderNo, u.Id })
  70. .Select((u, a, b) => new TenantOutput
  71. {
  72. Id = u.Id,
  73. OrgId = b.Id,
  74. Name = b.Name,
  75. UserId = a.Id,
  76. AdminAccount = a.Account,
  77. Phone = a.Phone,
  78. Host = u.Host,
  79. Email = a.Email,
  80. TenantType = u.TenantType,
  81. DbType = u.DbType,
  82. Connection = u.Connection,
  83. ConfigId = u.ConfigId,
  84. OrderNo = u.OrderNo,
  85. Remark = u.Remark,
  86. Status = u.Status,
  87. CreateTime = u.CreateTime,
  88. CreateUserName = u.CreateUserName,
  89. UpdateTime = u.UpdateTime,
  90. UpdateUserName = u.UpdateUserName,
  91. }, true)
  92. .ToPagedListAsync(input.Page, input.PageSize);
  93. }
  94. /// <summary>
  95. /// 获取租户列表
  96. /// </summary>
  97. /// <returns></returns>
  98. [AllowAnonymous]
  99. [DisplayName("获取租户列表"), HttpGet]
  100. public async Task<dynamic> GetList()
  101. {
  102. return await _sysTenantRep.AsQueryable()
  103. .LeftJoin<SysOrg>((u, a) => u.OrgId == a.Id).ClearFilter()
  104. .Where(u => u.Status == StatusEnum.Enable && u.IsDelete == false)
  105. .Select((u, a) => new
  106. {
  107. Label = $"{u.Title}-{a.Name}",
  108. Host = u.Host.ToLower(),
  109. Value = u.Id,
  110. }).ToListAsync();
  111. }
  112. /// <summary>
  113. /// 获取当前租户
  114. /// </summary>
  115. /// <returns></returns>
  116. [NonAction]
  117. public async Task<SysTenant> GetCurrentTenant()
  118. {
  119. var tenantId = long.Parse(App.User?.FindFirst(ClaimConst.TenantId)?.Value ?? "0");
  120. var host = App.HttpContext.Request.Host.Host.ToLower();
  121. return await _sysTenantRep.AsQueryable()
  122. .WhereIF(tenantId > 0, u => u.Id == tenantId)
  123. .WhereIF(!(tenantId > 0), u => SqlFunc.ToLower(u.Host).Equals(host))
  124. .FirstAsync();
  125. }
  126. /// <summary>
  127. /// 获取库隔离的租户列表
  128. /// </summary>
  129. /// <returns></returns>
  130. [NonAction]
  131. public async Task<List<SysTenant>> GetTenantDbList()
  132. {
  133. return await _sysTenantRep.GetListAsync(u => u.TenantType == TenantTypeEnum.Db && u.Status == StatusEnum.Enable);
  134. }
  135. /// <summary>
  136. /// 增加租户 🔖
  137. /// </summary>
  138. /// <param name="input"></param>
  139. /// <returns></returns>
  140. [UnitOfWork]
  141. [ApiDescriptionSettings(Name = "Add"), HttpPost]
  142. [DisplayName("增加租户")]
  143. public async Task AddTenant(AddTenantInput input)
  144. {
  145. var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name);
  146. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
  147. input.Host = input.Host?.ToLower();
  148. isExist = await _sysTenantRep.IsAnyAsync(u => !string.IsNullOrWhiteSpace(u.Host) && u.Host == input.Host);
  149. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
  150. isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.Account == input.AdminAccount);
  151. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
  152. // 从库配置判断
  153. if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections)) throw Oops.Oh(ErrorCodeEnum.D1302);
  154. switch (input.TenantType)
  155. {
  156. // Id隔离时设置与主库一致
  157. case TenantTypeEnum.Id:
  158. var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
  159. input.DbType = config.DbType;
  160. input.Connection = config.ConnectionString;
  161. break;
  162. case TenantTypeEnum.Db:
  163. if (string.IsNullOrWhiteSpace(input.Connection))
  164. throw Oops.Oh(ErrorCodeEnum.Z1004);
  165. break;
  166. default:
  167. throw Oops.Oh(ErrorCodeEnum.D3004);
  168. }
  169. if (input.EnableReg == YesNoEnum.N) input.RegWayId = null;
  170. var tenant = input.Adapt<TenantOutput>();
  171. // 设置logo
  172. SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
  173. tenant.Id = _sysTenantRep.InsertReturnEntity(tenant).Id;
  174. await InitNewTenant(tenant);
  175. await CacheTenant();
  176. }
  177. /// <summary>
  178. /// 设置logo
  179. /// </summary>
  180. /// <param name="tenant"></param>
  181. /// <param name="logoBase64"></param>
  182. /// <param name="logoFileName"></param>
  183. [NonAction]
  184. public void SetLogoUrl(SysTenant tenant, string logoBase64, string logoFileName)
  185. {
  186. // 旧图标文件相对路径
  187. var oldSysLogoRelativeFilePath = tenant.Logo ?? "";
  188. var oldSysLogoAbsoluteFilePath = Path.Combine(App.WebHostEnvironment.WebRootPath, oldSysLogoRelativeFilePath.TrimStart('/'));
  189. var groups = Regex.Match(logoBase64, @"data:image/(?<type>.+?);base64,(?<data>.+)").Groups;
  190. //var type = groups["type"].Value;
  191. var base64Data = groups["data"].Value;
  192. var binData = Convert.FromBase64String(base64Data);
  193. // 根据文件名取扩展名
  194. var ext = string.IsNullOrWhiteSpace(logoFileName) ? ".png" : Path.GetExtension(logoFileName);
  195. // 本地图标保存路径
  196. var fileName = $"{tenant.ViceTitle}-logo{ext}".ToLower();
  197. var path = _uploadOptions.Path.Replace("/{yyyy}/{MM}/{dd}", "");
  198. path = path.StartsWith("/") || Regex.IsMatch(path, "^[A-Z|a-z]:") ? path : Path.Combine(App.WebHostEnvironment.WebRootPath, path);
  199. var absoluteFilePath = Path.Combine(path, fileName);
  200. // 删除已存在文件
  201. if (File.Exists(oldSysLogoAbsoluteFilePath)) File.Delete(oldSysLogoAbsoluteFilePath);
  202. // 创建文件夹
  203. var absoluteFileDir = Path.GetDirectoryName(absoluteFilePath);
  204. if (!Directory.Exists(absoluteFileDir)) Directory.CreateDirectory(absoluteFileDir);
  205. // 保存图标文件
  206. File.WriteAllBytesAsync(absoluteFilePath, binData);
  207. // 保存图标配置
  208. tenant.Logo = $"/upload/{fileName}";
  209. }
  210. /// <summary>
  211. /// 设置租户状态 🔖
  212. /// </summary>
  213. /// <param name="input"></param>
  214. /// <returns></returns>
  215. [DisplayName("设置租户状态")]
  216. public async Task<int> SetStatus(TenantInput input)
  217. {
  218. var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id);
  219. if (tenant == null || tenant.ConfigId == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.Z1001);
  220. if (!Enum.IsDefined(typeof(StatusEnum), input.Status)) throw Oops.Oh(ErrorCodeEnum.D3005);
  221. tenant.Status = input.Status;
  222. return await _sysTenantRep.AsUpdateable(tenant).UpdateColumns(u => new { u.Status }).ExecuteCommandAsync();
  223. }
  224. /// <summary>
  225. /// 新增租户初始化
  226. /// </summary>
  227. /// <param name="tenant"></param>
  228. private async Task InitNewTenant(TenantOutput tenant)
  229. {
  230. var tenantId = tenant.Id;
  231. var tenantName = tenant.Name;
  232. // 初始化机构
  233. var newOrg = new SysOrg { TenantId = tenantId, Pid = 0, Name = tenantName, Code = tenantName, Remark = tenantName, };
  234. await _sysOrgRep.InsertAsync(newOrg);
  235. // 初始化默认角色
  236. var newRole = new SysRole { TenantId = tenantId, Name = CommonConst.DefaultBaseRoleName, Code = CommonConst.DefaultBaseRoleCode, DataScope = DataScopeEnum.Self, Remark = "此角色为系统自动创建角色" };
  237. var baseRole = await _sysRoleRep.InsertReturnEntityAsync(newRole);
  238. var baseRoleMenuIdList = GetBaseRoleMenuIdList().ToList();
  239. await _sysRoleMenuService.GrantRoleMenu(new RoleMenuInput { Id = baseRole.Id, MenuIdList = baseRoleMenuIdList.Select(u => u.MenuId).ToList() });
  240. // 初始化职位
  241. var newPos = new SysPos { TenantId = tenantId, Name = "管理员-" + tenantName, Code = tenantName, Remark = tenantName };
  242. await _sysPosRep.InsertAsync(newPos);
  243. // 初始化租户管理员账号
  244. var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
  245. var newUser = new SysUser
  246. {
  247. TenantId = tenantId,
  248. Account = tenant.AdminAccount,
  249. Password = CryptogramUtil.Encrypt(password),
  250. NickName = "系统管理员",
  251. Email = tenant.Email,
  252. Phone = tenant.Phone,
  253. AccountType = AccountTypeEnum.SysAdmin,
  254. OrgId = newOrg.Id,
  255. PosId = newPos.Id,
  256. Birthday = DateTime.Parse("2000-01-01"),
  257. RealName = "系统管理员",
  258. Remark = "系统管理员" + tenantName,
  259. };
  260. await _sysUserRep.InsertAsync(newUser);
  261. // 关联租户组织机构和管理员用户
  262. await _sysTenantRep.UpdateAsync(u => new SysTenant { UserId = newUser.Id, OrgId = newOrg.Id }, u => u.Id == tenantId);
  263. // 默认租户管理员角色菜单集合
  264. var menuList = GetTenantDefaultMenuList().ToList();
  265. await GrantMenu(new TenantMenuInput { Id = tenantId, MenuIdList = menuList.Select(u => u.MenuId).ToList() });
  266. }
  267. /// <summary>
  268. /// 获取租户默认菜单
  269. /// </summary>
  270. /// <param name="ignoreHome">如果某租户需要定制主页,可以忽略</param>
  271. /// <returns></returns>
  272. [NonAction]
  273. public IEnumerable<SysTenantMenu> GetTenantDefaultMenuList(bool ignoreHome = false)
  274. {
  275. var menuList = new List<SysMenu>();
  276. // 默认数据库配置
  277. var defaultConfig = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault();
  278. //从程序集中获取种子菜单数据,种子菜单存在于其他类库中,需要动态加载
  279. var menuSeedDataTypeList = GetSeedDataTypes(defaultConfig, nameof(SysMenuSeedData));
  280. var allMenuList = new List<SysMenu>();
  281. foreach (var menu in menuSeedDataTypeList)
  282. {
  283. var menuSeedDataList = ((IEnumerable)menu.GetMethod("HasData")?.Invoke(Activator.CreateInstance(menu), null))?.Cast<SysMenu>();
  284. if (menuSeedDataList != null)
  285. {
  286. allMenuList.AddRange(menuSeedDataList);
  287. }
  288. }
  289. //实现三个层级的菜单
  290. var topMenuList = allMenuList.Where(u => u.Pid == 0 && u.Type == MenuTypeEnum.Dir).ToList();
  291. menuList.AddRange(topMenuList);
  292. var childMenuList = allMenuList.ToChildList(u => u.Id, u => u.Pid, u => topMenuList.Select(p => p.Id).Contains(u.Pid));
  293. menuList.AddRange(childMenuList);
  294. var endMenuList = allMenuList.ToChildList(u => u.Id, u => u.Pid, u => childMenuList.Select(p => p.Id).Contains(u.Pid));
  295. if (endMenuList != null)
  296. {
  297. menuList.AddRange(endMenuList);
  298. }
  299. //是否需要排除首页菜单
  300. if (ignoreHome) menuList = menuList.Where(u => !(u.Type == MenuTypeEnum.Menu && u.Name == "home")).ToList();
  301. menuList = menuList.Distinct().ToList();
  302. return menuList.Select(u => new SysTenantMenu
  303. {
  304. Id = CommonUtil.GetFixedHashCode("" + SqlSugarConst.DefaultTenantId + u.Id, 1300000000000),
  305. TenantId = SqlSugarConst.DefaultTenantId,
  306. MenuId = u.Id
  307. });
  308. }
  309. /// <summary>
  310. /// 获取种子数据类型
  311. /// </summary>
  312. /// <param name="config">数据库连接配置</param>
  313. /// <returns>种子数据类型列表</returns>
  314. [NonAction]
  315. private List<Type> GetSeedDataTypes(DbConnectionConfig config, string typeName)
  316. {
  317. return App.EffectiveTypes
  318. .Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass && u.Name == typeName && u.GetInterfaces().Any(i => i.HasImplementedRawGeneric(typeof(ISqlSugarEntitySeedData<>))))
  319. .WhereIF(config.SeedSettings.EnableIncreSeed, u => u.IsDefined(typeof(IncreSeedAttribute), false))
  320. .OrderBy(u => u.GetCustomAttributes(typeof(SeedDataAttribute), false).Length > 0 ? ((SeedDataAttribute)u.GetCustomAttributes(typeof(SeedDataAttribute), false)[0]).Order : 0)
  321. .ToList();
  322. }
  323. /// <summary>
  324. /// 获取租户默认菜单
  325. /// </summary>
  326. /// <returns></returns>
  327. [NonAction]
  328. public IEnumerable<SysTenantMenu> GetBaseRoleMenuIdList()
  329. {
  330. var menuList = new List<SysMenu>();
  331. var allMenuList = new SysMenuSeedData().HasData().ToList();
  332. var dashboardMenu = allMenuList.First(u => u.Type == MenuTypeEnum.Dir && u.Title == "工作台");
  333. menuList.AddRange(allMenuList.ToChildList(u => u.Id, u => u.Pid, dashboardMenu.Id));
  334. var systemMenu = allMenuList.First(u => u.Type == MenuTypeEnum.Dir && u.Title == "系统管理");
  335. menuList.Add(systemMenu);
  336. menuList.AddRange(allMenuList.ToChildList(u => u.Id, u => u.Pid, u => u.Pid == systemMenu.Id && new[] { "机构管理", "个人中心" }.Contains(u.Title)));
  337. menuList = menuList.Where(u => !new[] { "增加", "编辑", "删除" }.Contains(u.Title)).ToList();
  338. return menuList.Select(u => new SysTenantMenu
  339. {
  340. Id = CommonUtil.GetFixedHashCode("" + SqlSugarConst.DefaultTenantId + u.Id, 1300000000000),
  341. TenantId = SqlSugarConst.DefaultTenantId,
  342. MenuId = u.Id
  343. });
  344. }
  345. /// <summary>
  346. /// 删除租户 🔖
  347. /// </summary>
  348. /// <param name="input"></param>
  349. /// <returns></returns>
  350. [UnitOfWork]
  351. [ApiDescriptionSettings(Name = "Delete"), HttpPost]
  352. [DisplayName("删除租户")]
  353. public async Task DeleteTenant(DeleteTenantInput input)
  354. {
  355. // 禁止删除默认租户
  356. if (input.Id.ToString() == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.D1023);
  357. // 若账号为开放接口绑定租户则禁止删除
  358. var isOpenAccessTenant = await _sysTenantRep.ChangeRepository<SqlSugarRepository<SysOpenAccess>>().IsAnyAsync(u => u.BindTenantId == input.Id);
  359. if (isOpenAccessTenant) throw Oops.Oh(ErrorCodeEnum.D1031);
  360. await _sysTenantRep.DeleteAsync(u => u.Id == input.Id);
  361. await CacheTenant(input.Id);
  362. // 删除与租户相关的表数据
  363. await _sysTenantMenuRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  364. await _sysTenantRep.Context.Deleteable<SysTenantConfigData>().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  365. var users = await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.Id).ToListAsync();
  366. var userIds = users.Select(u => u.Id).ToList();
  367. await _sysUserRep.AsDeleteable().Where(u => userIds.Contains(u.Id)).ExecuteCommandAsync();
  368. await _userRoleRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  369. await _sysUserExtOrgRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  370. await _sysTenantRep.Context.Deleteable<SysUserMenu>().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  371. await _sysTenantRep.Context.Deleteable<SysUserConfigData>().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  372. var roleIds = await _sysRoleRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.Id).Select(u => u.Id).ToListAsync();
  373. await _sysRoleRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  374. await _sysRoleMenuRep.AsDeleteable().Where(u => roleIds.Contains(u.RoleId)).ExecuteCommandAsync();
  375. await _sysTenantRep.Context.Deleteable<SysRoleOrg>().Where(u => roleIds.Contains(u.RoleId)).ExecuteCommandAsync();
  376. await _sysOrgRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  377. await _sysPosRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  378. }
  379. /// <summary>
  380. /// 更新租户 🔖
  381. /// </summary>
  382. /// <param name="input"></param>
  383. /// <returns></returns>
  384. [ApiDescriptionSettings(Name = "Update"), HttpPost]
  385. [DisplayName("更新租户")]
  386. public async Task UpdateTenant(UpdateTenantInput input)
  387. {
  388. var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name && u.Id != input.OrgId);
  389. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
  390. input.Host = input.Host?.ToLower();
  391. isExist = await _sysTenantRep.IsAnyAsync(u => !string.IsNullOrWhiteSpace(u.Host) && u.Host == input.Host && u.Id != input.Id);
  392. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
  393. isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.Account == input.AdminAccount && u.Id != input.UserId);
  394. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
  395. // Id隔离时设置与主库一致
  396. switch (input.TenantType)
  397. {
  398. case TenantTypeEnum.Id:
  399. var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
  400. input.DbType = config.DbType;
  401. input.Connection = config.ConnectionString;
  402. break;
  403. case TenantTypeEnum.Db:
  404. if (string.IsNullOrWhiteSpace(input.Connection))
  405. throw Oops.Oh(ErrorCodeEnum.Z1004);
  406. break;
  407. default:
  408. throw Oops.Oh(ErrorCodeEnum.D3004);
  409. }
  410. // 从库配置判断
  411. if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections))
  412. throw Oops.Oh(ErrorCodeEnum.D1302);
  413. // 设置logo
  414. var tenant = input.Adapt<SysTenant>();
  415. if (!string.IsNullOrWhiteSpace(input.LogoBase64)) SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
  416. // 更新租户信息
  417. await _sysTenantRep.AsUpdateable(tenant).IgnoreColumns(true).ExecuteCommandAsync();
  418. // 更新系统机构
  419. await _sysOrgRep.UpdateAsync(u => new SysOrg() { Name = input.Name }, u => u.Id == input.OrgId);
  420. // 更新系统用户
  421. await _sysUserRep.UpdateAsync(u => new SysUser() { Account = input.AdminAccount, Phone = input.Phone, Email = input.Email }, u => u.Id == input.UserId);
  422. await CacheTenant(input.Id);
  423. }
  424. /// <summary>
  425. /// 授权租户菜单 🔖
  426. /// </summary>
  427. /// <param name="input"></param>
  428. /// <returns></returns>
  429. [UnitOfWork]
  430. [DisplayName("授权租户菜单")]
  431. public async Task GrantMenu(TenantMenuInput input)
  432. {
  433. // 获取需要授权的菜单列表
  434. var menuList = await _sysTenantRep.Context.Queryable<SysMenu>()
  435. .Where(u => input.MenuIdList.Contains(u.Id))
  436. .InnerJoin<SysTenantMenu>((u, t) => t.TenantId == input.Id && u.Id == t.MenuId)
  437. .ToListAsync();
  438. // 检查是否存在重复菜单
  439. if (menuList.Where(u => u.Type != MenuTypeEnum.Btn).GroupBy(u => new { u.Pid, u.Title }).Any(u => u.Count() > 1) ||
  440. menuList.Where(u => u.Type == MenuTypeEnum.Btn).GroupBy(u => u.Permission).Any(u => u.Count() > 1))
  441. throw Oops.Oh(ErrorCodeEnum.D1304);
  442. // 检查路由是否重复
  443. if (menuList.Where(u => !string.IsNullOrWhiteSpace(u.Name)).GroupBy(u => u.Name).Any(u => u.Count() > 1))
  444. throw Oops.Oh(ErrorCodeEnum.D4009);
  445. // 删除旧记录
  446. await _sysTenantMenuRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  447. // 追加父级菜单
  448. var allIdList = await _sysTenantRep.Context.Queryable<SysMenu>().Select(u => new { u.Id, u.Pid }).ToListAsync();
  449. var pIdList = allIdList.ToChildList(u => u.Pid, u => u.Id, u => input.MenuIdList.Contains(u.Id)).Select(u => u.Pid).Distinct().ToList();
  450. input.MenuIdList = input.MenuIdList.Concat(pIdList).Distinct().Where(u => u != 0).ToList();
  451. // 保存租户菜单
  452. var sysTenantMenuList = input.MenuIdList.Select(menuId => new SysTenantMenu { TenantId = input.Id, MenuId = menuId }).ToList();
  453. await _sysTenantMenuRep.InsertRangeAsync(sysTenantMenuList);
  454. // 清除菜单权限缓存
  455. SysMenuService.DeleteMenuCache();
  456. }
  457. /// <summary>
  458. /// 获取租户菜单Id集合 🔖
  459. /// </summary>
  460. /// <param name="input"></param>
  461. /// <returns></returns>
  462. [DisplayName("获取租户菜单Id集合")]
  463. public async Task<List<long>> GetTenantMenuList([FromQuery] BaseIdInput input)
  464. {
  465. var menuIds = await _sysTenantMenuRep.AsQueryable().Where(u => u.TenantId == input.Id).Select(u => u.MenuId).ToListAsync();
  466. return await SysMenuService.ExcludeParentMenuOfFullySelected(menuIds);
  467. }
  468. /// <summary>
  469. /// 重置租户管理员密码 🔖
  470. /// </summary>
  471. /// <param name="input"></param>
  472. /// <returns></returns>
  473. [DisplayName("重置租户管理员密码")]
  474. public async Task<string> ResetPwd(TenantUserInput input)
  475. {
  476. var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
  477. var encryptPassword = CryptogramUtil.Encrypt(password);
  478. await _sysUserRep.UpdateAsync(u => new SysUser { Password = encryptPassword }, u => u.Id == input.UserId);
  479. return password;
  480. }
  481. /// <summary>
  482. /// 切换租户 🔖
  483. /// </summary>
  484. /// <param name="input"></param>
  485. /// <returns></returns>
  486. [UnitOfWork]
  487. [DisplayName("切换租户")]
  488. public async Task<LoginOutput> ChangeTenant(BaseIdInput input)
  489. {
  490. var userId = (App.HttpContext?.User.FindFirst(ClaimConst.UserId)?.Value)?.ToLong();
  491. _ = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  492. var user = await _sysUserRep.GetFirstAsync(u => u.Id == userId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  493. user.TenantId = input.Id;
  494. return await GetAccessTokenInNotSingleLogin(user);
  495. }
  496. /// <summary>
  497. /// 进入租管端 🔖
  498. /// </summary>
  499. /// <param name="input"></param>
  500. /// <returns></returns>
  501. [DisplayName("进入租管端")]
  502. public async Task<LoginOutput> GoTenant(BaseIdInput input)
  503. {
  504. var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  505. var user = await _sysUserRep.GetFirstAsync(u => u.Id == tenant.UserId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  506. return await GetAccessTokenInNotSingleLogin(user);
  507. }
  508. /// <summary>
  509. /// 同步授权菜单(用于版本更新后,同步授权数据) 🔖
  510. /// </summary>
  511. /// <param name="input"></param>
  512. /// <returns></returns>
  513. [UnitOfWork]
  514. [DisplayName("同步授权菜单")]
  515. public async Task SyncGrantMenu(BaseIdInput input)
  516. {
  517. var menuIdList = input.Id == SqlSugarConst.DefaultTenantId
  518. ? new SysMenuSeedData().HasData().Select(u => u.Id).ToList()
  519. : await _sysRoleRep.AsQueryable().ClearFilter()
  520. .InnerJoin<SysTenant>((u, t) => t.Id == input.Id && u.TenantId == t.Id)
  521. .InnerJoin<SysRoleMenu>((u, t, rm) => u.Id == rm.RoleId)
  522. .Select((u, t, rm) => rm.MenuId)
  523. .Distinct()
  524. .ToListAsync() ?? throw Oops.Oh(ErrorCodeEnum.D1019);
  525. var adminRole = await _sysRoleRep.AsQueryable().ClearFilter().FirstAsync(u => u.TenantId == input.Id && u.Code == "sys_admin");
  526. if (adminRole != null)
  527. {
  528. await _sysRoleRep.Context.Deleteable<SysUserRole>().Where(u => u.RoleId == adminRole.Id).ExecuteCommandAsync();
  529. await App.GetService<SysRoleService>().DeleteRole(new DeleteRoleInput { Id = adminRole.Id });
  530. }
  531. await GrantMenu(new TenantMenuInput { Id = input.Id, MenuIdList = menuIdList });
  532. }
  533. /// <summary>
  534. /// 在非单用户登录模式下获取登录令牌
  535. /// </summary>
  536. /// <param name="user"></param>
  537. /// <returns></returns>
  538. [NonAction]
  539. public async Task<LoginOutput> GetAccessTokenInNotSingleLogin(SysUser user)
  540. {
  541. // 使用非单用户模式登录
  542. var singleLogin = _sysCacheService.Get<bool>($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}");
  543. try
  544. {
  545. _sysCacheService.Set($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}", false);
  546. return await App.GetService<SysAuthService>().CreateToken(user);
  547. }
  548. finally
  549. {
  550. // 恢复单用户登录参数
  551. if (singleLogin) _sysCacheService.Set($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}", true);
  552. }
  553. }
  554. /// <summary>
  555. /// 缓存所有租户
  556. /// </summary>
  557. /// <param name="tenantId"></param>
  558. /// <returns></returns>
  559. [NonAction]
  560. public async Task CacheTenant(long tenantId = 0)
  561. {
  562. // 移除 ISqlSugarClient 中的库连接并排除默认主库
  563. if (tenantId > 0 && tenantId.ToString() != SqlSugarConst.MainConfigId)
  564. _sysTenantRep.AsTenant().RemoveConnection(tenantId);
  565. var tenantList = await _sysTenantRep.GetListAsync();
  566. // 对租户库连接进行SM2加密
  567. foreach (var tenant in tenantList.Where(tenant => !string.IsNullOrWhiteSpace(tenant.Connection)))
  568. tenant.Connection = CryptogramUtil.SM2Encrypt(tenant.Connection);
  569. _sysCacheService.Set(CacheConst.KeyTenant, tenantList);
  570. }
  571. /// <summary>
  572. /// 创建租户数据库 🔖
  573. /// </summary>
  574. /// <param name="input"></param>
  575. /// <returns></returns>
  576. [ApiDescriptionSettings(Name = "CreateDb"), HttpPost]
  577. [DisplayName("创建租户数据库")]
  578. public async Task CreateDb(TenantInput input)
  579. {
  580. var tenant = await _sysTenantRep.GetSingleAsync(u => u.Id == input.Id);
  581. if (tenant == null) return;
  582. if (tenant.DbType == SqlSugar.DbType.Oracle)
  583. throw Oops.Oh(ErrorCodeEnum.Z1002);
  584. if (string.IsNullOrWhiteSpace(tenant.Connection) || tenant.Connection.Length < 10)
  585. throw Oops.Oh(ErrorCodeEnum.Z1004);
  586. // 默认数据库配置
  587. var defaultConfig = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault();
  588. var config = new DbConnectionConfig
  589. {
  590. ConfigId = tenant.Id.ToString(),
  591. DbType = tenant.DbType,
  592. ConnectionString = tenant.Connection,
  593. DbSettings = new DbSettings()
  594. {
  595. EnableInitDb = true,
  596. EnableDiffLog = false,
  597. EnableUnderLine = defaultConfig!.DbSettings.EnableUnderLine,
  598. }
  599. };
  600. SqlSugarSetup.InitTenantDatabase(App.GetRequiredService<ISqlSugarClient>().AsTenant(), config);
  601. }
  602. /// <summary>
  603. /// 获取租户下的用户列表 🔖
  604. /// </summary>
  605. /// <param name="input"></param>
  606. /// <returns></returns>
  607. [DisplayName("获取租户下的用户列表")]
  608. public async Task<List<SysUser>> UserList(TenantIdInput input)
  609. {
  610. return await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.TenantId).ToListAsync();
  611. }
  612. /// <summary>
  613. /// 获取租户数据库连接
  614. /// </summary>
  615. /// <returns></returns>
  616. [NonAction]
  617. public SqlSugarScopeProvider GetTenantDbConnectionScope(long tenantId)
  618. {
  619. var iTenant = _sysTenantRep.AsTenant();
  620. // 若已存在租户库连接,则直接返回
  621. if (iTenant.IsAnyConnection(tenantId.ToString())) return iTenant.GetConnectionScope(tenantId.ToString());
  622. lock (iTenant)
  623. {
  624. // 从缓存里面获取租户信息
  625. var tenant = _sysCacheService.Get<List<SysTenant>>(CacheConst.KeyTenant)?.FirstOrDefault(u => u.Id == tenantId);
  626. if (tenant == null || tenant.TenantType == TenantTypeEnum.Id) return null;
  627. // 获取默认库连接配置
  628. var dbOptions = App.GetOptions<DbConnectionOptions>();
  629. var mainConnConfig = dbOptions.ConnectionConfigs.First(u => u.ConfigId.ToString() == SqlSugarConst.MainConfigId);
  630. // 设置租户库连接配置
  631. var tenantConnConfig = new DbConnectionConfig
  632. {
  633. ConfigId = tenant.Id.ToString(),
  634. DbType = tenant.DbType,
  635. TenantType = tenant.TenantType,
  636. IsAutoCloseConnection = true,
  637. ConnectionString = CryptogramUtil.SM2Decrypt(tenant.Connection), // 对租户库连接进行SM2解密
  638. DbSettings = new DbSettings()
  639. {
  640. EnableUnderLine = mainConnConfig.DbSettings.EnableUnderLine,
  641. },
  642. SlaveConnectionConfigs = JSON.IsValid(tenant.SlaveConnections) ? JSON.Deserialize<List<SlaveConnectionConfig>>(tenant.SlaveConnections) : null // 从库连接配置
  643. };
  644. iTenant.AddConnection(tenantConnConfig);
  645. var sqlSugarScopeProvider = iTenant.GetConnectionScope(tenantId.ToString());
  646. SqlSugarSetup.SetDbConfig(tenantConnConfig);
  647. SqlSugarSetup.SetDbAop(sqlSugarScopeProvider, dbOptions.EnableConsoleSql, dbOptions.SuperAdminIgnoreIDeletedFilter);
  648. return sqlSugarScopeProvider;
  649. }
  650. }
  651. }