定时任务调用Feign出错，重新定义拦截器

最新推荐文章于 2024-03-04 14:23:55 发布

名月摘下

最新推荐文章于 2024-03-04 14:23:55 发布

阅读量544

点赞数

文章标签： java

本文链接：https://blog.csdn.net/u012601890/article/details/132583918

版权

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());
        }
        }

    }

}