package com.chinaztt.mes.common.config;

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

import javax.servlet.http.HttpServletRequest;

/**
 * @Author 张宾
 * @Date 2023/10/31
 */
@Component
public class FeignInterceptor implements RequestInterceptor {
	@Override
	public void apply(RequestTemplate template) {
		RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
		if(requestAttributes != null){
			HttpServletRequest request = ((ServletRequestAttributes) requestAttributes).getRequest();
			String authorization = request.getHeader("Authorization");
			template.header("Authorization", authorization);
		}
	}
}