From 0bae3870045c3288e5b2e3038380240ea54a5cf6 Mon Sep 17 00:00:00 2001
From: wangyalei <wangyl@wealthgrow.cn>
Date: Fri, 15 Oct 2021 18:59:12 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=B3=E6=B3=A8=E5=88=97=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../community/util/HttpServletHelper.java     | 44 ++++++++++---------
 1 file changed, 24 insertions(+), 20 deletions(-)

diff --git a/community-service/src/main/java/com/tanpu/community/util/HttpServletHelper.java b/community-service/src/main/java/com/tanpu/community/util/HttpServletHelper.java
index 366daa6..7aefc39 100644
--- a/community-service/src/main/java/com/tanpu/community/util/HttpServletHelper.java
+++ b/community-service/src/main/java/com/tanpu/community/util/HttpServletHelper.java
@@ -1,5 +1,6 @@
 package com.tanpu.community.util;
 
+import com.tanpu.common.auth.UserHolder;
 import com.tanpu.common.redis.RedisHelper;
 import com.tanpu.community.api.constants.RedisKeyConstant;
 import lombok.extern.slf4j.Slf4j;
@@ -26,28 +27,31 @@ import java.util.Optional;
 public class HttpServletHelper {
     @Resource
     private RedisHelper redisHelper;
+    @Resource
+    private UserHolder userHolder;
 
     public String getCurrentUserId() {
-        String userId = null;
-        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
-        String clientEnv = request.getHeader(RedisKeyConstant.ENV);
-        if (isAndroid(clientEnv) || isIOS(clientEnv) || isPc(clientEnv) || isMiniProgram(clientEnv)) {
-            // android，ios，pc，小程序用token
-            String token = request.getHeader(RedisKeyConstant.TAMP_TOKEN);
-            // redis获取的value,都带有 ""，这里去掉
-            userId = StringUtils.remove(redisHelper.get(RedisKeyConstant.REDIS_PREFIX_TOKEN + token), "\"");
-        } else {
-            // 用cookie
-            Cookie[] cookies = request.getCookies();
-            if (cookies != null) {
-                Optional<Cookie> first = Arrays.stream(request.getCookies())
-                        .filter(cookie -> StringUtils.equalsIgnoreCase(RedisKeyConstant.SESSION_COOKIE_NAME, cookie.getName()))
-                        .findFirst();
-                if (first.isPresent()) {
-                    userId = first.get().getValue();
-                }
-            }
-        }
+//        String userId = null;
+//        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
+//        String clientEnv = request.getHeader(RedisKeyConstant.ENV);
+//        if (isAndroid(clientEnv) || isIOS(clientEnv) || isPc(clientEnv) || isMiniProgram(clientEnv)) {
+//            // android，ios，pc，小程序用token
+//            String token = request.getHeader(RedisKeyConstant.TAMP_TOKEN);
+//            // redis获取的value,都带有 ""，这里去掉
+//            userId = StringUtils.remove(redisHelper.get(RedisKeyConstant.REDIS_PREFIX_TOKEN + token), "\"");
+//        } else {
+//            // 用cookie
+//            Cookie[] cookies = request.getCookies();
+//            if (cookies != null) {
+//                Optional<Cookie> first = Arrays.stream(request.getCookies())
+//                        .filter(cookie -> StringUtils.equalsIgnoreCase(RedisKeyConstant.SESSION_COOKIE_NAME, cookie.getName()))
+//                        .findFirst();
+//                if (first.isPresent()) {
+//                    userId = first.get().getValue();
+//                }
+//            }
+//        }
+        String userId = userHolder.getUserId();
         log.debug("当前登录用户Id: {}", userId);
         return userId;
     }
-- 
2.18.1