58 lines
1.9 KiB
Java
58 lines
1.9 KiB
Java
/**
|
||
* Copyright (c) 2018-2028, Chill Zhuang 庄骞 (smallchill@163.com).
|
||
* <p>
|
||
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE;
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
* <p>
|
||
* http://www.gnu.org/licenses/lgpl.html
|
||
* <p>
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
package org.springblade.core.boot.feign;
|
||
|
||
import feign.RequestInterceptor;
|
||
import feign.RequestTemplate;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.web.context.request.RequestContextHolder;
|
||
import org.springframework.web.context.request.ServletRequestAttributes;
|
||
|
||
import javax.servlet.http.HttpServletRequest;
|
||
import java.util.Enumeration;
|
||
|
||
/**
|
||
* feign 传递Request header
|
||
*/
|
||
@Slf4j
|
||
public class BladeFeignRequestHeaderInterceptor implements RequestInterceptor {
|
||
|
||
@Override
|
||
public void apply(RequestTemplate requestTemplate) {
|
||
ServletRequestAttributes attrs = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
||
if (attrs != null) {
|
||
HttpServletRequest request = attrs.getRequest();
|
||
Enumeration<String> headerNames = request.getHeaderNames();
|
||
if (headerNames != null) {
|
||
while (headerNames.hasMoreElements()) {
|
||
String name = headerNames.nextElement();
|
||
String value = request.getHeader(name);
|
||
/**
|
||
* 遍历请求头里面的属性字段，将Authorization添加到新的请求头中转发到下游服务
|
||
* */
|
||
if ("Authorization".equals(name)) {
|
||
log.debug("添加自定义请求头key:" + name + ",value:" + value);
|
||
requestTemplate.header(name, value);
|
||
}
|
||
}
|
||
} else {
|
||
log.warn("FeignHeadConfiguration", "获取请求头失败！");
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|