Explorar o código

修复登录时隐藏租户,TenantId为0时,提示注册功能未开启禁止注册
登录时隐藏租户,查找对应租户信息

Signed-off-by: 高级CV工程师 <2535688890@qq.com>

高级CV工程师 hai 7 meses
pai
achega
438509ca87
Modificáronse 1 ficheiros con 6 adicións e 0 borrados
  1. 6 0
      Admin.NET/Admin.NET.Core/Service/Auth/SysAuthService.cs

+ 6 - 0
Admin.NET/Admin.NET.Core/Service/Auth/SysAuthService.cs

@@ -23,6 +23,7 @@ public class SysAuthService : IDynamicApiController, ITransient
     private readonly SysOnlineUserService _sysOnlineUserService;
     private readonly SysOnlineUserService _sysOnlineUserService;
     private readonly SysConfigService _sysConfigService;
     private readonly SysConfigService _sysConfigService;
     private readonly SysUserService _sysUserService;
     private readonly SysUserService _sysUserService;
+    private readonly SysTenantService _sysTenantService;
     private readonly SysSmsService _sysSmsService;
     private readonly SysSmsService _sysSmsService;
     private readonly SysLdapService _sysLdapService;
     private readonly SysLdapService _sysLdapService;
     private readonly ICaptcha _captcha;
     private readonly ICaptcha _captcha;
@@ -40,6 +41,7 @@ public class SysAuthService : IDynamicApiController, ITransient
         SysCacheService sysCacheService,
         SysCacheService sysCacheService,
         SysMenuService sysMenuService,
         SysMenuService sysMenuService,
         SysUserService sysUserService,
         SysUserService sysUserService,
+        SysTenantService sysTenantService,
         UserManager userManager,
         UserManager userManager,
         ICaptcha captcha)
         ICaptcha captcha)
     {
     {
@@ -49,6 +51,7 @@ public class SysAuthService : IDynamicApiController, ITransient
         _sysSmsService = sysSmsService;
         _sysSmsService = sysSmsService;
         _eventPublisher = eventPublisher;
         _eventPublisher = eventPublisher;
         _sysUserService = sysUserService;
         _sysUserService = sysUserService;
+        _sysTenantService = sysTenantService;
         _sysMenuService = sysMenuService;
         _sysMenuService = sysMenuService;
         _sysCacheService = sysCacheService;
         _sysCacheService = sysCacheService;
         _sysConfigService = sysConfigService;
         _sysConfigService = sysConfigService;
@@ -420,6 +423,9 @@ public class SysAuthService : IDynamicApiController, ITransient
         if (!_captcha.Validate(input.CodeId.ToString(), input.Code)) throw Oops.Oh(ErrorCodeEnum.D0008);
         if (!_captcha.Validate(input.CodeId.ToString(), input.Code)) throw Oops.Oh(ErrorCodeEnum.D0008);
         _captcha.Generate(input.CodeId.ToString());
         _captcha.Generate(input.CodeId.ToString());
 
 
+        // 登录时隐藏租户,查找对应租户信息
+        input.TenantId = input.TenantId <= 0? (await _sysTenantService.GetCurrentTenantSysInfo()).Id : input.TenantId;
+
         // 判断租户是否有效且启用注册功能
         // 判断租户是否有效且启用注册功能
         var tenant = await _sysUserRep.Context.Queryable<SysTenant>().FirstAsync(u => u.Id == input.TenantId && u.Status == StatusEnum.Enable);
         var tenant = await _sysUserRep.Context.Queryable<SysTenant>().FirstAsync(u => u.Id == input.TenantId && u.Status == StatusEnum.Enable);
         if (tenant?.EnableReg != YesNoEnum.Y) throw Oops.Oh(ErrorCodeEnum.D1034);
         if (tenant?.EnableReg != YesNoEnum.Y) throw Oops.Oh(ErrorCodeEnum.D1034);