liuchang/mall-ebtp-cloud
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加cookie判断

Browse Source
This commit is contained in:
付庆吉
2021-03-24 14:31:40 +08:00
parent f4565f49f8
commit 4c583f5ad2
1 changed files with 34 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
uboot-common/src/main/java/com/chinaunicom/mall/ebtp/common/config/FeignConfig.java
View File

@ -1,17 +1,18 @@
package com.chinaunicom.mall.ebtp.common.config; package com.chinaunicom.mall.ebtp.common.config;
import static com.chinaunicom.mall.ebtp.cloud.security.starter.common.Constants.AUTHORIZATION_HEADER; import feign.RequestInterceptor;
import static com.chinaunicom.mall.ebtp.cloud.security.starter.common.Constants.TOKEN_PREFIX; import feign.RequestTemplate;
import static com.chinaunicom.mall.ebtp.cloud.security.starter.common.Constants.CURRENT_ROLE_CODE;
import org.apache.commons.lang3.RegExUtils; import org.apache.commons.lang3.RegExUtils;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes; import org.springframework.web.context.request.ServletRequestAttributes;
import feign.RequestInterceptor; import javax.servlet.http.Cookie;
import feign.RequestTemplate; import java.util.Arrays;
import java.util.Optional;
import static com.chinaunicom.mall.ebtp.cloud.security.starter.common.Constants.*;
/** /**
* 通过拦截器来为header注入token * 通过拦截器来为header注入token
@ -19,29 +20,36 @@ import feign.RequestTemplate;
@Configuration @Configuration
public class FeignConfig implements RequestInterceptor { public class FeignConfig implements RequestInterceptor {
/** /**
* @param template * @param template
*/ */
@Override @Override
public void apply(RequestTemplate template) { public void apply(RequestTemplate template) {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
if (null != attributes) { if (null != attributes) {
final String header = attributes.getRequest().getHeader(AUTHORIZATION_HEADER);// 提取request头信息 final String header = attributes.getRequest().getHeader(AUTHORIZATION_HEADER);// 提取request头信息
// 检查请求头是否包含 Bearer 前缀 // 检查请求头是否包含 Bearer 前缀
if (StringUtils.startsWith(header, TOKEN_PREFIX)) { if (StringUtils.startsWith(header, TOKEN_PREFIX)) {
String authToken = RegExUtils.replaceAll(header, TOKEN_PREFIX, "");// 提取 token 信息 String authToken = RegExUtils.replaceAll(header, TOKEN_PREFIX, "");// 提取 token 信息
template.header(AUTHORIZATION_HEADER, String.format("%s%s", TOKEN_PREFIX, authToken)); template.header(AUTHORIZATION_HEADER, String.format("%s%s", TOKEN_PREFIX, authToken));
} } else {
final String currentRoleCode = attributes.getRequest().getHeader(CURRENT_ROLE_CODE);// 提取request头信息 //检查cookie
Optional<Cookie> optional = Arrays.stream(attributes.getRequest().getCookies())
.filter(cookie -> StringUtils.equals(cookie.getName(), "mall3_token"))
.findAny();
// 检查请求头是否包含 currentRoleCode optional.ifPresent(o -> template.header(AUTHORIZATION_HEADER, String.format("%s%s", TOKEN_PREFIX, o)));
if (StringUtils.isNotEmpty(currentRoleCode)) { }
template.header(CURRENT_ROLE_CODE, currentRoleCode); final String currentRoleCode = attributes.getRequest().getHeader(CURRENT_ROLE_CODE);// 提取request头信息
}
} // 检查请求头是否包含 currentRoleCode
} if (StringUtils.isNotEmpty(currentRoleCode)) {
template.header(CURRENT_ROLE_CODE, currentRoleCode);
}
}
}
} }