Commit 0bae3870 authored by 王亚雷's avatar 王亚雷

关注列表

parent 6e5af22e
package com.tanpu.community.util; package com.tanpu.community.util;
import com.tanpu.common.auth.UserHolder;
import com.tanpu.common.redis.RedisHelper; import com.tanpu.common.redis.RedisHelper;
import com.tanpu.community.api.constants.RedisKeyConstant; import com.tanpu.community.api.constants.RedisKeyConstant;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
...@@ -26,28 +27,31 @@ import java.util.Optional; ...@@ -26,28 +27,31 @@ import java.util.Optional;
public class HttpServletHelper { public class HttpServletHelper {
@Resource @Resource
private RedisHelper redisHelper; private RedisHelper redisHelper;
@Resource
private UserHolder userHolder;
public String getCurrentUserId() { public String getCurrentUserId() {
String userId = null; // String userId = null;
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest(); // HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
String clientEnv = request.getHeader(RedisKeyConstant.ENV); // String clientEnv = request.getHeader(RedisKeyConstant.ENV);
if (isAndroid(clientEnv) || isIOS(clientEnv) || isPc(clientEnv) || isMiniProgram(clientEnv)) { // if (isAndroid(clientEnv) || isIOS(clientEnv) || isPc(clientEnv) || isMiniProgram(clientEnv)) {
// android,ios,pc,小程序用token // // android,ios,pc,小程序用token
String token = request.getHeader(RedisKeyConstant.TAMP_TOKEN); // String token = request.getHeader(RedisKeyConstant.TAMP_TOKEN);
// redis获取的value,都带有 "",这里去掉 // // redis获取的value,都带有 "",这里去掉
userId = StringUtils.remove(redisHelper.get(RedisKeyConstant.REDIS_PREFIX_TOKEN + token), "\""); // userId = StringUtils.remove(redisHelper.get(RedisKeyConstant.REDIS_PREFIX_TOKEN + token), "\"");
} else { // } else {
// 用cookie // // 用cookie
Cookie[] cookies = request.getCookies(); // Cookie[] cookies = request.getCookies();
if (cookies != null) { // if (cookies != null) {
Optional<Cookie> first = Arrays.stream(request.getCookies()) // Optional<Cookie> first = Arrays.stream(request.getCookies())
.filter(cookie -> StringUtils.equalsIgnoreCase(RedisKeyConstant.SESSION_COOKIE_NAME, cookie.getName())) // .filter(cookie -> StringUtils.equalsIgnoreCase(RedisKeyConstant.SESSION_COOKIE_NAME, cookie.getName()))
.findFirst(); // .findFirst();
if (first.isPresent()) { // if (first.isPresent()) {
userId = first.get().getValue(); // userId = first.get().getValue();
} // }
} // }
} // }
String userId = userHolder.getUserId();
log.debug("当前登录用户Id: {}", userId); log.debug("当前登录用户Id: {}", userId);
return userId; return userId;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment