SysTenantService.cs 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623
  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 readonly SqlSugarRepository<SysTenant> _sysTenantRep;
  14. private readonly SqlSugarRepository<SysOrg> _sysOrgRep;
  15. private readonly SqlSugarRepository<SysRole> _sysRoleRep;
  16. private readonly SqlSugarRepository<SysPos> _sysPosRep;
  17. private readonly SqlSugarRepository<SysUser> _sysUserRep;
  18. private readonly SqlSugarRepository<SysUserExtOrg> _sysUserExtOrgRep;
  19. private readonly SqlSugarRepository<SysTenantMenu> _sysTenantMenuRep;
  20. private readonly SqlSugarRepository<SysRoleMenu> _sysRoleMenuRep;
  21. private readonly SqlSugarRepository<SysUserRole> _userRoleRep;
  22. private readonly SysConfigService _sysConfigService;
  23. private readonly SysCacheService _sysCacheService;
  24. private readonly UploadOptions _uploadOptions;
  25. public SysTenantService(SqlSugarRepository<SysTenant> sysTenantRep,
  26. SqlSugarRepository<SysOrg> sysOrgRep,
  27. SqlSugarRepository<SysRole> sysRoleRep,
  28. SqlSugarRepository<SysPos> sysPosRep,
  29. SqlSugarRepository<SysUser> sysUserRep,
  30. SqlSugarRepository<SysUserExtOrg> sysUserExtOrgRep,
  31. SqlSugarRepository<SysTenantMenu> sysTenantMenuRep,
  32. SqlSugarRepository<SysRoleMenu> sysRoleMenuRep,
  33. SqlSugarRepository<SysUserRole> userRoleRep,
  34. IOptions<UploadOptions> uploadOptions,
  35. SysConfigService sysConfigService,
  36. SysCacheService sysCacheService)
  37. {
  38. _sysTenantRep = sysTenantRep;
  39. _sysOrgRep = sysOrgRep;
  40. _sysRoleRep = sysRoleRep;
  41. _sysPosRep = sysPosRep;
  42. _sysUserRep = sysUserRep;
  43. _sysTenantMenuRep = sysTenantMenuRep;
  44. _sysUserExtOrgRep = sysUserExtOrgRep;
  45. _sysRoleMenuRep = sysRoleMenuRep;
  46. _userRoleRep = userRoleRep;
  47. _uploadOptions = uploadOptions.Value;
  48. _sysConfigService = sysConfigService;
  49. _sysCacheService = sysCacheService;
  50. }
  51. /// <summary>
  52. /// 获取租户分页列表 🔖
  53. /// </summary>
  54. /// <param name="input"></param>
  55. /// <returns></returns>
  56. [DisplayName("获取租户分页列表")]
  57. public async Task<SqlSugarPagedList<TenantOutput>> Page(PageTenantInput input)
  58. {
  59. return await _sysTenantRep.AsQueryable()
  60. .LeftJoin<SysUser>((u, a) => u.UserId == a.Id)
  61. .LeftJoin<SysOrg>((u, a, b) => u.OrgId == b.Id)
  62. .WhereIF(!string.IsNullOrWhiteSpace(input.Phone), (u, a) => a.Phone.Contains(input.Phone.Trim()))
  63. .WhereIF(!string.IsNullOrWhiteSpace(input.Name), (u, a, b) => b.Name.Contains(input.Name.Trim()))
  64. .Where(u => u.Id.ToString() != SqlSugarConst.MainConfigId) // 排除默认主库/主租户
  65. .OrderBy(u => new { u.OrderNo, u.Id })
  66. .Select((u, a, b) => new TenantOutput
  67. {
  68. Id = u.Id,
  69. OrgId = b.Id,
  70. Name = b.Name,
  71. UserId = a.Id,
  72. AdminAccount = a.Account,
  73. Phone = a.Phone,
  74. Host = u.Host,
  75. Email = a.Email,
  76. TenantType = u.TenantType,
  77. DbType = u.DbType,
  78. Connection = u.Connection,
  79. ConfigId = u.ConfigId,
  80. OrderNo = u.OrderNo,
  81. Remark = u.Remark,
  82. Status = u.Status,
  83. CreateTime = u.CreateTime,
  84. CreateUserName = u.CreateUserName,
  85. UpdateTime = u.UpdateTime,
  86. UpdateUserName = u.UpdateUserName,
  87. }, true)
  88. .ToPagedListAsync(input.Page, input.PageSize);
  89. }
  90. /// <summary>
  91. /// 获取租户列表
  92. /// </summary>
  93. /// <returns></returns>
  94. [AllowAnonymous]
  95. [DisplayName("获取租户列表"), HttpGet]
  96. public async Task<dynamic> GetList()
  97. {
  98. return await _sysTenantRep.AsQueryable()
  99. .LeftJoin<SysOrg>((u, a) => u.OrgId == a.Id)
  100. .Where(u => u.Status == StatusEnum.Enable)
  101. .Select((u, a) => new
  102. {
  103. Label = a.Name,
  104. Value = u.Id,
  105. Host = u.Host.ToLower()
  106. }).ToListAsync();
  107. }
  108. /// <summary>
  109. /// 获取当前租户
  110. /// </summary>
  111. /// <returns></returns>
  112. [NonAction]
  113. public async Task<SysTenant> GetCurrentTenant()
  114. {
  115. var tenantId = long.Parse(App.User?.FindFirst(ClaimConst.TenantId)?.Value ?? "0");
  116. var host = App.HttpContext.Request.Host.Host.ToLower();
  117. return await _sysTenantRep.AsQueryable()
  118. .WhereIF(tenantId > 0, u => u.Id == tenantId)
  119. .WhereIF(tenantId <= 0, u => SqlFunc.ToLower(u.Host).Equals(host))
  120. .FirstAsync();
  121. }
  122. /// <summary>
  123. /// 获取库隔离的租户列表
  124. /// </summary>
  125. /// <returns></returns>
  126. [NonAction]
  127. public async Task<List<SysTenant>> GetTenantDbList()
  128. {
  129. return await _sysTenantRep.GetListAsync(u => u.TenantType == TenantTypeEnum.Db && u.Status == StatusEnum.Enable);
  130. }
  131. /// <summary>
  132. /// 增加租户 🔖
  133. /// </summary>
  134. /// <param name="input"></param>
  135. /// <returns></returns>
  136. [UnitOfWork]
  137. [ApiDescriptionSettings(Name = "Add"), HttpPost]
  138. [DisplayName("增加租户")]
  139. public async Task AddTenant(AddTenantInput input)
  140. {
  141. var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name);
  142. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
  143. input.Host = input.Host.ToLower();
  144. isExist = await _sysTenantRep.IsAnyAsync(u => u.Host == input.Host);
  145. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
  146. isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.AccountType == AccountTypeEnum.SuperAdmin && u.Account == input.AdminAccount);
  147. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
  148. // 从库配置判断
  149. if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections)) throw Oops.Oh(ErrorCodeEnum.D1302);
  150. switch (input.TenantType)
  151. {
  152. // Id隔离时设置与主库一致
  153. case TenantTypeEnum.Id:
  154. var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
  155. input.DbType = config.DbType;
  156. input.Connection = config.ConnectionString;
  157. break;
  158. case TenantTypeEnum.Db:
  159. if (string.IsNullOrWhiteSpace(input.Connection))
  160. throw Oops.Oh(ErrorCodeEnum.Z1004);
  161. break;
  162. default:
  163. throw Oops.Oh(ErrorCodeEnum.D3004);
  164. }
  165. var tenant = input.Adapt<TenantOutput>();
  166. // 设置logo
  167. SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
  168. await _sysTenantRep.InsertAsync(tenant);
  169. await InitNewTenant(tenant);
  170. await CacheTenant();
  171. }
  172. /// <summary>
  173. /// 设置logo
  174. /// </summary>
  175. /// <param name="tenant"></param>
  176. /// <param name="logoBase64"></param>
  177. /// <param name="logoFileName"></param>
  178. [NonAction]
  179. public void SetLogoUrl(SysTenant tenant, string logoBase64, string logoFileName)
  180. {
  181. // 旧图标文件相对路径
  182. var oldSysLogoRelativeFilePath = tenant.Logo ?? "";
  183. var oldSysLogoAbsoluteFilePath = Path.Combine(App.WebHostEnvironment.WebRootPath, oldSysLogoRelativeFilePath.TrimStart('/'));
  184. var groups = Regex.Match(logoBase64, @"data:image/(?<type>.+?);base64,(?<data>.+)").Groups;
  185. //var type = groups["type"].Value;
  186. var base64Data = groups["data"].Value;
  187. var binData = Convert.FromBase64String(base64Data);
  188. // 根据文件名取扩展名
  189. var ext = string.IsNullOrWhiteSpace(logoFileName) ? ".png" : Path.GetExtension(logoFileName);
  190. // 本地图标保存路径
  191. var fileName = $"{tenant.ViceTitle}-logo{ext}".ToLower();
  192. var path = _uploadOptions.Path.Replace("/{yyyy}/{MM}/{dd}", "");
  193. path = path.StartsWith("/") || Regex.IsMatch(path, "^[A-Z|a-z]:") ? path : Path.Combine(App.WebHostEnvironment.WebRootPath, path);
  194. var absoluteFilePath = Path.Combine(path, fileName);
  195. // 删除已存在文件
  196. if (File.Exists(oldSysLogoAbsoluteFilePath)) File.Delete(oldSysLogoAbsoluteFilePath);
  197. // 创建文件夹
  198. var absoluteFileDir = Path.GetDirectoryName(absoluteFilePath);
  199. if (!Directory.Exists(absoluteFileDir)) Directory.CreateDirectory(absoluteFileDir);
  200. // 保存图标文件
  201. File.WriteAllBytesAsync(absoluteFilePath, binData);
  202. // 保存图标配置
  203. tenant.Logo = $"/upload/{fileName}";
  204. }
  205. /// <summary>
  206. /// 设置租户状态 🔖
  207. /// </summary>
  208. /// <param name="input"></param>
  209. /// <returns></returns>
  210. [DisplayName("设置租户状态")]
  211. public async Task<int> SetStatus(TenantInput input)
  212. {
  213. var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id);
  214. if (tenant == null || tenant.ConfigId == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.Z1001);
  215. if (!Enum.IsDefined(typeof(StatusEnum), input.Status)) throw Oops.Oh(ErrorCodeEnum.D3005);
  216. tenant.Status = input.Status;
  217. return await _sysTenantRep.AsUpdateable(tenant).UpdateColumns(u => new { u.Status }).ExecuteCommandAsync();
  218. }
  219. /// <summary>
  220. /// 新增租户初始化
  221. /// </summary>
  222. /// <param name="tenant"></param>
  223. private async Task InitNewTenant(TenantOutput tenant)
  224. {
  225. var tenantId = tenant.Id;
  226. var tenantName = tenant.Name;
  227. // 初始化机构
  228. var newOrg = new SysOrg { TenantId = tenantId, Pid = 0, Name = tenantName, Code = tenantName, Remark = tenantName, };
  229. await _sysOrgRep.InsertAsync(newOrg);
  230. // 初始化职位
  231. var newPos = new SysPos { TenantId = tenantId, Name = "管理员-" + tenantName, Code = tenantName, Remark = tenantName };
  232. await _sysPosRep.InsertAsync(newPos);
  233. // 初始化租户管理员账号
  234. var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
  235. var newUser = new SysUser
  236. {
  237. TenantId = tenantId,
  238. Account = tenant.AdminAccount,
  239. Password = CryptogramUtil.Encrypt(password),
  240. NickName = "系统管理员",
  241. Email = tenant.Email,
  242. Phone = tenant.Phone,
  243. AccountType = AccountTypeEnum.SysAdmin,
  244. OrgId = newOrg.Id,
  245. PosId = newPos.Id,
  246. Birthday = DateTime.Parse("2000-01-01"),
  247. RealName = "系统管理员",
  248. Remark = "系统管理员" + tenantName,
  249. };
  250. await _sysUserRep.InsertAsync(newUser);
  251. // 关联租户组织机构和管理员用户
  252. await _sysTenantRep.UpdateAsync(u => new SysTenant { UserId = newUser.Id, OrgId = newOrg.Id }, u => u.Id == tenantId);
  253. // 默认租户管理员角色菜单集合
  254. var menuList = new List<SysMenu>();
  255. var allMenuList = new SysMenuSeedData().HasData().ToList();
  256. var titleList = new List<string> { "工作台", "系统管理", "账号管理", "角色管理", "机构管理", "职位管理", "个人中心", "通知公告", "平台管理", "菜单管理","系统配置", "关于项目" };
  257. foreach (var menu in allMenuList.Where(u => titleList.Contains(u.Title)))
  258. {
  259. menuList.Add(menu);
  260. if (menu.Type == MenuTypeEnum.Menu) menuList.AddRange(allMenuList.Where(u => u.Pid == menu.Id));
  261. }
  262. await GrantMenu(new TenantMenuInput { Id = tenantId, MenuIdList = menuList.Select(u => u.Id).ToList() });
  263. }
  264. /// <summary>
  265. /// 删除租户 🔖
  266. /// </summary>
  267. /// <param name="input"></param>
  268. /// <returns></returns>
  269. [UnitOfWork]
  270. [ApiDescriptionSettings(Name = "Delete"), HttpPost]
  271. [DisplayName("删除租户")]
  272. public async Task DeleteTenant(DeleteTenantInput input)
  273. {
  274. // 禁止删除默认租户
  275. if (input.Id.ToString() == SqlSugarConst.MainConfigId) throw Oops.Oh(ErrorCodeEnum.D1023);
  276. // 若账号为开放接口绑定租户则禁止删除
  277. var isOpenAccessTenant = await _sysTenantRep.ChangeRepository<SqlSugarRepository<SysOpenAccess>>().IsAnyAsync(u => u.BindTenantId == input.Id);
  278. if (isOpenAccessTenant) throw Oops.Oh(ErrorCodeEnum.D1031);
  279. await _sysTenantRep.DeleteAsync(u => u.Id == input.Id);
  280. await CacheTenant(input.Id);
  281. // 删除与租户相关的表数据
  282. await _sysTenantMenuRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  283. var menuIds = await _sysTenantRep.Context.Queryable<SysMenu>().Select(u => u.Id).ToListAsync();
  284. await _sysTenantRep.Context.Deleteable<SysRoleMenu>().Where(u => menuIds.Contains(u.Id)).ExecuteCommandAsync();
  285. await _sysTenantRep.Context.Deleteable<SysUserMenu>().Where(u => menuIds.Contains(u.Id)).ExecuteCommandAsync();
  286. await _sysTenantRep.Context.Deleteable<SysMenu>().Where(u => menuIds.Contains(u.Id)).ExecuteCommandAsync();
  287. var users = await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.Id).ToListAsync();
  288. var userIds = users.Select(u => u.Id).ToList();
  289. await _sysUserRep.AsDeleteable().Where(u => userIds.Contains(u.Id)).ExecuteCommandAsync();
  290. await _userRoleRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  291. await _sysUserExtOrgRep.AsDeleteable().Where(u => userIds.Contains(u.UserId)).ExecuteCommandAsync();
  292. await _sysRoleRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  293. var roleIds = await _sysRoleRep.AsQueryable().ClearFilter()
  294. .Where(u => u.TenantId == input.Id).Select(u => u.Id).ToListAsync();
  295. await _sysRoleMenuRep.AsDeleteable().Where(u => roleIds.Contains(u.RoleId)).ExecuteCommandAsync();
  296. await _sysOrgRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  297. await _sysPosRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  298. }
  299. /// <summary>
  300. /// 更新租户 🔖
  301. /// </summary>
  302. /// <param name="input"></param>
  303. /// <returns></returns>
  304. [ApiDescriptionSettings(Name = "Update"), HttpPost]
  305. [DisplayName("更新租户")]
  306. public async Task UpdateTenant(UpdateTenantInput input)
  307. {
  308. var isExist = await _sysOrgRep.IsAnyAsync(u => u.Name == input.Name && u.Id != input.OrgId);
  309. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1300);
  310. input.Host = input.Host.ToLower();
  311. isExist = await _sysTenantRep.IsAnyAsync(u => u.Host == input.Host && u.Id != input.Id);
  312. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1303);
  313. isExist = await _sysUserRep.AsQueryable().ClearFilter().AnyAsync(u => u.AccountType == AccountTypeEnum.SuperAdmin && u.Account == input.AdminAccount && u.Id != input.UserId);
  314. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1301);
  315. // Id隔离时设置与主库一致
  316. switch (input.TenantType)
  317. {
  318. case TenantTypeEnum.Id:
  319. var config = _sysTenantRep.AsSugarClient().CurrentConnectionConfig;
  320. input.DbType = config.DbType;
  321. input.Connection = config.ConnectionString;
  322. break;
  323. case TenantTypeEnum.Db:
  324. if (string.IsNullOrWhiteSpace(input.Connection))
  325. throw Oops.Oh(ErrorCodeEnum.Z1004);
  326. break;
  327. default:
  328. throw Oops.Oh(ErrorCodeEnum.D3004);
  329. }
  330. // 从库配置判断
  331. if (!string.IsNullOrWhiteSpace(input.SlaveConnections) && !JSON.IsValid(input.SlaveConnections))
  332. throw Oops.Oh(ErrorCodeEnum.D1302);
  333. // 设置logo
  334. var tenant = input.Adapt<SysTenant>();
  335. if (!string.IsNullOrWhiteSpace(input.LogoBase64)) SetLogoUrl(tenant, input.LogoBase64, input.LogoFileName);
  336. // 更新租户信息
  337. await _sysTenantRep.AsUpdateable(tenant).IgnoreColumns(true).ExecuteCommandAsync();
  338. // 更新系统机构
  339. await _sysOrgRep.UpdateAsync(u => new SysOrg() { Name = input.Name }, u => u.Id == input.OrgId);
  340. // 更新系统用户
  341. await _sysUserRep.UpdateAsync(u => new SysUser() { Account = input.AdminAccount, Phone = input.Phone, Email = input.Email }, u => u.Id == input.UserId);
  342. await CacheTenant(input.Id);
  343. }
  344. /// <summary>
  345. /// 授权租户菜单 🔖
  346. /// </summary>
  347. /// <param name="input"></param>
  348. /// <returns></returns>
  349. [UnitOfWork]
  350. [DisplayName("授权租户菜单")]
  351. public async Task GrantMenu(TenantMenuInput input)
  352. {
  353. // 获取需要授权的菜单列表
  354. var menuList = await _sysTenantRep.Context.Queryable<SysMenu>().ClearFilter()
  355. .Where(u => input.MenuIdList.Contains(u.Id))
  356. .InnerJoin<SysTenantMenu>((u, t) => t.TenantId == input.Id && u.Id == t.MenuId)
  357. .ToListAsync();
  358. // 检查是否存在重复菜单
  359. if (menuList.Where(u => u.Type != MenuTypeEnum.Btn).GroupBy(u => new { u.Pid, u.Title }).Any(u => u.Count() > 1) ||
  360. menuList.Where(u => u.Type == MenuTypeEnum.Btn).GroupBy(u => u.Permission).Any(u => u.Count() > 1))
  361. throw Oops.Oh(ErrorCodeEnum.D1304);
  362. // 检查路由是否重复
  363. if (menuList.Where(u => !string.IsNullOrWhiteSpace(u.Name)).GroupBy(u => u.Name).Any(u => u.Count() > 1))
  364. throw Oops.Oh(ErrorCodeEnum.D4009);
  365. await _sysTenantMenuRep.AsDeleteable().Where(u => u.TenantId == input.Id).ExecuteCommandAsync();
  366. var sysTenantMenuList = input.MenuIdList.Select(menuId => new SysTenantMenu { TenantId = input.Id, MenuId = menuId }).ToList();
  367. await _sysTenantMenuRep.InsertRangeAsync(sysTenantMenuList);
  368. }
  369. /// <summary>
  370. /// 获取租户菜单Id集合 🔖
  371. /// </summary>
  372. /// <param name="input"></param>
  373. /// <returns></returns>
  374. [DisplayName("获取租户菜单Id集合")]
  375. public async Task<List<long>> GetTenantMenuList([FromQuery] BaseIdInput input)
  376. {
  377. return await _sysTenantMenuRep.AsQueryable().Where(u => u.TenantId == input.Id).Select(u => u.MenuId).ToListAsync();
  378. }
  379. /// <summary>
  380. /// 重置租户管理员密码 🔖
  381. /// </summary>
  382. /// <param name="input"></param>
  383. /// <returns></returns>
  384. [DisplayName("重置租户管理员密码")]
  385. public async Task<string> ResetPwd(TenantUserInput input)
  386. {
  387. var password = await _sysConfigService.GetConfigValue<string>(ConfigConst.SysPassword);
  388. var encryptPassword = CryptogramUtil.Encrypt(password);
  389. await _sysUserRep.UpdateAsync(u => new SysUser { Password = encryptPassword }, u => u.Id == input.UserId);
  390. return password;
  391. }
  392. /// <summary>
  393. /// 切换租户 🔖
  394. /// </summary>
  395. /// <param name="input"></param>
  396. /// <returns></returns>
  397. [UnitOfWork]
  398. [DisplayName("切换租户")]
  399. public async Task<LoginOutput> ChangeTenant(BaseIdInput input)
  400. {
  401. var userManager = App.GetService<UserManager>();
  402. _ = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  403. var user = await _sysUserRep.GetFirstAsync(u => u.Id == userManager.UserId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  404. user.TenantId = input.Id;
  405. return await GetAccessTokenInNotSingleLogin(user);
  406. }
  407. /// <summary>
  408. /// 进入租管端 🔖
  409. /// </summary>
  410. /// <param name="input"></param>
  411. /// <returns></returns>
  412. [DisplayName("进入租管端")]
  413. public async Task<LoginOutput> GoTenant(BaseIdInput input)
  414. {
  415. var tenant = await _sysTenantRep.GetFirstAsync(u => u.Id == input.Id) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  416. var user = await _sysUserRep.GetFirstAsync(u => u.Id == tenant.UserId) ?? throw Oops.Oh(ErrorCodeEnum.D1002);
  417. return await GetAccessTokenInNotSingleLogin(user);
  418. }
  419. /// <summary>
  420. /// 在非单用户登录模式下获取登录令牌
  421. /// </summary>
  422. /// <param name="user"></param>
  423. /// <returns></returns>
  424. [NonAction]
  425. public async Task<LoginOutput> GetAccessTokenInNotSingleLogin(SysUser user)
  426. {
  427. // 使用非单用户模式登录
  428. var singleLogin = _sysCacheService.Get<bool>($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}");
  429. try
  430. {
  431. return await App.GetService<SysAuthService>().CreateToken(user);
  432. }
  433. finally
  434. {
  435. // 恢复单用户登录参数
  436. if (singleLogin) _sysCacheService.Set($"{CacheConst.KeyConfig}{ConfigConst.SysSingleLogin}", true);
  437. }
  438. }
  439. /// <summary>
  440. /// 缓存所有租户
  441. /// </summary>
  442. /// <param name="tenantId"></param>
  443. /// <returns></returns>
  444. [NonAction]
  445. public async Task CacheTenant(long tenantId = 0)
  446. {
  447. // 移除 ISqlSugarClient 中的库连接并排除默认主库
  448. if (tenantId > 0 && tenantId.ToString() != SqlSugarConst.MainConfigId)
  449. _sysTenantRep.AsTenant().RemoveConnection(tenantId);
  450. var tenantList = await _sysTenantRep.GetListAsync();
  451. // 对租户库连接进行SM2加密
  452. foreach (var tenant in tenantList.Where(tenant => !string.IsNullOrWhiteSpace(tenant.Connection)))
  453. tenant.Connection = CryptogramUtil.SM2Encrypt(tenant.Connection);
  454. _sysCacheService.Set(CacheConst.KeyTenant, tenantList);
  455. }
  456. /// <summary>
  457. /// 创建租户数据库 🔖
  458. /// </summary>
  459. /// <param name="input"></param>
  460. /// <returns></returns>
  461. [ApiDescriptionSettings(Name = "CreateDb"), HttpPost]
  462. [DisplayName("创建租户数据库")]
  463. public async Task CreateDb(TenantInput input)
  464. {
  465. var tenant = await _sysTenantRep.GetSingleAsync(u => u.Id == input.Id);
  466. if (tenant == null) return;
  467. if (tenant.DbType == SqlSugar.DbType.Oracle)
  468. throw Oops.Oh(ErrorCodeEnum.Z1002);
  469. if (string.IsNullOrWhiteSpace(tenant.Connection) || tenant.Connection.Length < 10)
  470. throw Oops.Oh(ErrorCodeEnum.Z1004);
  471. // 默认数据库配置
  472. var defaultConfig = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault();
  473. var config = new DbConnectionConfig
  474. {
  475. ConfigId = tenant.Id.ToString(),
  476. DbType = tenant.DbType,
  477. ConnectionString = tenant.Connection,
  478. DbSettings = new DbSettings()
  479. {
  480. EnableInitDb = true,
  481. EnableDiffLog = false,
  482. EnableUnderLine = defaultConfig!.DbSettings.EnableUnderLine,
  483. }
  484. };
  485. SqlSugarSetup.InitTenantDatabase(App.GetRequiredService<ISqlSugarClient>().AsTenant(), config);
  486. }
  487. /// <summary>
  488. /// 获取租户下的用户列表 🔖
  489. /// </summary>
  490. /// <param name="input"></param>
  491. /// <returns></returns>
  492. [DisplayName("获取租户下的用户列表")]
  493. public async Task<List<SysUser>> UserList(TenantIdInput input)
  494. {
  495. return await _sysUserRep.AsQueryable().ClearFilter().Where(u => u.TenantId == input.TenantId).ToListAsync();
  496. }
  497. /// <summary>
  498. /// 获取租户数据库连接
  499. /// </summary>
  500. /// <returns></returns>
  501. [NonAction]
  502. public SqlSugarScopeProvider GetTenantDbConnectionScope(long tenantId)
  503. {
  504. var iTenant = _sysTenantRep.AsTenant();
  505. // 若已存在租户库连接,则直接返回
  506. if (iTenant.IsAnyConnection(tenantId.ToString())) return iTenant.GetConnectionScope(tenantId.ToString());
  507. lock (iTenant)
  508. {
  509. // 从缓存里面获取租户信息
  510. var tenant = _sysCacheService.Get<List<SysTenant>>(CacheConst.KeyTenant)?.FirstOrDefault(u => u.Id == tenantId);
  511. if (tenant == null || tenant.TenantType == TenantTypeEnum.Id) return null;
  512. // 获取默认库连接配置
  513. var dbOptions = App.GetOptions<DbConnectionOptions>();
  514. var mainConnConfig = dbOptions.ConnectionConfigs.First(u => u.ConfigId.ToString() == SqlSugarConst.MainConfigId);
  515. // 设置租户库连接配置
  516. var tenantConnConfig = new DbConnectionConfig
  517. {
  518. ConfigId = tenant.Id.ToString(),
  519. DbType = tenant.DbType,
  520. TenantType = tenant.TenantType,
  521. IsAutoCloseConnection = true,
  522. ConnectionString = CryptogramUtil.SM2Decrypt(tenant.Connection), // 对租户库连接进行SM2解密
  523. DbSettings = new DbSettings()
  524. {
  525. EnableUnderLine = mainConnConfig.DbSettings.EnableUnderLine,
  526. },
  527. SlaveConnectionConfigs = JSON.IsValid(tenant.SlaveConnections) ? JSON.Deserialize<List<SlaveConnectionConfig>>(tenant.SlaveConnections) : null // 从库连接配置
  528. };
  529. iTenant.AddConnection(tenantConnConfig);
  530. var sqlSugarScopeProvider = iTenant.GetConnectionScope(tenantId.ToString());
  531. SqlSugarSetup.SetDbConfig(tenantConnConfig);
  532. SqlSugarSetup.SetDbAop(sqlSugarScopeProvider, dbOptions.EnableConsoleSql);
  533. return sqlSugarScopeProvider;
  534. }
  535. }
  536. }