import feign.RequestTemplate;
import feign.RequestInterceptor;
import org.springframework.context.annotation.Configuration;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import java.util.Enumeration;
@Configuration
public class FeignConfiguration implements RequestInterceptor {
// @Override
// public void apply(RequestTemplate template) {
// ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder
// .getRequestAttributes();
// if (attributes == null) {
// RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(new MockHttpServletRequest()));//为空进行新建
// attributes = (ServletRequestAttributes) RequestContextHolder
// .getRequestAttributes();//重新赋值
// }
// HttpServletRequest request = attributes.getRequest();
// // 对消息头进行配置
// Enumeration<String> headerNames = request.getHeaderNames();
// if (headerNames != null) {
// /**
// * 判断有没有token,如果是定时任务进来,是没有token的,此时用默认token以调用到对应的Feign服务
// */
// boolean flag = true;
// while (headerNames.hasMoreElements()) {
// String name = headerNames.nextElement();
// String values = request.getHeader(name);
// if(name.equals("Authorization")){
// flag = false;
// }
// template.header(name, values);
// }
// if(flag){
// template.header("Authorization","Bearer 31ac2344-4ffc-452f-82e7-1b0b2f44d2a2");
// }
// }
// // 对请求体进行配置
// Enumeration<String> bodyNames = request.getParameterNames();
// StringBuffer body =new StringBuffer();
// if (bodyNames != null) {
// while (bodyNames.hasMoreElements()) {
// String name = bodyNames.nextElement();
// String values = request.getParameter(name);
// body.append(name).append("=").append(values).append("&");
// }
// }
// if(body.length()!=0) {
// body.deleteCharAt(body.length()-1);
// template.body(body.toString());
// }
// }
@Override
public void apply(RequestTemplate template) {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder
.getRequestAttributes();
if (attributes != null){
HttpServletRequest request = attributes.getRequest();
Enumeration<String> headerNames = request.getHeaderNames();
if (headerNames != null) {
while (headerNames.hasMoreElements()) {
String name = headerNames.nextElement();
String values = request.getHeader(name);
template.header(name, values);
}
}
}else {
if (attributes == null) {
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(new MockHttpServletRequest()));//为空进行新建
attributes = (ServletRequestAttributes) RequestContextHolder
.getRequestAttributes();//重新赋值
}
HttpServletRequest request = attributes.getRequest();
// 对消息头进行配置
Enumeration<String> headerNames = request.getHeaderNames();
if (headerNames != null) {
/**
* 判断有没有token,如果是定时任务进来,是没有token的,此时用默认token以调用到对应的Feign服务
*/
boolean flag = true;
while (headerNames.hasMoreElements()) {
String name = headerNames.nextElement();
String values = request.getHeader(name);
if(name.equals("Authorization")){
flag = false;
}
template.header(name, values);
}
if(flag){
template.header("Authorization","Bearer 31ac2344-4ffc-452f-82e7-1b0b2f44d2a2");
}
}
// 对请求体进行配置
Enumeration<String> bodyNames = request.getParameterNames();
StringBuffer body =new StringBuffer();
if (bodyNames != null) {
while (bodyNames.hasMoreElements()) {
String name = bodyNames.nextElement();
String values = request.getParameter(name);
body.append(name).append("=").append(values).append("&");
}
}
if(body.length()!=0) {
body.deleteCharAt(body.length()-1);
template.body(body.toString());
}
}
}
}
定时任务调用Feign出错,重新定义拦截器
最新推荐文章于 2024-03-04 14:23:55 发布