From 1400ca38b688f54678aa87b972198c7b78fbb273 Mon Sep 17 00:00:00 2001
From: Penny <2500338766@qq.com>
Date: Wed, 1 Mar 2023 20:28:56 +0800
Subject: [PATCH] =?UTF-8?q?feature-img-1.0:=E6=B3=A8=E9=94=80=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../auth/controller/CreatorController.java    |  2 +-
 .../auth/controller/FhUserController.java     |  2 +-
 .../auth/controller/TiktokUserController.java |  2 +-
 .../auth/controller/WxUserController.java     |  2 +-
 .../common/security/utils/SecurityUtils.java  | 42 ++++++++++++++++++-
 5 files changed, 45 insertions(+), 5 deletions(-)

diff --git a/bnyer-auth/src/main/java/com/bnyer/auth/controller/CreatorController.java b/bnyer-auth/src/main/java/com/bnyer/auth/controller/CreatorController.java
index 4df68a1..0ae54c7 100644
--- a/bnyer-auth/src/main/java/com/bnyer/auth/controller/CreatorController.java
+++ b/bnyer-auth/src/main/java/com/bnyer/auth/controller/CreatorController.java
@@ -44,7 +44,7 @@ public class CreatorController
     @PostMapping("/creatorLogout")
     public R<?> logout(HttpServletRequest request)
     {
-        String token = SecurityUtils.getToken(request);
+        String token = SecurityUtils.getCreatorToken(request);
         creatorTokenService.delLoginCreator(token);
         return R.ok();
     }
diff --git a/bnyer-auth/src/main/java/com/bnyer/auth/controller/FhUserController.java b/bnyer-auth/src/main/java/com/bnyer/auth/controller/FhUserController.java
index fda1d5b..bee0b77 100644
--- a/bnyer-auth/src/main/java/com/bnyer/auth/controller/FhUserController.java
+++ b/bnyer-auth/src/main/java/com/bnyer/auth/controller/FhUserController.java
@@ -41,7 +41,7 @@ public class FhUserController
     @ApiOperation("快手小程序用户注销")
     public R<?> logout(HttpServletRequest request)
     {
-        String token = SecurityUtils.getToken(request);
+        String token = SecurityUtils.getFhToken(request);
         //删除快手用户缓存
         fUserTokenService.delLoginFhUser(token);
         return R.ok();
diff --git a/bnyer-auth/src/main/java/com/bnyer/auth/controller/TiktokUserController.java b/bnyer-auth/src/main/java/com/bnyer/auth/controller/TiktokUserController.java
index 07222fc..5e05901 100644
--- a/bnyer-auth/src/main/java/com/bnyer/auth/controller/TiktokUserController.java
+++ b/bnyer-auth/src/main/java/com/bnyer/auth/controller/TiktokUserController.java
@@ -41,7 +41,7 @@ public class TiktokUserController
     @ApiOperation("抖音小程序用户注销")
     public R<?> logout(HttpServletRequest request)
     {
-        String token = SecurityUtils.getToken(request);
+        String token = SecurityUtils.getTiktokToken(request);
         //删除快手用户缓存
         tiktokUserTokenService.delLoginFhUser(token);
         return R.ok();
diff --git a/bnyer-auth/src/main/java/com/bnyer/auth/controller/WxUserController.java b/bnyer-auth/src/main/java/com/bnyer/auth/controller/WxUserController.java
index a992672..96dfecb 100644
--- a/bnyer-auth/src/main/java/com/bnyer/auth/controller/WxUserController.java
+++ b/bnyer-auth/src/main/java/com/bnyer/auth/controller/WxUserController.java
@@ -41,7 +41,7 @@ public class WxUserController
     @ApiOperation("微信小程序用户注销")
     public R<?> logout(HttpServletRequest request)
     {
-        String token = SecurityUtils.getToken(request);
+        String token = SecurityUtils.getWechatToken(request);
         //删除微信用户缓存
         WxUserTokenService.delLoginFhUser(token);
         return R.ok();
diff --git a/bnyer-common/bnyer-common-security/src/main/java/com/bnyer/common/security/utils/SecurityUtils.java b/bnyer-common/bnyer-common-security/src/main/java/com/bnyer/common/security/utils/SecurityUtils.java
index 52ce420..59d0686 100644
--- a/bnyer-common/bnyer-common-security/src/main/java/com/bnyer/common/security/utils/SecurityUtils.java
+++ b/bnyer-common/bnyer-common-security/src/main/java/com/bnyer/common/security/utils/SecurityUtils.java
@@ -58,7 +58,7 @@ public class SecurityUtils
     }
 
     /**
-     * 根据request获取请求token
+     * 根据request获取PC端请求token
      */
     public static String getToken(HttpServletRequest request)
     {
@@ -67,6 +67,46 @@ public class SecurityUtils
         return replaceTokenPrefix(token);
     }
 
+    /**
+     * 根据request获取请求艺术家请求token
+     */
+    public static String getCreatorToken(HttpServletRequest request)
+    {
+        // 从header获取token标识
+        String token = request.getHeader(TokenConstants.CREATOR_APP_AUTH);
+        return replaceTokenPrefix(token);
+    }
+
+    /**
+     * 根据request获取微信小程序用户请求token
+     */
+    public static String getWechatToken(HttpServletRequest request)
+    {
+        // 从header获取token标识
+        String token = request.getHeader(TokenConstants.WECHAT_APP_AUTH);
+        return replaceTokenPrefix(token);
+    }
+
+    /**
+     * 根据request获取抖音小程序用户请求token
+     */
+    public static String getTiktokToken(HttpServletRequest request)
+    {
+        // 从header获取token标识
+        String token = request.getHeader(TokenConstants.TIKTOK_APP_AUTH);
+        return replaceTokenPrefix(token);
+    }
+
+    /**
+     * 根据request获取快手小程序用户请求token
+     */
+    public static String getFhToken(HttpServletRequest request)
+    {
+        // 从header获取token标识
+        String token = request.getHeader(TokenConstants.FASTHAND_APP_AUTH);
+        return replaceTokenPrefix(token);
+    }
+
     /**
      * 裁剪token前缀
      */