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

解决feign请求头叠加问题

Browse Source
This commit is contained in:
ajaxfan
2021-04-23 16:36:40 +08:00
parent ee91af9a1a
commit c245275dd3
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -76,18 +76,20 @@ public class FeignConfig implements RequestInterceptor {
*/ */
private void injectToken(RequestTemplate template, ServletRequestAttributes attributes) { private void injectToken(RequestTemplate template, ServletRequestAttributes attributes) {
final String header = attributes.getRequest().getHeader(AUTHORIZATION_HEADER);// 提取request头信息 final String header = attributes.getRequest().getHeader(AUTHORIZATION_HEADER);// 提取request头信息
// 移除已存的头信息
template.removeHeader(AUTHORIZATION_HEADER);
// 检查请求头是否包含 Bearer 前缀 // 检查请求头是否包含 Bearer 前缀
if (StringUtils.startsWith(header, TOKEN_PREFIX)) { if (StringUtils.startsWith(header, TOKEN_PREFIX)) {
log.info("FeignConfig request header {from header}", AUTHORIZATION_HEADER, header); log.info("FeignConfig request header (from header) {} {}", AUTHORIZATION_HEADER, header);
template.header(AUTHORIZATION_HEADER, header); template.header(AUTHORIZATION_HEADER, header);
} else {// 检查cookie } else {// 检查cookie
Optional.ofNullable(attributes.getRequest().getCookies()).ifPresent(cookies -> { Optional.ofNullable(attributes.getRequest().getCookies()).ifPresent(cookies -> {
Stream.of(cookies).filter(item -> StringUtils.equals(item.getName(), COOKIE_TOKEN_CODE)).findFirst() Stream.of(cookies).filter(item -> StringUtils.equals(item.getName(), COOKIE_TOKEN_CODE)).findFirst()
.ifPresent(token -> { .ifPresent(token -> {
String authToken = token.getValue(); String authToken = token.getValue();
log.info("FeignConfig request header (from cookie): {} {}", log.info("FeignConfig request header (from cookie): {}",
String.format("%s%s", AUTHORIZATION_HEADER, TOKEN_PREFIX, authToken)); String.format("%s %s%s", AUTHORIZATION_HEADER, TOKEN_PREFIX, authToken));
template.header(AUTHORIZATION_HEADER, String.format("%s%s", TOKEN_PREFIX, authToken)); template.header(AUTHORIZATION_HEADER, String.format("%s%s", TOKEN_PREFIX, authToken));
}); });
}); });