update 优化 三方登录不同域名获取不到租户id问题

ruoyi-auth/src/main/java/org/dromara/auth/controller/TokenController.java

@@ -91,9 +91,12 @@ public class TokenController {
             return R.fail(MessageUtils.message("auth.grant.type.blocked"));
         }
         // 校验租户
-        sysLoginService.checkTenant(loginBody.getTenantId());
+        RemoteTenantVo tenant = sysLoginService.checkTenant(loginBody.getTenantId());
         // 登录
         LoginVo loginVo = IAuthStrategy.login(body, clientVo, grantType);
+        if (StringUtils.isNotBlank(tenant.getTenantId())) {
+            loginVo.setDomain(tenant.getDomain());
+        }
 
         Long userId = LoginHelper.getUserId();
         scheduledExecutorService.schedule(() -> {
@@ -126,7 +129,7 @@ public class TokenController {
      * @return 结果
      */
     @PostMapping("/social/callback")
-    public R<Void> socialCallback(@RequestBody SocialLoginBody loginBody) {
+    public R<LoginVo> socialCallback(@RequestBody SocialLoginBody loginBody) {
         // 获取第三方登录信息
         AuthResponse<AuthUser> response = SocialUtils.loginAuth(
             loginBody.getSource(), loginBody.getSocialCode(),
@@ -137,7 +140,13 @@ public class TokenController {
             return R.fail(response.getMsg());
         }
         sysLoginService.socialRegister(authUserData);
-        return R.ok();
+
+        LoginVo loginVo = new LoginVo();
+        RemoteTenantVo tenant = sysLoginService.checkTenant(LoginHelper.getTenantId());
+        if (StringUtils.isNotBlank(tenant.getTenantId())) {
+            loginVo.setDomain(tenant.getDomain());
+        }
+        return R.ok(loginVo);
     }
 
 

ruoyi-auth/src/main/java/org/dromara/auth/domain/vo/LoginVo.java

@@ -51,4 +51,9 @@ public class LoginVo {
      */
     private String openid;
 
+    /**
+     * 域名
+     */
+    private String domain;
+
 }

ruoyi-auth/src/main/java/org/dromara/auth/service/SysLoginService.java

@@ -242,17 +242,17 @@ public class SysLoginService {
      *
      * @param tenantId 租户ID
      */
-    public void checkTenant(String tenantId) {
+    public RemoteTenantVo checkTenant(String tenantId) {
         if (!TenantHelper.isEnable()) {
-            return;
-        }
-        if (TenantConstants.DEFAULT_TENANT_ID.equals(tenantId)) {
-            return;
+            return null;
         }
         if (StringUtils.isBlank(tenantId)) {
             throw new TenantException("tenant.number.not.blank");
         }
         RemoteTenantVo tenant = remoteTenantService.queryByTenantId(tenantId);
+        if (TenantConstants.DEFAULT_TENANT_ID.equals(tenantId)) {
+            return tenant;
+        }
         if (ObjectUtil.isNull(tenant)) {
             log.info("登录租户：{} 不存在.", tenantId);
             throw new TenantException("tenant.not.exists");
@@ -264,5 +264,6 @@ public class SysLoginService {
             log.info("登录租户：{} 已超过有效期.", tenantId);
             throw new TenantException("tenant.expired");
         }
+        return tenant;
     }
 }