feign呼び出しマイクロサービスによるheaderの転送

1292 ワード

我々のプロジェクト開発ではフロントコールバックグラウンドインタフェースはリクエストヘッダにtokenトークンを携帯するが、feignによって他のマイクロサービスを呼び出す場合、敏感ヘッダ情報は下流のマイクロサービスに伝達されないため、ブロックを構成してheaderの伝達を実現しなければならない.コードは以下の通りである.
package com.cyclothes.interceptor;


import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.util.Enumeration;

/**
 * spirng cloud feign header
 *
 * @author  
 * @date 2019/4/3
 */
@Component
public class FeignHeaderInterceptor implements RequestInterceptor {

    @Override
    public void apply(RequestTemplate template) {
        /**
         *  token
         */
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        Enumeration headerNames = request.getHeaderNames();
        if (headerNames != null) {
            while (headerNames.hasMoreElements()) {
                String name = headerNames.nextElement();
                String values = request.getHeader(name);
                template.header(name, values);
            }
        }
    }
}