From a5589e9d8f23f4f415d1e7787798942da5f37ee4 Mon Sep 17 00:00:00 2001
From: WenC <450292408@qq.com>
Date: Thu, 25 Apr 2024 15:38:12 +0800
Subject: [PATCH] =?UTF-8?q?=E9=80=80=E5=87=BA=E9=87=8D=E5=AE=9A=E5=90=91?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../base/controller/route/OAuthRouteController.java    | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/login-base/src/main/java/ink/wgink/login/base/controller/route/OAuthRouteController.java b/login-base/src/main/java/ink/wgink/login/base/controller/route/OAuthRouteController.java
index 103dc512..8fa09c27 100644
--- a/login-base/src/main/java/ink/wgink/login/base/controller/route/OAuthRouteController.java
+++ b/login-base/src/main/java/ink/wgink/login/base/controller/route/OAuthRouteController.java
@@ -66,8 +66,14 @@ public class OAuthRouteController {
     public void logout(HttpServletRequest request, HttpServletResponse response) {
         new SecurityContextLogoutHandler().logout(request, null, null);
         try {
-            String referer = request.getHeader("referer");
-            response.sendRedirect(referer.replace("/default-main", ""));
+            String redirectUri = request.getParameter("redirect_uri");
+            if(StringUtils.isNotBlank(redirectUri)) {
+                response.sendRedirect(redirectUri);
+            }
+//            String referer = request.getHeader("referer");
+//            if(StringUtils.isNotBlank(referer)) {
+//                response.sendRedirect(referer.replace("/default-main", ""));
+//            }
         } catch (IOException e) {
             e.printStackTrace();
         }